llm

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

type CompletionOptions struct {
	Temperature   *float32
	MaxTokens     *int64
	TopP          *float32
	StopSequences []string
}

type Config struct {
	Provider    ProviderType
	Model       string
	Temperature float32
	MaxTokens   int
	Timeout     time.Duration

	// Ollama specific
	OllamaURL string

	// Azure specific
	AzureOpenAIEndpoint string
	AzureOpenAIKey      string
	AzureOpenAIVersion  string

	// OpenAI specific
	OpenAIAPIKey string
}

type Message struct {
	Role    string // "system", "user", "assistant"
	Content string
}

type ModelInfo struct {
	Name            string
	Provider        ProviderType
	ContextSize     int
	MaxTokens       int
	CostPer1kTokens float64 // in USD
	Capabilities    []string
}

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

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

type Provider interface {
	// Complete generates text completion
	Complete(ctx context.Context, prompt string, options *CompletionOptions) (string, error)

	// Chat performs a chat completion with message history
	Chat(ctx context.Context, messages []*Message, options *CompletionOptions) (string, error)

	// GetModel returns information about the configured model
	GetModel() *ModelInfo

	// IsHealthy checks if the provider is available and healthy
	IsHealthy(ctx context.Context) (bool, error)

	// GetProvider returns the provider type
	GetProvider() ProviderType
}

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

type ProviderType string