feat: provision main agent on gateway

backend/app/services/agent_provisioning.py

@@ -37,6 +37,7 @@ DEFAULT_GATEWAY_FILES = frozenset(
         "IDENTITY.md",
         "USER.md",
         "HEARTBEAT.md",
+        "BOOT.md",
         "BOOTSTRAP.md",
         "MEMORY.md",
     }
@@ -44,6 +45,12 @@ DEFAULT_GATEWAY_FILES = frozenset(
 
 HEARTBEAT_LEAD_TEMPLATE = "HEARTBEAT_LEAD.md"
 HEARTBEAT_AGENT_TEMPLATE = "HEARTBEAT_AGENT.md"
+MAIN_TEMPLATE_MAP = {
+    "AGENTS.md": "MAIN_AGENTS.md",
+    "HEARTBEAT.md": "MAIN_HEARTBEAT.md",
+    "USER.md": "MAIN_USER.md",
+    "BOOT.md": "MAIN_BOOT.md",
+}
 
 
 def _repo_root() -> Path:
@@ -159,6 +166,51 @@ def _build_context(
     }
 
 
+def _build_main_context(
+    agent: Agent,
+    gateway: Gateway,
+    auth_token: str,
+    user: User | None,
+) -> dict[str, str]:
+    base_url = settings.base_url or "REPLACE_WITH_BASE_URL"
+    identity_profile: dict[str, Any] = {}
+    if isinstance(agent.identity_profile, dict):
+        identity_profile = agent.identity_profile
+    normalized_identity: dict[str, str] = {}
+    for key, value in identity_profile.items():
+        if value is None:
+            continue
+        if isinstance(value, list):
+            parts = [str(item).strip() for item in value if str(item).strip()]
+            if not parts:
+                continue
+            normalized_identity[key] = ", ".join(parts)
+            continue
+        text = str(value).strip()
+        if text:
+            normalized_identity[key] = text
+    identity_context = {
+        context_key: normalized_identity.get(field, DEFAULT_IDENTITY_PROFILE[field])
+        for field, context_key in IDENTITY_PROFILE_FIELDS.items()
+    }
+    return {
+        "agent_name": agent.name,
+        "agent_id": str(agent.id),
+        "session_key": agent.openclaw_session_id or "",
+        "base_url": base_url,
+        "auth_token": auth_token,
+        "main_session_key": gateway.main_session_key or "",
+        "workspace_root": gateway.workspace_root or "",
+        "user_name": (user.name or "") if user else "",
+        "user_preferred_name": (user.preferred_name or "") if user else "",
+        "user_pronouns": (user.pronouns or "") if user else "",
+        "user_timezone": (user.timezone or "") if user else "",
+        "user_notes": (user.notes or "") if user else "",
+        "user_context": (user.context or "") if user else "",
+        **identity_context,
+    }
+
+
 def _session_key(agent: Agent) -> str:
     if agent.openclaw_session_id:
         return agent.openclaw_session_id
@@ -211,6 +263,7 @@ def _render_agent_files(
     file_names: set[str],
     *,
     include_bootstrap: bool,
+    template_overrides: dict[str, str] | None = None,
 ) -> dict[str, str]:
     env = _template_env()
     overrides: dict[str, str] = {}
@@ -227,7 +280,11 @@ def _render_agent_files(
             rendered[name] = "# MEMORY\n\nBootstrap pending.\n"
             continue
         if name == "HEARTBEAT.md":
-            heartbeat_template = _heartbeat_template_name(agent)
+            heartbeat_template = (
+                template_overrides.get(name)
+                if template_overrides and name in template_overrides
+                else _heartbeat_template_name(agent)
+            )
             heartbeat_path = _templates_root() / heartbeat_template
             if heartbeat_path.exists():
                 rendered[name] = (
@@ -238,14 +295,39 @@ def _render_agent_files(
         if override:
             rendered[name] = env.from_string(override).render(**context).strip()
             continue
-        path = _templates_root() / name
+        template_name = (
+            template_overrides.get(name)
+            if template_overrides and name in template_overrides
+            else name
+        )
+        path = _templates_root() / template_name
         if path.exists():
-            rendered[name] = env.get_template(name).render(**context).strip()
+            rendered[name] = env.get_template(template_name).render(**context).strip()
             continue
         rendered[name] = ""
     return rendered
 
 
+async def _gateway_default_agent_id(
+    config: GatewayClientConfig,
+) -> str | None:
+    try:
+        payload = await openclaw_call("agents.list", config=config)
+    except OpenClawGatewayError:
+        return None
+    if not isinstance(payload, dict):
+        return None
+    default_id = payload.get("defaultId") or payload.get("default_id")
+    if default_id:
+        return default_id
+    agents = payload.get("agents") or []
+    if isinstance(agents, list) and agents:
+        first = agents[0]
+        if isinstance(first, dict):
+            return first.get("id")
+    return None
+
+
 async def _patch_gateway_agent_list(
     agent_id: str,
     workspace_path: str,
@@ -381,6 +463,54 @@ async def provision_agent(
         )
 
 
+async def provision_main_agent(
+    agent: Agent,
+    gateway: Gateway,
+    auth_token: str,
+    user: User | None,
+    *,
+    action: str = "provision",
+) -> None:
+    if not gateway.url:
+        return
+    if not gateway.main_session_key:
+        raise ValueError("gateway main_session_key is required")
+    client_config = GatewayClientConfig(url=gateway.url, token=gateway.token)
+    await ensure_session(gateway.main_session_key, config=client_config, label="Main Agent")
+
+    agent_id = await _gateway_default_agent_id(client_config)
+    if not agent_id:
+        raise OpenClawGatewayError("Unable to resolve gateway main agent id")
+
+    context = _build_main_context(agent, gateway, auth_token, user)
+    supported = await _supported_gateway_files(client_config)
+    existing_files = await _gateway_agent_files_index(agent_id, client_config)
+    include_bootstrap = action != "update"
+    if action == "update":
+        if not existing_files:
+            include_bootstrap = False
+        else:
+            entry = existing_files.get("BOOTSTRAP.md")
+            if entry and entry.get("missing") is True:
+                include_bootstrap = False
+
+    rendered = _render_agent_files(
+        context,
+        agent,
+        supported,
+        include_bootstrap=include_bootstrap,
+        template_overrides=MAIN_TEMPLATE_MAP,
+    )
+    for name, content in rendered.items():
+        if content == "":
+            continue
+        await openclaw_call(
+            "agents.files.set",
+            {"agentId": agent_id, "name": name, "content": content},
+            config=client_config,
+        )
+
+
 async def cleanup_agent(
     agent: Agent,
     gateway: Gateway,