Skip to content

Commit decbf05

Browse files
committed
fix: ensure params are initialized as a dictionary and remove top_n from params
1 parent cdd974d commit decbf05

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

  • apps/models_provider/impl/vllm_model_provider/model

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,10 @@ def __init__(self, **kwargs):
1919
super().__init__(**kwargs)
2020
self.api_key = kwargs.get('api_key')
2121
self.model = kwargs.get('model')
22-
self.params = kwargs.get('params')
22+
self.params = dict(kwargs.get('params') or {})
2323
self.api_url = kwargs.get('api_url')
2424
self.top_n = kwargs.get('top_n', 3)
25+
self.params.pop('top_n', None)
2526
self.client = cohere.ClientV2(kwargs.get('api_key'), base_url=kwargs.get('api_url'))
2627

2728
@staticmethod

0 commit comments

Comments
 (0)