From 7b31b86f106f4edddca300325182e47d44415552 Mon Sep 17 00:00:00 2001 From: chinu0609 Date: Wed, 22 Oct 2025 20:55:59 +0530 Subject: [PATCH 1/2] fix: reverting the lancedb chnage --- .../databases/vector/embeddings/OllamaEmbeddingEngine.py | 3 ++- .../infrastructure/databases/vector/lancedb/LanceDBAdapter.py | 2 -- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py index d18ab1269..fb10e57c9 100644 --- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py @@ -111,7 +111,8 @@ class OllamaEmbeddingEngine(EmbeddingEngine): self.endpoint, json=payload, headers=headers, timeout=60.0 ) as response: data = await response.json() - return data["embeddings"] + print(data) + return data["embeddings"][0] def get_vector_size(self) -> int: """ diff --git a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py index 116ebac36..31243306d 100644 --- a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +++ b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py @@ -182,8 +182,6 @@ class LanceDBAdapter(VectorDBInterface): properties = get_own_properties(data_point) properties["id"] = str(properties["id"]) - if isinstance(vector, list) and len(vector) == 1 and isinstance(vector[0], list): - vector = vector[0] return LanceDataPoint[str, self.get_data_point_schema(type(data_point))]( id=str(data_point.id), vector=vector, From e6ab4bbeee46046907dca7eb9ffb5895e812e7d8 Mon Sep 17 00:00:00 2001 From: chinu0609 Date: Wed, 22 Oct 2025 20:58:45 +0530 Subject: [PATCH 2/2] fix: reverting the lancedb chnage --- .../databases/vector/embeddings/OllamaEmbeddingEngine.py | 1 - 1 file changed, 1 deletion(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py index fb10e57c9..89027dcd1 100644 --- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py @@ -111,7 +111,6 @@ class OllamaEmbeddingEngine(EmbeddingEngine): self.endpoint, json=payload, headers=headers, timeout=60.0 ) as response: data = await response.json() - print(data) return data["embeddings"][0] def get_vector_size(self) -> int: