From 3a6bb778e2a386ca2be4ffc85f437086e786f123 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 20 Jan 2026 20:50:59 +0100 Subject: [PATCH] refactor: Handle embedding case in data --- .../databases/vector/embeddings/OllamaEmbeddingEngine.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py index 1d5e7fbfe..74d33a76c 100644 --- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py @@ -124,6 +124,8 @@ class OllamaEmbeddingEngine(EmbeddingEngine): data = await response.json() if "embeddings" in data: return data["embeddings"][0] + if "embedding" in data: + return data["embedding"] else: return data["data"][0]["embedding"]