feat(agents): Add heartbeat configuration and delete confirmation for agents

backend/app/integrations/openclaw_gateway.py

@@ -9,6 +9,7 @@ from uuid import uuid4
 
 import websockets
 
+from app.integrations.openclaw_gateway_protocol import PROTOCOL_VERSION
 
 
 class OpenClawGatewayError(RuntimeError):
@@ -64,23 +65,10 @@ async def _send_request(
     return await _await_response(ws, request_id)
 
 
-async def _handle_challenge(
-    ws: websockets.WebSocketClientProtocol,
-    first_message: str | bytes | None,
-    config: GatewayConfig,
-) -> None:
-    if not first_message:
-        return
-    if isinstance(first_message, bytes):
-        first_message = first_message.decode("utf-8")
-    data = json.loads(first_message)
-    if data.get("type") != "event" or data.get("event") != "connect.challenge":
-        return
-
-    connect_id = str(uuid4())
+def _build_connect_params(config: GatewayConfig) -> dict[str, Any]:
     params: dict[str, Any] = {
-        "minProtocol": 3,
-        "maxProtocol": 3,
+        "minProtocol": PROTOCOL_VERSION,
+        "maxProtocol": PROTOCOL_VERSION,
         "client": {
             "id": "gateway-client",
             "version": "1.0.0",
@@ -90,11 +78,26 @@ async def _handle_challenge(
     }
     if config.token:
         params["auth"] = {"token": config.token}
+    return params
+
+
+async def _ensure_connected(
+    ws: websockets.WebSocketClientProtocol,
+    first_message: str | bytes | None,
+    config: GatewayConfig,
+) -> None:
+    if first_message:
+        if isinstance(first_message, bytes):
+            first_message = first_message.decode("utf-8")
+        data = json.loads(first_message)
+        if data.get("type") != "event" or data.get("event") != "connect.challenge":
+            pass
+    connect_id = str(uuid4())
     response = {
         "type": "req",
         "id": connect_id,
         "method": "connect",
-        "params": params,
+        "params": _build_connect_params(config),
     }
     await ws.send(json.dumps(response))
     await _await_response(ws, connect_id)
@@ -114,7 +117,7 @@ async def openclaw_call(
                 first_message = await asyncio.wait_for(ws.recv(), timeout=2)
             except asyncio.TimeoutError:
                 first_message = None
-            await _handle_challenge(ws, first_message, config)
+            await _ensure_connected(ws, first_message, config)
             return await _send_request(ws, method, params)
     except OpenClawGatewayError:
         raise

backend/app/integrations/openclaw_gateway_protocol.py

@@ -0,0 +1,119 @@
+from __future__ import annotations
+
+PROTOCOL_VERSION = 3
+
+# NOTE: These are the base gateway methods from the OpenClaw gateway repo.
+# The gateway can expose additional methods at runtime via channel plugins.
+GATEWAY_METHODS = [
+    "health",
+    "logs.tail",
+    "channels.status",
+    "channels.logout",
+    "status",
+    "usage.status",
+    "usage.cost",
+    "tts.status",
+    "tts.providers",
+    "tts.enable",
+    "tts.disable",
+    "tts.convert",
+    "tts.setProvider",
+    "config.get",
+    "config.set",
+    "config.apply",
+    "config.patch",
+    "config.schema",
+    "exec.approvals.get",
+    "exec.approvals.set",
+    "exec.approvals.node.get",
+    "exec.approvals.node.set",
+    "exec.approval.request",
+    "exec.approval.resolve",
+    "wizard.start",
+    "wizard.next",
+    "wizard.cancel",
+    "wizard.status",
+    "talk.mode",
+    "models.list",
+    "agents.list",
+    "agents.files.list",
+    "agents.files.get",
+    "agents.files.set",
+    "skills.status",
+    "skills.bins",
+    "skills.install",
+    "skills.update",
+    "update.run",
+    "voicewake.get",
+    "voicewake.set",
+    "sessions.list",
+    "sessions.preview",
+    "sessions.patch",
+    "sessions.reset",
+    "sessions.delete",
+    "sessions.compact",
+    "last-heartbeat",
+    "set-heartbeats",
+    "wake",
+    "node.pair.request",
+    "node.pair.list",
+    "node.pair.approve",
+    "node.pair.reject",
+    "node.pair.verify",
+    "device.pair.list",
+    "device.pair.approve",
+    "device.pair.reject",
+    "device.token.rotate",
+    "device.token.revoke",
+    "node.rename",
+    "node.list",
+    "node.describe",
+    "node.invoke",
+    "node.invoke.result",
+    "node.event",
+    "cron.list",
+    "cron.status",
+    "cron.add",
+    "cron.update",
+    "cron.remove",
+    "cron.run",
+    "cron.runs",
+    "system-presence",
+    "system-event",
+    "send",
+    "agent",
+    "agent.identity.get",
+    "agent.wait",
+    "browser.request",
+    "chat.history",
+    "chat.abort",
+    "chat.send",
+]
+
+GATEWAY_EVENTS = [
+    "connect.challenge",
+    "agent",
+    "chat",
+    "presence",
+    "tick",
+    "talk.mode",
+    "shutdown",
+    "health",
+    "heartbeat",
+    "cron",
+    "node.pair.requested",
+    "node.pair.resolved",
+    "node.invoke.request",
+    "device.pair.requested",
+    "device.pair.resolved",
+    "voicewake.changed",
+    "exec.approval.requested",
+    "exec.approval.resolved",
+]
+
+GATEWAY_METHODS_SET = frozenset(GATEWAY_METHODS)
+GATEWAY_EVENTS_SET = frozenset(GATEWAY_EVENTS)
+
+
+def is_known_gateway_method(method: str) -> bool:
+    return method in GATEWAY_METHODS_SET