dsl

type Agent struct {
	Name          string          `yaml:"name"`
	DisplayName   string          `yaml:"display_name"`
	Title         string          `yaml:"title"`
	Avatar        string          `yaml:"avatar"`
	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 AgentTemplate struct {
	Version     string   `json:"version" yaml:"version"`
	Name        string   `json:"name" yaml:"name"`
	DisplayName string   `json:"display_name,omitempty" yaml:"display_name,omitempty"`
	Title       string   `json:"title,omitempty" yaml:"title,omitempty"`
	Model       string   `json:"model" yaml:"model"`
	System      string   `json:"system" yaml:"system"`
	Tools       []string `json:"tools,omitempty" yaml:"tools,omitempty"`
	Team        []string `json:"team,omitempty" yaml:"team,omitempty"`
	ExportedBy  string   `json:"exported_by,omitempty" yaml:"exported_by,omitempty"`
	ExportedAt  string   `json:"exported_at,omitempty" yaml:"exported_at,omitempty"`
}

type ChannelBackend interface {
	CreateChannel(id, name, description, createdBy string, team []string, mode string) error
	GetChannelByName(name string) (*ChannelInfo, error)
	ListAllChannels() ([]ChannelInfo, error)
	ListChannelsForAgent(agent string) ([]ChannelInfo, error)
	FindChannelForAgents(agent1, agent2 string) (channelID string, channelName string, err error)
	InsertChannelMessage(channelID, agent, role, content string, threadID *int64, metadata, sender string) (int64, error)
	RecentChannelMessages(channelID string, limit int) ([]ChannelMessage, error)
}

type ChannelDef struct {
	Description string   `yaml:"description"`
	Team        []string `yaml:"team"`
	Mode        string   `yaml:"mode"` // "" (default) or "social"
}

type ChannelInfo struct {
	ID   string
	Name string
	Team []string
}

type ChannelMessage struct {
	Agent   string
	Sender  string
	Content string
}

type ChannelPeerResolver func(callerAgent, targetAgent string) bool

type ChannelPostCallback func(channelName, agent, content string, msgID int64, threadID *int64)

type ChannelReactiveCallback func(channelName string, team []string, poster string, message string, depth int, triggerMsgID int64)

type Company struct {
	ID          string           `yaml:"id" json:"id"`
	Name        string           `yaml:"name" json:"name"`
	Description string           `yaml:"description,omitempty" json:"description,omitempty"`
	Location    string           `yaml:"location,omitempty" json:"location,omitempty"`
	LogoURL     string           `yaml:"logo_url,omitempty" json:"logo_url,omitempty"`
	AccentColor string           `yaml:"accent_color,omitempty" json:"accent_color,omitempty"`
	Siblings    []CompanySibling `yaml:"siblings,omitempty" json:"siblings,omitempty"`
}

type CompanySibling struct {
	Name string `yaml:"name" json:"name"`
	URL  string `yaml:"url" json:"url"`
	Icon string `yaml:"icon,omitempty" json:"icon,omitempty"`
}

type DelegateToolOpts struct {
	SendFn              SendFunc
	TeamResolver        TeamResolver
	ChannelPeerResolver ChannelPeerResolver // optional — allows delegation to channel peers
}

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 DelegationObserver func(ctx context.Context, fromAgent, toAgent, message, response string)

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

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 HeraCallbacks struct {
	OnAgentCreated func(agent *Agent) error
	OnAgentDeleted func(name string)
	ChannelBackend ChannelBackend // optional — auto-creates channels for team leads
}

type InboxBackend interface {
	InsertInboxItem(fromAgent, subject, body, priority string) (int64, error)
	ListInboxItems(status string, limit int) ([]InboxItem, error)
	ResolveInboxItem(id int64, resolution string) error
}

type InboxItem struct {
	ID         int64      `json:"id"`
	FromAgent  string     `json:"from_agent"`
	Subject    string     `json:"subject"`
	Body       string     `json:"body,omitempty"`
	Priority   string     `json:"priority"`
	Status     string     `json:"status"`
	Resolution string     `json:"resolution,omitempty"`
	CreatedAt  time.Time  `json:"created_at"`
	ResolvedAt *time.Time `json:"resolved_at,omitempty"`
}

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 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 TeamResolver func(ctx context.Context) []string

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
}