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)