From dc46304a8da9e4e36b5c22973a13ae25c7515335 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 22 Oct 2024 15:15:45 +0200 Subject: [PATCH] fix: Add missing await statement to LanceDBAdapter and PGVectorAdapter Added missing await statement to batch search for LanceDB and PGVector adapters Fix #COG-170 --- .../infrastructure/databases/vector/lancedb/LanceDBAdapter.py | 2 +- .../infrastructure/databases/vector/pgvector/PGVectorAdapter.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py index 3bb47fcc0..404634489 100644 --- a/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py +++ b/cognee/infrastructure/databases/vector/lancedb/LanceDBAdapter.py @@ -152,7 +152,7 @@ class LanceDBAdapter(VectorDBInterface): ): query_vectors = await self.embedding_engine.embed_text(query_texts) - return asyncio.gather( + return await asyncio.gather( *[self.search( collection_name = collection_name, query_vector = query_vector, diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py index bd10cf0ef..8b0ddcd8a 100644 --- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py @@ -200,7 +200,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface): ): query_vectors = await self.embedding_engine.embed_text(query_texts) - return asyncio.gather( + return await asyncio.gather( *[ self.search( collection_name=collection_name,