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

fix: Qdrant adapter faulty limit=0 behavior in search (#965)

Browse source 
<!-- .github/pull_request_template.md -->

## Description
Fixes Qdrant adapter faulty limit=0 behavior

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
This commit is contained in:
hajdul88 2025-06-11 14:49:50 +02:00 • committed by GitHub
parent cfe9c949a7
commit 1c5ea46378
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -391,13 +391,7 @@ class QDrantAdapter(VectorDBInterface):
if query_text is None and query_vector is None: if query_text is None and query_vector is None:
raise InvalidValueError(message="One of query_text or query_vector must be provided!") raise InvalidValueError(message="One of query_text or query_vector must be provided!")
if limit <= 0:
return []
if not await self.has_collection(collection_name): if not await self.has_collection(collection_name):
logger.warning(
f"Collection '{collection_name}' not found in QdrantAdapter.search; returning []."
)
return [] return []
if query_vector is None: if query_vector is None:
@ -405,6 +399,8 @@ class QDrantAdapter(VectorDBInterface):
try: try:
client = self.get_qdrant_client() client = self.get_qdrant_client()
if limit == 0:
collection_size = await client.count(collection_name=collection_name)
results = await client.search( results = await client.search(
collection_name=collection_name, collection_name=collection_name,
@ -414,7 +410,7 @@ class QDrantAdapter(VectorDBInterface):
if query_vector is not None if query_vector is not None
else (await self.embed_data([query_text]))[0], else (await self.embed_data([query_text]))[0],
), ),
limit=limit if limit > 0 else None, limit=limit if limit > 0 else collection_size.count,
with_vectors=with_vector, with_vectors=with_vector,
) )
@ -431,13 +427,6 @@ class QDrantAdapter(VectorDBInterface):
) )
for result in results for result in results
] ]
except UnexpectedResponse as error:
if "Collection not found" in str(error):
raise CollectionNotFoundError(
message=f"Collection {collection_name} not found!"
) from error
else:
raise error
finally: finally:
await client.close() await client.close()