@@ -54,6 +54,7 @@ def __init__(
5454 api_url : Optional [str ] = None ,
5555 model_id : Optional [str ] = None ,
5656 eval_type : Optional [str ] = None ,
57+ model_args : Optional [Dict [str , Any ]] = None ,
5758 system_prompt : Optional [str ] = None ,
5859 prompt_template : Optional [str ] = None ,
5960 generation_config : Optional [Dict [str , Any ]] = None ,
@@ -70,6 +71,7 @@ def __init__(
7071 api_base (str, optional): API base URL
7172 model_id (str, optional): Model ID for LLM
7273 eval_type (str, optional): Evaluation LLM type for the judge
74+ model_args (dict, optional): Additional model arguments for the judge
7375 system_prompt (str, optional): System prompt for the judge
7476 prompt_template (str, optional): Prompt template for the judge
7577 generation_config (dict, optional): Generation configuration for the judge
@@ -85,6 +87,7 @@ def __init__(
8587 self .eval_type = eval_type or EvalType .OPENAI_API
8688 self .system_prompt = system_prompt or os .environ .get ('JUDGE_SYSTEM_PROMPT' , None )
8789 self .generation_config = generation_config or {'temperature' : 0.0 , 'max_tokens' : 4096 }
90+ self .model_args = model_args or {}
8891
8992 # Default score mapping for A/B pattern
9093 self .score_type = score_type
@@ -112,6 +115,7 @@ def _init_server_adapter(self):
112115 base_url = self .api_url ,
113116 api_key = self .api_key ,
114117 config = GenerateConfig (** self .generation_config ),
118+ model_args = self .model_args ,
115119 )
116120
117121 def judge (
0 commit comments