diff --git a/llama_stack/apis/models/models.py b/llama_stack/apis/models/models.py index 6b3f7c1ca..d542517ba 100644 --- a/llama_stack/apis/models/models.py +++ b/llama_stack/apis/models/models.py @@ -29,4 +29,4 @@ class Models(Protocol): async def list_models(self) -> List[ModelServingSpec]: ... @webmethod(route="/models/get", method="GET") - async def get_model(self, core_model_id: str) -> Optional[ModelsGetResponse]: ... + async def get_model(self, core_model_id: str) -> Optional[ModelServingSpec]: ...