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

fix: rerank error

Browse Source 
--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, "model": self.model,
"query": query, "query": query,
"documents": texts, "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) response = requests.post(f"{self.api_base}/rerank", json=payload, headers=headers)