gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #1 from chinu0609/fix-for-newer-version-ollama

Browse source 
fix: Update code for Ollama API compatibility with newer version
This commit is contained in:
Chinmay Bhosale 2025-10-22 20:06:49 +05:30 • committed by GitHub
commit 8b9e30408c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/vector/embeddings/OllamaEmbeddingEngine.py
View file

@ -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:
"""

5
cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py
View file

@ -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)