dsl

type AgentDef struct {
	Model        string            `yaml:"model" json:"model"`
	Provider     string            `yaml:"provider,omitempty" json:"provider,omitempty"`
	SystemPrompt string            `yaml:"system_prompt,omitempty" json:"system_prompt,omitempty"`
	Temperature  float64           `yaml:"temperature,omitempty" json:"temperature,omitempty"`
	MaxTokens    int               `yaml:"max_tokens,omitempty" json:"max_tokens,omitempty"`
	Tools        []string          `yaml:"tools,omitempty" json:"tools,omitempty"` // 引用 tools 中定义的工具
	Metadata     map[string]string `yaml:"metadata,omitempty" json:"metadata,omitempty"`
}

type ChainStepDef struct {
	Steps []ChainStepEntry `yaml:"steps" json:"steps"`
}

type ChainStepEntry struct {
	Tool       string            `yaml:"tool" json:"tool"`
	Args       map[string]any    `yaml:"args,omitempty" json:"args,omitempty"`
	ArgMapping map[string]string `yaml:"arg_mapping,omitempty" json:"arg_mapping,omitempty"`
	OnError    string            `yaml:"on_error,omitempty" json:"on_error,omitempty"`
	MaxRetries int               `yaml:"max_retries,omitempty" json:"max_retries,omitempty"`
}

type ErrorDef struct {
	Strategy      string `yaml:"strategy" json:"strategy"` // fail_fast, skip, retry
	MaxRetries    int    `yaml:"max_retries,omitempty" json:"max_retries,omitempty"`
	RetryDelayMs  int    `yaml:"retry_delay_ms,omitempty" json:"retry_delay_ms,omitempty"`
	FallbackValue any    `yaml:"fallback_value,omitempty" json:"fallback_value,omitempty"`
}

type LoopDef struct {
	Type          string `yaml:"type" json:"type"`                                         // while, for, foreach
	Condition     string `yaml:"condition,omitempty" json:"condition,omitempty"`           // 条件表达式
	MaxIterations int    `yaml:"max_iterations,omitempty" json:"max_iterations,omitempty"` // 最大迭代次数
	Collection    string `yaml:"collection,omitempty" json:"collection,omitempty"`         // foreach 的集合表达式
	ItemVar       string `yaml:"item_var,omitempty" json:"item_var,omitempty"`             // foreach 的迭代变量名
}

type NodeDef struct {
	ID        string            `yaml:"id" json:"id"`
	Type      string            `yaml:"type" json:"type"`                             // action, condition, loop, parallel, subgraph, checkpoint
	Step      string            `yaml:"step,omitempty" json:"step,omitempty"`         // 引用 steps 中的步骤
	StepDef   *StepDef          `yaml:"step_def,omitempty" json:"step_def,omitempty"` // 内联步骤定义
	Next      []string          `yaml:"next,omitempty" json:"next,omitempty"`
	Condition string            `yaml:"condition,omitempty" json:"condition,omitempty"` // 条件表达式
	OnTrue    []string          `yaml:"on_true,omitempty" json:"on_true,omitempty"`
	OnFalse   []string          `yaml:"on_false,omitempty" json:"on_false,omitempty"`
	Loop      *LoopDef          `yaml:"loop,omitempty" json:"loop,omitempty"`
	Parallel  []string          `yaml:"parallel,omitempty" json:"parallel,omitempty"`
	SubGraph  *WorkflowNodesDef `yaml:"subgraph,omitempty" json:"subgraph,omitempty"`
	Error     *ErrorDef         `yaml:"error,omitempty" json:"error,omitempty"`
	Metadata  map[string]any    `yaml:"metadata,omitempty" json:"metadata,omitempty"`
}

type OrchestrationStepDef struct {
	Mode      string   `yaml:"mode" json:"mode"`
	AgentIDs  []string `yaml:"agent_ids" json:"agent_ids"`
	MaxRounds int      `yaml:"max_rounds,omitempty" json:"max_rounds,omitempty"`
	TimeoutMs int      `yaml:"timeout_ms,omitempty" json:"timeout_ms,omitempty"`
}

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

type StepDef struct {
	Type          string                `yaml:"type" json:"type"`                                     // llm, tool, human_input, code, passthrough, orchestration, agent
	Agent         string                `yaml:"agent,omitempty" json:"agent,omitempty"`               // Agent ID reference for agent steps.
	InlineAgent   *AgentDef             `yaml:"inline_agent,omitempty" json:"inline_agent,omitempty"` // Legacy reject-only field. Presence is invalid and must be rejected during validation; agent steps only support `agent`.
	Tool          string                `yaml:"tool,omitempty" json:"tool,omitempty"`
	Prompt        string                `yaml:"prompt,omitempty" json:"prompt,omitempty"`
	Config        map[string]any        `yaml:"config,omitempty" json:"config,omitempty"`
	Orchestration *OrchestrationStepDef `yaml:"orchestration,omitempty" json:"orchestration,omitempty"`
	Chain         *ChainStepDef         `yaml:"chain,omitempty" json:"chain,omitempty"`
	SubGraph      *WorkflowNodesDef     `yaml:"subgraph,omitempty" json:"subgraph,omitempty"`
}

type ToolDef struct {
	Type        string         `yaml:"type" json:"type"` // builtin, mcp, http, code
	Description string         `yaml:"description" json:"description"`
	Config      map[string]any `yaml:"config,omitempty" json:"config,omitempty"`
	InputSchema map[string]any `yaml:"input_schema,omitempty" json:"input_schema,omitempty"`
}

type Validator struct{}

type VariableDef struct {
	Type        string `yaml:"type" json:"type"`                                   // string, int, float, bool, list, map
	Default     any    `yaml:"default,omitempty" json:"default,omitempty"`         // 默认值
	Description string `yaml:"description,omitempty" json:"description,omitempty"` // 描述
	Required    bool   `yaml:"required,omitempty" json:"required,omitempty"`       // 是否必填
}

type WorkflowDSL struct {
	// Version DSL 版本
	Version string `yaml:"version" json:"version"`
	// Name 工作流名称
	Name string `yaml:"name" json:"name"`
	// Description 工作流描述
	Description string `yaml:"description" json:"description"`

	// Variables 全局变量定义
	Variables map[string]VariableDef `yaml:"variables,omitempty" json:"variables,omitempty"`

	// Agents Agent 定义
	Agents map[string]AgentDef `yaml:"agents,omitempty" json:"agents,omitempty"`

	// Tools 工具定义
	Tools map[string]ToolDef `yaml:"tools,omitempty" json:"tools,omitempty"`

	// Steps 步骤定义（可复用）
	Steps map[string]StepDef `yaml:"steps,omitempty" json:"steps,omitempty"`

	// Workflow 工作流节点定义
	Workflow WorkflowNodesDef `yaml:"workflow" json:"workflow"`

	// Metadata 元数据
	Metadata map[string]any `yaml:"metadata,omitempty" json:"metadata,omitempty"`
}

type WorkflowNodesDef struct {
	Entry string    `yaml:"entry" json:"entry"`
	Nodes []NodeDef `yaml:"nodes" json:"nodes"`
}