models

type DeepseekOptions struct {
	// API 地址
	BaseURL string `json:"baseURL,omitempty"`
	// API 密钥
	APIKey string `json:"apiKey"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}

type MinimaxOptions struct {
	// API 地址
	BaseURL string `json:"baseURL,omitempty"`
	// API 密钥
	APIKey string `json:"apiKey"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}

type ModelConfig struct {
	// 模型名称
	Name string `json:"name"`
	// 供应商名称
	Provider string `json:"provider,omitempty"`
	// 模型描述（可选）
	Description string `json:"description,omitempty"`

	// 是否支持推理、思考模式
	Reasoning bool `json:"reasoning,omitempty"`
	// 是否支持视觉、图片理解
	Vision bool `json:"vision,omitempty"`

	// 价格信息
	Cost ModelCost `json:"cost,omitempty"`
	// 上下文窗口大小
	ContextWindow int64 `json:"contextWindow,omitempty"`
	// 最大输出 Token 数
	MaxOutputTokens int64 `json:"maxOutputTokens,omitempty"`
}

type ModelCost struct {
	// 每千输入 Token 价格
	Input float64 `json:"input,omitempty"`
	// 每千输出 Token 价格
	Output float64 `json:"output,omitempty"`
	// 每千缓存 Token 价格
	Cached float64 `json:"cached,omitempty"`
}

type ModelProvider struct {
	Ollama           *OllamaOptions           `json:"ollama,omitempty"`
	ZAI              *ZAIOptions              `json:"z-ai,omitempty"`
	Qwen             *QwenOptions             `json:"qwen,omitempty"`
	Moonshot         *MoonshotOptions         `json:"moonshotai,omitempty"`
	Deepseek         *DeepseekOptions         `json:"deepseek,omitempty"`
	Minimax          *MinimaxOptions          `json:"minimax,omitempty"`
	OpenAICompatible *OpenAICompatibleOptions `json:"openaiCompatible,omitempty"`
}

type Models struct {
	// 主模型，用于回答用户问题
	Primary string `json:"primary,omitempty"`
	// 轻量模型，用于处理简单事务。为空时使用主模型
	Light string `json:"light,omitempty"`
	// 视觉模型，用于处理图片理解任务
	Vision string `json:"vision,omitempty"`
}

type MoonshotOptions struct {
	// API 地址
	BaseURL string `json:"baseURL,omitempty"`
	// API 密钥
	APIKey string `json:"apiKey"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}

type OllamaOptions struct {
	// Ollama 服务端地址
	//
	// 默认 http://localhost:11434
	ServerAddress string `json:"serverAddress,omitempty"`
	// 模型响应超时时间，秒
	//
	// 默认 300
	Timeout int `json:"timeout,omitempty"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}

type OpenAICompatibleOptions struct {
	// 供应商名
	Name string `json:"name"`
	// API 地址
	BaseURL string `json:"baseURL"`
	// API 密钥
	APIKey string `json:"apiKey"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}

type QwenOptions struct {
	// API 地址
	BaseURL string `json:"baseURL,omitempty"`
	// API 密钥
	APIKey string `json:"apiKey"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}

type ZAIOptions struct {
	// API 地址
	BaseURL string `json:"baseURL,omitempty"`
	// API 密钥
	APIKey string `json:"apiKey"`
	// 模型列表
	Models []ModelConfig `json:"models,omitempty"`
}