Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update milvus_multitenancy.py (#19695)

Browse Source
This commit is contained in:
Classic298
2025-12-02 21:38:06 +01:00
committed by GitHub
parent 12f237ff80
commit b29fdc2a0c
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
backend/open_webui/retrieval/vector/dbs/milvus_multitenancy.py
View File

@@ -262,15 +262,23 @@ class MilvusClient(VectorDBBase):
else: else:
expr.append(f"metadata['{key}'] == {value}") expr.append(f"metadata['{key}'] == {value}")
results = collection.query( iterator = collection.query_iterator(
expr=" and ".join(expr), expr=" and ".join(expr),
output_fields=["id", "text", "metadata"], output_fields=["id", "text", "metadata"],
limit=limit, limit=limit if limit else -1,
) )
ids = [res["id"] for res in results] all_results = []
documents = [res["text"] for res in results] while True:
metadatas = [res["metadata"] for res in results] batch = iterator.next()
if not batch:
iterator.close()
break
all_results.extend(batch)
ids = [res["id"] for res in all_results]
documents = [res["text"] for res in all_results]
metadatas = [res["metadata"] for res in all_results]
return GetResult(ids=[ids], documents=[documents], metadatas=[metadatas]) return GetResult(ids=[ids], documents=[documents], metadatas=[metadatas])