providers

type AnthropicProvider struct {
	*BaseProvider
}

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

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
	ToolCalls    []ToolCall
	ToolCallID   string
	CacheControl *CacheControl
}

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

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
	Stream           bool
	Tools            []Tool
	ToolChoice       any
	ResponseFormat   *ResponseFormat
	Stop             []string
	ReasoningEffort  string
	ReasoningSummary string
	UseResponsesAPI  bool

	// 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
	ToolCalls    []ToolCall
	Usage        Usage
	Model        string
	Provider     string
	FinishReason string
	Reasoning    *ReasoningData
}

type ResponseFormat struct {
	Type       string
	JSONSchema *JSONSchema
}

type StreamChunk struct {
	Type          string
	Content       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
}

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
}