From 4bca3aa486ce1d3eaccc6c7a1c038d8eca98b769 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Thu, 2 Jul 2026 14:22:20 +0800
Subject: [PATCH] refactor: enhance VllmBgeReranker initialization with
 optional parameters and default top_n value

---
 .../impl/vllm_model_provider/model/reranker.py              | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/apps/models_provider/impl/vllm_model_provider/model/reranker.py b/apps/models_provider/impl/vllm_model_provider/model/reranker.py
index c05b9f4790e..e40fa1783a0 100644
--- a/apps/models_provider/impl/vllm_model_provider/model/reranker.py
+++ b/apps/models_provider/impl/vllm_model_provider/model/reranker.py
@@ -35,12 +35,14 @@ def new_instance(model_type, model_name, model_credential: Dict[str, object], **
             if model_credential.get("api_url").endswith("/v1")
             else model_credential.get("api_url")
         )
+        optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
+        top_n = optional_params.pop("top_n", 3)
         return VllmBgeReranker(
             model=model_name,
             api_key=model_credential.get("api_key"),
             api_url=r_url,
-            params=model_kwargs,
-            **model_kwargs,
+            top_n=top_n,
+            params=optional_params,
         )
 
     def compress_documents(