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")