ソースを参照

fix web results all getting the same source id when bypassing embedding and retrieval

Jan Kessler 1 ヶ月 前
コミット
2729d8aab4
1 ファイル変更9 行追加8 行削除
  1. 9 8
      backend/open_webui/utils/middleware.py

+ 9 - 8
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: