providers

type APIKeyEntry struct {
	Key     string `json:"key" yaml:"key"`
	BaseURL string `json:"base_url,omitempty" yaml:"base_url,omitempty"`
	Weight  int    `json:"weight,omitempty" yaml:"weight,omitempty"`
}

type BaseProviderConfig struct {
	APIKey  string        `json:"api_key" yaml:"api_key"`
	APIKeys []APIKeyEntry `json:"api_keys,omitempty" yaml:"api_keys,omitempty"` // 多 API Key 支持，轮询使用
	BaseURL string        `json:"base_url" yaml:"base_url"`
	Model   string        `json:"model,omitempty" yaml:"model,omitempty"`
	Models  []string      `json:"models,omitempty" yaml:"models,omitempty"` // 可用模型白名单
	Timeout time.Duration `json:"timeout,omitempty" yaml:"timeout,omitempty"`
}

type ClaudeConfig struct {
	BaseProviderConfig `yaml:",inline"`
	AnthropicVersion   string `json:"anthropic_version,omitempty" yaml:"anthropic_version,omitempty"` // 默认 "2023-06-01"
}

type DeepSeekConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type DoubaoConfig struct {
	BaseProviderConfig `yaml:",inline"`
	AccessKey          string `json:"access_key,omitempty" yaml:"access_key,omitempty"` // 火山引擎 Access Key
	SecretKey          string `json:"secret_key,omitempty" yaml:"secret_key,omitempty"` // 火山引擎 Secret Key
	Region             string `json:"region,omitempty" yaml:"region,omitempty"`         // 区域，默认 cn-beijing
}

type GLMConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type GeminiConfig struct {
	BaseProviderConfig `yaml:",inline"`
	ProjectID          string                `json:"project_id,omitempty" yaml:"project_id,omitempty"`
	Region             string                `json:"region,omitempty" yaml:"region,omitempty"`
	AuthType           string                `json:"auth_type,omitempty" yaml:"auth_type,omitempty"` // "api_key"(默认) | "oauth"
	SafetySettings     []GeminiSafetySetting `json:"safety_settings,omitempty" yaml:"safety_settings,omitempty"`
}

type GeminiSafetySetting struct {
	Category  string `json:"category" yaml:"category"`
	Threshold string `json:"threshold" yaml:"threshold"`
}

type GrokConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type HunyuanConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type KimiConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type LlamaConfig struct {
	BaseProviderConfig `yaml:",inline"`
	Provider           string `json:"provider,omitempty" yaml:"provider,omitempty"` // together/replicate/openrouter
}

type MiniMaxConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type MistralConfig struct {
	BaseProviderConfig `yaml:",inline"`
}

type OpenAIConfig struct {
	BaseProviderConfig `yaml:",inline"`
	Organization       string `json:"organization,omitempty" yaml:"organization,omitempty"`
	UseResponsesAPI    bool   `json:"use_responses_api,omitempty" yaml:"use_responses_api,omitempty"` // 启用新的 Responses API (2025)
}

type PollConfig struct {
	Interval    time.Duration // 轮询间隔，默认 5s
	MaxAttempts int           // 最大尝试次数，0 表示无限（依赖 ctx 超时）
}

type PollFunc[T any] func(ctx context.Context) PollResult[T]

type PollResult[T any] struct {
	Done   bool  // 任务是否完成（成功或失败）
	Result *T    // 成功时的结果
	Err    error // 失败时的错误（设置 Done=true + Err 表示任务失败）
}

type ProviderCapability struct {
	Provider      string
	Image         bool
	Video         bool
	AudioGenerate bool
	AudioSTT      bool
	Embedding     bool
	FineTuning    bool
	Rerank        bool
}

type QwenConfig struct {
	BaseProviderConfig `yaml:",inline"`
}