Function get_openai_models

web/pgadmin/llm/__init__.py:514–543

Fetch available OpenAI models. Returns models that support function calling.

()

@blueprint.route("/models/openai", methods=["GET"], endpoint='models_openai')
@pga_login_required
def get_openai_models():
    """
    Fetch available OpenAI models.
    Returns models that support function calling.
    """
    from pgadmin.llm.utils import get_openai_api_key, get_openai_api_url

    api_key = get_openai_api_key()
    api_url = get_openai_api_url()
    if not api_key and not api_url:
        return make_json_response(
            data={'models': [], 'error': 'No API key configured'},
            status=200
        )

    try:
        models = _fetch_openai_models(api_key, api_url)
        return make_json_response(data={'models': models}, status=200)
    except LLMApiError as e:
        return make_json_response(
            data={'models': [], 'error': str(e)},
            status=200
        )
    except Exception:
        return make_json_response(
            data={'models': [],
                  'error': 'Failed to fetch OpenAI models. '
                           'Check the API key and URL configuration.'},
            status=200
        )


@blueprint.route(