model

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

type AudioContent struct {
	Data       []byte `json:"-"`
	Format     string `json:"format"`               // wav, mp3, ogg, etc.
	Transcript string `json:"transcript,omitempty"` // transcribed text (for input) or source text (for output)
}

type AudioProvider interface {
	// Transcribe converts audio data to text using a speech-to-text model (e.g., Whisper).
	Transcribe(ctx context.Context, audio AudioContent) (string, error)
	// Synthesize converts text to speech using a TTS model.
	// voice is the voice ID (e.g., "alloy", "echo", "fable", "onyx", "nova", "shimmer").
	Synthesize(ctx context.Context, text, voice string) (*AudioContent, error)
}

type AzureConfig struct {
	ProviderConfig
	Deployment string `json:"deployment"`
	APIVersion string `json:"api_version"`
}

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

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

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

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

type ChatRequest struct {
	Model       string           `json:"model"`
	Messages    []Message        `json:"messages"`
	MaxTokens   int              `json:"max_tokens,omitempty"`
	Temperature float64          `json:"temperature,omitempty"`
	TopP        float64          `json:"top_p,omitempty"`
	Stream      bool             `json:"stream,omitempty"`
	Tools       []ToolDefinition `json:"tools,omitempty"`
	Stop        []string         `json:"stop,omitempty"`
	// ResponseFormat controls output format. Values:
	//   "json_object"  — model returns valid JSON (no schema enforcement)
	//   "json_schema"  — model returns JSON conforming to Metadata["json_schema"]
	ResponseFormat string         `json:"response_format,omitempty"`
	Metadata       map[string]any `json:"metadata,omitempty"`
}

type ChatResponse struct {
	ID         string     `json:"id"`
	Content    string     `json:"content"`
	Role       string     `json:"role"`
	Usage      Usage      `json:"usage"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	StopReason StopReason `json:"stop_reason,omitempty"`
	// Delta is true when this is a partial streaming response.
	Delta bool `json:"delta,omitempty"`
}

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

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

type ContentPart struct {
	Type     string `json:"type"`                // "text", "image_url", "file", "audio"
	Text     string `json:"text,omitempty"`      // for type "text"
	ImageURL string `json:"image_url,omitempty"` // for type "image_url" — URL or base64 data URI
	MimeType string `json:"mime_type,omitempty"` // MIME type for image or file
	FileName string `json:"file_name,omitempty"` // original filename for attachments
	Data     []byte `json:"-"`                   // raw binary data (not serialized to JSON)
}

type EmbeddingRequest struct {
	Model string   `json:"model"`
	Input []string `json:"input"`
}

type EmbeddingResponse struct {
	Embeddings [][]float32 `json:"embeddings"`
	Usage      Usage       `json:"usage"`
}

type EmbeddingsProvider interface {
	Embed(ctx context.Context, req *EmbeddingRequest) (*EmbeddingResponse, error)
}

type EstimatingCounter struct {
	CharsPerToken float64
}

type FallbackProvider struct {

	// OnFallback is called when a provider fails and the next one is tried.
	// It receives the failed provider index, its name, and the error.
	OnFallback func(index int, name string, err error)
	// contains filtered or unexported fields
}

type FunctionDef struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Parameters  map[string]any `json:"parameters"` // JSON Schema
}

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

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

type Message struct {
	Role       string         `json:"role"` // system, user, assistant, tool
	Content    string         `json:"content"`
	Parts      []ContentPart  `json:"parts,omitempty"` // multi-modal content parts
	Audio      []AudioContent `json:"audio,omitempty"` // audio input/output attachments
	Name       string         `json:"name,omitempty"`
	ToolCallID string         `json:"tool_call_id,omitempty"`
	ToolCalls  []ToolCall     `json:"tool_calls,omitempty"`
}

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

type ModelRef struct {
	Provider string
	ModelID  string
}

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

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

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

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

type OpenAIAudioConfig struct {
	// APIKey is the OpenAI API key.
	APIKey string `json:"api_key"`
	// BaseURL overrides the default OpenAI base URL.
	BaseURL string `json:"base_url,omitempty"`
	// OrgID is the optional OpenAI organization ID.
	OrgID string `json:"org_id,omitempty"`
	// TranscribeModel is the Whisper model to use for transcription (default: "whisper-1").
	TranscribeModel string `json:"transcribe_model,omitempty"`
	// TTSModel is the TTS model to use for speech synthesis (default: "tts-1").
	TTSModel string `json:"tts_model,omitempty"`
	// TTSFormat is the output audio format for TTS (default: "mp3").
	// Supported: "mp3", "opus", "aac", "flac", "wav", "pcm".
	TTSFormat string `json:"tts_format,omitempty"`
	// TimeoutSec is the HTTP request timeout in seconds (default: 120).
	TimeoutSec int `json:"timeout_sec,omitempty"`
}

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

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

type Provider interface {
	// Chat sends a request and returns a complete response.
	Chat(ctx context.Context, req *ChatRequest) (*ChatResponse, error)
	// StreamChat returns a channel of partial responses for streaming.
	// The channel is closed when the response is complete.
	StreamChat(ctx context.Context, req *ChatRequest) (<-chan *ChatResponse, error)
	// Name returns a human-readable name for this provider.
	Name() string
	// Model returns the default model ID for this provider.
	Model() string
}

type ProviderConfig struct {
	APIKey        string `json:"api_key"`
	BaseURL       string `json:"base_url,omitempty"`
	Model         string `json:"model"`
	MaxRetries    int    `json:"max_retries,omitempty"`
	TimeoutSec    int    `json:"timeout_sec,omitempty"`
	OrgID         string `json:"org_id,omitempty"`
	ContextWindow int    `json:"context_window,omitempty"` // override default context window size for the model
}

type ProviderFactory func(ref ModelRef) (Provider, error)

type StopReason string

type SummarizationConfig struct {
	Threshold           float64 // fraction of context window that triggers summarization (default 0.8)
	PreserveRecentTurns int     // number of recent user/assistant pairs to keep verbatim
}

type SummarizationResult struct {
	Summary           string
	PreservedMessages []Message
	SummarizedCount   int
}

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

type TokenCounter interface {
	CountTokens(messages []Message) int
	CountString(s string) int
}

type ToolCall struct {
	ID        string `json:"id"`
	Name      string `json:"name"`
	Arguments string `json:"arguments"` // JSON string
}

type ToolDefinition struct {
	Type     string      `json:"type"` // "function"
	Function FunctionDef `json:"function"`
}

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