100100router = APIRouter ()
101101
102102
103- @router .get ("/models/" , response_model = list [dict ], tags = ["models" ])
103+ @router .get (
104+ "/models/" ,
105+ response_model = list [dict ],
106+ tags = ["models" ],
107+ name = "List running CogStack Model Serve instances with metadata from the tracking server" ,
108+ )
104109async def get_models (
105110 config : Annotated [Config , Depends (get_config )],
106111 verbose : Annotated [
@@ -120,7 +125,12 @@ async def get_models(
120125 return models
121126
122127
123- @router .get ("/models/{model_name}/info" , response_model = dict , tags = ["models" ])
128+ @router .get (
129+ "/models/{model_name}/info" ,
130+ response_model = dict ,
131+ tags = ["models" ],
132+ name = "Get information about a running CogStack Model Serve instance" ,
133+ )
124134async def get_model_info (model_name : str ):
125135 """Get information about a running model server through its `/info` API."""
126136 # FIXME: Enable SSL verification when certificates are properly set up
@@ -139,7 +149,12 @@ async def get_model_info(model_name: str):
139149 return response .json ()
140150
141151
142- @router .post ("/models/{model_name}" , response_model = dict , tags = ["models" ])
152+ @router .post (
153+ "/models/{model_name}" ,
154+ response_model = dict ,
155+ tags = ["models" ],
156+ name = "Deploy a CogStack Model Serve instance with a given model URI or tracking ID" ,
157+ )
143158async def deploy_model (
144159 config : Annotated [Config , Depends (get_config )],
145160 model_name : Annotated [str , Depends (validate_model_name )],
@@ -215,7 +230,12 @@ async def deploy_model(
215230 }
216231
217232
218- @router .post ("/models/{model_name}/tasks/{task}" , response_model = dict , tags = ["models" ])
233+ @router .post (
234+ "/models/{model_name}/tasks/{task}" ,
235+ response_model = dict ,
236+ tags = ["models" ],
237+ name = "Schedule a task for execution on a running CogStack Model Serve instance" ,
238+ )
219239async def execute_task (
220240 model_name : str ,
221241 task : str ,
0 commit comments