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: rerank error 

--bug=1052035 --user=王孝刚 【模型管理】SiliconFlow的重排模型，重排后的内容不正确 https://www.tapd.cn/57709429/s/1651239
This commit is contained in:
wxg0103 2025-02-07 10:02:33 +08:00 committed by wxg
parent 8b32944692
commit fbe899e326
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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