fix: rerank error

--bug=1052035 --user=王孝刚【模型管理】SiliconFlow的重排模型，重排后的内容不正确 https://www.tapd.cn/57709429/s/1651239
2025-02-07 10:02:33 +08:00 · 2025-02-07 10:02:33 +08:00 · fbe899e326
parent 8b32944692
commit fbe899e326
1 changed files with 2 additions and 1 deletions
--- 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)