From d80f08e042adbea18cb126a1c340ef3b39f9dccb Mon Sep 17 00:00:00 2001 From: Abhimanyu Saharan Date: Wed, 4 Feb 2026 16:08:28 +0530 Subject: [PATCH] feat(agents): Send wakeup message on provision Send a hello/wakeup message to the new agent session after\nprovisioning to prompt startup, while keeping the main agent\nuntouched.\n\nCo-Authored-By: Claude --- backend/app/api/agents.py | 48 +++++++++++++++++++++- backend/app/services/agent_provisioning.py | 5 ++- 2 files changed, 51 insertions(+), 2 deletions(-) diff --git a/backend/app/api/agents.py b/backend/app/api/agents.py index 8a611383..3c97a636 100644 --- a/backend/app/api/agents.py +++ b/backend/app/api/agents.py @@ -107,6 +107,25 @@ def _record_provisioning_failure(session: Session, agent: Agent, error: str) -> ) +def _record_wakeup_failure(session: Session, agent: Agent, error: str) -> None: + record_activity( + session, + event_type="agent.wakeup.failed", + message=f"Wakeup message failed: {error}", + agent_id=agent.id, + ) + + +async def _send_wakeup_message(agent: Agent, config: GatewayConfig) -> None: + session_key = agent.openclaw_session_id or _build_session_key(agent.name) + message = ( + f"Hello {agent.name}. Your workspace has been provisioned.\n\n" + "Start the agent, run BOOT.md, and if BOOTSTRAP.md exists run it once " + "then delete it. Begin heartbeats after startup." + ) + await send_message(message, session_key=session_key, config=config, deliver=True) + + @router.get("", response_model=list[AgentRead]) def list_agents( session: Session = Depends(get_session), @@ -150,11 +169,20 @@ async def create_agent( session.commit() try: await send_provisioning_message(agent, board, raw_token) + await _send_wakeup_message(agent, config) + record_activity( + session, + event_type="agent.wakeup.sent", + message=f"Wakeup message sent to {agent.name}.", + agent_id=agent.id, + ) except OpenClawGatewayError as exc: _record_provisioning_failure(session, agent, str(exc)) + _record_wakeup_failure(session, agent, str(exc)) session.commit() except Exception as exc: # pragma: no cover - unexpected provisioning errors _record_provisioning_failure(session, agent, str(exc)) + _record_wakeup_failure(session, agent, str(exc)) session.commit() return agent @@ -260,11 +288,20 @@ async def heartbeat_or_create_agent( session.commit() try: await send_provisioning_message(agent, board, raw_token) + await _send_wakeup_message(agent, config) + record_activity( + session, + event_type="agent.wakeup.sent", + message=f"Wakeup message sent to {agent.name}.", + agent_id=agent.id, + ) except OpenClawGatewayError as exc: _record_provisioning_failure(session, agent, str(exc)) + _record_wakeup_failure(session, agent, str(exc)) session.commit() except Exception as exc: # pragma: no cover - unexpected provisioning errors _record_provisioning_failure(session, agent, str(exc)) + _record_wakeup_failure(session, agent, str(exc)) session.commit() elif actor.actor_type == "agent" and actor.agent and actor.agent.id != agent.id: raise HTTPException(status_code=status.HTTP_403_FORBIDDEN) @@ -276,13 +313,22 @@ async def heartbeat_or_create_agent( session.refresh(agent) try: board = _require_board(session, str(agent.board_id) if agent.board_id else None) - _require_gateway_config(board) + config = _require_gateway_config(board) await send_provisioning_message(agent, board, raw_token) + await _send_wakeup_message(agent, config) + record_activity( + session, + event_type="agent.wakeup.sent", + message=f"Wakeup message sent to {agent.name}.", + agent_id=agent.id, + ) except OpenClawGatewayError as exc: _record_provisioning_failure(session, agent, str(exc)) + _record_wakeup_failure(session, agent, str(exc)) session.commit() except Exception as exc: # pragma: no cover - unexpected provisioning errors _record_provisioning_failure(session, agent, str(exc)) + _record_wakeup_failure(session, agent, str(exc)) session.commit() elif not agent.openclaw_session_id: board = _require_board(session, str(agent.board_id) if agent.board_id else None) diff --git a/backend/app/services/agent_provisioning.py b/backend/app/services/agent_provisioning.py index 27ffbe97..510c7224 100644 --- a/backend/app/services/agent_provisioning.py +++ b/backend/app/services/agent_provisioning.py @@ -108,11 +108,14 @@ def build_provisioning_message(agent: Agent, board: Board, auth_token: str) -> s f"Base URL: {base_url}\n" f"Auth token: {auth_token}\n\n" "Steps:\n" + "0) IMPORTANT: Do NOT replace or repurpose the main agent. Keep " + f"{context['main_session_key']} unchanged and its workspace intact.\n" "1) Create the workspace directory.\n" "2) Write the files below with the exact contents.\n" "3) Update TOOLS.md if BASE_URL/AUTH_TOKEN must change.\n" "4) Leave BOOTSTRAP.md in place; the agent should run it on first start and delete it.\n" - "5) Register agent id in OpenClaw so it uses this workspace path.\n\n" + "5) Register agent id in OpenClaw so it uses this workspace path " + "(never overwrite the main agent session).\n\n" "Files:" + file_blocks )