@kb-labs/agent-entry 2.118.2 → 2.119.0-canary.cee505c72

package/dist/rest/handlers/correct-handler.js

@@ -1,17 +1,14 @@
-import { defineHandler, useAnalytics, rethrowForRest, useCache, usePlatform } from '@kb-labs/sdk';
+import { defineHandler, useAnalytics, rethrowForRest, useCache } from '@kb-labs/sdk';
 import { AGENT_ANALYTICS_EVENTS } from '@kb-labs/agent-contracts';
 
 // src/rest/handlers/correct-handler.ts
 var CACHE_PREFIX = "agent:run:";
-var EVENT_TOPIC_PREFIX = "agent:events:";
 var CACHE_TTL = 36e5;
 var RunManagerImpl = class {
-  /** Live agents and listeners (not cacheable) */
+  /** Live agents (not cacheable) */
   activeRuns = /* @__PURE__ */ new Map();
-  /** Session-level listeners: sessionId → Set<callback> — receive events from ALL runs in session */
+  /** Session-level listeners: sessionId → Set<callback> — receive updates from ALL runs in session */
   sessionListeners = /* @__PURE__ */ new Map();
-  /** sessionId per runId — set when run is registered */
-  runSessionMap = /* @__PURE__ */ new Map();
   /**
    * Generate unique run ID
    */
@@ -21,7 +18,7 @@ var RunManagerImpl = class {
   /**
    * Register a new run
    */
-  async register(runId, task, agent, sessionManager, sessionId) {
+  async register(runId, task, agent, sessionManager) {
     const now = (/* @__PURE__ */ new Date()).toISOString();
     const run = {
       runId,
@@ -29,15 +26,9 @@ var RunManagerImpl = class {
       status: "pending",
       agent,
       sessionManager,
-      startedAt: now,
-      listeners: /* @__PURE__ */ new Set(),
-      lastSeq: 0,
-      eventBuffer: []
+      startedAt: now
     };
     this.activeRuns.set(runId, run);
-    if (sessionId) {
-      this.runSessionMap.set(runId, sessionId);
-    }
     await this.saveToCache(run);
     return run;
   }
@@ -100,101 +91,42 @@ var RunManagerImpl = class {
     };
     await cache.set(`${CACHE_PREFIX}${run.runId}`, state, CACHE_TTL);
   }
-  /** Track subscriptions for cleanup */
-  subscriptions = /* @__PURE__ */ new Map();
   /**
-   * Add event listener to run (uses eventBus for cross-process)
+   * Broadcast turn deltas to all session-level listeners. Callers MUST await
+   * persistence (sessionManager.addEvent) before calling this, so listeners
+   * never observe a delta whose underlying turn state isn't on disk yet —
+   * that ordering is what makes gap-recovery / reconnect resume correct.
    */
-  addListener(runId, callback) {
-    const platform = usePlatform();
-    if (!platform?.eventBus) {
-      const run2 = this.activeRuns.get(runId);
-      if (run2) {
-        run2.listeners.add(callback);
-        return true;
-      }
-      return false;
-    }
-    const topic = `${EVENT_TOPIC_PREFIX}${runId}`;
-    const unsubscribe = platform.eventBus.subscribe(topic, async (event) => {
-      try {
-        callback(event);
-      } catch {
-      }
-    });
-    if (!this.subscriptions.has(runId)) {
-      this.subscriptions.set(runId, /* @__PURE__ */ new Map());
-    }
-    this.subscriptions.get(runId).set(callback, unsubscribe);
-    const run = this.activeRuns.get(runId);
-    if (run) {
-      run.listeners.add(callback);
+  broadcastSessionDeltas(sessionId, deltas) {
+    if (deltas.length === 0) {
+      return;
     }
-    return true;
+    this.notifySessionListeners(sessionId, { kind: "deltas", deltas });
   }
   /**
-   * Remove event listener from run
+   * Broadcast a run's terminal outcome to session-level listeners. Called
+   * directly from the run's own completion handler (after `agent.execute()`
+   * resolves) rather than derived reactively from an `agent:end` event —
+   * the completion handler already has definitive success/summary/duration,
+   * and this guarantees the message fires exactly once, deterministically.
    */
-  removeListener(runId, callback) {
-    const runSubs = this.subscriptions.get(runId);
-    if (runSubs) {
-      const unsubscribe = runSubs.get(callback);
-      if (unsubscribe) {
-        unsubscribe();
-        runSubs.delete(callback);
-      }
-      if (runSubs.size === 0) {
-        this.subscriptions.delete(runId);
-      }
-    }
-    const run = this.activeRuns.get(runId);
-    if (run) {
-      run.listeners.delete(callback);
-    }
+  broadcastRunCompleted(sessionId, payload) {
+    this.notifySessionListeners(sessionId, { kind: "run:completed", ...payload });
   }
-  /**
-   * Broadcast event to all listeners of a run (uses eventBus for cross-process)
-   * Assigns monotonic sequence number for reliable ordering
-   * @returns The event with seq assigned (for persistence)
-   */
-  broadcast(runId, event) {
-    const platform = usePlatform();
-    const run = this.activeRuns.get(runId);
-    let seqEvent = event;
-    if (run) {
-      run.lastSeq++;
-      seqEvent = { ...event, seq: run.lastSeq, runId };
-      run.eventBuffer.push(seqEvent);
-    }
-    if (platform?.eventBus) {
-      const topic = `${EVENT_TOPIC_PREFIX}${runId}`;
-      platform.eventBus.publish(topic, seqEvent).catch(() => {
-      });
-    }
-    if (run) {
-      for (const listener of run.listeners) {
-        try {
-          listener(seqEvent);
-        } catch {
-        }
-      }
+  notifySessionListeners(sessionId, payload) {
+    const listeners = this.sessionListeners.get(sessionId);
+    if (!listeners) {
+      return;
     }
-    const sessionId = this.runSessionMap.get(runId);
-    if (sessionId) {
-      const sListeners = this.sessionListeners.get(sessionId);
-      if (sListeners) {
-        for (const listener of sListeners) {
-          try {
-            listener(seqEvent);
-          } catch {
-          }
-        }
+    for (const listener of listeners) {
+      try {
+        listener(payload);
+      } catch {
       }
     }
-    return seqEvent;
   }
   /**
-   * Add a session-level listener — receives events from ALL runs in this session.
+   * Add a session-level listener — receives updates from ALL runs in this session.
    */
   addSessionListener(sessionId, callback) {
     if (!this.sessionListeners.has(sessionId)) {
@@ -214,21 +146,6 @@ var RunManagerImpl = class {
       }
     }
   }
-  /**
-   * Get replay buffer for a run (all events emitted since start).
-   * Used by WS handler to send missed events on late connection.
-   * Optionally filter by afterSeq to only get events the client hasn't seen.
-   */
-  getEventBuffer(runId, afterSeq) {
-    const run = this.activeRuns.get(runId);
-    if (!run) {
-      return [];
-    }
-    if (afterSeq != null) {
-      return run.eventBuffer.filter((e) => e.seq != null && e.seq > afterSeq);
-    }
-    return [...run.eventBuffer];
-  }
   /**
    * List all active runs
    */

package/dist/rest/handlers/correct-handler.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/rest/run-manager.ts","../../../src/rest/handlers/correct-handler.ts"],"names":["run"],"mappings":";;;;AAaA,IAAM,YAAA,GAAe,YAAA;AACrB,IAAM,kBAAA,GAAqB,eAAA;AAC3B,IAAM,SAAA,GAAY,IAAA;AAqClB,IAAM,iBAAN,MAAqB;AAAA;AAAA,EAEX,UAAA,uBAAyC,GAAA,EAAI;AAAA;AAAA,EAG7C,gBAAA,uBAA6D,GAAA,EAAI;AAAA;AAAA,EAGjE,aAAA,uBAAyC,GAAA,EAAI;AAAA;AAAA;AAAA;AAAA,EAKrD,aAAA,GAAwB;AACtB,IAAA,OAAO,CAAA,IAAA,EAAO,IAAA,CAAK,GAAA,EAAK,IAAI,IAAA,CAAK,MAAA,EAAO,CAAE,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAA,CACJ,KAAA,EACA,IAAA,EACA,KAAA,EACA,gBACA,SAAA,EACoB;AACpB,IAAA,MAAM,GAAA,GAAA,iBAAM,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAEnC,IAAA,MAAM,GAAA,GAAiB;AAAA,MACrB,KAAA;AAAA,MACA,IAAA;AAAA,MACA,MAAA,EAAQ,SAAA;AAAA,MACR,KAAA;AAAA,MACA,cAAA;AAAA,MACA,SAAA,EAAW,GAAA;AAAA,MACX,SAAA,sBAAe,GAAA,EAAI;AAAA,MACnB,OAAA,EAAS,CAAA;AAAA,MACT,aAAa;AAAC,KAChB;AAGA,IAAA,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,GAAG,CAAA;AAG9B,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,KAAA,EAAO,SAAS,CAAA;AAAA,IACzC;AAGA,IAAA,MAAM,IAAA,CAAK,YAAY,GAAG,CAAA;AAE1B,IAAA,OAAO,GAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,KAAA,EAAsC;AACxC,IAAA,OAAO,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,OAAO,KAAA,EAAiC;AAE5C,IAAA,IAAI,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,QAAA,CAAS,KAAK,CAAA;AACvC,IAAA,OAAO,KAAA,KAAU,IAAA;AAAA,EACnB;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,SAAS,KAAA,EAAyC;AACtD,IAAA,MAAM,QAAQ,QAAA,EAAS;AACvB,IAAA,IAAI,CAAC,KAAA,EAAO;AAAC,MAAA,OAAO,IAAA;AAAA,IAAK;AAEzB,IAAA,OAAO,MAAM,GAAA,CAAc,CAAA,EAAG,YAAY,CAAA,EAAG,KAAK,CAAA,CAAE,CAAA;AAAA,EACtD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAA,CAAa,KAAA,EAAe,MAAA,EAAmB,KAAA,EAA0C;AAC7F,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,GAAA,EAAK;AACP,MAAA,GAAA,CAAI,MAAA,GAAS,MAAA;AACb,MAAA,IAAI,KAAA,EAAO;AACT,QAAA,MAAA,CAAO,MAAA,CAAO,KAAK,KAAK,CAAA;AAAA,MAC1B;AACA,MAAA,MAAM,IAAA,CAAK,YAAY,GAAG,CAAA;AAAA,IAC5B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,YAAY,GAAA,EAA+B;AACvD,IAAA,MAAM,QAAQ,QAAA,EAAS;AACvB,IAAA,IAAI,CAAC,KAAA,EAAO;AAAC,MAAA;AAAA,IAAO;AAEpB,IAAA,MAAM,KAAA,GAAkB;AAAA,MACtB,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,QAAQ,GAAA,CAAI,MAAA;AAAA,MACZ,WAAW,GAAA,CAAI,SAAA;AAAA,MACf,aAAa,GAAA,CAAI,WAAA;AAAA,MACjB,YAAY,GAAA,CAAI,UAAA;AAAA,MAChB,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,OAAO,GAAA,CAAI;AAAA,KACb;AAEA,IAAA,MAAM,KAAA,CAAM,IAAI,CAAA,EAAG,YAAY,GAAG,GAAA,CAAI,KAAK,CAAA,CAAA,EAAI,KAAA,EAAO,SAAS,CAAA;AAAA,EACjE;AAAA;AAAA,EAGQ,aAAA,uBAAsE,GAAA,EAAI;AAAA;AAAA;AAAA;AAAA,EAKlF,WAAA,CAAY,OAAe,QAAA,EAAuC;AAChE,IAAA,MAAM,WAAW,WAAA,EAAY;AAC7B,IAAA,IAAI,CAAC,UAAU,QAAA,EAAU;AAEvB,MAAA,MAAMA,IAAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,MAAA,IAAIA,IAAAA,EAAK;AACP,QAAAA,IAAAA,CAAI,SAAA,CAAU,GAAA,CAAI,QAAQ,CAAA;AAC1B,QAAA,OAAO,IAAA;AAAA,MACT;AACA,MAAA,OAAO,KAAA;AAAA,IACT;AAGA,IAAA,MAAM,KAAA,GAAQ,CAAA,EAAG,kBAAkB,CAAA,EAAG,KAAK,CAAA,CAAA;AAC3C,IAAA,MAAM,cAAc,QAAA,CAAS,QAAA,CAAS,SAAA,CAAsB,KAAA,EAAO,OAAO,KAAA,KAAU;AAClF,MAAA,IAAI;AACF,QAAA,QAAA,CAAS,KAAK,CAAA;AAAA,MAChB,CAAA,CAAA,MAAQ;AAAA,MAER;AAAA,IACF,CAAC,CAAA;AAGD,IAAA,IAAI,CAAC,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,KAAK,CAAA,EAAG;AAClC,MAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,KAAA,kBAAO,IAAI,KAAK,CAAA;AAAA,IACzC;AACA,IAAA,IAAA,CAAK,cAAc,GAAA,CAAI,KAAK,CAAA,CAAG,GAAA,CAAI,UAAU,WAAW,CAAA;AAGxD,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,GAAA,EAAK;AACP,MAAA,GAAA,CAAI,SAAA,CAAU,IAAI,QAAQ,CAAA;AAAA,IAC5B;AAEA,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,cAAA,CAAe,OAAe,QAAA,EAAoC;AAEhE,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,KAAK,CAAA;AAC5C,IAAA,IAAI,OAAA,EAAS;AACX,MAAA,MAAM,WAAA,GAAc,OAAA,CAAQ,GAAA,CAAI,QAAQ,CAAA;AACxC,MAAA,IAAI,WAAA,EAAa;AACf,QAAA,WAAA,EAAY;AACZ,QAAA,OAAA,CAAQ,OAAO,QAAQ,CAAA;AAAA,MACzB;AACA,MAAA,IAAI,OAAA,CAAQ,SAAS,CAAA,EAAG;AACtB,QAAA,IAAA,CAAK,aAAA,CAAc,OAAO,KAAK,CAAA;AAAA,MACjC;AAAA,IACF;AAGA,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,GAAA,EAAK;AACP,MAAA,GAAA,CAAI,SAAA,CAAU,OAAO,QAAQ,CAAA;AAAA,IAC/B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,SAAA,CAAU,OAAe,KAAA,EAA+B;AACtD,IAAA,MAAM,WAAW,WAAA,EAAY;AAC7B,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAGrC,IAAA,IAAI,QAAA,GAAW,KAAA;AACf,IAAA,IAAI,GAAA,EAAK;AACP,MAAA,GAAA,CAAI,OAAA,EAAA;AACJ,MAAA,QAAA,GAAW,EAAE,GAAG,KAAA,EAAO,GAAA,EAAK,GAAA,CAAI,SAAS,KAAA,EAAM;AAG/C,MAAA,GAAA,CAAI,WAAA,CAAY,KAAK,QAAQ,CAAA;AAAA,IAC/B;AAGA,IAAA,IAAI,UAAU,QAAA,EAAU;AACtB,MAAA,MAAM,KAAA,GAAQ,CAAA,EAAG,kBAAkB,CAAA,EAAG,KAAK,CAAA,CAAA;AAC3C,MAAA,QAAA,CAAS,SAAS,OAAA,CAAQ,KAAA,EAAO,QAAQ,CAAA,CAAE,MAAM,MAAM;AAAA,MAEvD,CAAC,CAAA;AAAA,IACH;AAGA,IAAA,IAAI,GAAA,EAAK;AACP,MAAA,KAAA,MAAW,QAAA,IAAY,IAAI,SAAA,EAAW;AACpC,QAAA,IAAI;AACF,UAAA,QAAA,CAAS,QAAQ,CAAA;AAAA,QACnB,CAAA,CAAA,MAAQ;AAAA,QAER;AAAA,MACF;AAAA,IACF;AAGA,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,KAAK,CAAA;AAC9C,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACtD,MAAA,IAAI,UAAA,EAAY;AACd,QAAA,KAAA,MAAW,YAAY,UAAA,EAAY;AACjC,UAAA,IAAI;AACF,YAAA,QAAA,CAAS,QAAQ,CAAA;AAAA,UACnB,CAAA,CAAA,MAAQ;AAAA,UAER;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,IAAA,OAAO,QAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,kBAAA,CAAmB,WAAmB,QAAA,EAAoC;AACxE,IAAA,IAAI,CAAC,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA,EAAG;AACzC,MAAA,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAA,kBAAW,IAAI,KAAK,CAAA;AAAA,IAChD;AACA,IAAA,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA,CAAG,IAAI,QAAQ,CAAA;AAAA,EACpD;AAAA;AAAA;AAAA;AAAA,EAKA,qBAAA,CAAsB,WAAmB,QAAA,EAAoC;AAC3E,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACrD,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,SAAA,CAAU,OAAO,QAAQ,CAAA;AACzB,MAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AACxB,QAAA,IAAA,CAAK,gBAAA,CAAiB,OAAO,SAAS,CAAA;AAAA,MACxC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,cAAA,CAAe,OAAe,QAAA,EAAiC;AAC7D,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,CAAC,GAAA,EAAK;AAAC,MAAA,OAAO,EAAC;AAAA,IAAE;AAErB,IAAA,IAAI,YAAY,IAAA,EAAM;AACpB,MAAA,OAAO,GAAA,CAAI,WAAA,CAAY,MAAA,CAAO,CAAC,CAAA,KAAM,EAAE,GAAA,IAAO,IAAA,IAAQ,CAAA,CAAE,GAAA,GAAM,QAAQ,CAAA;AAAA,IACxE;AACA,IAAA,OAAO,CAAC,GAAG,GAAA,CAAI,WAAW,CAAA;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA,EAKA,UAAA,GAA2F;AACzF,IAAA,OAAO,KAAA,CAAM,KAAK,IAAA,CAAK,UAAA,CAAW,QAAQ,CAAA,CAAE,IAAI,CAAA,CAAA,MAAM;AAAA,MACpD,OAAO,CAAA,CAAE,KAAA;AAAA,MACT,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,QAAQ,CAAA,CAAE,MAAA;AAAA,MACV,WAAW,CAAA,CAAE;AAAA,KACf,CAAE,CAAA;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,YAAY,KAAA,EAAwB;AAClC,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,CAAC,GAAA,IAAO,GAAA,CAAI,WAAW,SAAA,IAAa,CAAC,IAAI,KAAA,EAAO;AAClD,MAAA,OAAO,KAAA;AAAA,IACT;AACA,IAAA,GAAA,CAAI,MAAM,WAAA,EAAY;AACtB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,OAAA,GAAgB;AACd,IAAA,KAAA,MAAW,CAAC,KAAA,EAAO,GAAG,CAAA,IAAK,KAAK,UAAA,EAAY;AAC1C,MAAA,IAAI,GAAA,CAAI,WAAW,WAAA,IAAe,GAAA,CAAI,WAAW,QAAA,IAAY,GAAA,CAAI,WAAW,SAAA,EAAW;AAErF,QAAA,IAAA,CAAK,UAAA,CAAW,OAAO,KAAK,CAAA;AAAA,MAC9B;AAAA,IACF;AAAA,EACF;AACF,CAAA;AAQA,IAAM,UAAA,GAAa,0BAAA;AACnB,IAAI,CAAE,UAAA,CAAuC,UAAU,CAAA,EAAG;AACxD,EAAC,UAAA,CAAuC,UAAU,CAAA,GAAI,IAAI,cAAA,EAAe;AAC3E;AACO,IAAM,UAAA,GAAc,WAAuC,UAAU,CAAA;;;AClX5E,IAAO,0BAAQ,aAAA,CAAc;AAAA,EAC3B,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EAC6B;AAC7B,IAAA,IAAI;AACF,MAAA,MAAM,EAAE,KAAA,EAAM,GAAI,KAAA,CAAM,MAAA;AACxB,MAAA,MAAM,OAAO,KAAA,CAAM,IAAA;AAEnB,MAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAClB,QAAA,MAAM,IAAI,MAAM,gCAAgC,CAAA;AAAA,MAClD;AAEA,MAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,CAAA,qCAAA,EAAwC,KAAK,CAAA,EAAA,EAAK,IAAA,CAAK,OAAA,CAAQ,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,GAAA,CAAK,CAAA;AAEzG,MAAA,MAAM,GAAA,GAAM,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAEhC,MAAA,IAAI,CAAC,GAAA,EAAK;AACR,QAAA,OAAO;AAAA,UACL,YAAA,EAAc,EAAA;AAAA,UACd,UAAU,EAAC;AAAA,UACX,MAAA,EAAQ,OAAO,KAAK,CAAA,UAAA,CAAA;AAAA,UACpB,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAEA,MAAA,IAAI,GAAA,CAAI,WAAW,SAAA,EAAW;AAC5B,QAAA,OAAO;AAAA,UACL,YAAA,EAAc,EAAA;AAAA,UACd,UAAU,EAAC;AAAA,UACX,MAAA,EAAQ,CAAA,IAAA,EAAO,KAAK,CAAA,yBAAA,EAA4B,IAAI,MAAM,CAAA,CAAA,CAAA;AAAA,UAC1D,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAGA,MAAA,MAAM,YAAA,GAAe,CAAA,KAAA,EAAQ,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AACjF,MAAA,GAAA,CAAI,KAAA,EAAO,iBAAA,CAAkB,IAAA,CAAK,OAAO,CAAA;AACzC,MAAA,MAAM,MAAA,GAA6B;AAAA,QACjC,YAAA;AAAA,QACA,QAAA,EAAU,CAAC,GAAA,CAAI,KAAA,EAAO,WAAW,KAAK,CAAA;AAAA,QACtC,MAAA,EAAQ,gCAAA;AAAA,QACR,OAAA,EAAS,CAAC,CAAC,GAAA,CAAI;AAAA,OACjB;AAGA,MAAA,MAAM,YAAY,YAAA,EAAa;AAC/B,MAAA,MAAM,SAAA,EAAW,KAAA;AAAA,QACf,MAAA,CAAO,OAAA,GAAU,sBAAA,CAAuB,kBAAA,GAAqB,sBAAA,CAAuB,mBAAA;AAAA,QACpF;AAAA,UACE,KAAA;AAAA,UACA,cAAc,MAAA,CAAO,YAAA;AAAA,UACrB,UAAU,MAAA,CAAO,QAAA;AAAA,UACjB,aAAA,EAAe,KAAK,OAAA,CAAQ,MAAA;AAAA,UAC5B,eAAe,IAAA,CAAK;AAAA;AACtB,OACF;AAEA,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,QAClB,CAAA,6BAAA,EAAgC,OAAO,YAAY,CAAA,YAAA,EAAe,OAAO,QAAA,CAAS,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,OAC9F;AAEA,MAAA,OAAO,MAAA;AAAA,IACT,SAAS,GAAA,EAAK;AACZ,MAAA,cAAA,CAAe,GAAG,CAAA;AAAA,IACpB;AAAA,EACF;AACF,CAAC","file":"correct-handler.js","sourcesContent":["/**\n * Run Manager - tracks active agent runs for REST/WS API\n *\n * Uses platform cache for persistence and in-memory Map for active agents.\n * Cache stores serializable run state, Map stores live agent references.\n * Uses platform eventBus for cross-process event broadcasting.\n */\n\nimport type { IAgentRunner } from '@kb-labs/agent-sdk';\nimport type { SessionManager } from '@kb-labs/agent-core';\nimport type { AgentEvent, AgentEventCallback } from '@kb-labs/agent-contracts';\nimport { useCache, usePlatform } from '@kb-labs/sdk';\n\nconst CACHE_PREFIX = 'agent:run:';\nconst EVENT_TOPIC_PREFIX = 'agent:events:';\nconst CACHE_TTL = 3600000; // 1 hour\n\n/**\n * Run status\n */\nexport type RunStatus = 'pending' | 'running' | 'completed' | 'failed' | 'stopped';\n\n/**\n * Serializable run state (stored in cache)\n */\nexport interface RunState {\n  runId: string;\n  task: string;\n  status: RunStatus;\n  startedAt: string;\n  completedAt?: string;\n  durationMs?: number;\n  summary?: string;\n  error?: string;\n}\n\n/**\n * Active run with live agent (in-memory only)\n */\nexport interface ActiveRun extends RunState {\n  agent?: IAgentRunner;\n  sessionManager: SessionManager;\n  listeners: Set<AgentEventCallback>;\n  /** Monotonic sequence counter for event ordering */\n  lastSeq: number;\n  /** Replay buffer: all events emitted since run start (for late WS connections) */\n  eventBuffer: AgentEvent[];\n}\n\n/**\n * Run Manager implementation\n */\nclass RunManagerImpl {\n  /** Live agents and listeners (not cacheable) */\n  private activeRuns: Map<string, ActiveRun> = new Map();\n\n  /** Session-level listeners: sessionId → Set<callback> — receive events from ALL runs in session */\n  private sessionListeners: Map<string, Set<AgentEventCallback>> = new Map();\n\n  /** sessionId per runId — set when run is registered */\n  private runSessionMap: Map<string, string> = new Map();\n\n  /**\n   * Generate unique run ID\n   */\n  generateRunId(): string {\n    return `run-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;\n  }\n\n  /**\n   * Register a new run\n   */\n  async register(\n    runId: string,\n    task: string,\n    agent: IAgentRunner | undefined,\n    sessionManager: SessionManager,\n    sessionId?: string\n  ): Promise<ActiveRun> {\n    const now = new Date().toISOString();\n\n    const run: ActiveRun = {\n      runId,\n      task,\n      status: 'pending',\n      agent,\n      sessionManager,\n      startedAt: now,\n      listeners: new Set(),\n      lastSeq: 0,\n      eventBuffer: [],\n    };\n\n    // Store in memory (for live agent)\n    this.activeRuns.set(runId, run);\n\n    // Track sessionId → runId mapping for session-level listeners\n    if (sessionId) {\n      this.runSessionMap.set(runId, sessionId);\n    }\n\n    // Store serializable state in cache\n    await this.saveToCache(run);\n\n    return run;\n  }\n\n  /**\n   * Get run by ID (from memory first, then cache for state)\n   */\n  get(runId: string): ActiveRun | undefined {\n    return this.activeRuns.get(runId);\n  }\n\n  /**\n   * Check if run exists (in memory or cache)\n   */\n  async exists(runId: string): Promise<boolean> {\n    // Check memory first\n    if (this.activeRuns.has(runId)) {\n      return true;\n    }\n    // Fallback to cache\n    const state = await this.getState(runId);\n    return state !== null;\n  }\n\n  /**\n   * Get run state from cache (for completed runs or cross-process access)\n   */\n  async getState(runId: string): Promise<RunState | null> {\n    const cache = useCache();\n    if (!cache) {return null;}\n\n    return cache.get<RunState>(`${CACHE_PREFIX}${runId}`);\n  }\n\n  /**\n   * Update run status\n   */\n  async updateStatus(runId: string, status: RunStatus, extra?: Partial<RunState>): Promise<void> {\n    const run = this.activeRuns.get(runId);\n    if (run) {\n      run.status = status;\n      if (extra) {\n        Object.assign(run, extra);\n      }\n      await this.saveToCache(run);\n    }\n  }\n\n  /**\n   * Save run state to cache\n   */\n  private async saveToCache(run: ActiveRun): Promise<void> {\n    const cache = useCache();\n    if (!cache) {return;}\n\n    const state: RunState = {\n      runId: run.runId,\n      task: run.task,\n      status: run.status,\n      startedAt: run.startedAt,\n      completedAt: run.completedAt,\n      durationMs: run.durationMs,\n      summary: run.summary,\n      error: run.error,\n    };\n\n    await cache.set(`${CACHE_PREFIX}${run.runId}`, state, CACHE_TTL);\n  }\n\n  /** Track subscriptions for cleanup */\n  private subscriptions: Map<string, Map<AgentEventCallback, () => void>> = new Map();\n\n  /**\n   * Add event listener to run (uses eventBus for cross-process)\n   */\n  addListener(runId: string, callback: AgentEventCallback): boolean {\n    const platform = usePlatform();\n    if (!platform?.eventBus) {\n      // Fallback to in-memory only\n      const run = this.activeRuns.get(runId);\n      if (run) {\n        run.listeners.add(callback);\n        return true;\n      }\n      return false;\n    }\n\n    // Subscribe to eventBus topic for this run\n    const topic = `${EVENT_TOPIC_PREFIX}${runId}`;\n    const unsubscribe = platform.eventBus.subscribe<AgentEvent>(topic, async (event) => {\n      try {\n        callback(event);\n      } catch {\n        // Ignore callback errors\n      }\n    });\n\n    // Track subscription for cleanup\n    if (!this.subscriptions.has(runId)) {\n      this.subscriptions.set(runId, new Map());\n    }\n    this.subscriptions.get(runId)!.set(callback, unsubscribe);\n\n    // Also add to in-memory if run exists locally\n    const run = this.activeRuns.get(runId);\n    if (run) {\n      run.listeners.add(callback);\n    }\n\n    return true;\n  }\n\n  /**\n   * Remove event listener from run\n   */\n  removeListener(runId: string, callback: AgentEventCallback): void {\n    // Unsubscribe from eventBus\n    const runSubs = this.subscriptions.get(runId);\n    if (runSubs) {\n      const unsubscribe = runSubs.get(callback);\n      if (unsubscribe) {\n        unsubscribe();\n        runSubs.delete(callback);\n      }\n      if (runSubs.size === 0) {\n        this.subscriptions.delete(runId);\n      }\n    }\n\n    // Also remove from in-memory\n    const run = this.activeRuns.get(runId);\n    if (run) {\n      run.listeners.delete(callback);\n    }\n  }\n\n  /**\n   * Broadcast event to all listeners of a run (uses eventBus for cross-process)\n   * Assigns monotonic sequence number for reliable ordering\n   * @returns The event with seq assigned (for persistence)\n   */\n  broadcast(runId: string, event: AgentEvent): AgentEvent {\n    const platform = usePlatform();\n    const run = this.activeRuns.get(runId);\n\n    // Assign sequence number for ordering\n    let seqEvent = event;\n    if (run) {\n      run.lastSeq++;\n      seqEvent = { ...event, seq: run.lastSeq, runId };\n\n      // Store in replay buffer (for late WS connections)\n      run.eventBuffer.push(seqEvent);\n    }\n\n    // Publish to eventBus for cross-process delivery\n    if (platform?.eventBus) {\n      const topic = `${EVENT_TOPIC_PREFIX}${runId}`;\n      platform.eventBus.publish(topic, seqEvent).catch(() => {\n        // Ignore publish errors\n      });\n    }\n\n    // Also notify local in-memory listeners (for same-process)\n    if (run) {\n      for (const listener of run.listeners) {\n        try {\n          listener(seqEvent);\n        } catch {\n          // Ignore listener errors\n        }\n      }\n    }\n\n    // Notify session-level listeners (persistent connections)\n    const sessionId = this.runSessionMap.get(runId);\n    if (sessionId) {\n      const sListeners = this.sessionListeners.get(sessionId);\n      if (sListeners) {\n        for (const listener of sListeners) {\n          try {\n            listener(seqEvent);\n          } catch {\n            // Ignore listener errors\n          }\n        }\n      }\n    }\n\n    return seqEvent;\n  }\n\n  /**\n   * Add a session-level listener — receives events from ALL runs in this session.\n   */\n  addSessionListener(sessionId: string, callback: AgentEventCallback): void {\n    if (!this.sessionListeners.has(sessionId)) {\n      this.sessionListeners.set(sessionId, new Set());\n    }\n    this.sessionListeners.get(sessionId)!.add(callback);\n  }\n\n  /**\n   * Remove a session-level listener.\n   */\n  removeSessionListener(sessionId: string, callback: AgentEventCallback): void {\n    const listeners = this.sessionListeners.get(sessionId);\n    if (listeners) {\n      listeners.delete(callback);\n      if (listeners.size === 0) {\n        this.sessionListeners.delete(sessionId);\n      }\n    }\n  }\n\n  /**\n   * Get replay buffer for a run (all events emitted since start).\n   * Used by WS handler to send missed events on late connection.\n   * Optionally filter by afterSeq to only get events the client hasn't seen.\n   */\n  getEventBuffer(runId: string, afterSeq?: number): AgentEvent[] {\n    const run = this.activeRuns.get(runId);\n    if (!run) {return [];}\n\n    if (afterSeq != null) {\n      return run.eventBuffer.filter((e) => e.seq != null && e.seq > afterSeq);\n    }\n    return [...run.eventBuffer];\n  }\n\n  /**\n   * List all active runs\n   */\n  listActive(): Array<{ runId: string; task: string; status: RunStatus; startedAt: string }> {\n    return Array.from(this.activeRuns.values()).map(r => ({\n      runId: r.runId,\n      task: r.task,\n      status: r.status,\n      startedAt: r.startedAt,\n    }));\n  }\n\n  /**\n   * Request graceful stop of a running agent (and its child agents via propagated AbortSignal).\n   * Agent finishes its current tool call then exits at the next iteration boundary.\n   */\n  requestStop(runId: string): boolean {\n    const run = this.activeRuns.get(runId);\n    if (!run || run.status !== 'running' || !run.agent) {\n      return false;\n    }\n    run.agent.requestStop();\n    return true;\n  }\n\n  /**\n   * Clean up completed runs from memory (cache handles its own TTL)\n   */\n  cleanup(): void {\n    for (const [runId, run] of this.activeRuns) {\n      if (run.status === 'completed' || run.status === 'failed' || run.status === 'stopped') {\n        // Remove from memory but keep in cache\n        this.activeRuns.delete(runId);\n      }\n    }\n  }\n}\n\n/**\n * Singleton instance — stored on globalThis so all bundled modules share one instance.\n * When tsup compiles multiple entry points, each gets its own module scope,\n * so a plain `export const RunManager = new RunManagerImpl()` creates separate instances.\n * Using globalThis ensures run-handler.js and session-stream-handler.js share one RunManager.\n */\nconst GLOBAL_KEY = '__kb_agent_run_manager__';\nif (!(globalThis as Record<string, unknown>)[GLOBAL_KEY]) {\n  (globalThis as Record<string, unknown>)[GLOBAL_KEY] = new RunManagerImpl();\n}\nexport const RunManager = (globalThis as Record<string, unknown>)[GLOBAL_KEY] as RunManagerImpl;\n","/**\n * POST /run/:runId/correct handler\n *\n * Injects user correction into running agent(s)\n */\n\nimport { defineHandler, rethrowForRest, useAnalytics, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport type { CorrectionRequest, CorrectionResponse } from '@kb-labs/agent-contracts';\nimport { AGENT_ANALYTICS_EVENTS } from '@kb-labs/agent-contracts';\nimport { RunManager } from '../run-manager.js';\n\nexport default defineHandler({\n  async execute(\n    ctx: PluginContextV3,\n    input: RestInput<CorrectionRequest>\n  ): Promise<CorrectionResponse> {\n    try {\n      const { runId } = input.params as { runId: string };\n      const body = input.body as CorrectionRequest | undefined;\n\n      if (!body?.message) {\n        throw new Error('Correction message is required');\n      }\n\n      ctx.platform.logger.info(`[correct-handler] Correction for run ${runId}: ${body.message.slice(0, 50)}...`);\n\n      const run = RunManager.get(runId);\n\n      if (!run) {\n        return {\n          correctionId: '',\n          routedTo: [],\n          reason: `Run ${runId} not found`,\n          applied: false,\n        };\n      }\n\n      if (run.status !== 'running') {\n        return {\n          correctionId: '',\n          routedTo: [],\n          reason: `Run ${runId} is not running (status: ${run.status})`,\n          applied: false,\n        };\n      }\n\n      // Inject correction into the running agent via the public IAgentRunner API\n      const correctionId = `corr-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;\n      run.agent?.injectUserContext(body.message);\n      const result: CorrectionResponse = {\n        correctionId,\n        routedTo: [run.agent?.agentId ?? runId],\n        reason: 'Injected via injectUserContext',\n        applied: !!run.agent,\n      };\n\n      // Track correction\n      const analytics = useAnalytics();\n      await analytics?.track(\n        result.applied ? AGENT_ANALYTICS_EVENTS.CORRECTION_APPLIED : AGENT_ANALYTICS_EVENTS.CORRECTION_REJECTED,\n        {\n          runId,\n          correctionId: result.correctionId,\n          routedTo: result.routedTo,\n          messageLength: body.message.length,\n          targetAgentId: body.targetAgentId,\n        }\n      );\n\n      ctx.platform.logger.info(\n        `[correct-handler] Correction ${result.correctionId} routed to: ${result.routedTo.join(', ')}`\n      );\n\n      return result;\n    } catch (err) {\n      rethrowForRest(err);\n    }\n  },\n});\n"]}
+{"version":3,"sources":["../../../src/rest/run-manager.ts","../../../src/rest/handlers/correct-handler.ts"],"names":[],"mappings":";;;;AAYA,IAAM,YAAA,GAAe,YAAA;AACrB,IAAM,SAAA,GAAY,IAAA;AAqDlB,IAAM,iBAAN,MAAqB;AAAA;AAAA,EAEX,UAAA,uBAAyC,GAAA,EAAI;AAAA;AAAA,EAG7C,gBAAA,uBAAkE,GAAA,EAAI;AAAA;AAAA;AAAA;AAAA,EAK9E,aAAA,GAAwB;AACtB,IAAA,OAAO,CAAA,IAAA,EAAO,IAAA,CAAK,GAAA,EAAK,IAAI,IAAA,CAAK,MAAA,EAAO,CAAE,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAA,CACJ,KAAA,EACA,IAAA,EACA,OACA,cAAA,EACoB;AACpB,IAAA,MAAM,GAAA,GAAA,iBAAM,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAEnC,IAAA,MAAM,GAAA,GAAiB;AAAA,MACrB,KAAA;AAAA,MACA,IAAA;AAAA,MACA,MAAA,EAAQ,SAAA;AAAA,MACR,KAAA;AAAA,MACA,cAAA;AAAA,MACA,SAAA,EAAW;AAAA,KACb;AAGA,IAAA,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,GAAG,CAAA;AAG9B,IAAA,MAAM,IAAA,CAAK,YAAY,GAAG,CAAA;AAE1B,IAAA,OAAO,GAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,KAAA,EAAsC;AACxC,IAAA,OAAO,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,OAAO,KAAA,EAAiC;AAE5C,IAAA,IAAI,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,QAAA,CAAS,KAAK,CAAA;AACvC,IAAA,OAAO,KAAA,KAAU,IAAA;AAAA,EACnB;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,SAAS,KAAA,EAAyC;AACtD,IAAA,MAAM,QAAQ,QAAA,EAAS;AACvB,IAAA,IAAI,CAAC,KAAA,EAAO;AAAC,MAAA,OAAO,IAAA;AAAA,IAAK;AAEzB,IAAA,OAAO,MAAM,GAAA,CAAc,CAAA,EAAG,YAAY,CAAA,EAAG,KAAK,CAAA,CAAE,CAAA;AAAA,EACtD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAA,CAAa,KAAA,EAAe,MAAA,EAAmB,KAAA,EAA0C;AAC7F,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,GAAA,EAAK;AACP,MAAA,GAAA,CAAI,MAAA,GAAS,MAAA;AACb,MAAA,IAAI,KAAA,EAAO;AACT,QAAA,MAAA,CAAO,MAAA,CAAO,KAAK,KAAK,CAAA;AAAA,MAC1B;AACA,MAAA,MAAM,IAAA,CAAK,YAAY,GAAG,CAAA;AAAA,IAC5B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,YAAY,GAAA,EAA+B;AACvD,IAAA,MAAM,QAAQ,QAAA,EAAS;AACvB,IAAA,IAAI,CAAC,KAAA,EAAO;AAAC,MAAA;AAAA,IAAO;AAEpB,IAAA,MAAM,KAAA,GAAkB;AAAA,MACtB,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,QAAQ,GAAA,CAAI,MAAA;AAAA,MACZ,WAAW,GAAA,CAAI,SAAA;AAAA,MACf,aAAa,GAAA,CAAI,WAAA;AAAA,MACjB,YAAY,GAAA,CAAI,UAAA;AAAA,MAChB,SAAS,GAAA,CAAI,OAAA;AAAA,MACb,OAAO,GAAA,CAAI;AAAA,KACb;AAEA,IAAA,MAAM,KAAA,CAAM,IAAI,CAAA,EAAG,YAAY,GAAG,GAAA,CAAI,KAAK,CAAA,CAAA,EAAI,KAAA,EAAO,SAAS,CAAA;AAAA,EACjE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,sBAAA,CAAuB,WAAmB,MAAA,EAA2B;AACnE,IAAA,IAAI,MAAA,CAAO,WAAW,CAAA,EAAG;AAAE,MAAA;AAAA,IAAQ;AACnC,IAAA,IAAA,CAAK,uBAAuB,SAAA,EAAW,EAAE,IAAA,EAAM,QAAA,EAAU,QAAQ,CAAA;AAAA,EACnE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,qBAAA,CAAsB,WAAmB,OAAA,EAA0F;AACjI,IAAA,IAAA,CAAK,uBAAuB,SAAA,EAAW,EAAE,MAAM,eAAA,EAAiB,GAAG,SAAS,CAAA;AAAA,EAC9E;AAAA,EAEQ,sBAAA,CAAuB,WAAmB,OAAA,EAAwC;AACxF,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACrD,IAAA,IAAI,CAAC,SAAA,EAAW;AAAE,MAAA;AAAA,IAAQ;AAC1B,IAAA,KAAA,MAAW,YAAY,SAAA,EAAW;AAChC,MAAA,IAAI;AACF,QAAA,QAAA,CAAS,OAAO,CAAA;AAAA,MAClB,CAAA,CAAA,MAAQ;AAAA,MAER;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,kBAAA,CAAmB,WAAmB,QAAA,EAAyC;AAC7E,IAAA,IAAI,CAAC,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA,EAAG;AACzC,MAAA,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAA,kBAAW,IAAI,KAAK,CAAA;AAAA,IAChD;AACA,IAAA,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA,CAAG,IAAI,QAAQ,CAAA;AAAA,EACpD;AAAA;AAAA;AAAA;AAAA,EAKA,qBAAA,CAAsB,WAAmB,QAAA,EAAyC;AAChF,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AACrD,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,SAAA,CAAU,OAAO,QAAQ,CAAA;AACzB,MAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AACxB,QAAA,IAAA,CAAK,gBAAA,CAAiB,OAAO,SAAS,CAAA;AAAA,MACxC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,UAAA,GAA2F;AACzF,IAAA,OAAO,KAAA,CAAM,KAAK,IAAA,CAAK,UAAA,CAAW,QAAQ,CAAA,CAAE,IAAI,CAAA,CAAA,MAAM;AAAA,MACpD,OAAO,CAAA,CAAE,KAAA;AAAA,MACT,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,QAAQ,CAAA,CAAE,MAAA;AAAA,MACV,WAAW,CAAA,CAAE;AAAA,KACf,CAAE,CAAA;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,YAAY,KAAA,EAAwB;AAClC,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AACrC,IAAA,IAAI,CAAC,GAAA,IAAO,GAAA,CAAI,WAAW,SAAA,IAAa,CAAC,IAAI,KAAA,EAAO;AAClD,MAAA,OAAO,KAAA;AAAA,IACT;AACA,IAAA,GAAA,CAAI,MAAM,WAAA,EAAY;AACtB,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,OAAA,GAAgB;AACd,IAAA,KAAA,MAAW,CAAC,KAAA,EAAO,GAAG,CAAA,IAAK,KAAK,UAAA,EAAY;AAC1C,MAAA,IAAI,GAAA,CAAI,WAAW,WAAA,IAAe,GAAA,CAAI,WAAW,QAAA,IAAY,GAAA,CAAI,WAAW,SAAA,EAAW;AAErF,QAAA,IAAA,CAAK,UAAA,CAAW,OAAO,KAAK,CAAA;AAAA,MAC9B;AAAA,IACF;AAAA,EACF;AACF,CAAA;AAQA,IAAM,UAAA,GAAa,0BAAA;AACnB,IAAI,CAAE,UAAA,CAAuC,UAAU,CAAA,EAAG;AACxD,EAAC,UAAA,CAAuC,UAAU,CAAA,GAAI,IAAI,cAAA,EAAe;AAC3E;AACO,IAAM,UAAA,GAAc,WAAuC,UAAU,CAAA;;;AC5Q5E,IAAO,0BAAQ,aAAA,CAAc;AAAA,EAC3B,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EAC6B;AAC7B,IAAA,IAAI;AACF,MAAA,MAAM,EAAE,KAAA,EAAM,GAAI,KAAA,CAAM,MAAA;AACxB,MAAA,MAAM,OAAO,KAAA,CAAM,IAAA;AAEnB,MAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAClB,QAAA,MAAM,IAAI,MAAM,gCAAgC,CAAA;AAAA,MAClD;AAEA,MAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,CAAA,qCAAA,EAAwC,KAAK,CAAA,EAAA,EAAK,IAAA,CAAK,OAAA,CAAQ,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,GAAA,CAAK,CAAA;AAEzG,MAAA,MAAM,GAAA,GAAM,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAEhC,MAAA,IAAI,CAAC,GAAA,EAAK;AACR,QAAA,OAAO;AAAA,UACL,YAAA,EAAc,EAAA;AAAA,UACd,UAAU,EAAC;AAAA,UACX,MAAA,EAAQ,OAAO,KAAK,CAAA,UAAA,CAAA;AAAA,UACpB,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAEA,MAAA,IAAI,GAAA,CAAI,WAAW,SAAA,EAAW;AAC5B,QAAA,OAAO;AAAA,UACL,YAAA,EAAc,EAAA;AAAA,UACd,UAAU,EAAC;AAAA,UACX,MAAA,EAAQ,CAAA,IAAA,EAAO,KAAK,CAAA,yBAAA,EAA4B,IAAI,MAAM,CAAA,CAAA,CAAA;AAAA,UAC1D,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAGA,MAAA,MAAM,YAAA,GAAe,CAAA,KAAA,EAAQ,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AACjF,MAAA,GAAA,CAAI,KAAA,EAAO,iBAAA,CAAkB,IAAA,CAAK,OAAO,CAAA;AACzC,MAAA,MAAM,MAAA,GAA6B;AAAA,QACjC,YAAA;AAAA,QACA,QAAA,EAAU,CAAC,GAAA,CAAI,KAAA,EAAO,WAAW,KAAK,CAAA;AAAA,QACtC,MAAA,EAAQ,gCAAA;AAAA,QACR,OAAA,EAAS,CAAC,CAAC,GAAA,CAAI;AAAA,OACjB;AAGA,MAAA,MAAM,YAAY,YAAA,EAAa;AAC/B,MAAA,MAAM,SAAA,EAAW,KAAA;AAAA,QACf,MAAA,CAAO,OAAA,GAAU,sBAAA,CAAuB,kBAAA,GAAqB,sBAAA,CAAuB,mBAAA;AAAA,QACpF;AAAA,UACE,KAAA;AAAA,UACA,cAAc,MAAA,CAAO,YAAA;AAAA,UACrB,UAAU,MAAA,CAAO,QAAA;AAAA,UACjB,aAAA,EAAe,KAAK,OAAA,CAAQ,MAAA;AAAA,UAC5B,eAAe,IAAA,CAAK;AAAA;AACtB,OACF;AAEA,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,QAClB,CAAA,6BAAA,EAAgC,OAAO,YAAY,CAAA,YAAA,EAAe,OAAO,QAAA,CAAS,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,OAC9F;AAEA,MAAA,OAAO,MAAA;AAAA,IACT,SAAS,GAAA,EAAK;AACZ,MAAA,cAAA,CAAe,GAAG,CAAA;AAAA,IACpB;AAAA,EACF;AACF,CAAC","file":"correct-handler.js","sourcesContent":["/**\n * Run Manager - tracks active agent runs for REST/WS API\n *\n * Uses platform cache for persistence and in-memory Map for active agents.\n * Cache stores serializable run state, Map stores live agent references.\n */\n\nimport type { IAgentRunner } from '@kb-labs/agent-sdk';\nimport type { SessionManager } from '@kb-labs/agent-core';\nimport type { Turn, TurnDelta } from '@kb-labs/agent-contracts';\nimport { useCache } from '@kb-labs/sdk';\n\nconst CACHE_PREFIX = 'agent:run:';\nconst CACHE_TTL = 3600000; // 1 hour\n\n/**\n * Run status\n */\nexport type RunStatus = 'pending' | 'running' | 'completed' | 'failed' | 'stopped';\n\n/**\n * Serializable run state (stored in cache)\n */\nexport interface RunState {\n  runId: string;\n  task: string;\n  status: RunStatus;\n  startedAt: string;\n  completedAt?: string;\n  durationMs?: number;\n  summary?: string;\n  error?: string;\n}\n\n/**\n * Active run with live agent (in-memory only)\n */\nexport interface ActiveRun extends RunState {\n  agent?: IAgentRunner;\n  sessionManager: SessionManager;\n}\n\n/**\n * Payload delivered to session-level listeners — the sole live-update\n * channel for a session's WS connection(s). `deltas` carries incremental\n * turn changes (see TurnDelta); `run:completed` is a terminal signal\n * carrying the finished run's outcome and, when available, the final\n * state of the assistant turn it produced.\n */\nexport type SessionBroadcastPayload =\n  | { kind: 'deltas'; deltas: TurnDelta[] }\n  | {\n      kind: 'run:completed';\n      runId: string;\n      success: boolean;\n      summary: string;\n      durationMs: number;\n      seq: number;\n      turn?: Turn;\n    };\n\ntype SessionListenerCallback = (payload: SessionBroadcastPayload) => void;\n\n/**\n * Run Manager implementation\n */\nclass RunManagerImpl {\n  /** Live agents (not cacheable) */\n  private activeRuns: Map<string, ActiveRun> = new Map();\n\n  /** Session-level listeners: sessionId → Set<callback> — receive updates from ALL runs in session */\n  private sessionListeners: Map<string, Set<SessionListenerCallback>> = new Map();\n\n  /**\n   * Generate unique run ID\n   */\n  generateRunId(): string {\n    return `run-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;\n  }\n\n  /**\n   * Register a new run\n   */\n  async register(\n    runId: string,\n    task: string,\n    agent: IAgentRunner | undefined,\n    sessionManager: SessionManager,\n  ): Promise<ActiveRun> {\n    const now = new Date().toISOString();\n\n    const run: ActiveRun = {\n      runId,\n      task,\n      status: 'pending',\n      agent,\n      sessionManager,\n      startedAt: now,\n    };\n\n    // Store in memory (for live agent)\n    this.activeRuns.set(runId, run);\n\n    // Store serializable state in cache\n    await this.saveToCache(run);\n\n    return run;\n  }\n\n  /**\n   * Get run by ID (from memory first, then cache for state)\n   */\n  get(runId: string): ActiveRun | undefined {\n    return this.activeRuns.get(runId);\n  }\n\n  /**\n   * Check if run exists (in memory or cache)\n   */\n  async exists(runId: string): Promise<boolean> {\n    // Check memory first\n    if (this.activeRuns.has(runId)) {\n      return true;\n    }\n    // Fallback to cache\n    const state = await this.getState(runId);\n    return state !== null;\n  }\n\n  /**\n   * Get run state from cache (for completed runs or cross-process access)\n   */\n  async getState(runId: string): Promise<RunState | null> {\n    const cache = useCache();\n    if (!cache) {return null;}\n\n    return cache.get<RunState>(`${CACHE_PREFIX}${runId}`);\n  }\n\n  /**\n   * Update run status\n   */\n  async updateStatus(runId: string, status: RunStatus, extra?: Partial<RunState>): Promise<void> {\n    const run = this.activeRuns.get(runId);\n    if (run) {\n      run.status = status;\n      if (extra) {\n        Object.assign(run, extra);\n      }\n      await this.saveToCache(run);\n    }\n  }\n\n  /**\n   * Save run state to cache\n   */\n  private async saveToCache(run: ActiveRun): Promise<void> {\n    const cache = useCache();\n    if (!cache) {return;}\n\n    const state: RunState = {\n      runId: run.runId,\n      task: run.task,\n      status: run.status,\n      startedAt: run.startedAt,\n      completedAt: run.completedAt,\n      durationMs: run.durationMs,\n      summary: run.summary,\n      error: run.error,\n    };\n\n    await cache.set(`${CACHE_PREFIX}${run.runId}`, state, CACHE_TTL);\n  }\n\n  /**\n   * Broadcast turn deltas to all session-level listeners. Callers MUST await\n   * persistence (sessionManager.addEvent) before calling this, so listeners\n   * never observe a delta whose underlying turn state isn't on disk yet —\n   * that ordering is what makes gap-recovery / reconnect resume correct.\n   */\n  broadcastSessionDeltas(sessionId: string, deltas: TurnDelta[]): void {\n    if (deltas.length === 0) { return; }\n    this.notifySessionListeners(sessionId, { kind: 'deltas', deltas });\n  }\n\n  /**\n   * Broadcast a run's terminal outcome to session-level listeners. Called\n   * directly from the run's own completion handler (after `agent.execute()`\n   * resolves) rather than derived reactively from an `agent:end` event —\n   * the completion handler already has definitive success/summary/duration,\n   * and this guarantees the message fires exactly once, deterministically.\n   */\n  broadcastRunCompleted(sessionId: string, payload: Omit<Extract<SessionBroadcastPayload, { kind: 'run:completed' }>, 'kind'>): void {\n    this.notifySessionListeners(sessionId, { kind: 'run:completed', ...payload });\n  }\n\n  private notifySessionListeners(sessionId: string, payload: SessionBroadcastPayload): void {\n    const listeners = this.sessionListeners.get(sessionId);\n    if (!listeners) { return; }\n    for (const listener of listeners) {\n      try {\n        listener(payload);\n      } catch {\n        // Ignore listener errors — one bad WS connection shouldn't affect others\n      }\n    }\n  }\n\n  /**\n   * Add a session-level listener — receives updates from ALL runs in this session.\n   */\n  addSessionListener(sessionId: string, callback: SessionListenerCallback): void {\n    if (!this.sessionListeners.has(sessionId)) {\n      this.sessionListeners.set(sessionId, new Set());\n    }\n    this.sessionListeners.get(sessionId)!.add(callback);\n  }\n\n  /**\n   * Remove a session-level listener.\n   */\n  removeSessionListener(sessionId: string, callback: SessionListenerCallback): void {\n    const listeners = this.sessionListeners.get(sessionId);\n    if (listeners) {\n      listeners.delete(callback);\n      if (listeners.size === 0) {\n        this.sessionListeners.delete(sessionId);\n      }\n    }\n  }\n\n  /**\n   * List all active runs\n   */\n  listActive(): Array<{ runId: string; task: string; status: RunStatus; startedAt: string }> {\n    return Array.from(this.activeRuns.values()).map(r => ({\n      runId: r.runId,\n      task: r.task,\n      status: r.status,\n      startedAt: r.startedAt,\n    }));\n  }\n\n  /**\n   * Request graceful stop of a running agent (and its child agents via propagated AbortSignal).\n   * Agent finishes its current tool call then exits at the next iteration boundary.\n   */\n  requestStop(runId: string): boolean {\n    const run = this.activeRuns.get(runId);\n    if (!run || run.status !== 'running' || !run.agent) {\n      return false;\n    }\n    run.agent.requestStop();\n    return true;\n  }\n\n  /**\n   * Clean up completed runs from memory (cache handles its own TTL)\n   */\n  cleanup(): void {\n    for (const [runId, run] of this.activeRuns) {\n      if (run.status === 'completed' || run.status === 'failed' || run.status === 'stopped') {\n        // Remove from memory but keep in cache\n        this.activeRuns.delete(runId);\n      }\n    }\n  }\n}\n\n/**\n * Singleton instance — stored on globalThis so all bundled modules share one instance.\n * When tsup compiles multiple entry points, each gets its own module scope,\n * so a plain `export const RunManager = new RunManagerImpl()` creates separate instances.\n * Using globalThis ensures run-handler.js and session-stream-handler.js share one RunManager.\n */\nconst GLOBAL_KEY = '__kb_agent_run_manager__';\nif (!(globalThis as Record<string, unknown>)[GLOBAL_KEY]) {\n  (globalThis as Record<string, unknown>)[GLOBAL_KEY] = new RunManagerImpl();\n}\nexport const RunManager = (globalThis as Record<string, unknown>)[GLOBAL_KEY] as RunManagerImpl;\n","/**\n * POST /run/:runId/correct handler\n *\n * Injects user correction into running agent(s)\n */\n\nimport { defineHandler, rethrowForRest, useAnalytics, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport type { CorrectionRequest, CorrectionResponse } from '@kb-labs/agent-contracts';\nimport { AGENT_ANALYTICS_EVENTS } from '@kb-labs/agent-contracts';\nimport { RunManager } from '../run-manager.js';\n\nexport default defineHandler({\n  async execute(\n    ctx: PluginContextV3,\n    input: RestInput<CorrectionRequest>\n  ): Promise<CorrectionResponse> {\n    try {\n      const { runId } = input.params as { runId: string };\n      const body = input.body as CorrectionRequest | undefined;\n\n      if (!body?.message) {\n        throw new Error('Correction message is required');\n      }\n\n      ctx.platform.logger.info(`[correct-handler] Correction for run ${runId}: ${body.message.slice(0, 50)}...`);\n\n      const run = RunManager.get(runId);\n\n      if (!run) {\n        return {\n          correctionId: '',\n          routedTo: [],\n          reason: `Run ${runId} not found`,\n          applied: false,\n        };\n      }\n\n      if (run.status !== 'running') {\n        return {\n          correctionId: '',\n          routedTo: [],\n          reason: `Run ${runId} is not running (status: ${run.status})`,\n          applied: false,\n        };\n      }\n\n      // Inject correction into the running agent via the public IAgentRunner API\n      const correctionId = `corr-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;\n      run.agent?.injectUserContext(body.message);\n      const result: CorrectionResponse = {\n        correctionId,\n        routedTo: [run.agent?.agentId ?? runId],\n        reason: 'Injected via injectUserContext',\n        applied: !!run.agent,\n      };\n\n      // Track correction\n      const analytics = useAnalytics();\n      await analytics?.track(\n        result.applied ? AGENT_ANALYTICS_EVENTS.CORRECTION_APPLIED : AGENT_ANALYTICS_EVENTS.CORRECTION_REJECTED,\n        {\n          runId,\n          correctionId: result.correctionId,\n          routedTo: result.routedTo,\n          messageLength: body.message.length,\n          targetAgentId: body.targetAgentId,\n        }\n      );\n\n      ctx.platform.logger.info(\n        `[correct-handler] Correction ${result.correctionId} routed to: ${result.routedTo.join(', ')}`\n      );\n\n      return result;\n    } catch (err) {\n      rethrowForRest(err);\n    }\n  },\n});\n"]}

package/dist/rest/handlers/execute-session-plan-handler.js

@@ -1,4 +1,4 @@
-import { defineHandler, useAnalytics, useCache, useConfig, rethrowForRest, usePlatform } from '@kb-labs/sdk';
+import { defineHandler, useAnalytics, useCache, useConfig, rethrowForRest } from '@kb-labs/sdk';
 import { bootstrapAgentSDK, SessionManager, createSessionMemoryBridge, createCoreToolPack, PlanDocumentService } from '@kb-labs/agent-core';
 import fs, { promises } from 'fs';
 import { createDefaultResponseRequirementsSelector } from '@kb-labs/agent-runtime';
@@ -10,15 +10,12 @@ import { AGENT_ANALYTICS_EVENTS, AGENTS_WS_CHANNELS, AGENTS_WS_BASE_PATH } from
 
 // src/rest/handlers/execute-session-plan-handler.ts
 var CACHE_PREFIX = "agent:run:";
-var EVENT_TOPIC_PREFIX = "agent:events:";
 var CACHE_TTL = 36e5;
 var RunManagerImpl = class {
-  /** Live agents and listeners (not cacheable) */
+  /** Live agents (not cacheable) */
   activeRuns = /* @__PURE__ */ new Map();
-  /** Session-level listeners: sessionId → Set<callback> — receive events from ALL runs in session */
+  /** Session-level listeners: sessionId → Set<callback> — receive updates from ALL runs in session */
   sessionListeners = /* @__PURE__ */ new Map();
-  /** sessionId per runId — set when run is registered */
-  runSessionMap = /* @__PURE__ */ new Map();
   /**
    * Generate unique run ID
    */
@@ -28,7 +25,7 @@ var RunManagerImpl = class {
   /**
    * Register a new run
    */
-  async register(runId, task, agent, sessionManager, sessionId) {
+  async register(runId, task, agent, sessionManager) {
     const now = (/* @__PURE__ */ new Date()).toISOString();
     const run = {
       runId,
@@ -36,15 +33,9 @@ var RunManagerImpl = class {
       status: "pending",
       agent,
       sessionManager,
-      startedAt: now,
-      listeners: /* @__PURE__ */ new Set(),
-      lastSeq: 0,
-      eventBuffer: []
+      startedAt: now
     };
     this.activeRuns.set(runId, run);
-    if (sessionId) {
-      this.runSessionMap.set(runId, sessionId);
-    }
     await this.saveToCache(run);
     return run;
   }
@@ -107,101 +98,42 @@ var RunManagerImpl = class {
     };
     await cache.set(`${CACHE_PREFIX}${run.runId}`, state, CACHE_TTL);
   }
-  /** Track subscriptions for cleanup */
-  subscriptions = /* @__PURE__ */ new Map();
   /**
-   * Add event listener to run (uses eventBus for cross-process)
+   * Broadcast turn deltas to all session-level listeners. Callers MUST await
+   * persistence (sessionManager.addEvent) before calling this, so listeners
+   * never observe a delta whose underlying turn state isn't on disk yet —
+   * that ordering is what makes gap-recovery / reconnect resume correct.
    */
-  addListener(runId, callback) {
-    const platform = usePlatform();
-    if (!platform?.eventBus) {
-      const run2 = this.activeRuns.get(runId);
-      if (run2) {
-        run2.listeners.add(callback);
-        return true;
-      }
-      return false;
-    }
-    const topic = `${EVENT_TOPIC_PREFIX}${runId}`;
-    const unsubscribe = platform.eventBus.subscribe(topic, async (event) => {
-      try {
-        callback(event);
-      } catch {
-      }
-    });
-    if (!this.subscriptions.has(runId)) {
-      this.subscriptions.set(runId, /* @__PURE__ */ new Map());
-    }
-    this.subscriptions.get(runId).set(callback, unsubscribe);
-    const run = this.activeRuns.get(runId);
-    if (run) {
-      run.listeners.add(callback);
+  broadcastSessionDeltas(sessionId, deltas) {
+    if (deltas.length === 0) {
+      return;
     }
-    return true;
+    this.notifySessionListeners(sessionId, { kind: "deltas", deltas });
   }
   /**
-   * Remove event listener from run
+   * Broadcast a run's terminal outcome to session-level listeners. Called
+   * directly from the run's own completion handler (after `agent.execute()`
+   * resolves) rather than derived reactively from an `agent:end` event —
+   * the completion handler already has definitive success/summary/duration,
+   * and this guarantees the message fires exactly once, deterministically.
    */
-  removeListener(runId, callback) {
-    const runSubs = this.subscriptions.get(runId);
-    if (runSubs) {
-      const unsubscribe = runSubs.get(callback);
-      if (unsubscribe) {
-        unsubscribe();
-        runSubs.delete(callback);
-      }
-      if (runSubs.size === 0) {
-        this.subscriptions.delete(runId);
-      }
-    }
-    const run = this.activeRuns.get(runId);
-    if (run) {
-      run.listeners.delete(callback);
-    }
+  broadcastRunCompleted(sessionId, payload) {
+    this.notifySessionListeners(sessionId, { kind: "run:completed", ...payload });
   }
-  /**
-   * Broadcast event to all listeners of a run (uses eventBus for cross-process)
-   * Assigns monotonic sequence number for reliable ordering
-   * @returns The event with seq assigned (for persistence)
-   */
-  broadcast(runId, event) {
-    const platform = usePlatform();
-    const run = this.activeRuns.get(runId);
-    let seqEvent = event;
-    if (run) {
-      run.lastSeq++;
-      seqEvent = { ...event, seq: run.lastSeq, runId };
-      run.eventBuffer.push(seqEvent);
-    }
-    if (platform?.eventBus) {
-      const topic = `${EVENT_TOPIC_PREFIX}${runId}`;
-      platform.eventBus.publish(topic, seqEvent).catch(() => {
-      });
-    }
-    if (run) {
-      for (const listener of run.listeners) {
-        try {
-          listener(seqEvent);
-        } catch {
-        }
-      }
+  notifySessionListeners(sessionId, payload) {
+    const listeners = this.sessionListeners.get(sessionId);
+    if (!listeners) {
+      return;
     }
-    const sessionId = this.runSessionMap.get(runId);
-    if (sessionId) {
-      const sListeners = this.sessionListeners.get(sessionId);
-      if (sListeners) {
-        for (const listener of sListeners) {
-          try {
-            listener(seqEvent);
-          } catch {
-          }
-        }
+    for (const listener of listeners) {
+      try {
+        listener(payload);
+      } catch {
       }
     }
-    return seqEvent;
   }
   /**
-   * Add a session-level listener — receives events from ALL runs in this session.
+   * Add a session-level listener — receives updates from ALL runs in this session.
    */
   addSessionListener(sessionId, callback) {
     if (!this.sessionListeners.has(sessionId)) {
@@ -221,21 +153,6 @@ var RunManagerImpl = class {
       }
     }
   }
-  /**
-   * Get replay buffer for a run (all events emitted since start).
-   * Used by WS handler to send missed events on late connection.
-   * Optionally filter by afterSeq to only get events the client hasn't seen.
-   */
-  getEventBuffer(runId, afterSeq) {
-    const run = this.activeRuns.get(runId);
-    if (!run) {
-      return [];
-    }
-    if (afterSeq != null) {
-      return run.eventBuffer.filter((e) => e.seq != null && e.seq > afterSeq);
-    }
-    return [...run.eventBuffer];
-  }
   /**
    * List all active runs
    */
@@ -398,7 +315,7 @@ var run_handler_default = defineHandler({
         ctx.platform.logger.info(`[run-handler] Continuing session ${sessionId} (workingDir=${workingDir})`);
       }
       ctx.platform.logger.info(`[run-handler] Starting run ${runId} for task: ${body.task}`);
-      await sessionManager.createUserTurn(sessionId, body.task, runId);
+      const userTurn = await sessionManager.createUserTurn(sessionId, body.task, runId, body.clientId);
       await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_STARTED, {
         runId,
         sessionId,
@@ -435,21 +352,22 @@ var run_handler_default = defineHandler({
         tier: body.tier ?? "medium",
         tokenBudget: agentsConfig?.tokenBudget,
         onEvent: (event) => {
-          const seqEvent = RunManager.broadcast(runId, event);
           void sessionManager.addEvent(finalSessionId, {
-            ...seqEvent,
+            ...event,
             sessionId: finalSessionId,
             runId,
             metadata: {
-              ...seqEvent.metadata,
+              ...event.metadata,
               sessionId: finalSessionId,
               runId,
               workingDir
             }
+          }).then((deltas) => {
+            RunManager.broadcastSessionDeltas(finalSessionId, deltas);
           });
         }
       });
-      const run = await RunManager.register(runId, body.task, agent, sessionManager, finalSessionId);
+      const run = await RunManager.register(runId, body.task, agent, sessionManager);
       await RunManager.updateStatus(runId, "running");
       void (async () => {
         try {
@@ -469,6 +387,17 @@ var run_handler_default = defineHandler({
             summary: result.summary,
             error: result.error
           });
+          const finalTurns = await sessionManager.getTurns(finalSessionId);
+          const finalTurn = finalTurns.find((t) => t.type === "assistant" && t.metadata.runId === runId);
+          const seq = await sessionManager.getCurrentSessionSeq(finalSessionId);
+          RunManager.broadcastRunCompleted(finalSessionId, {
+            runId,
+            success: result.success,
+            summary: result.summary,
+            durationMs,
+            seq,
+            turn: finalTurn
+          });
           await analytics?.track(
             result.success ? AGENT_ANALYTICS_EVENTS.RUN_COMPLETED : AGENT_ANALYTICS_EVENTS.RUN_FAILED,
             {
@@ -492,6 +421,14 @@ var run_handler_default = defineHandler({
             durationMs,
             error: errorMsg
           });
+          const seq = await sessionManager.getCurrentSessionSeq(finalSessionId);
+          RunManager.broadcastRunCompleted(finalSessionId, {
+            runId,
+            success: false,
+            summary: errorMsg,
+            durationMs,
+            seq
+          });
           await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_FAILED, {
             runId,
             durationMs,
@@ -507,7 +444,8 @@ var run_handler_default = defineHandler({
         sessionId: finalSessionId,
         eventsPath,
         status: "started",
-        startedAt: run.startedAt
+        startedAt: run.startedAt,
+        userTurn
       };
     } catch (err) {
       rethrowForRest(err);