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

fix: Add missing await statement to LanceDBAdapter and PGVectorAdapter

Browse source 
Added missing await statement to batch search for LanceDB and PGVector adapters

Fix #COG-170
This commit is contained in:
Igor Ilic 2024-10-22 15:15:45 +02:00
parent 195929e245
commit dc46304a8d
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -152,7 +152,7 @@ class LanceDBAdapter(VectorDBInterface):
): ):
query_vectors = await self.embedding_engine.embed_text(query_texts) query_vectors = await self.embedding_engine.embed_text(query_texts)
return asyncio.gather( return await asyncio.gather(
*[self.search( *[self.search(
collection_name = collection_name, collection_name = collection_name,
query_vector = query_vector, query_vector = query_vector,

2
cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
View file

@ -200,7 +200,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
): ):
query_vectors = await self.embedding_engine.embed_text(query_texts) query_vectors = await self.embedding_engine.embed_text(query_texts)
return asyncio.gather( return await asyncio.gather(
*[ *[
self.search( self.search(
collection_name=collection_name, collection_name=collection_name,