diff --git a/llama_stack/apis/eval/eval.py b/llama_stack/apis/eval/eval.py index 60d2567b9..0f059761b 100644 --- a/llama_stack/apis/eval/eval.py +++ b/llama_stack/apis/eval/eval.py @@ -71,7 +71,7 @@ class Eval(Protocol): @webmethod(route="/eval/evaluate_batch", method="POST") async def evaluate_task( self, - eval_task_def: Union[str, EvalTaskDef], # type: ignore + eval_task_id: str, eval_task_config: EvalTaskConfig, # type: ignore ) -> Job: ...