Function _fetch_ollama_models

web/pgadmin/llm/__init__.py:924–980

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

(api_url)

def _fetch_ollama_models(api_url):
    """
    Fetch models from Ollama 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

    # Normalize URL
    api_url = api_url.rstrip('/')

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

    url = f'{api_url}/api/tags'

    req = urllib.request.Request(url)

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

    models = []
    for model in data.get('models', []):
        name = model.get('name', '')
        if name:
            # Format size if available
            size = model.get('size', 0)
            if size:
                size_gb = size / (1024 ** 3)
                label = f"{name} ({size_gb:.1f} GB)"
            else:
                label = name

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

    # Sort alphabetically
    models.sort(key=lambda x: x['value'])

    return models