From 53729c64053e90f21579dbc54b09dabce66466fd Mon Sep 17 00:00:00 2001 From: Maytown Date: Mon, 12 May 2025 12:45:21 +0200 Subject: [PATCH] Fix: adjusted multiple embedders/rerankers to be only loaded once --- backend/open_webui/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py index 0f1ab97b4..d91d39111 100644 --- a/backend/open_webui/main.py +++ b/backend/open_webui/main.py @@ -723,13 +723,13 @@ app.state.YOUTUBE_LOADER_TRANSLATION = None try: - app.state.ef["DEFAULT"] = get_ef( + app.state.ef[app.state.config.RAG_EMBEDDING_MODEL] = get_ef( app.state.config.RAG_EMBEDDING_ENGINE, app.state.config.RAG_EMBEDDING_MODEL, RAG_EMBEDDING_MODEL_AUTO_UPDATE, ) - app.state.rf["DEFAULT"] = get_rf( + app.state.rf[app.state.config.RAG_RERANKING_MODEL] = get_rf( app.state.config.RAG_RERANKING_MODEL, RAG_RERANKING_MODEL_AUTO_UPDATE, ) @@ -738,7 +738,7 @@ except Exception as e: pass -app.state.EMBEDDING_FUNCTION["DEFAULT"] = get_embedding_function( +app.state.EMBEDDING_FUNCTION[app.state.config.RAG_EMBEDDING_MODEL] = get_embedding_function( app.state.config.RAG_EMBEDDING_ENGINE, app.state.config.RAG_EMBEDDING_MODEL, app.state.ef,