diff options
Diffstat (limited to 'internal/llm/ollama.go')
| -rw-r--r-- | internal/llm/ollama.go | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/internal/llm/ollama.go b/internal/llm/ollama.go index a22dd7b..a878b62 100644 --- a/internal/llm/ollama.go +++ b/internal/llm/ollama.go @@ -40,6 +40,19 @@ type ollamaChatResponse struct { Error string `json:"error,omitempty"` } +func init() { + RegisterProvider("ollama", ollamaProviderFactory) +} + +func ollamaProviderFactory(cfg Config, _ ProviderKeys) (Client, error) { + return newOllamaWithTimeout( + cfg.OllamaBaseURL, + cfg.OllamaModel, + withDefaultTemperature(cfg.OllamaTemperature, 0.2), + cfg.RequestTimeout, + ), nil +} + // Constructor (kept among the first functions by convention) func newOllama(baseURL, model string, defaultTemp *float64) Client { return newOllamaWithTimeout(baseURL, model, defaultTemp, 0) |
