From 8af9ad3f30825131dec6a6eda596baec7cbdb097 Mon Sep 17 00:00:00 2001 From: expruc Date: Sat, 9 Aug 2025 22:04:41 +0300 Subject: [PATCH] updated query function with scroll too --- backend/open_webui/retrieval/vector/dbs/qdrant.py | 6 +++--- .../open_webui/retrieval/vector/dbs/qdrant_multitenancy.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/backend/open_webui/retrieval/vector/dbs/qdrant.py b/backend/open_webui/retrieval/vector/dbs/qdrant.py index 248f51b187..7997865467 100644 --- a/backend/open_webui/retrieval/vector/dbs/qdrant.py +++ b/backend/open_webui/retrieval/vector/dbs/qdrant.py @@ -179,12 +179,12 @@ class QdrantClient(VectorDBBase): ) ) - points = self.client.query_points( + points = self.client.scroll( collection_name=f"{self.collection_prefix}_{collection_name}", - query_filter=models.Filter(should=field_conditions), + scroll_filter=models.Filter(should=field_conditions), limit=limit, ) - return self._result_to_get_result(points.points) + return self._result_to_get_result(points[0]) except Exception as e: log.exception(f"Error querying a collection '{collection_name}': {e}") return None diff --git a/backend/open_webui/retrieval/vector/dbs/qdrant_multitenancy.py b/backend/open_webui/retrieval/vector/dbs/qdrant_multitenancy.py index ddfec98cc7..c120f07632 100644 --- a/backend/open_webui/retrieval/vector/dbs/qdrant_multitenancy.py +++ b/backend/open_webui/retrieval/vector/dbs/qdrant_multitenancy.py @@ -289,12 +289,12 @@ class QdrantClient(VectorDBBase): tenant_filter = _tenant_filter(tenant_id) field_conditions = [_metadata_filter(k, v) for k, v in filter.items()] combined_filter = models.Filter(must=[tenant_filter, *field_conditions]) - points = self.client.query_points( + points = self.client.scroll( collection_name=mt_collection, - query_filter=combined_filter, + scroll_filter=combined_filter, limit=limit, ) - return self._result_to_get_result(points.points) + return self._result_to_get_result(points[0]) def get(self, collection_name: str) -> Optional[GetResult]: """