diff --git a/apps/models_provider/impl/vllm_model_provider/model/reranker.py b/apps/models_provider/impl/vllm_model_provider/model/reranker.py index 193a661e2b9..b4a36cc6d47 100644 --- a/apps/models_provider/impl/vllm_model_provider/model/reranker.py +++ b/apps/models_provider/impl/vllm_model_provider/model/reranker.py @@ -48,6 +48,6 @@ def compress_documents(self, documents: Sequence[Document], query: str, callback return [] ds = [d.page_content for d in documents] - result = self.client.rerank(model=self.model, query=query, documents=ds, top_n=self.top_n, **self.params) + result = self.client.rerank(model=self.model, query=query, documents=ds, top_n=self.top_n) return [Document(page_content=d.document.get('text'), metadata={'relevance_score': d.relevance_score}) for d in result.results]