agent-relay-server 0.10.11 → 0.10.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay-server",
-  "version": "0.10.11",
+  "version": "0.10.13",
   "description": "Lightweight HTTP message relay for inter-agent communication across machines",
   "module": "src/index.ts",
   "type": "module",

package/scripts/orchestrator-spawn-smoke.ts

@@ -4,7 +4,7 @@ type Orchestrator = {
   status: string;
   providers: string[];
   baseDir: string;
-  managedAgents?: Array<{ label?: string; tmuxSession: string }>;
+  managedAgents?: Array<{ agentId?: string; label?: string; tmuxSession: string }>;
 };
 
 type Agent = {
@@ -115,7 +115,7 @@ for (const provider of providers) {
   });
   console.log(`registered ${provider}: ${agent.id}`);
 
-  await waitFor(`waiting for ${provider} managed session`, async () => {
+  const managed = await waitFor(`waiting for ${provider} managed session`, async () => {
     const latest = (await api<Orchestrator[]>("GET", "/orchestrators")).find((orch) => orch.id === orchestrator.id);
     return latest?.managedAgents?.find((entry) => entry.label === label || entry.tmuxSession.includes(label)) || null;
   });
@@ -127,6 +127,17 @@ for (const provider of providers) {
     const current = await apiOptional<Agent>("GET", `/agents/${encodeURIComponent(agent.id)}`);
     return current ? null : true;
   });
+  await api("POST", `/orchestrators/${encodeURIComponent(orchestrator.id)}/actions`, {
+    action: "shutdown",
+    agentId: agent.id,
+    tmuxSession: managed.tmuxSession,
+    reason: "smoke-cleanup",
+  });
+  await waitFor(`waiting for ${provider} managed session cleanup`, async () => {
+    const latest = (await api<Orchestrator[]>("GET", "/orchestrators")).find((orch) => orch.id === orchestrator.id);
+    const stillManaged = latest?.managedAgents?.some((entry) => entry.tmuxSession === managed.tmuxSession || entry.agentId === agent.id);
+    return stillManaged ? null : true;
+  });
   console.log(`shutdown ${provider}: ${label}`);
 }
 

package/src/bus.ts

@@ -3,6 +3,7 @@ import { getAgent, getDb, heartbeat, markReady, orphanTasksForAgent, setStatus,
 import { getOldestOutboxCursor, getOutboxCursor, replayEvents, type BusEvent } from "./bus-outbox";
 import { emitRelayEvent, subscribeRelayEvents, type RelayEvent } from "./events";
 import { createCommand, getCommand, updateCommand } from "./commands-db";
+import { getLifecycleManager } from "./lifecycle-manager";
 import { applyCommandToRecipe } from "./recipe-runner";
 import {
   BusProtocolError,
@@ -230,6 +231,10 @@ function handleRegister(ws: BusWebSocket, frame: RegisterFrame): void {
       meta: payload.meta,
     });
     epoch = agent.epoch;
+    getLifecycleManager().onAgentRegistered(agent.id, {
+      policyName: stringMeta(payload.meta, "policyName"),
+      spawnRequestId: stringMeta(payload.meta, "spawnRequestId"),
+    });
     emitAgentStatusEvent(agent.id);
   }
 

package/src/routes.ts

@@ -1552,7 +1552,7 @@ const postOrchestratorAction: Handler = async (req, params) => {
     const command = createCommand({
       type: action === "restart" ? "agent.restart" : "agent.shutdown",
       source: "system",
-      target: agentId || orch.agentId,
+      target: orch.agentId,
       correlationId: spawnRequestId,
       params: { action, agentId, policyName, spawnRequestId, tmuxSession, reason, requestedBy: "dashboard", requestedAt: Date.now(), orchestratorId: orch.id },
     });