agent-relay-orchestrator 0.10.1 → 0.10.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay-orchestrator",
-  "version": "0.10.1",
+  "version": "0.10.3",
   "description": "Agent Relay orchestrator — manages agent lifecycle across hosts",
   "type": "module",
   "bin": {

package/src/index.ts

@@ -2,8 +2,9 @@
 import { loadConfig, initConfigFile } from "./config";
 import { createRelayClient } from "./relay";
 import { createControlHandler } from "./control";
-import { recoverExistingSessions, hasSession } from "./tmux";
+import { hasSession } from "./tmux";
 import { startApiServer } from "./api";
+import { recoverManagedAgents } from "./recovery";
 
 const args = process.argv.slice(2);
 
@@ -62,12 +63,7 @@ async function startup(): Promise<void> {
   relay.startHeartbeatLoop();
 
   // Recover existing tmux sessions
-  const recovered = await recoverExistingSessions(config);
-  if (recovered.length > 0) {
-    console.error(`[orchestrator] Recovered ${recovered.length} existing session(s)`);
-    control.setManagedAgents(recovered);
-    await relay.updateManagedAgents(recovered);
-  }
+  await recoverManagedAgents(config, control, relay);
 
   // Start polling for command requests
   startPolling();

package/src/recovery.ts

@@ -0,0 +1,22 @@
+import type { OrchestratorConfig } from "./config";
+import type { ManagedAgentReport, RelayClient } from "./relay";
+import { recoverExistingSessions } from "./tmux";
+
+interface ManagedAgentControl {
+  setManagedAgents(agents: ManagedAgentReport[]): void;
+}
+
+export async function recoverManagedAgents(
+  config: OrchestratorConfig,
+  control: ManagedAgentControl,
+  relay: Pick<RelayClient, "updateManagedAgents">,
+  recover: (config: OrchestratorConfig) => Promise<ManagedAgentReport[]> = recoverExistingSessions,
+): Promise<ManagedAgentReport[]> {
+  const recovered = await recover(config);
+  if (recovered.length > 0) {
+    console.error(`[orchestrator] Recovered ${recovered.length} existing session(s)`);
+  }
+  control.setManagedAgents(recovered);
+  await relay.updateManagedAgents(recovered);
+  return recovered;
+}