From b47cb7462d1585794a28261e3e7a2981bba4aac0 Mon Sep 17 00:00:00 2001 From: chinu0609 Date: Wed, 22 Oct 2025 19:55:45 +0530 Subject: [PATCH] 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)