UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-16 21:42:50 +00:00
Code Issues Actions3 Packages Projects Releases Wiki Activity

add NO_LIMIT var

Browse Source
This commit is contained in:
Robin Bially 2024-10-09 18:33:39 +02:00
parent 54dc94317c
commit 34150fc3ed
1 changed files with 5 additions and 3 deletions

8
backend/open_webui/apps/retrieval/vector/dbs/qdrant.py
View File

@ -7,6 +7,8 @@ from qdrant_client.models import models
from open_webui.apps.retrieval.vector.main import VectorItem, SearchResult, GetResult from open_webui.apps.retrieval.vector.main import VectorItem, SearchResult, GetResult
from open_webui.config import QDRANT_URI from open_webui.config import QDRANT_URI
NO_LIMIT = 999999999
class QdrantClient: class QdrantClient:
def __init__(self): def __init__(self):
self.collection_prefix = "open-webui" self.collection_prefix = "open-webui"
@ -73,7 +75,7 @@ class QdrantClient:
) -> Optional[SearchResult]: ) -> Optional[SearchResult]:
# Search for the nearest neighbor items based on the vectors and return 'limit' number of results. # Search for the nearest neighbor items based on the vectors and return 'limit' number of results.
if limit is None: if limit is None:
limit=10000000 # otherwise qdrant would set limit to 10! limit = NO_LIMIT # otherwise qdrant would set limit to 10!
query_response = self.client.query_points( query_response = self.client.query_points(
collection_name=f"{self.collection_prefix}_{collection_name}", collection_name=f"{self.collection_prefix}_{collection_name}",
@ -94,7 +96,7 @@ class QdrantClient:
return None return None
try: try:
if limit is None: if limit is None:
limit=10000000 # otherwise qdrant would set limit to 10! limit = NO_LIMIT # otherwise qdrant would set limit to 10!
field_conditions = [] field_conditions = []
for key, value in filter.items(): for key, value in filter.items():
@ -115,7 +117,7 @@ class QdrantClient:
# Get all the items in the collection. # Get all the items in the collection.
points = self.client.query_points( points = self.client.query_points(
collection_name=f"{self.collection_prefix}_{collection_name}", collection_name=f"{self.collection_prefix}_{collection_name}",
limit=10000000 # default is 10 limit=NO_LIMIT # otherwise qdrant would set limit to 10!
) )
return self._result_to_get_result(points.points) return self._result_to_get_result(points.points)