From 08849dfd5e7641e5639d3efb7eb4940ba7bc1d63 Mon Sep 17 00:00:00 2001 From: Zanie Blue Date: Fri, 17 Oct 2025 16:27:26 -0500 Subject: [PATCH 1/4] Remove graphrag submodule --- notebooks/data/graphrag | 1 - 1 file changed, 1 deletion(-) delete mode 160000 notebooks/data/graphrag diff --git a/notebooks/data/graphrag b/notebooks/data/graphrag deleted file mode 160000 index 130b84db9..000000000 --- a/notebooks/data/graphrag +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 130b84db9270734756d16918e5c86034777140fc From b47cb7462d1585794a28261e3e7a2981bba4aac0 Mon Sep 17 00:00:00 2001 From: chinu0609 Date: Wed, 22 Oct 2025 19:55:45 +0530 Subject: [PATCH 2/4] fix: Update code for Ollama API compatibility with newer version --- .../databases/vector/embeddings/OllamaEmbeddingEngine.py | 2 +- .../databases/vector/lancedb/LanceDBAdapter.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py index e79ba3f6a..d18ab1269 100644 --- a/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py @@ -111,7 +111,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine): self.endpoint, json=payload, headers=headers, timeout=60.0 ) as response: data = await response.json() - return data["embedding"] + return data["embeddings"] 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 91f20898e..116ebac36 100644 --- a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +++ b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py @@ -181,7 +181,9 @@ class LanceDBAdapter(VectorDBInterface): def create_lance_data_point(data_point: DataPoint, vector: list[float]) -> LanceDataPoint: 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, @@ -324,7 +326,6 @@ class LanceDBAdapter(VectorDBInterface): def get_data_point_schema(self, model_type: BaseModel): related_models_fields = [] - for field_name, field_config in model_type.model_fields.items(): if hasattr(field_config, "model_fields"): related_models_fields.append(field_name) From 7b31b86f106f4edddca300325182e47d44415552 Mon Sep 17 00:00:00 2001 From: chinu0609 Date: Wed, 22 Oct 2025 20:55:59 +0530 Subject: [PATCH 3/4] 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 4/4] 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: