llm

type AnthropicConfig struct {
	APIKey       string
	IsOAuthToken bool   // True if APIKey is an OAuth access token (uses Bearer auth)
	BaseURL      string // Optional custom endpoint
	Model        string
	MaxTokens    int
	Thinking     ThinkingConfig
	Retry        RetryConfig
}

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

type ChatRequest struct {
	Messages  []Message `json:"messages"`
	Tools     []ToolDef `json:"tools,omitempty"`
	MaxTokens int       `json:"max_tokens,omitempty"`
}

type ChatResponse struct {
	Content      string             `json:"content"`
	Thinking     string             `json:"thinking,omitempty"`
	ToolCalls    []ToolCallResponse `json:"tool_calls,omitempty"`
	StopReason   string             `json:"stop_reason"`
	InputTokens  int                `json:"input_tokens"`
	OutputTokens int                `json:"output_tokens"`
	Model        string             `json:"model"`
	TTFTMs       int64              `json:"ttft_ms,omitempty"` // Time to first token in milliseconds
}

type GoogleConfig struct {
	APIKey    string
	Model     string
	MaxTokens int
	Thinking  ThinkingConfig
	Retry     RetryConfig
}

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

type Message struct {
	Role       string             `json:"role"` // user, assistant, tool, system
	Content    string             `json:"content"`
	ToolCalls  []ToolCallResponse `json:"tool_calls,omitempty"`
	ToolCallID string             `json:"tool_call_id,omitempty"` // For tool result messages
}

type MockProvider struct {

	// ChatFunc can be overridden for custom behavior
	ChatFunc func(ctx context.Context, req ChatRequest) (*ChatResponse, error)
	// contains filtered or unexported fields
}

type OllamaCloudConfig struct {
	APIKey    string
	BaseURL   string // defaults to https://ollama.com
	Model     string
	MaxTokens int
	Thinking  ThinkingConfig
	Retry     RetryConfig
}

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

type OpenAICompatConfig struct {
	APIKey       string
	BaseURL      string
	Model        string
	MaxTokens    int
	ProviderName string // For logging/identification
	Thinking     ThinkingConfig
	Retry        RetryConfig
}

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

type OpenAIConfig struct {
	APIKey    string
	BaseURL   string // Optional custom endpoint
	Model     string
	MaxTokens int
	Thinking  ThinkingConfig
	Retry     RetryConfig
}

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

type Provider interface {
	// Chat sends a chat request and returns the response.
	Chat(ctx context.Context, req ChatRequest) (*ChatResponse, error)
}

type ProviderConfig struct {
	Provider     string         `json:"provider"` // anthropic, openai, google, groq, mistral, openai-compat
	Model        string         `json:"model"`
	APIKey       string         `json:"api_key"`
	IsOAuthToken bool           `json:"is_oauth_token"` // True if APIKey is an OAuth access token (Anthropic only)
	MaxTokens    int            `json:"max_tokens"`
	BaseURL      string         `json:"base_url"` // Custom API endpoint (for OpenRouter, LiteLLM, Ollama, LMStudio)
	Thinking     ThinkingConfig `json:"thinking"` // Thinking/reasoning configuration
	RetryConfig  RetryConfig    `json:"retry"`    // Retry configuration
}

type ProviderFactory interface {
	// GetProvider returns a provider for the given profile name.
	// Empty profile name returns the default provider.
	GetProvider(profile string) (Provider, error)
}

type RetryConfig struct {
	MaxRetries  int           `json:"max_retries"`  // Max retry attempts (default 5)
	MaxBackoff  time.Duration `json:"max_backoff"`  // Max backoff duration (default 60s)
	InitBackoff time.Duration `json:"init_backoff"` // Initial backoff (default 1s)
}

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

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

type ThinkingConfig struct {
	// Level: "auto", "off", "low", "medium", "high"
	// Auto uses heuristic classifier to determine level per-request.
	Level ThinkingLevel

	// BudgetTokens for Anthropic extended thinking (optional, 0 = provider default)
	BudgetTokens int64
}

type ThinkingLevel string

type ToolCallResponse struct {
	ID   string                 `json:"id"`
	Name string                 `json:"name"`
	Args map[string]interface{} `json:"args"`
}

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

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