Browse Source

Fix if checks

Marko Henning 1 tháng trước cách đây
mục cha
commit
cd02ff2e07
2 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 1 1
      backend/open_webui/main.py
  2. 1 1
      backend/open_webui/routers/retrieval.py

+ 1 - 1
backend/open_webui/main.py

@@ -924,7 +924,7 @@ try:
         app.state.config.RAG_EMBEDDING_MODEL,
         RAG_EMBEDDING_MODEL_AUTO_UPDATE,
     )
-    if ENABLE_RAG_HYBRID_SEARCH and not BYPASS_EMBEDDING_AND_RETRIEVAL:
+    if app.state.config.ENABLE_RAG_HYBRID_SEARCH and not app.state.config.BYPASS_EMBEDDING_AND_RETRIEVAL:
         app.state.rf = get_rf(
             app.state.config.RAG_RERANKING_ENGINE,
             app.state.config.RAG_RERANKING_MODEL,

+ 1 - 1
backend/open_webui/routers/retrieval.py

@@ -852,7 +852,7 @@ async def update_rag_config(
                 import torch
                 if torch.cuda.is_available():
                     torch.cuda.empty_cache()
-            if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH and not request.app.state.BYPASS_EMBEDDING_AND_RETRIEVAL:
+            if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH and not request.app.state.config.BYPASS_EMBEDDING_AND_RETRIEVAL:
                 request.app.state.rf = get_rf(
                     request.app.state.config.RAG_RERANKING_ENGINE,
                     request.app.state.config.RAG_RERANKING_MODEL,