types

type BaseConfig struct {
	Provider   string `json:"provider"`
	APIKey     string `json:"api_key"`
	BaseURL    string `json:"base_url,omitempty"`
	Timeout    int    `json:"timeout,omitempty"` // in seconds
	MaxRetries int    `json:"max_retries,omitempty"`
}

type CompletionRequest struct {
	Messages       []*Message             `json:"messages"`
	Model          string                 `json:"model"`
	MaxTokens      int                    `json:"max_tokens,omitempty"`
	Temperature    float64                `json:"temperature,omitempty"`
	TopP           float64                `json:"top_p,omitempty"`
	TopK           int                    `json:"top_k,omitempty"`
	Seed           *int                   `json:"seed,omitempty"`
	Stop           []string               `json:"stop,omitempty"`
	Stream         bool                   `json:"stream,omitempty"`
	Tools          []Tool                 `json:"tools,omitempty"`
	GroundingTools []GroundingTool        `json:"grounding_tools,omitempty"` // Google-specific: URL context, Google Search
	ToolChoice     interface{}            `json:"tool_choice,omitempty"`
	ThinkingConfig *ThinkingConfig        `json:"thinking_config,omitempty"`
	ResponseFormat *ResponseFormat        `json:"response_format,omitempty"`
	Metadata       map[string]interface{} `json:"metadata,omitempty"`
}

type CompletionResponse struct {
	ID           string                 `json:"id"`
	Model        string                 `json:"model"`
	Provider     string                 `json:"provider"`
	Message      *Message               `json:"message,omitempty"`
	FinishReason string                 `json:"finish_reason,omitempty"`
	Usage        *Usage                 `json:"usage,omitempty"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
	Created      int64                  `json:"created,omitempty"`
}

type Config interface {
	GetProvider() string
	Validate() error
}

type Error struct {
	Code     string                 `json:"code"`
	Message  string                 `json:"message"`
	Provider string                 `json:"provider"`
	Details  map[string]interface{} `json:"details,omitempty"`
	Cause    error                  `json:"-"`
}

type GroundingTool struct {
	Type string `json:"type"` // "url_context" or "google_search"
}

type ImageContent struct {
	URL    string `json:"url,omitempty"`
	Base64 string `json:"base64,omitempty"`
	Detail string `json:"detail,omitempty"` // "low", "high", "auto"
}

type Message struct {
	ID         string                 `json:"id,omitempty"`
	Role       Role                   `json:"role"`
	Content    []MessageContent       `json:"content,omitempty"`
	TextData   string                 `json:"text_data,omitempty"` // For simple text messages
	ToolCalls  []ToolCall             `json:"tool_calls,omitempty"`
	ToolResult *ToolResult            `json:"tool_result,omitempty"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
	Timestamp  time.Time              `json:"timestamp,omitempty"`
}

type MessageContent interface {
	Type() string
}

type Model struct {
	ID           string                 `json:"id"`
	Name         string                 `json:"name"`
	Provider     string                 `json:"provider"`
	Description  string                 `json:"description,omitempty"`
	MaxTokens    int                    `json:"max_tokens,omitempty"`
	InputCost    float64                `json:"input_cost,omitempty"`   // Cost per 1M tokens
	OutputCost   float64                `json:"output_cost,omitempty"`  // Cost per 1M tokens
	Capabilities []string               `json:"capabilities,omitempty"` // e.g., "chat", "completion", "vision", "tools"
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
}

type ModelCapability string

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

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

	// Initialize sets up the provider with configuration
	Initialize(config Config) error

	// GetModels returns available models for this provider
	GetModels(ctx context.Context) ([]*Model, error)

	// Complete performs a completion request
	Complete(ctx context.Context, req *CompletionRequest) (*CompletionResponse, error)

	// Stream performs a streaming completion request
	Stream(ctx context.Context, req *CompletionRequest, callback StreamCallback) error

	// EstimateTokens estimates the token count for given messages
	EstimateTokens(ctx context.Context, messages []*Message, model string) (int, error)

	// ValidateModel checks if a model is supported by this provider
	ValidateModel(model string) error

	// Close cleans up resources
	Close() error
}

type ResponseFormat struct {
	Type   string                 `json:"type"` // "text" or "json_object"
	Schema map[string]interface{} `json:"schema,omitempty"`
}

type Role string

type StreamCallback func(ctx context.Context, response *StreamResponse) error

type StreamResponse struct {
	ID           string                 `json:"id"`
	Model        string                 `json:"model"`
	Provider     string                 `json:"provider"`
	Delta        *Message               `json:"delta,omitempty"`
	FinishReason string                 `json:"finish_reason,omitempty"`
	Usage        *Usage                 `json:"usage,omitempty"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
}

type TextContent struct {
	Text string `json:"text"`
}

type ThinkingConfig struct {
	// Optional. Indicates whether to include thoughts in the response. If true, thoughts
	// are returned only if the model supports thought and thoughts are available.
	IncludeThoughts bool `json:"includeThoughts,omitempty"`
	// Optional. Indicates the thinking budget in tokens.
	ThinkingBudget *int32 `json:"thinkingBudget,omitempty"`
}

type Tool struct {
	Type     string        `json:"type"`
	Function *ToolFunction `json:"function,omitempty"`
}

type ToolCall struct {
	ID       string                 `json:"id"`
	Type     string                 `json:"type"`
	Function ToolCallFunction       `json:"function"`
	Args     map[string]interface{} `json:"args,omitempty"`
}

type ToolCallFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

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

type ToolResult struct {
	ToolCallID string `json:"tool_call_id"`
	Content    string `json:"content"`
	Error      string `json:"error,omitempty"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}