Updated embedder model

phidatahq · Nov 19, 2024 · 0a07b04 · 0a07b04
1 parent a4159db
commit 0a07b04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cookbook/providers/ollama_tools/knowledge.py b/cookbook/providers/ollama_tools/knowledge.py
@@ -3,7 +3,7 @@
 
 Run Ollama Server: `ollama serve`
 Pull required models:
-`ollama pull openhermes`
+`ollama pull nomic-embed-text`
 `ollama pull llama3.1:8b`
 
 If you haven't deployed database yet, run:
@@ -25,7 +25,7 @@
     vector_db=PgVector(
         table_name="recipes",
         db_url=db_url,
-        embedder=OllamaEmbedder(host="http://localhost:11434"),
+        embedder=OllamaEmbedder(model="nomic-embed-text", host="http://localhost:11434", dimensions=768),
     ),
 )
 knowledge_base.load(recreate=False)  # Comment out after first run