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

fix: reverting the lancedb chnage

Browse source
This commit is contained in:
chinu0609 2025-10-22 20:55:59 +05:30
parent b47cb7462d
commit 7b31b86f10
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -111,7 +111,8 @@ 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["embeddings"] print(data)
return data["embeddings"][0]
def get_vector_size(self) -> int: def get_vector_size(self) -> int:
""" """

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

@ -182,8 +182,6 @@ class LanceDBAdapter(VectorDBInterface):
properties = get_own_properties(data_point) properties = get_own_properties(data_point)
properties["id"] = str(properties["id"]) 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))]( 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,