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 <noreply@anthropic.com>
This commit is contained in:
@@ -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])
|
@router.get("", response_model=list[AgentRead])
|
||||||
def list_agents(
|
def list_agents(
|
||||||
session: Session = Depends(get_session),
|
session: Session = Depends(get_session),
|
||||||
@@ -150,11 +169,20 @@ async def create_agent(
|
|||||||
session.commit()
|
session.commit()
|
||||||
try:
|
try:
|
||||||
await send_provisioning_message(agent, board, raw_token)
|
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:
|
except OpenClawGatewayError as exc:
|
||||||
_record_provisioning_failure(session, agent, str(exc))
|
_record_provisioning_failure(session, agent, str(exc))
|
||||||
|
_record_wakeup_failure(session, agent, str(exc))
|
||||||
session.commit()
|
session.commit()
|
||||||
except Exception as exc: # pragma: no cover - unexpected provisioning errors
|
except Exception as exc: # pragma: no cover - unexpected provisioning errors
|
||||||
_record_provisioning_failure(session, agent, str(exc))
|
_record_provisioning_failure(session, agent, str(exc))
|
||||||
|
_record_wakeup_failure(session, agent, str(exc))
|
||||||
session.commit()
|
session.commit()
|
||||||
return agent
|
return agent
|
||||||
|
|
||||||
@@ -260,11 +288,20 @@ async def heartbeat_or_create_agent(
|
|||||||
session.commit()
|
session.commit()
|
||||||
try:
|
try:
|
||||||
await send_provisioning_message(agent, board, raw_token)
|
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:
|
except OpenClawGatewayError as exc:
|
||||||
_record_provisioning_failure(session, agent, str(exc))
|
_record_provisioning_failure(session, agent, str(exc))
|
||||||
|
_record_wakeup_failure(session, agent, str(exc))
|
||||||
session.commit()
|
session.commit()
|
||||||
except Exception as exc: # pragma: no cover - unexpected provisioning errors
|
except Exception as exc: # pragma: no cover - unexpected provisioning errors
|
||||||
_record_provisioning_failure(session, agent, str(exc))
|
_record_provisioning_failure(session, agent, str(exc))
|
||||||
|
_record_wakeup_failure(session, agent, str(exc))
|
||||||
session.commit()
|
session.commit()
|
||||||
elif actor.actor_type == "agent" and actor.agent and actor.agent.id != agent.id:
|
elif actor.actor_type == "agent" and actor.agent and actor.agent.id != agent.id:
|
||||||
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN)
|
raise HTTPException(status_code=status.HTTP_403_FORBIDDEN)
|
||||||
@@ -276,13 +313,22 @@ async def heartbeat_or_create_agent(
|
|||||||
session.refresh(agent)
|
session.refresh(agent)
|
||||||
try:
|
try:
|
||||||
board = _require_board(session, str(agent.board_id) if agent.board_id else None)
|
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_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:
|
except OpenClawGatewayError as exc:
|
||||||
_record_provisioning_failure(session, agent, str(exc))
|
_record_provisioning_failure(session, agent, str(exc))
|
||||||
|
_record_wakeup_failure(session, agent, str(exc))
|
||||||
session.commit()
|
session.commit()
|
||||||
except Exception as exc: # pragma: no cover - unexpected provisioning errors
|
except Exception as exc: # pragma: no cover - unexpected provisioning errors
|
||||||
_record_provisioning_failure(session, agent, str(exc))
|
_record_provisioning_failure(session, agent, str(exc))
|
||||||
|
_record_wakeup_failure(session, agent, str(exc))
|
||||||
session.commit()
|
session.commit()
|
||||||
elif not agent.openclaw_session_id:
|
elif not agent.openclaw_session_id:
|
||||||
board = _require_board(session, str(agent.board_id) if agent.board_id else None)
|
board = _require_board(session, str(agent.board_id) if agent.board_id else None)
|
||||||
|
|||||||
@@ -108,11 +108,14 @@ def build_provisioning_message(agent: Agent, board: Board, auth_token: str) -> s
|
|||||||
f"Base URL: {base_url}\n"
|
f"Base URL: {base_url}\n"
|
||||||
f"Auth token: {auth_token}\n\n"
|
f"Auth token: {auth_token}\n\n"
|
||||||
"Steps:\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"
|
"1) Create the workspace directory.\n"
|
||||||
"2) Write the files below with the exact contents.\n"
|
"2) Write the files below with the exact contents.\n"
|
||||||
"3) Update TOOLS.md if BASE_URL/AUTH_TOKEN must change.\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"
|
"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
|
"Files:" + file_blocks
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user