From 1fdfa0859873e368381bf58a7c01c9df5b0f22ed Mon Sep 17 00:00:00 2001 From: ili16 Date: Fri, 26 Jan 2024 13:11:08 +0000 Subject: [PATCH] use env var to determine ollama model --- ollama/ollama.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ollama/ollama.go b/ollama/ollama.go index 3f0b887..8b13ace 100644 --- a/ollama/ollama.go +++ b/ollama/ollama.go @@ -14,7 +14,7 @@ func CreateEmbedding(prompt string) ([]float32, error) { url := os.Getenv("OLLAMA_URL") + "/api/embeddings" data := map[string]interface{}{ - "model": "codellama:13b-instruct", + "model": os.Getenv("OLLAMA_MODEL"), "prompt": prompt, }