diff --git a/llama_stack/apis/eval_tasks/eval_tasks.py b/llama_stack/apis/eval_tasks/eval_tasks.py index cfa00034c..870673e58 100644 --- a/llama_stack/apis/eval_tasks/eval_tasks.py +++ b/llama_stack/apis/eval_tasks/eval_tasks.py @@ -37,7 +37,7 @@ class EvalTasks(Protocol): eval_task_id: str, dataset_id: str, scoring_functions: List[str], - provider_id: Optional[str] = None, provider_eval_task_id: Optional[str] = None, + provider_id: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, ) -> None: ... diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index 2ad9bc282..b0091f5a0 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -367,8 +367,8 @@ class EvalTasksRoutingTable(CommonRoutingTableImpl, EvalTasks): dataset_id: str, scoring_functions: List[str], metadata: Optional[Dict[str, Any]] = None, - provider_id: Optional[str] = None, provider_eval_task_id: Optional[str] = None, + provider_id: Optional[str] = None, ) -> None: if metadata is None: metadata = {}