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

fix: await retrieve in qdrant adapter

Browse source
This commit is contained in:
Boris Arzentar 2024-04-26 18:45:37 +02:00
parent ac1a4cf076
commit 7652f4001d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py
View file

@ -111,8 +111,8 @@ class QDrantAdapter(VectorDBInterface):
return result return result
async def retrieve(self, collection_name: str, data_id: str): async def retrieve(self, collection_name: str, data_id: str):
client = await self.get_qdrant_client() client = self.get_qdrant_client()
results = client.retrieve(collection_name, [data_id], with_payload = True) results = await client.retrieve(collection_name, [data_id], with_payload = True)
await client.close() await client.close()
return results[0] if len(results) > 0 else None return results[0] if len(results) > 0 else None