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

Merge pull request #12507 from Ithanil/fix_web_result_collection_source_ids

Browse Source 
fix: fix web results all getting the same source id when using embedding and retrieval
This commit is contained in:
Timothy Jaeryang Baek
2025-04-06 15:43:21 -07:00
committed by GitHub
parent 66cd40f959 4476060044
commit 9825d03602
2 changed files with 27 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
backend/open_webui/utils/middleware.py
View File

@@ -399,15 +399,16 @@ async def chat_web_search_handler(
all_results.append(results)
files = form_data.get("files", [])
if results.get("collection_name"):
files.append(
{
"collection_name": results["collection_name"],
"name": searchQuery,
"type": "web_search",
"urls": results["filenames"],
}
)
if results.get("collection_names"):
for col_idx, collection_name in enumerate(results.get("collection_names")):
files.append(
{
"collection_name": collection_name,
"name": searchQuery,
"type": "web_search",
"urls": [results["filenames"][col_idx]],
}
)
elif results.get("docs"):
# Invoked when bypass embedding and retrieval is set to True
docs = results["docs"]