瀏覽代碼

Enable RAG_WEB_SEARCH_CONCURRENT_REQUESTS with asynchronous optimization for improved performance

yeounhak 5 月之前
父節點
當前提交
8c161c797b
共有 1 個文件被更改,包括 4 次插入2 次删除
  1. 4 2
      backend/open_webui/apps/retrieval/main.py

+ 4 - 2
backend/open_webui/apps/retrieval/main.py

@@ -1283,9 +1283,11 @@ def process_web_search(form_data: SearchForm, user=Depends(get_verified_user)):
         urls = [result.link for result in web_results]
 
         loader = get_web_loader(
-            urls, verify_ssl=app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION
+            urls,
+            verify_ssl=app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
+            requests_per_second=app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS
         )
-        docs = loader.load()
+        docs = loader.aload()
 
         save_docs_to_vector_db(docs, collection_name, overwrite=True)