Skip to content

datarobot_genai.core.agents.render

render

Pure string rendering for AG-UI event types (no I/O).

render_ag_ui_event 

render_ag_ui_event(event: Event) -> str | None

Render an AG-UI event.

Parameters

event: The AG-UI event to render.

Source code in datarobot_genai/core/agents/render.py 
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
def render_ag_ui_event(event: Event) -> str | None:
    """Render an AG-UI event.

    Parameters
    ----------
    event:
        The AG-UI event to render.
    """
    event_type = event.type

    delta = str(getattr(event, "delta", "") or "")
    name = str(
        getattr(event, "tool_call_name", "")
        or getattr(event, "step_name", "")
        or getattr(event, "name", "")
        or ""
    )
    content = str(getattr(event, "content", "") or "")
    message = str(getattr(event, "message", "") or "Unknown error")

    rendered = render_event(
        event_type,
        delta=delta,
        name=name,
        content=content,
        message=message,
    )
    return rendered

render_event 

render_event(event_type: EventType, *, delta: str = '', name: str = '', content: str = '', message: str = '') -> str | None

Build a terminal-friendly string for a single event, or None to skip.

Parameters

event_type: The AG-UI :class:~ag_ui.core.events.EventType for this event. delta: Text delta for content/args events. name: Name for tool-call-start, step-started, or custom events. content: Content for tool-call-result events. message: Error message for run-error events.

Returns
A terminal friendly string if the event was rendered, otherwise None.
Print with print(rendered, end='') or click.echo(rendered, nl=False)
Source code in datarobot_genai/core/agents/render.py 
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
def render_event(
    event_type: EventType,
    *,
    delta: str = "",
    name: str = "",
    content: str = "",
    message: str = "",
) -> str | None:
    """Build a terminal-friendly string for a single event, or None to skip.

    Parameters
    ----------
    event_type:
        The AG-UI :class:`~ag_ui.core.events.EventType` for this event.
    delta:
        Text delta for content/args events.
    name:
        Name for tool-call-start, step-started, or custom events.
    content:
        Content for tool-call-result events.
    message:
        Error message for run-error events.

    Returns
    -------
        A terminal friendly string if the event was rendered, otherwise None.
        Print with print(rendered, end='') or click.echo(rendered, nl=False)
    """
    rendered: str | None = None
    if event_type in (EventType.TEXT_MESSAGE_CONTENT, EventType.TEXT_MESSAGE_CHUNK):
        rendered = f"{Fore.CYAN}{delta}{Style.RESET_ALL}"
    elif event_type == EventType.TEXT_MESSAGE_END:
        rendered = "\n"
    elif event_type == EventType.TEXT_MESSAGE_START:
        rendered = None
    elif event_type in (EventType.REASONING_MESSAGE_CONTENT, EventType.REASONING_MESSAGE_CHUNK):
        rendered = f"{Fore.YELLOW}{delta}{Style.RESET_ALL}"
    elif event_type in (EventType.REASONING_END, EventType.REASONING_MESSAGE_END):
        rendered = "\n"
    elif event_type in (EventType.REASONING_START, EventType.REASONING_MESSAGE_START):
        rendered = None
    elif event_type == EventType.TOOL_CALL_START:
        rendered = (
            f"\n{Fore.MAGENTA}\u25b6 Tool Call: {Fore.MAGENTA}{Style.DIM}{name}{Style.RESET_ALL}"
            f"\n{Fore.MAGENTA}  Arguments: {Style.RESET_ALL}"
        )
    elif event_type == EventType.TOOL_CALL_ARGS:
        rendered = f"{Fore.MAGENTA}{Style.DIM}{delta}{Style.RESET_ALL}"
    elif event_type == EventType.TOOL_CALL_END:
        rendered = "\n"
    elif event_type == EventType.TOOL_CALL_RESULT:
        if len(content) > TOOL_RESULT_MAX_LEN:
            content = content[:TOOL_RESULT_MAX_LEN] + "\u2026"
        rendered = f"{Fore.MAGENTA}  Result: {Fore.MAGENTA}{Style.DIM}{content}{Style.RESET_ALL}\n"
    elif event_type == EventType.STEP_STARTED:
        rendered = f"{Style.DIM}Step: {name}{Style.RESET_ALL}\n"
    elif event_type == EventType.STEP_FINISHED:
        rendered = None
    elif event_type == EventType.RUN_STARTED:
        rendered = f"{Style.DIM}Run started{Style.RESET_ALL}\n"
    elif event_type == EventType.RUN_FINISHED:
        rendered = f"\n{Fore.GREEN}\u2705 Run finished.{Style.RESET_ALL}\n"
    elif event_type == EventType.RUN_ERROR:
        rendered = f"{Fore.RED}\u274c Run failed: {message}{Style.RESET_ALL}\n"
    elif event_type == EventType.CUSTOM:
        if name != "Heartbeat":
            rendered = f"{Style.DIM}[{name}]{Style.RESET_ALL}\n"
    else:
        logger.debug(f"Unhandled event type: {event_type.value}")

    return rendered