Przeglądaj źródła

add NO_LIMIT var

Robin Bially 7 miesięcy temu
rodzic
commit
34150fc3ed

+ 5 - 3
backend/open_webui/apps/retrieval/vector/dbs/qdrant.py

@@ -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)