dsl

type Agent struct {
	Name          string          `yaml:"name"`
	Extends       string          `yaml:"extends"`
	Model         string          `yaml:"model"`
	FallbackModel string          `yaml:"fallback_model"`
	System        string          `yaml:"system"`
	Temperature   *float64        `yaml:"temperature"`
	Budget        string          `yaml:"budget"` // e.g., "$0.50"
	Tools         []string        `yaml:"tools"`
	Knowledge     []string        `yaml:"knowledge"`
	Team          []string        `yaml:"team"`
	Supervision   *SupervisionDef `yaml:"supervision"`
	Retry         *RetryDef       `yaml:"retry"`
	Skills        *SkillsDef      `yaml:"skills"`
	Delegation    *DelegationDef  `yaml:"delegation"`
}

type DelegationContext struct {
	CallerAgent string
	Messages    []llm.Message
}

type DelegationDef struct {
	ContextWindow int      `yaml:"context_window"` // number of recent messages to forward
	IncludeRoles  []string `yaml:"include_roles"`  // filter by role (user, assistant, system)
	Blackboard    bool     `yaml:"blackboard"`     // enable shared blackboard for team
}

type Document struct {
	Name        string               `yaml:"name"`
	Description string               `yaml:"description"`
	Agents      map[string]*Agent    `yaml:"agents"`
	Workflows   map[string]*Workflow `yaml:"workflows"`
	Tools       map[string]*ToolDef  `yaml:"tools"`
	Settings    *Settings            `yaml:"settings"`
}

type ExecutionContext struct {
	// Inputs are the workflow input values
	Inputs map[string]any

	// Variables holds step outputs and set values
	Variables map[string]any

	// CurrentStep is the index of the executing step
	CurrentStep int

	// LoopState for loop iterations
	LoopState *LoopState

	// StartTime is when execution began
	StartTime time.Time

	// Timeout for the entire workflow
	Timeout time.Duration
}

type GlobalSkillsDef struct {
	Directories []string `yaml:"directories"`
}

type GroupResolver func(ctx context.Context) *vega.ProcessGroup

type Input struct {
	Type        string   `yaml:"type"`
	Description string   `yaml:"description"`
	Required    bool     `yaml:"required"`
	Default     any      `yaml:"default"`
	Enum        []string `yaml:"enum"`
	Min         *float64 `yaml:"min"`
	Max         *float64 `yaml:"max"`
}

type Interpreter struct {
	// contains filtered or unexported fields
}

type InterpreterOption func(*Interpreter)

type LoggingDef struct {
	Level string `yaml:"level"` // debug, info, warn, error
	File  string `yaml:"file"`
}

type LoopState struct {
	Index int
	Count int
	Item  any
	First bool
	Last  bool
}

type MCPDef struct {
	Servers []MCPServerDef `yaml:"servers"`
}

type MCPServerDef struct {
	Name         string            `yaml:"name"`
	Transport    string            `yaml:"transport"`
	Command      string            `yaml:"command"`
	Args         []string          `yaml:"args"`
	Env          map[string]string `yaml:"env"`
	URL          string            `yaml:"url"`
	Headers      map[string]string `yaml:"headers"`
	Timeout      string            `yaml:"timeout"`
	FromRegistry bool              `yaml:"-"` // resolved from registry, not serialized
}

type MotherCallbacks struct {
	OnAgentCreated func(agent *Agent) error
	OnAgentDeleted func(name string)
}

type Parser struct {
	// BaseDir for resolving relative paths
	BaseDir string
}

type RateLimitDef struct {
	RequestsPerMinute int `yaml:"requests_per_minute"`
	TokensPerMinute   int `yaml:"tokens_per_minute"`
}

type Repeat struct {
	Steps []Step `yaml:"steps"`
	Until string `yaml:"until"`
	Max   int    `yaml:"max"`
}

type RetryDef struct {
	MaxAttempts int    `yaml:"max_attempts"`
	Backoff     string `yaml:"backoff"` // linear, exponential, constant
}

type ScheduledJob struct {
	Name      string `json:"name"`
	Cron      string `json:"cron"`    // standard 5-field cron expression
	AgentName string `json:"agent"`   // agent to message on schedule
	Message   string `json:"message"` // message to send
	Enabled   bool   `json:"enabled"`
}

type SchedulerBackend interface {
	AddJob(job ScheduledJob) error
	RemoveJob(name string) error
	ListJobs() []ScheduledJob
}

type SendFunc func(ctx context.Context, agent string, message string) (string, error)

type Settings struct {
	DefaultModel       string           `yaml:"default_model"`
	DefaultTemperature *float64         `yaml:"default_temperature"`
	Sandbox            string           `yaml:"sandbox"`
	Budget             string           `yaml:"budget"`
	Supervision        *SupervisionDef  `yaml:"supervision"`
	RateLimit          *RateLimitDef    `yaml:"rate_limit"`
	Logging            *LoggingDef      `yaml:"logging"`
	Tracing            *TracingDef      `yaml:"tracing"`
	MCP                *MCPDef          `yaml:"mcp"`
	Skills             *GlobalSkillsDef `yaml:"skills"`
}

type SkillsDef struct {
	Directories []string `yaml:"directories"`
	Include     []string `yaml:"include"`
	Exclude     []string `yaml:"exclude"`
	MaxActive   int      `yaml:"max_active"`
}

type Step struct {
	// Agent step fields
	Agent           string `yaml:"-"` // Extracted from key
	Action          string `yaml:"-"` // Extracted from key
	Send            string `yaml:"send"`
	Save            string `yaml:"save"`
	Timeout         string `yaml:"timeout"`
	Budget          string `yaml:"budget"`
	Retry           int    `yaml:"retry"`
	If              string `yaml:"if"`
	ContinueOnError bool   `yaml:"continue_on_error"`
	Format          string `yaml:"format"` // json, yaml, etc.

	// Control flow fields
	Condition string `yaml:"-"` // For if steps
	Then      []Step `yaml:"then"`
	Else      []Step `yaml:"else"`

	// Loop fields
	ForEach string  `yaml:"for"` // "item in items"
	Repeat  *Repeat `yaml:"repeat"`

	// Parallel fields
	Parallel []Step `yaml:"parallel"`

	// Sub-workflow fields
	Workflow string         `yaml:"workflow"`
	With     map[string]any `yaml:"with"`

	// Special fields
	Set    map[string]any `yaml:"set"`
	Return string         `yaml:"return"`
	Try    []Step         `yaml:"try"`
	Catch  []Step         `yaml:"catch"`

	// Raw for flexible parsing
	Raw map[string]any `yaml:"-"`
}

type SupervisionDef struct {
	Strategy    string `yaml:"strategy"` // restart, stop, escalate
	MaxRestarts int    `yaml:"max_restarts"`
	Window      string `yaml:"window"` // e.g., "10m"
}

type ToolDef struct {
	Name           string      `yaml:"name"`
	Description    string      `yaml:"description"`
	Params         []ToolParam `yaml:"params"`
	Implementation *ToolImpl   `yaml:"implementation"`
	Include        []string    `yaml:"include"` // For loading from files
}

type ToolImpl struct {
	Type    string            `yaml:"type"` // http, exec, file_read, file_write, builtin
	Method  string            `yaml:"method"`
	URL     string            `yaml:"url"`
	Headers map[string]string `yaml:"headers"`
	Query   map[string]string `yaml:"query"`
	Body    any               `yaml:"body"`
	Command string            `yaml:"command"`
	Timeout string            `yaml:"timeout"`
}

type ToolParam struct {
	Name        string   `yaml:"name"`
	Type        string   `yaml:"type"`
	Description string   `yaml:"description"`
	Required    bool     `yaml:"required"`
	Default     any      `yaml:"default"`
	Enum        []string `yaml:"enum"`
}

type TracingDef struct {
	Enabled  bool   `yaml:"enabled"`
	Exporter string `yaml:"exporter"` // otlp, jaeger, json
	Endpoint string `yaml:"endpoint"`
}

type ValidationError struct {
	File    string
	Line    int
	Column  int
	Field   string
	Message string
	Hint    string
}

type Workflow struct {
	Description string            `yaml:"description"`
	Inputs      map[string]*Input `yaml:"inputs"`
	Steps       []Step            `yaml:"steps"`
	Output      any               `yaml:"output"` // string or map
}