provider

type AnthropicFactory struct{}

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

type CompleteResponse struct {
	Message types.Message
	Usage   *TokenUsage
}

type CustomClaudeFactory struct{}

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

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

type DoubaoConfig struct {
	// EndpointID 模型端点 ID（必需）
	EndpointID string
}

type DoubaoFactory struct {
	Config *DoubaoConfig
}

type DoubaoProvider struct {
	*OpenAICompatibleProvider
	// contains filtered or unexported fields
}

type Factory interface {
	Create(config *types.ModelConfig) (Provider, error)
}

type GLMFactory struct{}

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

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

type GeminiBlob struct {
	MimeType string `json:"mimeType"`
	Data     string `json:"data"` // base64 编码
}

type GeminiContent struct {
	Role  string       `json:"role,omitempty"`
	Parts []GeminiPart `json:"parts"`
}

type GeminiFactory struct{}

type GeminiFunctionCall struct {
	Name string         `json:"name"`
	Args map[string]any `json:"args"`
}

type GeminiFunctionDeclaration struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Parameters  map[string]any `json:"parameters"`
}

type GeminiFunctionResponse struct {
	Name     string         `json:"name"`
	Response map[string]any `json:"response"`
}

type GeminiPart struct {
	// 文本内容
	Text string `json:"text,omitempty"`

	// 内联数据（图片、音频等）
	InlineData *GeminiBlob `json:"inlineData,omitempty"`

	// 函数调用
	FunctionCall *GeminiFunctionCall `json:"functionCall,omitempty"`

	// 函数响应
	FunctionResponse *GeminiFunctionResponse `json:"functionResponse,omitempty"`
}

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

type GeminiTool struct {
	FunctionDeclarations []GeminiFunctionDeclaration `json:"functionDeclarations"`
}

type GroqFactory struct{}

type GroqProvider struct {
	*OpenAICompatibleProvider
}

type MistralFactory struct{}

type MistralProvider struct {
	*OpenAICompatibleProvider
}

type MoonshotFactory struct{}

type MoonshotProvider struct {
	*OpenAICompatibleProvider
}

type MultiProviderFactory struct{}

type OllamaFactory struct{}

type OllamaProvider struct {
	*OpenAICompatibleProvider
}

type OpenAICompatibleOptions struct {
	// 是否需要 API Key
	RequireAPIKey bool

	// 默认模型名称
	DefaultModel string

	// 是否支持推理模型
	SupportReasoning bool

	// 是否支持 Prompt Caching
	SupportPromptCache bool

	// 是否支持多模态
	SupportVision bool
	SupportAudio  bool

	// 超时配置
	Timeout time.Duration

	// 重试配置
	MaxRetries int
	RetryDelay time.Duration
	RetryOn429 bool // 是否在 429 时重试
	RetryOn500 bool // 是否在 5xx 时重试

	// 自定义请求头
	CustomHeaders map[string]string
}

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

type OpenAIFactory struct{}

type OpenAIProvider struct {
	*OpenAICompatibleProvider
}

type OpenRouterConfig struct {
	// 优先使用的 provider (如 "OpenAI", "Anthropic")
	ProviderPreference string

	// Provider 优先级顺序
	ProviderOrder []string

	// 是否允许 fallback 到其他 provider
	AllowFallbacks bool

	// 应用名称（用于统计）
	AppName string

	// 站点 URL（用于统计）
	SiteURL string
}

type OpenRouterFactory struct {
	Config *OpenRouterConfig
}

type OpenRouterProvider struct {
	*OpenAICompatibleProvider
	// contains filtered or unexported fields
}

type Provider interface {
	// Stream 流式对话
	Stream(ctx context.Context, messages []types.Message, opts *StreamOptions) (<-chan StreamChunk, error)

	// Complete 非流式对话(阻塞式,返回完整响应)
	Complete(ctx context.Context, messages []types.Message, opts *StreamOptions) (*CompleteResponse, error)

	// Config 返回配置
	Config() *types.ModelConfig

	// Capabilities 返回模型能力
	Capabilities() ProviderCapabilities

	// SetSystemPrompt 设置系统提示词
	SetSystemPrompt(prompt string) error

	// GetSystemPrompt 获取系统提示词
	GetSystemPrompt() string

	// Close 关闭连接
	Close() error
}

type ProviderCapabilities struct {
	// 基础能力
	SupportToolCalling  bool // 是否支持工具调用
	SupportSystemPrompt bool // 是否支持独立 system prompt
	SupportStreaming    bool // 是否支持流式输出

	// 多模态能力
	SupportVision bool // 是否支持视觉（图片）
	SupportAudio  bool // 是否支持音频
	SupportVideo  bool // 是否支持视频

	// 高级能力
	SupportReasoning        bool // 是否支持推理模型（o1/o3/R1）
	SupportPromptCache      bool // 是否支持 Prompt Caching
	SupportJSONMode         bool // 是否支持 JSON 模式
	SupportFunctionCall     bool // 是否支持 Function Calling
	SupportStructuredOutput bool // 是否支持结构化输出（JSON Schema）

	// 限制
	MaxTokens       int // 最大 token 数
	MaxToolsPerCall int // 单次最多调用工具数
	MaxImageSize    int // 最大图片大小（字节）

	// Tool Calling 格式
	ToolCallingFormat string // "anthropic" | "openai" | "qwen" | "custom"

	// 推理模型特性
	ReasoningTokensIncluded bool // reasoning tokens 是否包含在总 token 中

	// Prompt Caching 特性
	CacheMinTokens int // 最小缓存 Token 数
}

type ProviderFactory interface {
	Create(config *types.ModelConfig) (Provider, error)
}

type ReasoningTrace struct {
	Step         int     `json:"step"`
	Thought      string  `json:"thought"`
	ThoughtDelta string  `json:"thought_delta,omitempty"`
	Type         string  `json:"type,omitempty"` // "thinking", "reflection", "conclusion"
	Confidence   float64 `json:"confidence,omitempty"`
}

type ResponseFormat struct {
	Type   ResponseFormatType `json:"type"`
	Name   string             `json:"name,omitempty"`   // JSON Schema 名称（仅用于 json_schema 类型）
	Schema map[string]any     `json:"schema,omitempty"` // JSON Schema 定义（仅用于 json_schema 类型）
	Strict bool               `json:"strict,omitempty"` // 是否严格模式（OpenAI）
}

type ResponseFormatType string

type StreamChunk struct {
	// Type 块类型
	Type string `json:"type"`

	// Index 内容块索引（用于兼容 Anthropic 格式）
	Index int `json:"index,omitempty"`

	// Delta 增量数据（通用，兼容旧版）
	Delta any `json:"delta,omitempty"`

	// TextDelta 文本增量（新增，明确类型）
	TextDelta string `json:"text_delta,omitempty"`

	// ToolCall 工具调用增量（新增）
	ToolCall *ToolCallDelta `json:"tool_call,omitempty"`

	// Reasoning 推理过程（新增）
	Reasoning *ReasoningTrace `json:"reasoning,omitempty"`

	// Usage Token使用情况
	Usage *TokenUsage `json:"usage,omitempty"`

	// Error 错误信息（新增）
	Error *StreamError `json:"error,omitempty"`

	// FinishReason 完成原因（新增）
	FinishReason string `json:"finish_reason,omitempty"`
}

type StreamChunkType string

type StreamError struct {
	Code    string `json:"code"`
	Message string `json:"message"`
	Type    string `json:"type,omitempty"`
	Param   string `json:"param,omitempty"`
}

type StreamOptions struct {
	Tools       []ToolSchema
	MaxTokens   int
	Temperature float64
	System      string

	// ToolChoice 工具选择策略（Anthropic API 支持）
	// 可选值: nil (默认), "auto", "any", 或指定工具名
	ToolChoice *ToolChoiceOption `json:"tool_choice,omitempty"`

	// ResponseFormat 响应格式（用于结构化输出）
	// 支持 JSON Schema 强制输出特定格式的响应
	ResponseFormat *ResponseFormat `json:"response_format,omitempty"`

	// Thinking Extended Thinking 配置（Claude 专属）
	// 启用后模型会在响应前进行深度思考，思考过程会通过流式事件返回
	Thinking *ThinkingConfig `json:"thinking,omitempty"`
}

type ThinkingConfig struct {
	// Enabled 是否启用 extended thinking
	Enabled bool `json:"enabled"`

	// BudgetTokens 思考过程的 token 预算
	// Claude 建议范围: 1024 - 32000
	BudgetTokens int `json:"budget_tokens,omitempty"`
}

type TokenUsage struct {
	// 基础统计
	InputTokens  int64 `json:"input_tokens"`
	OutputTokens int64 `json:"output_tokens"`
	TotalTokens  int64 `json:"total_tokens,omitempty"`

	// 推理模型特殊统计
	ReasoningTokens int64 `json:"reasoning_tokens,omitempty"`

	// Prompt Caching 统计
	CachedTokens        int64 `json:"cached_tokens,omitempty"`
	CacheCreationTokens int64 `json:"cache_creation_tokens,omitempty"`
	CacheReadTokens     int64 `json:"cache_read_tokens,omitempty"`

	// 成本估算（新增）
	EstimatedCost float64 `json:"estimated_cost,omitempty"` // 估算成本 (USD)

	// 请求元数据（新增）
	RequestID string `json:"request_id,omitempty"` // API 请求 ID
	Model     string `json:"model,omitempty"`      // 使用的模型
	Provider  string `json:"provider,omitempty"`   // Provider 类型

	// 时间统计（新增）
	LatencyMs        int64 `json:"latency_ms,omitempty"`          // 请求总耗时 (ms)
	TimeToFirstToken int64 `json:"time_to_first_token,omitempty"` // 首 token 时间 (ms)
}

type ToolCallDelta struct {
	Index          int    `json:"index"`
	ID             string `json:"id,omitempty"`
	Type           string `json:"type,omitempty"`
	Name           string `json:"name,omitempty"`
	ArgumentsDelta string `json:"arguments_delta,omitempty"`
}

type ToolChoiceOption struct {
	// Type 选择类型: "auto", "any", "tool"
	Type string `json:"type"`

	// Name 当 Type="tool" 时，指定工具名称
	Name string `json:"name,omitempty"`

	// DisableParallelToolUse 禁用并行工具调用
	DisableParallelToolUse bool `json:"disable_parallel_tool_use,omitempty"`
}

type ToolExample struct {
	Description string         `json:"description"`
	Input       map[string]any `json:"input"`
	Output      any            `json:"output,omitempty"`
}

type ToolSchema struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	InputSchema map[string]any `json:"input_schema"`

	// InputExamples 工具使用示例，帮助 LLM 更准确地调用工具
	// 参考 Anthropic 的 Tool Use Examples 功能
	InputExamples []ToolExample `json:"input_examples,omitempty"`

	// AllowedCallers 指定哪些上下文可以调用此工具 (PTC 支持)
	// 可选值: ["direct"], ["code_execution_20250825"], 或两者组合
	// 默认: nil 或 ["direct"] - 仅 LLM 直接调用
	AllowedCallers []string `json:"allowed_callers,omitempty"`
}