From 7652f4001d65eb44be2516bcfd6d6928afcf3ebf Mon Sep 17 00:00:00 2001 From: Boris Arzentar Date: Fri, 26 Apr 2024 18:45:37 +0200 Subject: [PATCH] fix: await retrieve in qdrant adapter --- .../infrastructure/databases/vector/qdrant/QDrantAdapter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py b/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py index 4cba001e1..2d994d3ee 100644 --- a/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py +++ b/cognee/infrastructure/databases/vector/qdrant/QDrantAdapter.py @@ -111,8 +111,8 @@ class QDrantAdapter(VectorDBInterface): return result async def retrieve(self, collection_name: str, data_id: str): - client = await self.get_qdrant_client() - results = client.retrieve(collection_name, [data_id], with_payload = True) + client = self.get_qdrant_client() + results = await client.retrieve(collection_name, [data_id], with_payload = True) await client.close() return results[0] if len(results) > 0 else None