Function _fetch_anthropic_models

web/pgadmin/llm/__init__.py:779–851

Fetch models from Anthropic API. Returns a list of model options with label and value.

(api_key, api_url='')

def _fetch_anthropic_models(api_key, api_url=''):
    """
    Fetch models from Anthropic API.
    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.anthropic.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 = {
        'anthropic-version': '2023-06-01'
    }
    if api_key:
        headers['x-api-key'] = 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 Anthropic API: {e.reason}'
        )

    models = []
    seen = set()

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

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

        # Create a user-friendly label
        if display_name and display_name != model_id:
            label = f"{display_name} ({model_id})"
        else:
            label = model_id