zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: siliconCloud rerank error

Browse Source 
--bug=1054256 --user=王孝刚 【模型】添加硅基流动的重排序模型失败 https://www.tapd.cn/57709429/s/1679612
This commit is contained in:
wxg0103 2025-04-02 15:17:04 +08:00 committed by wxg
parent f6994e16b9
commit 44c1d35b1f
1 changed files with 0 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
apps/setting/models_provider/impl/siliconCloud_model_provider/model/reranker.py
View File

@ -36,26 +36,6 @@ class SiliconCloudReranker(MaxKBBaseModel, BaseDocumentCompressor):
top_n=model_kwargs.get('top_n', 3) 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) -> \ def compress_documents(self, documents: Sequence[Document], query: str, callbacks: Optional[Callbacks] = None) -> \
Sequence[Document]: Sequence[Document]:
if not documents: if not documents: