From 44c1d35b1fc936e783408d36090242a9f64078bf Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Wed, 2 Apr 2025 15:17:04 +0800 Subject: [PATCH] fix: siliconCloud rerank error MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --bug=1054256 --user=王孝刚 【模型】添加硅基流动的重排序模型失败 https://www.tapd.cn/57709429/s/1679612 --- .../model/reranker.py | 20 ------------------- 1 file changed, 20 deletions(-) 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 b07bd98d8..ef85cec58 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 @@ -36,26 +36,6 @@ class SiliconCloudReranker(MaxKBBaseModel, BaseDocumentCompressor): top_n=model_kwargs.get('top_n', 3) ) - def __init__( - self, api_base: Optional[str] = None, model: Optional[str] = None, top_n=3, - api_key: Optional[str] = None - ): - super().__init__() - - if not api_base: - raise ValueError(_('Please provide server URL')) - - if not model: - raise ValueError(_('Please provide the model')) - - if not api_key: - raise ValueError(_('Please provide the API Key')) - - self.api_base = api_base - self.model = model - self.api_key = api_key - self.top_n = top_n - def compress_documents(self, documents: Sequence[Document], query: str, callbacks: Optional[Callbacks] = None) -> \ Sequence[Document]: if not documents: