From 42071cb8e85f093c6dbfb9e0fe2b0fc47129ae64 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Tue, 18 Nov 2025 15:27:26 -0500 Subject: [PATCH] refac --- backend/open_webui/routers/retrieval.py | 4 ++-- backend/open_webui/utils/middleware.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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