fix: rerank error

wxg0103 · wxg0103 · commit fbe899e326d8 · 2025-02-07T10:18:14.000+08:00
--bug=1052035 --user=王孝刚 【模型管理】SiliconFlow的重排模型，重排后的内容不正确 https://www.tapd.cn/57709429/s/1651239
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
@@ -73,7 +73,8 @@ def compress_documents(self, documents: Sequence[Document], query: str, callback
             "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)

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,8 @@ def compress_documents(self, documents: Sequence[Document], query: str, callback`
`73`	`73`	`"model": self.model,`
`74`	`74`	`"query": query,`
`75`	`75`	`"documents": texts,`
`76`		`- "top_n": self.top_n`
	`76`	`+ "top_n": self.top_n,`
	`77`	`+ "return_documents": True,`
`77`	`78`	`}`
`78`	`79`
`79`	`80`	`response = requests.post(f"{self.api_base}/rerank", json=payload, headers=headers)`