diff --git a/extensions/openai/models.py b/extensions/openai/models.py index 9c086a17..115149b0 100644 --- a/extensions/openai/models.py +++ b/extensions/openai/models.py @@ -18,6 +18,15 @@ def list_models(): return {'model_names': get_available_models()} +def list_models_openai_format(): + """Returns model list in OpenAI API format""" + model_names = get_available_models() + return { + "object": "list", + "data": [model_info_dict(name) for name in model_names] + } + + def model_info_dict(model_name: str) -> dict: return { "id": model_name, diff --git a/extensions/openai/script.py b/extensions/openai/script.py index f3b5b093..9440cb1e 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()['model_names'] + response = OAImodels.list_models_openai_format() else: model_name = path[len('/v1/models/'):] response = OAImodels.model_info_dict(model_name)