From fbe899e326d840c1fbbcb25f8923b5e1705ab36b Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Fri, 7 Feb 2025 10:02:33 +0800
Subject: [PATCH] fix: rerank error
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

--bug=1052035 --user=王孝刚 【模型管理】SiliconFlow的重排模型，重排后的内容不正确 https://www.tapd.cn/57709429/s/1651239
---
 .../impl/siliconCloud_model_provider/model/reranker.py         | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/setting/models_provider/impl/siliconCloud_model_provider/model/reranker.py b/apps/setting/models_provider/impl/siliconCloud_model_provider/model/reranker.py
index 41755c4f9..b07bd98d8 100644
--- a/apps/setting/models_provider/impl/siliconCloud_model_provider/model/reranker.py
+++ b/apps/setting/models_provider/impl/siliconCloud_model_provider/model/reranker.py
@@ -73,7 +73,8 @@ class SiliconCloudReranker(MaxKBBaseModel, BaseDocumentCompressor):
             "model": self.model,
             "query": query,
             "documents": texts,
-            "top_n": self.top_n
+            "top_n": self.top_n,
+            "return_documents": True,
         }
 
         response = requests.post(f"{self.api_base}/rerank", json=payload, headers=headers)