Function refresh_ollama_models

web/pgadmin/llm/__init__.py:655–695

Fetch available Ollama models using a provided API URL. Used by the preferences refresh button to load models before saving.

()

)
@pga_login_required
def refresh_ollama_models():
    """
    Fetch available Ollama models using a provided API URL.
    Used by the preferences refresh button to load models before saving.
    """
    from pgadmin.llm.utils import validate_api_url

    data = request.get_json(force=True, silent=True) or {}
    api_url = data.get('api_url', '')

    if not api_url:
        return make_json_response(
            data={'models': [], 'error': 'No API URL provided'},
            status=200
        )

    if not validate_api_url(api_url):
        return make_json_response(
            data={'models': [],
                  'error': 'API URL is not in the allowed list. '
                           'Contact your administrator to update '
                           'ALLOWED_LLM_API_URLS in the server '
                           'configuration.'},
            status=200
        )

    try:
        models = _fetch_ollama_models(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 Ollama models. '
                           'Check the API URL configuration.'},
            status=200
        )


@blueprint.route("/models/docker", methods=["GET"], endpoint='models_docker')