From 5fe3ddb27d706f2141959f15ccc40cac975cf578 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 10 Feb 2025 09:41:01 -0800 Subject: [PATCH] update eval_task_id -> task_id --- llama_stack/apis/eval_tasks/eval_tasks.py | 4 ++-- llama_stack/distribution/routers/routing_tables.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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,