Function save_template

apps/agent/src/templates.py:99–137

Save a generated UI as a reusable template. Call this when the user wants to save a widget/visualization they liked for reuse later. Args: name: Short name for the template (e.g. "Invoice", "Dashboard") description: What the template displays or does html: The r

(
    name: str,
    description: str,
    html: str,
    data_description: str,
    runtime: ToolRuntime,
)

@tool
def save_template(
    name: str,
    description: str,
    html: str,
    data_description: str,
    runtime: ToolRuntime,
) -> Command:
    """
    Save a generated UI as a reusable template.
    Call this when the user wants to save a widget/visualization they liked for reuse later.

    Args:
        name: Short name for the template (e.g. "Invoice", "Dashboard")
        description: What the template displays or does
        html: The raw HTML string of the widget to save as a template
        data_description: Description of the data shape this template expects
    """
    templates = list(runtime.state.get("templates", []))

    template: UITemplate = {
        "id": str(uuid.uuid4()),
        "name": name,
        "description": description,
        "html": html,
        "data_description": data_description,
        "created_at": datetime.now().isoformat(),
        "version": 1,
    }
    templates.append(template)

    return Command(update={
        "templates": templates,
        "messages": [
            ToolMessage(
                content=f"Template '{name}' saved successfully (id: {template['id']})",
                tool_call_id=runtime.tool_call_id,
            )
        ],
    })


@tool