From 2729d8aab478de76ba427cf4401ee9f361b364d7 Mon Sep 17 00:00:00 2001 From: Jan Kessler Date: Sun, 6 Apr 2025 13:37:42 +0200 Subject: [PATCH] fix web results all getting the same source id when bypassing embedding and retrieval --- backend/open_webui/utils/middleware.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 72f1f30ce..98965ce68 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -395,14 +395,15 @@ async def chat_web_search_handler( } ) elif results.get("docs"): - files.append( - { - "docs": results.get("docs", []), - "name": searchQuery, - "type": "web_search", - "urls": results["filenames"], - } - ) + for doc_idx, doc in enumerate(results["docs"]): + files.append( + { + "docs": [doc], + "name": searchQuery, + "type": "web_search", + "urls": [results["filenames"][doc_idx]], + } + ) form_data["files"] = files except Exception as e: