Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-07-14 14:05:06 +04:00
parent 0013f5c1fc
commit 18bd83413b
3 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/retrieval/utils.py
+2
View File
@@ -446,6 +446,8 @@ def get_embedding_function(
def get_reranking_function(reranking_engine, reranking_model, reranking_function): def get_reranking_function(reranking_engine, reranking_model, reranking_function):
if reranking_function is None:
return None
if reranking_engine == "external": if reranking_engine == "external":
return lambda sentences, user=None: reranking_function.predict( return lambda sentences, user=None: reranking_function.predict(
sentences, user=user sentences, user=user
backend/open_webui/routers/retrieval.py
+6 -2
View File
@@ -2049,8 +2049,12 @@ def query_doc_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,
backend/open_webui/utils/middleware.py
+8 -2
View File
@@ -652,8 +652,14 @@ async def chat_completion_files_handler(
query, prefix=prefix, user=user query, prefix=prefix, user=user
), ),
k=request.app.state.config.TOP_K, k=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=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,