genai

type GenerateStats struct {
	DoneReason         string
	TotalDuration      time.Duration
	LoadDuration       time.Duration
	PromptTokens       int64
	PromptEvalDuration time.Duration
	TokenCount         int64
	EvalDuration       time.Duration
}

type GenerativeAI interface {
	Chat(messages []Message) (string, GenerateStats, error)
	Complete(prompt string) (string, GenerateStats, error)
}

type Message struct {
	Role    string
	Content string
}

type Mode int

type Ollama struct {
	Client  *api.Client
	Model   string
	Options map[string]any
}

type OllamaConfig struct {
	BaseURL string
	Model   string
	Options map[string]any
}

type OpenAI struct {
	Client           *openai.Client
	Model            string
	FrequencyPenalty float64
	MaxTokens        int64
	PresencePenalty  float64
	ReasoningEffort  string
	Stop             string
	Temperature      float64
	TopP             float64
}

type OpenAIConfig struct {
	BaseURL          string
	APIKey           string
	Model            string
	FrequencyPenalty float64
	MaxTokens        int64
	PresencePenalty  float64
	ReasoningEffort  string
	Stop             string
	Temperature      float64
	TopP             float64
}

type Provider int

type ProviderConfig interface {
	Validate() error
}

type ProviderFactory func(ProviderConfig) (GenerativeAI, error)