provider

type Capabilities struct {
	Tools       bool // Supports tool/function calling
	Vision      bool // Supports image input
	Streaming   bool // Supports streaming responses
	Reasoning   bool // Supports extended thinking/reasoning
	Temperature bool // Supports temperature parameter
}

type EmbeddingProvider interface {
	EmbeddingModel(modelID string) model.EmbeddingModel
}

type ImageProvider interface {
	ImageModel(modelID string) model.ImageModel
}

type LanguageProvider interface {
	LanguageModel(modelID string) model.LanguageModel
}

type ModelInfo struct {
	ID           string
	Name         string
	Provider     string
	ContextLimit int
	OutputLimit  int
	Capabilities Capabilities
}

type Options struct {
	// APIKey is the API key for authentication.
	APIKey string

	// BaseURL overrides the default API endpoint.
	BaseURL string

	// Headers are additional HTTP headers to send.
	Headers map[string]string

	// Organization is the organization ID (for providers that support it).
	Organization string

	// Project is the project ID (for providers that support it, e.g., OpenAI).
	Project string
}

type Provider interface {
	// ID returns the provider identifier (e.g., "openai", "anthropic").
	ID() string

	// Model returns a language model instance for the given model ID.
	// This is the primary method for getting a model.
	Model(modelID string) stream.Model

	// LanguageModel returns a language model instance.
	// Returns nil if the model is not supported.
	LanguageModel(modelID string) model.LanguageModel

	// ImageModel returns an image generation model instance.
	// Returns nil if the provider doesn't support image generation.
	ImageModel(modelID string) model.ImageModel

	// EmbeddingModel returns an embedding model instance.
	// Returns nil if the provider doesn't support embeddings.
	EmbeddingModel(modelID string) model.EmbeddingModel

	// SpeechModel returns a text-to-speech model instance.
	// Returns nil if the provider doesn't support speech generation.
	SpeechModel(modelID string) model.SpeechModel

	// TranscriptionModel returns a speech-to-text model instance.
	// Returns nil if the provider doesn't support transcription.
	TranscriptionModel(modelID string) model.TranscriptionModel

	// RerankingModel returns a reranking model instance.
	// Returns nil if the provider doesn't support reranking.
	RerankingModel(modelID string) model.RerankingModel

	// Models returns a list of available model IDs.
	Models() []string
}

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

type RerankingProvider interface {
	RerankingModel(modelID string) model.RerankingModel
}

type SpeechProvider interface {
	SpeechModel(modelID string) model.SpeechModel
}

type StreamInput struct {
	*stream.Input
	Context context.Context
}

type TranscriptionProvider interface {
	TranscriptionModel(modelID string) model.TranscriptionModel
}