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

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-10-09 16:16:24 -05:00
parent b93e9b1698
commit 4e763e8aa8
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
backend/open_webui/retrieval/utils.py
View file

@ -268,6 +268,13 @@ def merge_and_sort_query_results(query_results: list[dict], k: int) -> dict:
combined = dict() # To store documents with unique document hashes combined = dict() # To store documents with unique document hashes
for data in query_results: for data in query_results:
if (
len(data.get("distances", [])) == 0
or len(data.get("documents", [])) == 0
or len(data.get("metadatas", [])) == 0
):
continue
distances = data["distances"][0] distances = data["distances"][0]
documents = data["documents"][0] documents = data["documents"][0]
metadatas = data["metadatas"][0] metadatas = data["metadatas"][0]