Skip to content

Commit 0e00a5e

Browse files
committed
fix: remove unnecessary params from rerank method call in reranker.py
1 parent 01dcae3 commit 0e00a5e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • apps/models_provider/impl/vllm_model_provider/model

apps/models_provider/impl/vllm_model_provider/model/reranker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,6 @@ def compress_documents(self, documents: Sequence[Document], query: str, callback
4848
return []
4949

5050
ds = [d.page_content for d in documents]
51-
result = self.client.rerank(model=self.model, query=query, documents=ds, top_n=self.top_n, **self.params)
51+
result = self.client.rerank(model=self.model, query=query, documents=ds, top_n=self.top_n)
5252
return [Document(page_content=d.document.get('text'), metadata={'relevance_score': d.relevance_score}) for d in
5353
result.results]

0 commit comments

Comments
 (0)