diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
index e1237a6577..29c57ed2ea 100644
--- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
@@ -97,6 +97,7 @@ async def _get_embedding(self, prompt: str) -> List[float]:
         payload = {
             "model": self.model,
             "prompt": prompt,
+            "input": prompt
         }
         headers = {}
         api_key = os.getenv("LLM_API_KEY")