Function apply_template

apps/agent/src/templates.py:162–214

Retrieve a saved template's HTML so you can adapt it with new data. After calling this, generate a NEW widget in the same style and render via generateSandboxedUi. This tool automatically checks for a pending_template in state (set by the frontend when the user picks a template fro

(runtime: ToolRuntime, name: str = "", template_id: str = "")

@tool
def apply_template(runtime: ToolRuntime, name: str = "", template_id: str = ""):
    """
    Retrieve a saved template's HTML so you can adapt it with new data.
    After calling this, generate a NEW widget in the same style and render via generateSandboxedUi.

    This tool automatically checks for a pending_template in state (set by the
    frontend when the user picks a template from the library). If pending_template
    is present, it takes priority over name/template_id arguments.

    Also searches built-in seed templates, so users can apply them by name in chat
    even if the frontend hasn't pushed them into agent state yet.

    Args:
        name: The name of the template to apply (fallback if no pending_template)
        template_id: The ID of the template to apply (fallback if no pending_template)
    """
    state_templates = runtime.state.get("templates", [])
    state_ids = {t["id"] for t in state_templates}
    templates = [*state_templates, *(s for s in SEED_TEMPLATES if s["id"] not in state_ids)]

    # Check pending_template from frontend first — this is the most reliable source
    pending = runtime.state.get("pending_template")
    if pending and pending.get("id"):
        template_id = pending["id"]

    # Look up by ID first
    if template_id:
        for t in templates:
            if t["id"] == template_id:
                return {
                    "name": t["name"],
                    "description": t["description"],
                    "html": t["html"],
                    "data_description": t.get("data_description", ""),
                    "usage_note": TEMPLATE_USAGE_NOTE,
                }
        return {"error": f"Template with id '{template_id}' not found"}

    # Look up by name (most recent match)
    if name:
        matches = [t for t in templates if t["name"].lower() == name.lower()]
        if matches:
            t = max(matches, key=lambda x: x.get("created_at", ""))
            return {
                "name": t["name"],
                "description": t["description"],
                "html": t["html"],
                "data_description": t.get("data_description", ""),
                "usage_note": TEMPLATE_USAGE_NOTE,
            }
        return {"error": f"No template named '{name}' found"}

    return {"error": "Provide either a name or template_id"}


@tool