15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: siliconCloud rerank error 

--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)
)
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: