summaryrefslogtreecommitdiff
path: root/internal/llm/ollama.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/llm/ollama.go')
-rw-r--r--internal/llm/ollama.go13
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)