@asm-agent/coding-agent 0.8.7 → 0.9.0

package/dist/modes/daemon/daemon-supervisor.js

@@ -66,6 +66,7 @@ const WORKER_STARTUP_GATE_FD = 3;
 const DAEMON_COMMAND_TYPES = new Set([
     "ack_result",
     "list",
+    "list_agent_peers",
     "list_saved_sessions",
     "create",
     "attach",
@@ -415,7 +416,6 @@ export class DaemonSupervisor {
     commandJournal;
     streamReconstructor = new CompactAssistantStreamReconstructor();
     compactCatchupInProgress = new Set();
-    agentPeerSyncQueue = Promise.resolve();
     pendingSessionNames = new Set();
     catalog;
     settingsManager;
@@ -501,7 +501,6 @@ export class DaemonSupervisor {
             if (adoptionFailed) {
                 throw adoptionFailure;
             }
-            await this.syncAgentPeers().catch((error) => this.log(`Could not synchronize agent peers: ${String(error)}`));
             for (const worker of this.workers.values()) {
                 this.scheduleOwnedWorkerCleanup(worker);
             }
@@ -1203,6 +1202,21 @@ export class DaemonSupervisor {
                 return undefined;
             case "list":
                 return this.handleList(client, command);
+            case "list_agent_peers": {
+                const requester = [...this.workers.values()].find((worker) => worker.descriptor.authenticationToken === command.workerToken);
+                if (!requester)
+                    throw new Error("Worker authentication failed");
+                const peers = [...this.workers.values()]
+                    .filter((worker) => worker !== requester &&
+                    this.isLiveWorker(worker) &&
+                    worker.descriptor.lifecycle === "ready" &&
+                    worker.client !== undefined)
+                    .flatMap((worker) => {
+                    const root = worker.summaries.get(worker.descriptor.rootActiveSessionId);
+                    return root ? [this.agentPeerSummary(root)] : [];
+                });
+                return success(command.id, command.type, { peers });
+            }
             case "list_saved_sessions":
                 return this.handleSavedSessionList(client, command);
             case "create": {
@@ -1216,7 +1230,6 @@ export class DaemonSupervisor {
                     const response = await this.forwardToWorker(worker, withoutSupervisorCreateFields(command));
                     if (response.success && isSessionSummary(response.data)) {
                         await this.refreshWorkerSummaries(worker);
-                        await this.syncAgentPeers().catch(() => undefined);
                         return { ...response, id: command.id, data: this.publicSummary(worker, response.data) };
                     }
                     return responseWithId(response, command.id);
@@ -1771,7 +1784,6 @@ export class DaemonSupervisor {
         await Promise.all([...this.workers.values()]
             .filter((worker) => !this.isWorkerStopping(worker))
             .map((worker) => this.refreshWorkerSummaries(worker).catch(() => undefined)));
-        await this.syncAgentPeers().catch((error) => this.log(`Could not synchronize agent peers: ${String(error)}`));
         const clientOwnedWorkers = [...this.workers.values()].filter((worker) => !this.isVisibleWorker(worker));
         // Stopping workers stay listed (with an honest workerState) because this
         // list also feeds busy-daemon safety checks in daemon-launch.
@@ -1931,7 +1943,6 @@ export class DaemonSupervisor {
             this.invalidateWorkerSessionInputPauses(worker, "Session worker stopped while input was paused");
             this.workers.delete(worker.descriptor.workerId);
             this.deleteWorkerDescriptor(worker);
-            await this.syncAgentPeers().catch(() => undefined);
             return true;
         }
         // Fail fast before waiting on anything: only a confirmed-dead process is
@@ -1981,7 +1992,6 @@ export class DaemonSupervisor {
         }
         worker.launchEnv = undefined;
         worker.transientCreateCommand = undefined;
-        await this.syncAgentPeers().catch((error) => this.log(`Could not synchronize agent peers: ${String(error)}`));
     }
     async launchWorker(command, existing, ownerClientId) {
         await this.assertRecoveryAllowed();
@@ -2065,7 +2075,6 @@ export class DaemonSupervisor {
                 rootActiveSessionId,
                 ownerClientId: existing?.descriptor.ownerClientId ?? ownerClientId,
                 sessionDir: createCommand.config?.sessionDir,
-                telemetryDisabled: createCommand.config?.telemetryDisabled,
                 createdAt: existing?.descriptor.createdAt ?? now,
                 updatedAt: now,
                 lifecycle: "starting",
@@ -2157,7 +2166,6 @@ export class DaemonSupervisor {
                 worker.launchEnv = undefined;
                 worker.transientCreateCommand = undefined;
             }
-            await this.syncAgentPeers().catch((error) => this.log(`Could not synchronize agent peers: ${String(error)}`));
             this.broadcastHeartbeatsChanged();
             return worker;
         }
@@ -2369,7 +2377,6 @@ export class DaemonSupervisor {
         worker.descriptor.lifecycle = "recovering";
         worker.descriptor.lastError = error.message;
         this.persistWorker(worker);
-        void this.syncAgentPeers().catch(() => undefined);
         void this.recoverWorker(worker);
     }
     isWorkerRecoveryEligible(worker) {
@@ -2417,7 +2424,6 @@ export class DaemonSupervisor {
             worker.descriptor.lifecycle = "recovering";
             worker.descriptor.lastError = disconnectError.message;
             this.persistWorker(worker);
-            void this.syncAgentPeers().catch(() => undefined);
             void this.recoverWorker(worker);
             return;
         }
@@ -2608,7 +2614,6 @@ export class DaemonSupervisor {
                             worker.descriptor.lifecycle = "ready";
                             worker.descriptor.consecutiveFailures = 0;
                             this.persistWorker(worker);
-                            await this.syncAgentPeers().catch((error) => this.log(`Could not synchronize agent peers after worker recovery: ${String(error)}`));
                             this.broadcastHeartbeatsChanged();
                             return;
                         }
@@ -2634,7 +2639,6 @@ export class DaemonSupervisor {
                         worker.descriptor.lifecycle = "failed";
                         worker.descriptor.lastError = "Waiting for a client with fresh runtime context";
                         this.persistWorker(worker);
-                        await this.syncAgentPeers().catch(() => undefined);
                         return;
                     }
                     const safeToKillWorkerProcess = processAlive && processIdentityMatches && worker.descriptor.processStartId !== undefined;
@@ -2671,7 +2675,6 @@ export class DaemonSupervisor {
             }
             worker.descriptor.lifecycle = "failed";
             this.persistWorker(worker);
-            await this.syncAgentPeers().catch(() => undefined);
             this.log(`Worker ${worker.descriptor.workerId} failed after three recovery attempts`);
         })().finally(() => {
             worker.recovery = undefined;
@@ -2893,29 +2896,6 @@ export class DaemonSupervisor {
             ignoreSessionId: target.id,
         });
     }
-    syncAgentPeers() {
-        const sync = this.agentPeerSyncQueue
-            .catch(() => undefined)
-            .then(async () => {
-            const readyWorkers = [...this.workers.values()].filter((worker) => this.isLiveWorker(worker) && worker.descriptor.lifecycle === "ready" && worker.client !== undefined);
-            await Promise.all(readyWorkers.map(async (worker) => {
-                const peers = [
-                    ...readyWorkers
-                        .filter((candidate) => candidate !== worker)
-                        .flatMap((candidate) => {
-                        const root = candidate.summaries.get(candidate.descriptor.rootActiveSessionId);
-                        return root ? [this.agentPeerSummary(root)] : [];
-                    }),
-                ];
-                const response = await worker.client.requestWorker({ type: "worker_sync_agent_peers", peers }, 5000);
-                if (!response.success) {
-                    throw new Error(response.error);
-                }
-            }));
-        });
-        this.agentPeerSyncQueue = sync;
-        return sync;
-    }
     isVisibleWorker(worker) {
         return worker.descriptor.ownerClientId === undefined;
     }
@@ -3102,7 +3082,6 @@ export class DaemonSupervisor {
             if (ownedWorker.descriptor.ownerClientId !== this.protocolClientId(client)) {
                 throw new Error(`Unknown active session: ${command.activeSessionId}`);
             }
-            this.assertTelemetryAttachAllowed(ownedWorker, command.telemetryDisabled);
             ownedWorker.launchEnv = command.launchEnv ?? ownedWorker.launchEnv;
             if (!ownedWorker.client || ownedWorker.descriptor.lifecycle !== "ready") {
                 if (command.recoveryConfig) {
@@ -3110,7 +3089,6 @@ export class DaemonSupervisor {
                         ...ownedWorker.descriptor.createCommand,
                         config: {
                             ...command.recoveryConfig,
-                            ...(ownedWorker.descriptor.telemetryDisabled === true ? { telemetryDisabled: true } : {}),
                         },
                         env: command.env,
                         launchEnv: command.launchEnv,
@@ -3130,7 +3108,6 @@ export class DaemonSupervisor {
             }
         }
         const match = await this.findWorkerForClient(client, command.activeSessionId);
-        this.assertTelemetryAttachAllowed(match.worker, command.telemetryDisabled);
         this.requireAvailableWorkerClient(match.worker);
         const activeSessionId = match.summary.activeSessionId ?? match.summary.id;
         const duplicateValidation = this.currentSnapshotGeneration(match.worker, activeSessionId)?.validation;
@@ -3277,11 +3254,6 @@ export class DaemonSupervisor {
             throw error;
         }
     }
-    assertTelemetryAttachAllowed(worker, telemetryDisabled) {
-        if (telemetryDisabled && worker.descriptor.telemetryDisabled !== true) {
-            throw new Error("Cannot attach to this active agent while telemetry is disabled for the current invocation. Stop the agent and retry so it can restart without telemetry.");
-        }
-    }
     cacheLoadedSnapshot(worker, activeSessionId, loaded, observedSnapshotId) {
         const currentTranscript = worker.transcriptCaches.get(activeSessionId);
         const currentGeneration = this.currentSnapshotGeneration(worker, activeSessionId);
@@ -3861,16 +3833,12 @@ export class DaemonSupervisor {
             this.writeSerialized(client, publicPayload);
         }
         if (outboundType === "session_replaced" || outboundType === "session_closed") {
-            void this.refreshWorkerSummaries(worker)
-                .then(() => this.syncAgentPeers())
-                .catch(() => undefined);
+            void this.refreshWorkerSummaries(worker).catch(() => undefined);
         }
         else if (sessionEventType === "turn_start" ||
             sessionEventType === "turn_end" ||
             sessionEventType === "rlm_child_update") {
-            void this.refreshWorkerSummaries(worker)
-                .then(() => this.syncAgentPeers())
-                .catch(() => undefined);
+            void this.refreshWorkerSummaries(worker).catch(() => undefined);
         }
         if (decodedOutbound?.type === "session_closed" &&
             decodedOutbound.reason === "shutdown" &&
@@ -3884,7 +3852,6 @@ export class DaemonSupervisor {
                 this.invalidateWorkerSessionInputPauses(worker, "Session worker stopped while input was paused");
                 this.workers.delete(worker.descriptor.workerId);
                 this.deleteWorkerDescriptor(worker);
-                void this.syncAgentPeers().catch(() => undefined);
             }
         }
     }
@@ -4380,7 +4347,6 @@ export class DaemonSupervisor {
             this.deleteWorkerDescriptor(worker);
         }
         if (!this.shuttingDown) {
-            void this.syncAgentPeers().catch(() => undefined);
             this.broadcastHeartbeatsChanged();
         }
     }