dsl

type AgentConfig struct {
	Model       string         `json:"model"`
	ModelConfig ModelConfig    `json:"model_config"`
	Prompt      PromptConfig   `json:"prompt"`
	Tools       []AgentToolRef `json:"tools"`
	ToolChoice  string         `json:"tool_choice"`
	Output      OutputConfig   `json:"output"`
	Limits      LimitsConfig   `json:"limits"`
}

type AgentToolRef struct {
	Name   string `json:"name"`
	When   string `json:"when,omitempty"`
	Unless string `json:"unless,omitempty"`
}

type AgentTools struct {
	Toolset    *tools.Toolset
	ToolChoice niro.ToolChoice
	Options    tools.ToolStreamOptions
}

type CompileOptions struct {
	// ToolHandlers maps tool name to handler for "code" type tools. Used when a tool
	// has no "type" or type "code" (args only, handler from here).
	ToolHandlers map[string]tools.ToolHandler
	// ToolTypes maps type name to builder (e.g. "http" -> HTTPToolType). Merged with
	// default registry (RegisterToolType). Option entries override defaults.
	ToolTypes map[string]ToolTypeBuilder
}

type CompiledAgentConfig struct {
	Model          string
	Options        niro.Options
	PromptTemplate string
	ToolRefs       []AgentToolRef
	ToolChoice     niro.ToolChoice
	Limits         LimitsConfig
	Output         OutputConfig
}

type CompiledWorkflow struct {
	Type      string   // "fan" | "race" | "sequence" | "fan_then"
	Agents    []string // agent names (for fan_then: parallel agents)
	ThenAgent string   // for fan_then: synthesizer agent
	// contains filtered or unexported fields
}

type DSLDefinition struct {
	// Tools is map of tool name -> raw spec. Each spec can have "type" (e.g. "http", "code")
	// and type-specific fields. Omit "type" or use "code" for custom handlers from ToolHandlers.
	Tools   map[string]json.RawMessage `json:"tools"`
	Schemas map[string]json.RawMessage `json:"schemas"`
	Agents  map[string]AgentConfig     `json:"agents"`
}

type HTTPToolSpec struct {
	Type        string            `json:"type"`
	Description string            `json:"description,omitempty"`
	Input       json.RawMessage   `json:"input"`
	Output      string            `json:"output,omitempty"` // "text" | "json" (default): text = raw body; json = cases/parse or decode
	Method      string            `json:"method"`
	URL         string            `json:"url"`
	Headers     map[string]string `json:"headers,omitempty"`
	Timeout     string            `json:"timeout,omitempty"`
	Cases       []httpCase        `json:"cases,omitempty"`
	Retry       *struct {
		MaxRetries int    `json:"max_retries"`
		Backoff    string `json:"backoff"`
	} `json:"retry,omitempty"`
}

type HandoffSpec struct {
	Type        string         `json:"type"`
	Description string         `json:"description"`
	Allow       []string       `json:"allow,omitempty"` // shorthand: target enum when args not used
	Args        map[string]any `json:"args,omitempty"`  // full args; if present, used for schema
}

type LimitsConfig struct {
	MaxToolCalls     int `json:"max_tool_calls"`
	MaxParallelTools int `json:"max_parallel_tools"`
}

type ModelConfig struct {
	Temperature    *float64 `json:"temperature"`
	MaxTokens      int      `json:"max_tokens"`
	ThinkingBudget *int     `json:"thinking_budget"`
}

type NiroDefinition struct {
	// Agents holds compiled config per agent name.
	Agents map[string]*CompiledAgentConfig
	// Toolset is the shared toolset from DSL tools + optional handlers.
	Toolset *tools.Toolset
}

type OutputConfig struct {
	Allow      []string          `json:"allow"`
	Modalities []string          `json:"modalities"`
	Schemas    map[string]string `json:"schemas"`
	Stream     bool              `json:"stream"`
}

type PromptConfig struct {
	Template string `json:"template"`
}

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

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

type RunnerOption func(*Runner)

type ToolTypeBuilder interface {
	// Build returns a ToolDefinition for the given tool name and spec.
	// The spec is the raw JSON value for this tool from the DSL (e.g. {"type":"http","url":"...","args":{...}}).
	Build(name string, spec json.RawMessage) (tools.ToolDefinition, error)
}

type ToolTypeBuilderFunc func(name string, spec json.RawMessage) (tools.ToolDefinition, error)

type WorkflowDefinition struct {
	// Single-workflow form: name, description, type, agents
	Name        string   `json:"name,omitempty"`
	Description string   `json:"description,omitempty"`
	Type        string   `json:"type,omitempty"` // "fan" | "race" | "sequence"
	Agents      []string `json:"agents,omitempty"`

	// Multi-workflow form: workflows map
	Workflows map[string]WorkflowNode `json:"workflows,omitempty"`
}

type WorkflowNode struct {
	Description string   `json:"description,omitempty"`
	Type        string   `json:"type"`                 // "fan" | "race" | "sequence" | "fan_then"
	Agent       string   `json:"agent,omitempty"`      // single agent by name (entry point)
	Agents      []string `json:"agents,omitempty"`     // agent names (for fan/sequence/fan_then)
	ThenAgent   string   `json:"then_agent,omitempty"` // for fan_then: agent that synthesizes parallel results
}