ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 20:35:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac/fix: reranking function

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-07-16 13:56:02 +04:00
parent ae716bddee
commit abe280f0a3
2 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
backend/open_webui/routers/retrieval.py
View file

@ -2050,11 +2050,13 @@ def query_doc_handler(
), ),
k=form_data.k if form_data.k else request.app.state.config.TOP_K, k=form_data.k if form_data.k else request.app.state.config.TOP_K,
reranking_function=( reranking_function=(
lambda sentences: ( (
request.app.state.RERANKING_FUNCTION(sentences, user=user) lambda sentences: request.app.state.RERANKING_FUNCTION(
if request.app.state.RERANKING_FUNCTION sentences, user=user
else None )
) )
if request.app.state.RERANKING_FUNCTION
else None
), ),
k_reranker=form_data.k_reranker k_reranker=form_data.k_reranker
or request.app.state.config.TOP_K_RERANKER, or request.app.state.config.TOP_K_RERANKER,
@ -2112,8 +2114,14 @@ def query_collection_handler(
query, prefix=prefix, user=user query, prefix=prefix, user=user
), ),
k=form_data.k if form_data.k else request.app.state.config.TOP_K, k=form_data.k if form_data.k else request.app.state.config.TOP_K,
reranking_function=lambda sentences: request.app.state.RERANKING_FUNCTION( reranking_function=(
sentences, user=user (
lambda sentences: request.app.state.RERANKING_FUNCTION(
sentences, user=user
)
)
if request.app.state.RERANKING_FUNCTION
else None
), ),
k_reranker=form_data.k_reranker k_reranker=form_data.k_reranker
or request.app.state.config.TOP_K_RERANKER, or request.app.state.config.TOP_K_RERANKER,

8
backend/open_webui/utils/middleware.py
View file

@ -653,13 +653,13 @@ async def chat_completion_files_handler(
), ),
k=request.app.state.config.TOP_K, k=request.app.state.config.TOP_K,
reranking_function=( reranking_function=(
lambda sentences: ( (
request.app.state.RERANKING_FUNCTION( lambda sentences: request.app.state.RERANKING_FUNCTION(
sentences, user=user sentences, user=user
) )
if request.app.state.RERANKING_FUNCTION
else None
) )
if request.app.state.RERANKING_FUNCTION
else None
), ),
k_reranker=request.app.state.config.TOP_K_RERANKER, k_reranker=request.app.state.config.TOP_K_RERANKER,
r=request.app.state.config.RELEVANCE_THRESHOLD, r=request.app.state.config.RELEVANCE_THRESHOLD,