From dcdc42fa06ba56eec5ca09b305147a27ee08ff39 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Thu, 19 Jun 2025 07:57:17 -0700 Subject: [PATCH] Fix the /v1/models output format (closes #7089) --- extensions/openai/models.py | 2 +- extensions/openai/script.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions/openai/models.py b/extensions/openai/models.py index f8d9a1e8..9c086a17 100644 --- a/extensions/openai/models.py +++ b/extensions/openai/models.py @@ -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: diff --git a/extensions/openai/script.py b/extensions/openai/script.py index 3d8d5f73..f3b5b093 100644 --- a/extensions/openai/script.py +++ b/extensions/openai/script.py @@ -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)