config

type Chat struct {
	Desc              string        `yaml:"desc"`
	System            string        `yaml:"system"`
	Model             string        `yaml:"model"`
	MaxMessageRounds  int           `yaml:"maxMessageRounds"`
	FullMessageRounds int           `yaml:"fullMessageRounds,omitempty"`
	MaxIterations     int           `yaml:"maxIterations"`
	MaxRetries        int           `yaml:"maxRetries"`
	MCPServers        []string      `yaml:"mcpServers,omitempty"`
	Skill             *Skill        `yaml:"skill,omitempty"`
	Tools             []string      `yaml:"tools,omitempty"`
	Default           bool          `yaml:"default"`
	Hooks             *SessionHooks `yaml:"hooks,omitempty"`
	Persistence       bool          `yaml:"persistence"`
}

type Config struct {
	Chats         map[string]Chat      `yaml:"chats,omitempty"`
	Providers     map[string]Provider  `yaml:"providers,omitempty"`
	Models        map[string]Model     `yaml:"models,omitempty"`
	MCPServers    map[string]MCPServer `yaml:"mcpServers,omitempty"`
	Tools         map[string]Tool      `yaml:"tools,omitempty"`
	SystemPrompts map[string]string    `yaml:"systemPrompts,omitempty"`
}

type MCPServer struct {
	Type string `yaml:"type"`
	// for stdio
	Cmd  string            `yaml:"cmd,omitempty"`
	Args []string          `yaml:"args,omitempty"`
	Env  map[string]string `yaml:"env,omitempty"`
	// for sse & streamable-http
	URL               string            `yaml:"url,omitempty"`
	Headers           map[string]string `yaml:"headers,omitempty"`
	AutoApproval      bool              `yaml:"autoApproval"`
	AutoApprovalTools []string          `yaml:"autoApprovalTools"`
	// Tool filtering: include only these tools (if non-empty, only these tools are kept)
	Include []string `yaml:"include,omitempty"`
	// Tool filtering: exclude these tools (if non-empty, these tools are removed)
	Exclude []string `yaml:"exclude,omitempty"`
	// NoConcurrent: if true, all tools from this server share a single mutex,
	// meaning no two tools from this server can run concurrently.
	NoConcurrent bool `yaml:"noConcurrent,omitempty"`
	// NoConcurrentTools: list of specific tool names that should NOT be called concurrently.
	// Each listed tool gets its own mutex, so different tools don't block each other.
	NoConcurrentTools []string `yaml:"noConcurrentTools,omitempty"`
	// LowercaseTools: if true, all discovered tool names are lowercased before
	// filtering (include/exclude/autoApprovalTools/noConcurrentTools) and registration.
	LowercaseTools bool `yaml:"lowercaseTools,omitempty"`
}

type MixedModel struct {
	ModelParams `yaml:",inline"`
	Weight      int `yaml:"weight,omitempty"` // weight for weighted random selection (default: 1)
}

type Model struct {
	ModelParams `yaml:",inline"`
	Mixed       []MixedModel `yaml:"mixed,omitempty"`
}

type ModelParams struct {
	Provider        string         `yaml:"provider"`
	Model           string         `yaml:"model"`
	Thinking        bool           `yaml:"thinking"`
	ReasoningEffort *string        `yaml:"reasoningEffort"`
	MaxTokens       int            `yaml:"maxTokens,omitempty"`
	Temperature     float64        `yaml:"temperature,omitempty"`
	TopP            float64        `yaml:"topP,omitempty"`
	TopK            int            `yaml:"topK,omitempty"`
	ExtraBody       map[string]any `yaml:"extraBody"`
}

type Provider struct {
	Type    string            `yaml:"type"`
	BaseURL string            `yaml:"baseUrl,omitempty"`
	APIKey  string            `yaml:"apiKey,omitempty"`
	Headers map[string]string `yaml:"headers,omitempty"`
	Timeout int               `yaml:"timeout,omitempty"` // in seconds
}

type SessionHookConfig struct {
	Enabled    bool              `yaml:"enabled"`
	Type       string            `yaml:"type,omitempty"`    // "script" or "http", default is "script"
	ScriptPath string            `yaml:"scriptPath"`        // used when type is "script"
	URL        string            `yaml:"url,omitempty"`     // used when type is "http"
	Method     string            `yaml:"method,omitempty"`  // HTTP method for http type, default is "POST"
	Headers    map[string]string `yaml:"headers,omitempty"` // HTTP headers for http type
	Args       []string          `yaml:"args,omitempty"`
	Timeout    int               `yaml:"timeout,omitempty"` // in seconds, default is 30
	Env        map[string]string `yaml:"env,omitempty"`     // environment variables for the hook script
}

type SessionHooks struct {
	Keep          *SessionHookConfig `yaml:"keep,omitempty"`
	GenModelInput *SessionHookConfig `yaml:"genModelInput,omitempty"`
}

type Skill struct {
	Dir               string   `yaml:"dir"`
	WorkDir           string   `yaml:"workDir"`
	Timeout           int      `yaml:"timeout"`
	AutoApproval      bool     `yaml:"autoApproval"`
	AutoApprovalTools []string `yaml:"autoApprovalTools"`
}

type Tool struct {
	Category          string                 `yaml:"category"`
	Params            map[string]interface{} `yaml:"params"`
	AutoApproval      bool                   `yaml:"autoApproval"`
	AutoApprovalTools []string               `yaml:"autoApprovalTools"`
}