浏览代码

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

Jan Kessler 3 月之前
父节点
当前提交
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"):
                 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
                 form_data["files"] = files
         except Exception as e:
         except Exception as e: