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

Merge branch 'dev' into refactor/update-web-parsing

Browse source
This commit is contained in:
Vasilije 2025-10-22 18:05:54 +02:00 • committed by GitHub
commit 6c9b3d6385
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 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

@ -124,7 +124,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
self.endpoint, json=payload, headers=headers, timeout=60.0 self.endpoint, json=payload, headers=headers, timeout=60.0
) as response: ) as response:
data = await response.json() data = await response.json()
return data["embedding"] return data["embeddings"][0]
def get_vector_size(self) -> int: def get_vector_size(self) -> int:
""" """

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

@ -181,7 +181,7 @@ class LanceDBAdapter(VectorDBInterface):
def create_lance_data_point(data_point: DataPoint, vector: list[float]) -> LanceDataPoint: def create_lance_data_point(data_point: DataPoint, vector: list[float]) -> LanceDataPoint:
properties = get_own_properties(data_point) properties = get_own_properties(data_point)
properties["id"] = str(properties["id"]) properties["id"] = str(properties["id"])
return LanceDataPoint[str, self.get_data_point_schema(type(data_point))]( return LanceDataPoint[str, self.get_data_point_schema(type(data_point))](
id=str(data_point.id), id=str(data_point.id),
vector=vector, vector=vector,
@ -324,7 +324,6 @@ class LanceDBAdapter(VectorDBInterface):
def get_data_point_schema(self, model_type: BaseModel): def get_data_point_schema(self, model_type: BaseModel):
related_models_fields = [] related_models_fields = []
for field_name, field_config in model_type.model_fields.items(): for field_name, field_config in model_type.model_fields.items():
if hasattr(field_config, "model_fields"): if hasattr(field_config, "model_fields"):
related_models_fields.append(field_name) related_models_fields.append(field_name)