From c821c3ecb06e19872bea5378bbcee6deeed066bb Mon Sep 17 00:00:00 2001
From: Marko Henning <marko.henning@posteo.de>
Date: Thu, 21 Aug 2025 13:40:56 +0200
Subject: [PATCH] Formatting

---
 backend/open_webui/main.py              |  5 ++++-
 backend/open_webui/routers/retrieval.py | 17 ++++++++++++-----
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py
index 461600351a..e82347088f 100644
--- a/backend/open_webui/main.py
+++ b/backend/open_webui/main.py
@@ -924,7 +924,10 @@ try:
         app.state.config.RAG_EMBEDDING_MODEL,
         RAG_EMBEDDING_MODEL_AUTO_UPDATE,
     )
-    if app.state.config.ENABLE_RAG_HYBRID_SEARCH and not app.state.config.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,
diff --git a/backend/open_webui/routers/retrieval.py b/backend/open_webui/routers/retrieval.py
index 51a81b1fd7..d75677cb43 100644
--- a/backend/open_webui/routers/retrieval.py
+++ b/backend/open_webui/routers/retrieval.py
@@ -280,14 +280,16 @@ async def update_embedding_config(
     log.info(
         f"Updating embedding model: {request.app.state.config.RAG_EMBEDDING_MODEL} to {form_data.embedding_model}"
     )
-    if request.app.state.config.RAG_EMBEDDING_ENGINE == '':
+    if request.app.state.config.RAG_EMBEDDING_ENGINE == "":
         # unloads current internal embedding model and clears VRAM cache
         request.app.state.ef = None
         request.app.state.EMBEDDING_FUNCTION = None
         import gc
+
         gc.collect()
-        if DEVICE_TYPE == 'cuda':
+        if DEVICE_TYPE == "cuda":
             import torch
+
             if torch.cuda.is_available():
                 torch.cuda.empty_cache()
     try:
@@ -815,14 +817,16 @@ async def update_rag_config(
     )
 
     # Reranking settings
-    if request.app.state.config.RAG_RERANKING_ENGINE == '':
+    if request.app.state.config.RAG_RERANKING_ENGINE == "":
         # Unloading the internal reranker and clear VRAM memory
         request.app.state.rf = None
         request.app.state.RERANKING_FUNCTION = None
         import gc
+
         gc.collect()
-        if DEVICE_TYPE == 'cuda':
+        if DEVICE_TYPE == "cuda":
             import torch
+
             if torch.cuda.is_available():
                 torch.cuda.empty_cache()
     request.app.state.config.RAG_RERANKING_ENGINE = (
@@ -854,7 +858,10 @@ async def update_rag_config(
         )
 
         try:
-            if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH and not request.app.state.config.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,