Fix the /v1/models output format (closes #7089)

This commit is contained in:
oobabooga 2025-06-19 07:57:17 -07:00
parent 9c6913ad61
commit dcdc42fa06
2 changed files with 2 additions and 2 deletions

View file

@ -15,7 +15,7 @@ def get_current_model_info():
def list_models():
return {'model_names': get_available_models()[1:]}
return {'model_names': get_available_models()}
def model_info_dict(model_name: str) -> dict:

View file

@ -180,7 +180,7 @@ async def handle_models(request: Request):
is_list = request.url.path.split('?')[0].split('#')[0] == '/v1/models'
if is_list:
response = OAImodels.list_models()
response = OAImodels.list_models()['model_names']
else:
model_name = path[len('/v1/models/'):]
response = OAImodels.model_info_dict(model_name)