소스 검색

Merge pull request #15789 from azurewtl/hotfix-web-config-clean-reranker

- fix: keep reranker_model config been removed by web search config
Tim Jaeryang Baek 9 달 전
부모
커밋
6bb5248812
1개의 변경된 파일5개의 추가작업 그리고 1개의 파일을 삭제
  1. 5 1
      backend/open_webui/routers/retrieval.py

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

@@ -815,7 +815,11 @@ async def update_rag_config(
         f"Updating reranking model: {request.app.state.config.RAG_RERANKING_MODEL} to {form_data.RAG_RERANKING_MODEL}"
     )
     try:
-        request.app.state.config.RAG_RERANKING_MODEL = form_data.RAG_RERANKING_MODEL
+        request.app.state.config.RAG_RERANKING_MODEL = (
+            form_data.RAG_RERANKING_MODEL
+            if form_data.RAG_RERANKING_MODEL is not None
+            else request.app.state.config.RAG_RERANKING_MODEL
+        )
 
         try:
             request.app.state.rf = get_rf(