Function _fetch_openai_models

web/pgadmin/llm/__init__.py:854–921

Fetch models from OpenAI API or any OpenAI-compatible endpoint. Returns a list of model options with label and value.

(api_key, api_url='')

def _fetch_openai_models(api_key, api_url=''):
    """
    Fetch models from OpenAI API or any OpenAI-compatible endpoint.
    Returns a list of model options with label and value.
    """
    import urllib.request
    import urllib.error
    from pgadmin.llm.utils import validate_api_url

    base_url = (api_url or 'https://api.openai.com/v1').rstrip('/')

    if not validate_api_url(base_url):
        raise LLMApiError(
            'API URL is not in the allowed list. '
            'Check the ALLOWED_LLM_API_URLS configuration.'
        )

    url = f'{base_url}/models'

    headers = {
        'Content-Type': 'application/json'
    }
    if api_key:
        headers['Authorization'] = f'Bearer {api_key}'

    req = urllib.request.Request(url, headers=headers)

    try:
        with urllib.request.urlopen(
            req, timeout=30, context=SSL_CONTEXT
        ) as response:
            data = json.loads(response.read().decode('utf-8'))
    except urllib.error.HTTPError as e:
        if e.code == 401:
            raise LLMApiError('Invalid API key')
        raise LLMApiError(f'API error: {e.code}')
    except urllib.error.URLError as e:
        raise LLMApiError(
            f'Cannot connect to OpenAI API: {e.reason}'
        )

    models = []
    seen = set()

    for model in data.get('data', []):
        model_id = model.get('id', '')

        # Skip if already seen or empty
        if not model_id or model_id in seen:
            continue
        seen.add(model_id)

        models.append({
            'label': model_id,
            'value': model_id
        })

    if not models and api_url: