diff --git a/llama_stack/apis/eval_tasks/eval_tasks.py b/llama_stack/apis/eval_tasks/eval_tasks.py index 3600589d1..f36d44c88 100644 --- a/llama_stack/apis/eval_tasks/eval_tasks.py +++ b/llama_stack/apis/eval_tasks/eval_tasks.py @@ -51,13 +51,13 @@ class EvalTasks(Protocol): @webmethod(route="/eval/tasks/{task_id}", method="GET") async def get_eval_task( self, - eval_task_id: str, + task_id: str, ) -> Optional[EvalTask]: ... @webmethod(route="/eval/tasks", method="POST") async def register_eval_task( self, - eval_task_id: str, + task_id: str, dataset_id: str, scoring_functions: List[str], provider_eval_task_id: Optional[str] = None, diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index 68fafd8ee..3f5dea66d 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -432,8 +432,8 @@ class EvalTasksRoutingTable(CommonRoutingTableImpl, EvalTasks): async def list_eval_tasks(self) -> ListEvalTasksResponse: return ListEvalTasksResponse(data=await self.get_all_with_type("eval_task")) - async def get_eval_task(self, eval_task_id: str) -> Optional[EvalTask]: - return await self.get_object_by_identifier("eval_task", eval_task_id) + async def get_eval_task(self, task_id: str) -> Optional[EvalTask]: + return await self.get_object_by_identifier("eval_task", task_id) async def register_eval_task( self,