diff --git a/backend/open_webui/routers/retrieval.py b/backend/open_webui/routers/retrieval.py
index b7ddaf6af1..7ebeb05bfd 100644
--- a/backend/open_webui/routers/retrieval.py
+++ b/backend/open_webui/routers/retrieval.py
@@ -2345,8 +2345,8 @@ def query_collection_handler(
                 k=form_data.k if form_data.k else request.app.state.config.TOP_K,
                 reranking_function=(
                     (
-                        lambda sentences: request.app.state.RERANKING_FUNCTION(
-                            sentences, user=user
+                        lambda query, documents: request.app.state.RERANKING_FUNCTION(
+                            query, documents, user=user
                         )
                     )
                     if request.app.state.RERANKING_FUNCTION
diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index cb96b2f885..9a3b7bea70 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -984,8 +984,8 @@ async def chat_completion_files_handler(
                         k=request.app.state.config.TOP_K,
                         reranking_function=(
                             (
-                                lambda sentences: request.app.state.RERANKING_FUNCTION(
-                                    sentences, user=user
+                                lambda query, documents: request.app.state.RERANKING_FUNCTION(
+                                    query, documents, user=user
                                 )
                             )
                             if request.app.state.RERANKING_FUNCTION