瀏覽代碼

Fix if checks

Marko Henning 1 月之前
父節點
當前提交
cd02ff2e07
共有 2 個文件被更改,包括 2 次插入2 次删除
  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,