From 4dafbbccfc9cf377a66ebb6ba667796179b2b4a7 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 12 Apr 2025 22:55:24 -0700 Subject: [PATCH] fix: rag template display issue --- backend/open_webui/routers/retrieval.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/routers/retrieval.py b/backend/open_webui/routers/retrieval.py index d46b8b393..87f978767 100644 --- a/backend/open_webui/routers/retrieval.py +++ b/backend/open_webui/routers/retrieval.py @@ -353,7 +353,7 @@ async def get_rag_config(request: Request, user=Depends(get_admin_user)): return { "status": True, # RAG settings - "TEMPLATE": request.app.state.config.RAG_TEMPLATE, + "RAG_TEMPLATE": request.app.state.config.RAG_TEMPLATE, "TOP_K": request.app.state.config.TOP_K, "BYPASS_EMBEDDING_AND_RETRIEVAL": request.app.state.config.BYPASS_EMBEDDING_AND_RETRIEVAL, "RAG_FULL_CONTEXT": request.app.state.config.RAG_FULL_CONTEXT, @@ -470,7 +470,7 @@ class WebConfig(BaseModel): class ConfigForm(BaseModel): # RAG settings - TEMPLATE: Optional[str] = None + RAG_TEMPLATE: Optional[str] = None TOP_K: Optional[int] = None BYPASS_EMBEDDING_AND_RETRIEVAL: Optional[bool] = None RAG_FULL_CONTEXT: Optional[bool] = None @@ -712,8 +712,15 @@ async def update_rag_config( return { "status": True, + # RAG settings + "RAG_TEMPLATE": request.app.state.config.RAG_TEMPLATE, + "TOP_K": request.app.state.config.TOP_K, "BYPASS_EMBEDDING_AND_RETRIEVAL": request.app.state.config.BYPASS_EMBEDDING_AND_RETRIEVAL, "RAG_FULL_CONTEXT": request.app.state.config.RAG_FULL_CONTEXT, + # Hybrid search settings + "ENABLE_RAG_HYBRID_SEARCH": request.app.state.config.ENABLE_RAG_HYBRID_SEARCH, + "TOP_K_RERANKER": request.app.state.config.TOP_K_RERANKER, + "RELEVANCE_THRESHOLD": request.app.state.config.RELEVANCE_THRESHOLD, # Content extraction settings "CONTENT_EXTRACTION_ENGINE": request.app.state.config.CONTENT_EXTRACTION_ENGINE, "PDF_EXTRACT_IMAGES": request.app.state.config.PDF_EXTRACT_IMAGES,