Update OllamaEmbeddingEngine.py

Compatible private embeddeding model (openai format)
2025-09-04 15:27:36 +08:00 · 2025-09-04 15:27:36 +08:00 · ac8f30f45c
commit ac8f30f45c
parent 8b6aaff554
1 changed files with 1 additions and 0 deletions
--- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
@ -97,6 +97,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
        payload = {
            "model": self.model,
            "prompt": prompt,
+            "input": prompt
        }
        headers = {}
        api_key = os.getenv("LLM_API_KEY")