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,