From ac8f30f45cc0bcba3977c97ff7b32134d761189f Mon Sep 17 00:00:00 2001
From: "Gao,Wei" <kaovip@gmail.com>
Date: Thu, 4 Sep 2025 15:27:36 +0800
Subject: [PATCH] Update OllamaEmbeddingEngine.py

Compatible private embeddeding model (openai format)
---
 .../databases/vector/embeddings/OllamaEmbeddingEngine.py         | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
index e1237a657..29c57ed2e 100644
--- 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")