ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

updated query function with scroll too

Browse source
This commit is contained in:
expruc 2025-08-09 22:04:41 +03:00
parent 88abd01b87
commit 8af9ad3f30
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
backend/open_webui/retrieval/vector/dbs/qdrant.py
View file

@ -179,12 +179,12 @@ class QdrantClient(VectorDBBase):
) )
) )
points = self.client.query_points( points = self.client.scroll(
collection_name=f"{self.collection_prefix}_{collection_name}", collection_name=f"{self.collection_prefix}_{collection_name}",
query_filter=models.Filter(should=field_conditions), scroll_filter=models.Filter(should=field_conditions),
limit=limit, limit=limit,
) )
return self._result_to_get_result(points.points) return self._result_to_get_result(points[0])
except Exception as e: except Exception as e:
log.exception(f"Error querying a collection '{collection_name}': {e}") log.exception(f"Error querying a collection '{collection_name}': {e}")
return None return None

6
backend/open_webui/retrieval/vector/dbs/qdrant_multitenancy.py
View file

@ -289,12 +289,12 @@ class QdrantClient(VectorDBBase):
tenant_filter = _tenant_filter(tenant_id) tenant_filter = _tenant_filter(tenant_id)
field_conditions = [_metadata_filter(k, v) for k, v in filter.items()] field_conditions = [_metadata_filter(k, v) for k, v in filter.items()]
combined_filter = models.Filter(must=[tenant_filter, *field_conditions]) combined_filter = models.Filter(must=[tenant_filter, *field_conditions])
points = self.client.query_points( points = self.client.scroll(
collection_name=mt_collection, collection_name=mt_collection,
query_filter=combined_filter, scroll_filter=combined_filter,
limit=limit, 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]: def get(self, collection_name: str) -> Optional[GetResult]:
""" """