Update with user defined embed functionality from ollama

TheAiSingularity · Aug 14, 2024 · 4c25f02 · 4c25f02
1 parent 7487893
commit 4c25f02
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphrag/llm/openai/openai_embeddings_llm.py b/graphrag/llm/openai/openai_embeddings_llm.py
@@ -28,6 +28,6 @@ async def _execute_llm(
         }
         embedding_list = []
         for inp in input:
-            embedding = ollama.embeddings(model="nomic-embed-text", prompt=inp)
+            embedding = ollama.embeddings(model=self._configuration.model, prompt=inp)
             embedding_list.append(embedding["embedding"])
-        return embedding_list
+        return embedding_list