Add new file

statemachine/types.go

@@ -0,0 +1,122 @@
+package statemachine
+
+import "time"
+
+// ModelSpec defines LLM model configuration.
+type ModelSpec struct {
+	ModelID  string // e.g. "claude-opus-5", "claude-sonnet-5"
+	Thinking string // "adaptive" or ""
+	Effort   string // "low", "medium", "high", "xhigh", "max"
+}
+
+// PromptSpec defines a prompt template with variables and model.
+type PromptSpec struct {
+	TemplateRef string            // e.g. "planner/default.tmpl"
+	RawTemplate string            // overrides TemplateRef if non-empty
+	Variables   map[string]any    // template variables
+	Model       ModelSpec         // which LLM to use
+	LessonsRef  string            // key into lessons store
+}
+
+// PiRetryPolicy defines retry and timeout settings for Pi command execution.
+type PiRetryPolicy struct {
+	ScheduleToCloseTimeout time.Duration // default: 5m
+	InitialInterval        time.Duration // default: 2s
+	MaximumInterval        time.Duration // default: 30s
+	BackoffCoefficient     float64       // default: 2.0
+	StreamTimeout          time.Duration // default: 30s
+	StreamTimeoutMax       time.Duration // default: 2m
+}
+
+// ActivityTuning defines timeouts and retry counts for activities.
+type ActivityTuning struct {
+	ImplementerBaseTimeout time.Duration // default: 10m
+	ImplementerMaxRetries  int           // default: 3
+	JudgeTimeout           time.Duration // default: 5m
+	PiRetry                PiRetryPolicy
+}
+
+// OrchestratorConfig holds all runtime configuration for the orchestrator.
+type OrchestratorConfig struct {
+	SystemPrompt string                    // shared prompt prefix
+	Skills       []SkillRef                // required skill sources
+	RolePrompts  map[string]PromptSpec     // per-role: "planner", "judge", "implementer"
+	Tuning       ActivityTuning
+}
+
+// OrchestratorInput is the input to the Orchestrator workflow.
+type OrchestratorInput struct {
+	TargetRepoPath      string
+	RemoteURL           string
+	Milestone           string // e.g. "T0"
+	Config              OrchestratorConfig
+	DryRun              bool
+	CycleCount          int
+	MaxCyclesBeforeCAN  int // default: 100
+}
+
+// OrchestratorOutput is the output of the Orchestrator workflow.
+type OrchestratorOutput struct {
+	MilestoneComplete bool
+	Done              bool
+	LastError         string
+}
+
+// TaskUnitInput is the input to the TaskUnit workflow.
+type TaskUnitInput struct {
+	TaskID          string
+	TargetRepoPath  string
+	JudgeSpec       PromptSpec
+	ImplementerSpec PromptSpec
+	BaseTimeout     time.Duration
+	MaxJudgeRetries int
+}
+
+// TaskUnitOutput is the output of the TaskUnit workflow.
+type TaskUnitOutput struct {
+	TaskID   string
+	Verdict  string // "pass" or "fail"
+	Critique string
+	Branch   string
+}
+
+// SkillRef references a skill source.
+type SkillRef struct {
+	Name string // skill identifier
+	URL  string // source to clone
+}
+
+// Default values for types.
+const (
+	defaultScheduleToCloseTimeout = 5 * time.Minute
+	defaultInitialInterval         = 2 * time.Second
+	defaultMaximumInterval         = 30 * time.Second
+	defaultBackoffCoefficient      = 2.0
+	defaultStreamTimeout           = 30 * time.Second
+	defaultStreamTimeoutMax        = 2 * time.Minute
+	defaultImplementerBaseTimeout  = 10 * time.Minute
+	defaultImplementerMaxRetries   = 3
+	defaultJudgeTimeout            = 5 * time.Minute
+)
+
+// NewPiRetryPolicy returns a PiRetryPolicy with defaults.
+func NewPiRetryPolicy() PiRetryPolicy {
+	return PiRetryPolicy{
+		ScheduleToCloseTimeout: defaultScheduleToCloseTimeout,
+		InitialInterval:        defaultInitialInterval,
+		MaximumInterval:        defaultMaximumInterval,
+		BackoffCoefficient:     defaultBackoffCoefficient,
+		StreamTimeout:          defaultStreamTimeout,
+		StreamTimeoutMax:       defaultStreamTimeoutMax,
+	}
+}
+
+// NewActivityTuning returns an ActivityTuning with defaults.
+func NewActivityTuning() ActivityTuning {
+	return ActivityTuning{
+		ImplementerBaseTimeout: defaultImplementerBaseTimeout,
+		ImplementerMaxRetries:  defaultImplementerMaxRetries,
+		JudgeTimeout:           defaultJudgeTimeout,
+		PiRetry:                NewPiRetryPolicy(),
+	}
+}