diff --git a/backend/open_webui/apps/retrieval/main.py b/backend/open_webui/apps/retrieval/main.py
index 471643285..cfbc5beee 100644
--- a/backend/open_webui/apps/retrieval/main.py
+++ b/backend/open_webui/apps/retrieval/main.py
@@ -608,9 +608,7 @@ async def update_rag_config(form_data: ConfigUpdateForm, user=Depends(get_admin_
         app.state.config.BRAVE_SEARCH_API_KEY = (
             form_data.web.search.brave_search_api_key
         )
-        app.state.config.KAGI_SEARCH_API_KEY = (
-            form_data.web.search.kagi_search_api_key
-        )
+        app.state.config.KAGI_SEARCH_API_KEY = form_data.web.search.kagi_search_api_key
         app.state.config.MOJEEK_SEARCH_API_KEY = (
             form_data.web.search.mojeek_search_api_key
         )
@@ -1390,8 +1388,7 @@ def query_doc_handler(
         else:
             return query_doc(
                 collection_name=form_data.collection_name,
-                query=form_data.query,
-                embedding_function=app.state.EMBEDDING_FUNCTION,
+                query_embedding=app.state.EMBEDDING_FUNCTION(form_data.query),
                 k=form_data.k if form_data.k else app.state.config.TOP_K,
             )
     except Exception as e: