provider

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

type AzureProvider struct {
	Endpoint       string
	DeploymentName string
	APIVersion     string
}

type CustomProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}

type DeepSeekModel struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type DeepSeekModelsResponse struct {
	Object string          `json:"object"`
	Data   []DeepSeekModel `json:"data"`
}

type DeepSeekProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}

type GeminiModel struct {
	Name                       string   `json:"name"`
	BaseModelID                string   `json:"baseModelId,omitempty"`
	Version                    string   `json:"version"`
	DisplayName                string   `json:"displayName"`
	Description                string   `json:"description"`
	InputTokenLimit            int      `json:"inputTokenLimit"`
	OutputTokenLimit           int      `json:"outputTokenLimit"`
	SupportedGenerationMethods []string `json:"supportedGenerationMethods"`
}

type GeminiModelsResponse struct {
	Models []GeminiModel `json:"models"`
}

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

type GrokModel struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type GrokModelsResponse struct {
	Object string      `json:"object"`
	Data   []GrokModel `json:"data"`
}

type GrokProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}

type MiniMaxModel struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type MiniMaxModelsResponse struct {
	Object string         `json:"object"`
	Data   []MiniMaxModel `json:"data"`
}

type MiniMaxProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}

type OllamaModel struct {
	Name       string    `json:"name"`
	Model      string    `json:"model"`
	ModifiedAt time.Time `json:"modified_at"`
	Size       int64     `json:"size"`
	Digest     string    `json:"digest"`
	Details    struct {
		Format            string   `json:"format"`
		Family            string   `json:"family"`
		Families          []string `json:"families"`
		ParameterSize     string   `json:"parameter_size"`
		QuantizationLevel string   `json:"quantization_level"`
	} `json:"details"`
}

type OllamaProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}

type OllamaTagsResponse struct {
	Models []OllamaModel `json:"models"`
}

type OpenAIModel struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type OpenAIModelsResponse struct {
	Data   []OpenAIModel `json:"data"`
	Object string        `json:"object"`
}

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

type OpenRouterModel struct {
	ID               string                     `json:"id"`
	Name             string                     `json:"name"`
	Description      string                     `json:"description"`
	ContextLength    int                        `json:"context_length"`
	Pricing          OpenRouterModelPricing     `json:"pricing"`
	TopProvider      OpenRouterModelTopProvider `json:"top_provider"`
	PerRequestLimits *OpenRouterModelLimits     `json:"per_request_limits,omitempty"`
}

type OpenRouterModelInfo struct {
	ID            string
	Name          string
	Description   string
	ContextLength int
	InputPrice    float64 // Price per 1M tokens
	OutputPrice   float64 // Price per 1M tokens
	Provider      string  // Extracted from ID (e.g., "openai" from "openai/gpt-4o")
}

type OpenRouterModelLimits struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
}

type OpenRouterModelPricing struct {
	Prompt     string `json:"prompt"`     // Price per token for prompt
	Completion string `json:"completion"` // Price per token for completion
}

type OpenRouterModelTopProvider struct {
	ContextLength  int  `json:"context_length"`
	MaxCompletions int  `json:"max_completion_tokens"`
	IsModerated    bool `json:"is_moderated"`
}

type OpenRouterModelsResponse struct {
	Data []OpenRouterModel `json:"data"`
}

type OpenRouterProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}

type Provider interface {
	// Name returns the provider name.
	Name() string

	// GetHeaders returns the HTTP headers for API requests.
	GetHeaders(apiKey string) http.Header

	// GetEndpointURL returns the full URL for chat completions.
	GetEndpointURL() string

	// TransformModelID transforms a model ID for the provider.
	TransformModelID(modelID string) string

	// SupportsStreaming indicates if the provider supports SSE streaming.
	SupportsStreaming() bool

	// RequiresTransformation indicates if the provider needs Anthropic->OpenAI translation.
	RequiresTransformation() bool
}

type QwenModel struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type QwenModelsResponse struct {
	Object string      `json:"object"`
	Data   []QwenModel `json:"data"`
}

type QwenProvider struct {
	BaseURL string
	// contains filtered or unexported fields
}