From f7e012d21658eb24a9e17462598195e909061d69 Mon Sep 17 00:00:00 2001 From: Hantaek Lim <143519383+LHANTAEK@users.noreply.github.com> Date: Sat, 8 Mar 2025 20:08:48 +0900 Subject: [PATCH] Fix: reranker OFF logic to preserve user setting (#15235) Co-authored-by: crazywoola <427733928@qq.com> --- .../workflow/nodes/knowledge-retrieval/utils.ts | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/utils.ts b/web/app/components/workflow/nodes/knowledge-retrieval/utils.ts index fd4a5806eb..68fe3d11e9 100644 --- a/web/app/components/workflow/nodes/knowledge-retrieval/utils.ts +++ b/web/app/components/workflow/nodes/knowledge-retrieval/utils.ts @@ -154,7 +154,8 @@ export const getMultipleRetrievalConfig = ( result.reranking_mode = RerankingModeEnum.RerankingModel if (!result.reranking_model?.provider || !result.reranking_model?.model) { if (rerankModelIsValid) { - result.reranking_enable = true + result.reranking_enable = reranking_enable !== false + result.reranking_model = { provider: validRerankModel?.provider || '', model: validRerankModel?.model || '', @@ -168,7 +169,7 @@ export const getMultipleRetrievalConfig = ( } } else { - result.reranking_enable = true + result.reranking_enable = reranking_enable !== false } } @@ -176,7 +177,8 @@ export const getMultipleRetrievalConfig = ( if (!reranking_mode) { if (validRerankModel?.provider && validRerankModel?.model) { result.reranking_mode = RerankingModeEnum.RerankingModel - result.reranking_enable = true + result.reranking_enable = reranking_enable !== false + result.reranking_model = { provider: validRerankModel.provider, model: validRerankModel.model, @@ -194,7 +196,8 @@ export const getMultipleRetrievalConfig = ( if (reranking_mode === RerankingModeEnum.WeightedScore && weights && shouldSetWeightDefaultValue) { if (rerankModelIsValid) { result.reranking_mode = RerankingModeEnum.RerankingModel - result.reranking_enable = true + result.reranking_enable = reranking_enable !== false + result.reranking_model = { provider: validRerankModel.provider || '', model: validRerankModel.model || '',