Function generate_security_report_stream

web/pgadmin/llm/__init__.py:1117–1162

Stream a security report for the specified server via SSE.

(sid)

@pgCSRFProtect.exempt
@pga_login_required
def generate_security_report_stream(sid):
    """
    Stream a security report for the specified server via SSE.
    """
    from pgadmin.llm.utils import is_llm_enabled
    from pgadmin.llm.reports.generator import (
        generate_report_streaming, create_sse_response
    )
    from pgadmin.utils.driver import get_driver

    if not is_llm_enabled():
        return make_json_response(
            success=0,
            errormsg=gettext(
                'LLM is not configured. Please configure an LLM provider '
                'in Preferences > AI.'
            )
        )

    try:
        driver = get_driver(config.PG_DEFAULT_DRIVER)
        manager = driver.connection_manager(sid)
        conn = manager.connection()

        if not conn.connected():
            return make_json_response(
                success=0,
                errormsg=gettext('Server is not connected.')
            )

        context = {}
        generator = generate_report_streaming(
            report_type='security',
            scope='server',
            conn=conn,
            manager=manager,
            context=context
        )

        return create_sse_response(generator)

    except Exception as e:
        return make_json_response(
            success=0,
            errormsg=str(e)
        )


def _gather_security_config(conn, manager):