use env var to determine ollama model

rwth-acis · Jan 26, 2024 · 1fdfa08 · 1fdfa08
1 parent c4975de
commit 1fdfa08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
 	}