From 72c20c256d80885ebf4cbd3375e28491471d6d8b Mon Sep 17 00:00:00 2001 From: "Gao,Wei" Date: Wed, 5 Nov 2025 17:44:08 +0800 Subject: [PATCH] Handle multiple response formats in OllamaEmbeddingEngine The OllamaEmbeddingEngine is compatible with OpenAI --- .../databases/vector/embeddings/OllamaEmbeddingEngine.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py index b0e763991..f4510f374 100644 --- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py @@ -124,7 +124,10 @@ class OllamaEmbeddingEngine(EmbeddingEngine): self.endpoint, json=payload, headers=headers, timeout=60.0 ) as response: data = await response.json() - return data["embeddings"][0] + if "embeddings" in data: + return data["embeddings"][0] + else: + return data["data"][0]["embedding"] def get_vector_size(self) -> int: """