Jelajahi Sumber

Added parameter to: app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION to get_web_loader(verify_ssl).
Invert variable value to send to UI , not ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION.

Davi Sclifo Zucon 6 bulan lalu
induk
melakukan
35a9140567
1 mengubah file dengan 5 tambahan dan 4 penghapusan
  1. 5 4
      backend/open_webui/apps/retrieval/main.py

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

@@ -428,7 +428,7 @@ async def get_rag_config(user=Depends(get_admin_user)):
             "translation": app.state.YOUTUBE_LOADER_TRANSLATION,
             "translation": app.state.YOUTUBE_LOADER_TRANSLATION,
         },
         },
         "web": {
         "web": {
-            "web_loader_ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
+            "web_loader_ssl_verification": not app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
             "search": {
             "search": {
                 "enabled": app.state.config.ENABLE_RAG_WEB_SEARCH,
                 "enabled": app.state.config.ENABLE_RAG_WEB_SEARCH,
                 "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
                 "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
@@ -536,8 +536,9 @@ async def update_rag_config(form_data: ConfigUpdateForm, user=Depends(get_admin_
         app.state.YOUTUBE_LOADER_TRANSLATION = form_data.youtube.translation
         app.state.YOUTUBE_LOADER_TRANSLATION = form_data.youtube.translation
 
 
     if form_data.web is not None:
     if form_data.web is not None:
+        # if "Bypass SSL verification for Websites" == True then web_loader_ssl_verification = False.
         app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION = (
         app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION = (
-            form_data.web.web_loader_ssl_verification
+            not form_data.web.web_loader_ssl_verification
         )
         )
 
 
         app.state.config.ENABLE_RAG_WEB_SEARCH = form_data.web.search.enabled
         app.state.config.ENABLE_RAG_WEB_SEARCH = form_data.web.search.enabled
@@ -592,7 +593,7 @@ async def update_rag_config(form_data: ConfigUpdateForm, user=Depends(get_admin_
             "translation": app.state.YOUTUBE_LOADER_TRANSLATION,
             "translation": app.state.YOUTUBE_LOADER_TRANSLATION,
         },
         },
         "web": {
         "web": {
-            "web_loader_ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
+            "web_loader_ssl_verification": not app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
             "search": {
             "search": {
                 "enabled": app.state.config.ENABLE_RAG_WEB_SEARCH,
                 "enabled": app.state.config.ENABLE_RAG_WEB_SEARCH,
                 "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
                 "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
@@ -1229,7 +1230,7 @@ def process_web_search(form_data: SearchForm, user=Depends(get_verified_user)):
 
 
         urls = [result.link for result in web_results]
         urls = [result.link for result in web_results]
 
 
-        loader = get_web_loader(urls)
+        loader = get_web_loader(urls, verify_ssl=app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION)
         docs = loader.load()
         docs = loader.load()
 
 
         save_docs_to_vector_db(docs, collection_name, overwrite=True)
         save_docs_to_vector_db(docs, collection_name, overwrite=True)