harness.Run now takes a scope key. Concurrent runs of one skill with one scope share a single in-flight run and its result, errors included, so only one Generate call is made; different scopes run independently. A process-wide Guard (NewGuard, DefaultMaxCalls = 2) bounds concurrent LLM calls with a context-aware semaphore. Each skill declares Timeout() (DefaultTimeout = 60 s), enforced via context over the whole run; a run past it fails with an error matching llm.ErrTimeout. A caller whose context ends gets the context error; the shared run is cancelled, and its slot freed, once no caller waits for it. llm.Fake results can now block (Release) and signal entry (Started), so tests hold calls in flight. Maestro task: 8f533721-32c9-4f73-af9f-9a9f0e94abf8 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package harness
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
"code.cubecraftlabs.com/Maniacs_Mansion/Meal-Quest/internal/ai/llm"
|
|
)
|
|
|
|
// echoSkill is the trivial test skill: it asks the model to repeat a word.
|
|
// Its prompt is prompts/echo.v1.tmpl.
|
|
type echoSkill struct{}
|
|
|
|
type echoInput struct {
|
|
Word string
|
|
Context []ContextSection
|
|
}
|
|
|
|
type echoOutput struct {
|
|
Echo string `json:"echo"`
|
|
}
|
|
|
|
var _ Skill[echoInput, echoOutput] = echoSkill{}
|
|
|
|
func (echoSkill) Name() string { return "echo" }
|
|
|
|
func (echoSkill) Version() int { return 1 }
|
|
|
|
func (echoSkill) PromptInput(in echoInput) (PromptInput, error) {
|
|
if in.Word == "" {
|
|
return PromptInput{}, errors.New("no word to echo")
|
|
}
|
|
return PromptInput{Data: in, Context: in.Context}, nil
|
|
}
|
|
|
|
func (echoSkill) Schema() llm.JSONSchema {
|
|
return llm.JSONSchema{Name: "echo", Schema: json.RawMessage(
|
|
`{"type":"object","properties":{"echo":{"type":"string"}},"required":["echo"],"additionalProperties":false}`)}
|
|
}
|
|
|
|
func (echoSkill) Validate(out echoOutput) []string {
|
|
if out.Echo == "" {
|
|
return []string{"echo must not be empty"}
|
|
}
|
|
if out.Echo != strings.ToLower(out.Echo) {
|
|
return []string{"echo must be lower case"}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (echoSkill) PostProcess(out echoOutput) (echoOutput, error) {
|
|
out.Echo = strings.TrimSpace(out.Echo)
|
|
return out, nil
|
|
}
|
|
|
|
func (echoSkill) Timeout() time.Duration { return DefaultTimeout }
|