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)