providers

type AnthropicProvider struct {
	*BaseProvider
}

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

type BedrockProvider struct {
	*BaseProvider
	// contains filtered or unexported fields
}

type BuiltinFactory func(ProviderConfig) Provider

type CacheControl struct {
	Type string // "ephemeral" or "persistent"
	TTL  *int   // Time to live in seconds (optional)
}

type DeepSeekProvider struct {
	*BaseProvider
}

type FunctionCall struct {
	Name      string
	Arguments string
}

type FunctionDef struct {
	Name        string
	Description string
	Parameters  any
}

type GLMProvider struct {
	*BaseProvider
}

type GeminiProvider struct {
	*BaseProvider
}

type HTTPDoer interface {
	Do(req *http.Request) (*http.Response, error)
}

type JSONSchema struct {
	Name        string
	Description string
	Schema      any
	Strict      *bool
}

type Message struct {
	Role         string
	Content      string
	Contents     []MessageContent
	ToolCalls    []ToolCall
	ToolCallID   string
	CacheControl *CacheControl
}

type MessageContent struct {
	Type        string
	Text        string
	ImageURL    *MessageImageURL
	Annotations []map[string]interface{}
	Logprobs    []map[string]interface{}
}

type MessageImageURL struct {
	URL string
}

type ModelInfo struct {
	ID              string
	Provider        string
	Name            string
	ContextWindow   int
	MaxOutputTokens int
	Capabilities    []string
}

type OpenAICompatMessage struct {
	Role             string           `json:"role"`
	Content          interface{}      `json:"content,omitempty"` // string or array (needed by OpenRouter)
	ToolCalls        []openaiToolCall `json:"tool_calls,omitempty"`
	ToolCallID       string           `json:"tool_call_id,omitempty"`
	ReasoningContent string           `json:"reasoning_content,omitempty"` // used by GLM/Qwen
}

type OpenAIProvider struct {
	*BaseProvider
}

type OpenRouterProvider struct {
	*BaseProvider
}

type Provider interface {
	Name() string
	Validate() error
	SupportsModel(model string) bool
	Models() []ModelInfo
	Chat(ctx context.Context, req *Request) (*Response, error)
	Stream(ctx context.Context, req *Request) (StreamReader, error)
}

type ProviderConfig struct {
	APIKey     string           `json:"api_key"`
	BaseURL    string           `json:"base_url,omitempty"`
	Timeout    time.Duration    `json:"timeout,omitempty"`
	Extra      map[string]any   `json:"extra,omitempty"`
	Resilience ResilienceConfig `json:"resilience,omitempty"`
	HTTPClient HTTPDoer         `json:"-"`
}

type QwenProvider struct {
	*BaseProvider
}

type ReasoningChunk struct {
	Summary string
	Content string
}

type ReasoningData struct {
	Summary    string
	Content    string
	TokensUsed int
}

type Request struct {
	Model             string
	Messages          []Message
	MaxTokens         *int
	Temperature       *float64
	TopP              *float64
	TopLogProbs       *int
	Stream            bool
	Tools             []Tool
	ToolChoice        any
	ResponseFormat    *ResponseFormat
	Stop              []string
	ReasoningEffort   string
	ReasoningSummary  string
	UseResponsesAPI   bool
	ServiceTier       string
	Store             *bool
	ParallelToolCalls *bool
	SafetyIdentifier  string

	ResponsesParams *ResponsesParams

	// Provider-specific extensions
	Extra map[string]interface{}
}

type ResilienceConfig struct {
	MaxRetries     int           `json:"max_retries"`
	InitialDelay   time.Duration `json:"initial_delay"`
	MaxDelay       time.Duration `json:"max_delay"`
	Multiplier     float64       `json:"multiplier"`
	Jitter         bool          `json:"jitter"`
	RequestTimeout time.Duration `json:"request_timeout"`
	ConnectTimeout time.Duration `json:"connect_timeout"`
}

type Response struct {
	Content      string
	Contents     []MessageContent
	ToolCalls    []ToolCall
	Usage        Usage
	Model        string
	Provider     string
	FinishReason string
	Reasoning    *ReasoningData
}

type ResponseFormat struct {
	Type       string
	JSONSchema *JSONSchema
}

type ResponsesParams struct {
	Instructions         string
	Conversation         string
	PreviousResponseID   string
	Metadata             map[string]string
	Store                *bool
	MaxOutputTokens      *int
	MaxInputTokens       *int
	MaxToolCalls         *int
	ParallelToolCalls    *bool
	Include              []string
	SafetyIdentifier     string
	ServiceTier          string
	Temperature          *float64
	TopP                 *float64
	ToolChoice           any
	ResponseFormat       *ResponseFormat
	PromptCacheKey       string
	PromptCacheRetention string
	Background           *bool
	Prompt               map[string]interface{}
	ModelOverride        string
}

type StreamChunk struct {
	Type          string
	Content       string
	ContentIndex  *int
	OutputIndex   *int
	ItemID        string
	ToolCallDelta *ToolCallDelta
	FinishReason  string
	Model         string
	Provider      string
	Done          bool
	Reasoning     *ReasoningChunk
	Usage         *Usage
}

type StreamReader interface {
	Next() (*StreamChunk, error)
	Close() error
}

type Tool struct {
	Type     string
	Function FunctionDef
}

type ToolCall struct {
	ID       string
	Type     string
	Function FunctionCall
}

type ToolCallDelta struct {
	Index          int
	ID             string
	Type           string
	FunctionName   string
	ArgumentsDelta string
	OutputIndex    *int
	ItemID         string
}

type Usage struct {
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
	ReasoningTokens  int

	// Cache-related token statistics
	CacheCreationInputTokens int // Tokens written to cache
	CacheReadInputTokens     int // Tokens read from cache
}