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

package/dist/ws/session-stream-handler.js

@@ -1,17 +1,14 @@
-import { defineWebSocket, useCache, usePlatform } from '@kb-labs/sdk';
+import { defineWebSocket, useCache } from '@kb-labs/sdk';
 import { SessionManager } from '@kb-labs/agent-core';
 
 // src/ws/session-stream-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
    */
@@ -268,37 +185,15 @@ if (!globalThis[GLOBAL_KEY]) {
   globalThis[GLOBAL_KEY] = new RunManagerImpl();
 }
 var RunManager = globalThis[GLOBAL_KEY];
-function getTurnIdFromEvent(event) {
-  if ((event.type === "agent:start" || event.type === "agent:end" || event.type === "agent:error") && !event.parentAgentId && event.agentId) {
-    return `turn-${event.agentId}`;
-  }
-  return null;
-}
-function getTurnSignature(turn) {
-  return `${turn.id}:${turn.status}:${turn.completedAt || ""}:${turn.steps.length}`;
-}
-async function resolveTurnForEvent(sessionManager, sessionId, event) {
-  const turns = await sessionManager.getTurns(sessionId);
-  if (turns.length === 0) {
-    return null;
-  }
-  const explicitTurnId = getTurnIdFromEvent(event);
-  if (explicitTurnId) {
-    const explicit = turns.find((turn) => turn.id === explicitTurnId);
-    if (explicit) {
-      return explicit;
-    }
-  }
-  const assistantTurns = turns.filter((turn) => turn.type === "assistant").sort((a, b) => b.sequence - a.sequence);
-  return assistantTurns[0] || null;
-}
 var connectionState = /* @__PURE__ */ new WeakMap();
 var session_stream_handler_default = defineWebSocket({
   path: "/session/:sessionId",
   description: "Persistent session event stream (all runs)",
   handler: {
     async onConnect(ctx, sender) {
-      const sessionId = ctx.hostContext.params?.sessionId;
+      const hostContext = ctx.hostContext;
+      const sessionId = hostContext.params?.sessionId;
+      const afterSeq = hostContext.query?.afterSeq ? Number(hostContext.query.afterSeq) : void 0;
       if (!sessionId) {
         await sender.send({
           type: "error",
@@ -308,9 +203,10 @@ var session_stream_handler_default = defineWebSocket({
         sender.close(4e3, "Missing session ID");
         return;
       }
-      ctx.platform.logger.info(`[session-ws] Client connected to session ${sessionId}`);
+      ctx.platform.logger.info(
+        afterSeq != null ? `[session-ws] Client connected to session ${sessionId} (resume after seq ${afterSeq})` : `[session-ws] Client connected to session ${sessionId}`
+      );
       const sessionManager = new SessionManager(ctx.cwd);
-      const lastTurnSignatures = /* @__PURE__ */ new Map();
       try {
         await sender.send({
           type: "connection:ready",
@@ -322,64 +218,58 @@ var session_stream_handler_default = defineWebSocket({
         throw err;
       }
       try {
-        const snapshot = await sessionManager.getConversationSnapshot(sessionId);
+        const { turns, seq } = await sessionManager.getProjection(sessionId);
+        const completedTurns = turns.filter((t) => t.status !== "streaming");
+        const activeTurns = turns.filter((t) => t.status === "streaming");
         await sender.send({
           type: "conversation:snapshot",
           payload: {
             sessionId,
-            completedTurns: snapshot.completedTurns,
-            activeTurns: snapshot.activeTurns,
-            totalTurns: snapshot.totalTurns,
+            completedTurns,
+            activeTurns,
+            totalTurns: turns.length,
+            seq,
             timestamp: (/* @__PURE__ */ new Date()).toISOString()
           },
           timestamp: Date.now()
         });
         ctx.platform.logger.info(
-          `[session-ws] Sent snapshot: ${snapshot.completedTurns.length} completed + ${snapshot.activeTurns.length} active turns`
+          `[session-ws] Sent snapshot: ${completedTurns.length} completed + ${activeTurns.length} active turns at seq ${seq}`
         );
       } catch (err) {
         ctx.platform.logger.error(`[session-ws] Failed to send snapshot: ${err}`);
       }
-      const eventCallback = async (event) => {
-        const evtSessionId = event.sessionId || event.metadata?.sessionId;
-        if (evtSessionId && evtSessionId !== sessionId) {
-          return;
-        }
-        const targetSessionId = evtSessionId || sessionId;
-        try {
-          const turn = await resolveTurnForEvent(sessionManager, targetSessionId, event);
-          if (turn) {
-            const signature = getTurnSignature(turn);
-            const previousSignature = lastTurnSignatures.get(turn.id);
-            if (previousSignature !== signature) {
-              lastTurnSignatures.set(turn.id, signature);
-              await sender.send({
-                type: "turn:snapshot",
-                payload: { sessionId: targetSessionId, turn, sequenceNumber: turn.sequence },
-                timestamp: Date.now()
-              });
-            }
+      const callback = (payload) => {
+        if (payload.kind === "deltas") {
+          for (const delta of payload.deltas) {
+            void sender.send({
+              type: "turn:delta",
+              payload: { sessionId, delta },
+              timestamp: Date.now()
+            }).catch((err) => {
+              ctx.platform.logger.error(`[session-ws] Failed to send turn:delta: ${err}`);
+            });
           }
-        } catch (err) {
-          ctx.platform.logger.error(`[session-ws] Failed to process turn snapshot: ${err}`);
-        }
-        if (event.type === "agent:end" && !event.parentAgentId) {
-          const runId = event.runId || event.metadata?.runId;
-          await sender.send({
-            type: "run:completed",
-            payload: {
-              runId: runId ?? "unknown",
-              success: event.data.success,
-              summary: event.data.summary,
-              durationMs: event.data.durationMs
-            },
-            timestamp: Date.now()
-          });
+          return;
         }
+        void sender.send({
+          type: "run:completed",
+          payload: {
+            runId: payload.runId,
+            success: payload.success,
+            summary: payload.summary,
+            durationMs: payload.durationMs,
+            seq: payload.seq,
+            turn: payload.turn
+          },
+          timestamp: Date.now()
+        }).catch((err) => {
+          ctx.platform.logger.error(`[session-ws] Failed to send run:completed: ${err}`);
+        });
       };
-      connectionState.set(ctx, { sessionId, callback: eventCallback });
+      connectionState.set(ctx, { sessionId, callback });
       try {
-        RunManager.addSessionListener(sessionId, eventCallback);
+        RunManager.addSessionListener(sessionId, callback);
       } catch (err) {
         ctx.platform.logger.error(`[session-ws] Failed to register session listener: ${String(err)}`);
         throw err;

package/dist/ws/session-stream-handler.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/rest/run-manager.ts","../../src/ws/session-stream-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;AChW5E,SAAS,mBAAmB,KAAA,EAAkC;AAC5D,EAAA,IAAA,CAAK,KAAA,CAAM,IAAA,KAAS,aAAA,IAAiB,KAAA,CAAM,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,IAAA,KAAS,aAAA,KAAkB,CAAC,KAAA,CAAM,aAAA,IAAiB,MAAM,OAAA,EAAS;AACzI,IAAA,OAAO,CAAA,KAAA,EAAQ,MAAM,OAAO,CAAA,CAAA;AAAA,EAC9B;AACA,EAAA,OAAO,IAAA;AACT;AAEA,SAAS,iBAAiB,IAAA,EAAoB;AAC5C,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,EAAE,CAAA,CAAA,EAAI,IAAA,CAAK,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,WAAA,IAAe,EAAE,CAAA,CAAA,EAAI,IAAA,CAAK,MAAM,MAAM,CAAA,CAAA;AACjF;AAEA,eAAe,mBAAA,CACb,cAAA,EACA,SAAA,EACA,KAAA,EACsB;AACtB,EAAA,MAAM,KAAA,GAAQ,MAAM,cAAA,CAAe,QAAA,CAAS,SAAS,CAAA;AACrD,EAAA,IAAI,KAAA,CAAM,WAAW,CAAA,EAAG;AACtB,IAAA,OAAO,IAAA;AAAA,EACT;AAEA,EAAA,MAAM,cAAA,GAAiB,mBAAmB,KAAK,CAAA;AAC/C,EAAA,IAAI,cAAA,EAAgB;AAClB,IAAA,MAAM,WAAW,KAAA,CAAM,IAAA,CAAK,CAAC,IAAA,KAAe,IAAA,CAAK,OAAO,cAAc,CAAA;AACtE,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAO,QAAA;AAAA,IACT;AAAA,EACF;AAGA,EAAA,MAAM,iBAAiB,KAAA,CACpB,MAAA,CAAO,CAAC,IAAA,KAAe,KAAK,IAAA,KAAS,WAAW,CAAA,CAChD,IAAA,CAAK,CAAC,CAAA,EAAS,CAAA,KAAY,CAAA,CAAE,QAAA,GAAW,EAAE,QAAQ,CAAA;AACrD,EAAA,OAAO,cAAA,CAAe,CAAC,CAAA,IAAK,IAAA;AAC9B;AAQA,IAAM,eAAA,uBAAsB,OAAA,EAAwC;AAEpE,IAAO,iCAAQ,eAAA,CAAuD;AAAA,EACpE,IAAA,EAAM,qBAAA;AAAA,EACN,WAAA,EAAa,4CAAA;AAAA,EAEb,OAAA,EAAS;AAAA,IACP,MAAM,SAAA,CAAU,GAAA,EAAsB,MAAA,EAAoC;AACxE,MAAA,MAAM,SAAA,GAAa,GAAA,CAAI,WAAA,CAAoD,MAAA,EAAQ,SAAA;AAEnF,MAAA,IAAI,CAAC,SAAA,EAAW;AACd,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,OAAA;AAAA,UACN,OAAA,EAAS,EAAE,IAAA,EAAM,oBAAA,EAAsB,SAAS,wBAAA,EAAyB;AAAA,UACzE,SAAA,EAAW,KAAK,GAAA;AAAI,SACE,CAAA;AACxB,QAAA,MAAA,CAAO,KAAA,CAAM,KAAM,oBAAoB,CAAA;AACvC,QAAA;AAAA,MACF;AAEA,MAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,CAAA,yCAAA,EAA4C,SAAS,CAAA,CAAE,CAAA;AAEhF,MAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACjD,MAAA,MAAM,kBAAA,uBAAyB,GAAA,EAAoB;AAGnD,MAAA,IAAI;AACF,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,kBAAA;AAAA,UACN,OAAA,EAAS,EAAE,KAAA,EAAO,SAAA,EAAW,8BAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY,EAAE;AAAA,UACnE,SAAA,EAAW,KAAK,GAAA;AAAI,SACY,CAAA;AAAA,MACpC,SAAS,GAAA,EAAK;AACZ,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,iDAAiD,MAAA,CAAO,GAAG,CAAC,CAAA,CAAE,CAAA;AACxF,QAAA,MAAM,GAAA;AAAA,MACR;AAGA,MAAA,IAAI;AACF,QAAA,MAAM,QAAA,GAAW,MAAM,cAAA,CAAe,uBAAA,CAAwB,SAAS,CAAA;AACvE,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,uBAAA;AAAA,UACN,OAAA,EAAS;AAAA,YACP,SAAA;AAAA,YACA,gBAAgB,QAAA,CAAS,cAAA;AAAA,YACzB,aAAa,QAAA,CAAS,WAAA;AAAA,YACtB,YAAY,QAAA,CAAS,UAAA;AAAA,YACrB,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,WACpC;AAAA,UACA,SAAA,EAAW,KAAK,GAAA;AAAI,SACiB,CAAA;AACvC,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,UAClB,+BAA+B,QAAA,CAAS,cAAA,CAAe,MAAM,CAAA,aAAA,EAAgB,QAAA,CAAS,YAAY,MAAM,CAAA,aAAA;AAAA,SAC1G;AAAA,MACF,SAAS,GAAA,EAAK;AACZ,QAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,KAAA,CAAM,CAAA,sCAAA,EAAyC,GAAG,CAAA,CAAE,CAAA;AAAA,MAC1E;AAGA,MAAA,MAAM,aAAA,GAAgB,OAAO,KAAA,KAAyD;AAEpF,QAAA,MAAM,YAAA,GAAgB,KAAA,CAAiC,SAAA,IACjD,KAAA,CAAM,QAAA,EAAU,SAAA;AAGtB,QAAA,IAAI,YAAA,IAAgB,iBAAiB,SAAA,EAAW;AAC9C,UAAA;AAAA,QACF;AAEA,QAAA,MAAM,kBAAkB,YAAA,IAAgB,SAAA;AAExC,QAAA,IAAI;AACF,UAAA,MAAM,IAAA,GAAO,MAAM,mBAAA,CAAoB,cAAA,EAAgB,iBAAiB,KAAK,CAAA;AAC7E,UAAA,IAAI,IAAA,EAAM;AACR,YAAA,MAAM,SAAA,GAAY,iBAAiB,IAAI,CAAA;AACvC,YAAA,MAAM,iBAAA,GAAoB,kBAAA,CAAmB,GAAA,CAAI,IAAA,CAAK,EAAE,CAAA;AACxD,YAAA,IAAI,sBAAsB,SAAA,EAAW;AACnC,cAAA,kBAAA,CAAmB,GAAA,CAAI,IAAA,CAAK,EAAA,EAAI,SAAS,CAAA;AAEzC,cAAA,MAAM,OAAO,IAAA,CAAK;AAAA,gBAChB,IAAA,EAAM,eAAA;AAAA,gBACN,SAAS,EAAE,SAAA,EAAW,iBAAiB,IAAA,EAAM,cAAA,EAAgB,KAAK,QAAA,EAAS;AAAA,gBAC3E,SAAA,EAAW,KAAK,GAAA;AAAI,eACS,CAAA;AAAA,YACjC;AAAA,UACF;AAAA,QACF,SAAS,GAAA,EAAK;AACZ,UAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,KAAA,CAAM,CAAA,8CAAA,EAAiD,GAAG,CAAA,CAAE,CAAA;AAAA,QAClF;AAGA,QAAA,IAAI,KAAA,CAAM,IAAA,KAAS,WAAA,IAAe,CAAC,MAAM,aAAA,EAAe;AACtD,UAAA,MAAM,KAAA,GAAS,KAAA,CAA6B,KAAA,IAAU,KAAA,CAAM,QAAA,EAAU,KAAA;AACtE,UAAA,MAAM,OAAO,IAAA,CAAK;AAAA,YAChB,IAAA,EAAM,eAAA;AAAA,YACN,OAAA,EAAS;AAAA,cACP,OAAO,KAAA,IAAS,SAAA;AAAA,cAChB,OAAA,EAAS,MAAM,IAAA,CAAK,OAAA;AAAA,cACpB,OAAA,EAAS,MAAM,IAAA,CAAK,OAAA;AAAA,cACpB,UAAA,EAAY,MAAM,IAAA,CAAK;AAAA,aACzB;AAAA,YACA,SAAA,EAAW,KAAK,GAAA;AAAI,WACS,CAAA;AAAA,QACjC;AAAA,MACF,CAAA;AAGA,MAAA,eAAA,CAAgB,IAAI,GAAA,EAAe,EAAE,SAAA,EAAW,QAAA,EAAU,eAAe,CAAA;AAGzE,MAAA,IAAI;AACF,QAAA,UAAA,CAAW,kBAAA,CAAmB,WAAW,aAAa,CAAA;AAAA,MACxD,SAAS,GAAA,EAAK;AACZ,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,qDAAqD,MAAA,CAAO,GAAG,CAAC,CAAA,CAAE,CAAA;AAC5F,QAAA,MAAM,GAAA;AAAA,MACR;AACA,MAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,CAAA,qDAAA,EAAwD,SAAS,CAAA,CAAE,CAAA;AAAA,IAC9F,CAAA;AAAA,IAEA,MAAM,SAAA,CAAU,GAAA,EAAsB,OAAA,EAAwB,MAAA,EAAoC;AAEhG,MAAA,IAAI,OAAA,CAAQ,SAAS,MAAA,EAAQ;AAC3B,QAAA,MAAM,SAAA,GAAY,eAAA,CAAgB,GAAA,CAAI,GAAa,GAAG,SAAA,IAAa,EAAA;AACnE,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,kBAAA;AAAA,UACN,OAAA,EAAS,EAAE,KAAA,EAAO,SAAA,IAAa,EAAA,EAAI,8BAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY,EAAE;AAAA,UACzE,SAAA,EAAW,KAAK,GAAA;AAAI,SACrB,CAAA;AAAA,MACH;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,aAAa,GAAA,EAAsB;AACvC,MAAA,MAAM,KAAA,GAAQ,eAAA,CAAgB,GAAA,CAAI,GAAa,CAAA;AAC/C,MAAA,IAAI,KAAA,EAAO;AACT,QAAA,UAAA,CAAW,qBAAA,CAAsB,KAAA,CAAM,SAAA,EAAW,KAAA,CAAM,QAAQ,CAAA;AAChE,QAAA,eAAA,CAAgB,OAAO,GAAa,CAAA;AAAA,MACtC;AAEA,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA,CAAK,CAAA,8CAAA,EAAiD,KAAA,EAAO,SAAS,CAAA,CAAE,CAAA;AAAA,IAC9F,CAAA;AAAA,IAEA,MAAM,OAAA,CAAQ,GAAA,EAAsB,KAAA,EAAc,MAAA,EAAoC;AACpF,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,CAAA,8BAAA,EAAiC,KAAA,CAAM,OAAO,CAAA,CAAE,CAAA;AAC1E,MAAA,MAAM,OAAO,IAAA,CAAK;AAAA,QAChB,IAAA,EAAM,OAAA;AAAA,QACN,SAAS,EAAE,IAAA,EAAM,gBAAA,EAAkB,OAAA,EAAS,MAAM,OAAA,EAAQ;AAAA,QAC1D,SAAA,EAAW,KAAK,GAAA;AAAI,OACE,CAAA;AAAA,IAC1B;AAAA;AAEJ,CAAC","file":"session-stream-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 * WebSocket handler for session-level event streaming\n *\n * Path: /session/:sessionId\n * Single persistent connection per session — streams turn:snapshot for ALL runs in the session.\n * Unlike /events/:runId which closes when a run ends, this stays open for the entire session.\n */\n\n/* eslint-disable @typescript-eslint/consistent-type-imports */\n\nimport {\n  defineWebSocket,\n  type PluginContextV3,\n  type TypedSender,\n} from '@kb-labs/sdk';\nimport type {\n  ServerMessage,\n  ClientMessage,\n  ConnectionReadyMessage,\n  RunCompletedMessage,\n  ErrorMessage,\n  TurnSnapshotMessage,\n  ConversationSnapshotMessage,\n  AgentEvent,\n  Turn,\n} from '@kb-labs/agent-contracts';\nimport { RunManager } from '../rest/run-manager.js';\nimport { SessionManager } from '@kb-labs/agent-core';\n\nfunction getTurnIdFromEvent(event: AgentEvent): string | null {\n  if ((event.type === 'agent:start' || event.type === 'agent:end' || event.type === 'agent:error') && !event.parentAgentId && event.agentId) {\n    return `turn-${event.agentId}`;\n  }\n  return null;\n}\n\nfunction getTurnSignature(turn: Turn): string {\n  return `${turn.id}:${turn.status}:${turn.completedAt || ''}:${turn.steps.length}`;\n}\n\nasync function resolveTurnForEvent(\n  sessionManager: SessionManager,\n  sessionId: string,\n  event: AgentEvent\n): Promise<Turn | null> {\n  const turns = await sessionManager.getTurns(sessionId);\n  if (turns.length === 0) {\n    return null;\n  }\n\n  const explicitTurnId = getTurnIdFromEvent(event);\n  if (explicitTurnId) {\n    const explicit = turns.find((turn: Turn) => turn.id === explicitTurnId);\n    if (explicit) {\n      return explicit;\n    }\n  }\n\n  // For tool/llm/status events, send latest assistant turn snapshot.\n  const assistantTurns = turns\n    .filter((turn: Turn) => turn.type === 'assistant')\n    .sort((a: Turn, b: Turn) => b.sequence - a.sequence);\n  return assistantTurns[0] || null;\n}\n\ninterface SessionConnectionState {\n  sessionId: string;\n  callback: import('@kb-labs/agent-contracts').AgentEventCallback;\n}\n\n/** Per-connection state keyed by the opaque ctx object — avoids mutating ctx */\nconst connectionState = new WeakMap<object, SessionConnectionState>();\n\nexport default defineWebSocket<unknown, ClientMessage, ServerMessage>({\n  path: '/session/:sessionId',\n  description: 'Persistent session event stream (all runs)',\n\n  handler: {\n    async onConnect(ctx: PluginContextV3, sender: TypedSender<ServerMessage>) {\n      const sessionId = (ctx.hostContext as { params?: { sessionId?: string } }).params?.sessionId;\n\n      if (!sessionId) {\n        await sender.send({\n          type: 'error',\n          payload: { code: 'MISSING_SESSION_ID', message: 'Session ID is required' },\n          timestamp: Date.now(),\n        } satisfies ErrorMessage);\n        sender.close(4000, 'Missing session ID');\n        return;\n      }\n\n      ctx.platform.logger.info(`[session-ws] Client connected to session ${sessionId}`);\n\n      const sessionManager = new SessionManager(ctx.cwd);\n      const lastTurnSignatures = new Map<string, string>();\n\n      // Send connection:ready immediately\n      try {\n        await sender.send({\n          type: 'connection:ready',\n          payload: { runId: sessionId, connectedAt: new Date().toISOString() },\n          timestamp: Date.now(),\n        } satisfies ConnectionReadyMessage);\n      } catch (err) {\n        ctx.platform.logger.error(`[session-ws] Failed to send connection:ready: ${String(err)}`);\n        throw err;\n      }\n\n      // Send conversation:snapshot (history)\n      try {\n        const snapshot = await sessionManager.getConversationSnapshot(sessionId);\n        await sender.send({\n          type: 'conversation:snapshot',\n          payload: {\n            sessionId,\n            completedTurns: snapshot.completedTurns,\n            activeTurns: snapshot.activeTurns,\n            totalTurns: snapshot.totalTurns,\n            timestamp: new Date().toISOString(),\n          },\n          timestamp: Date.now(),\n        } satisfies ConversationSnapshotMessage);\n        ctx.platform.logger.info(\n          `[session-ws] Sent snapshot: ${snapshot.completedTurns.length} completed + ${snapshot.activeTurns.length} active turns`\n        );\n      } catch (err) {\n        ctx.platform.logger.error(`[session-ws] Failed to send snapshot: ${err}`);\n      }\n\n      // Session-level event callback — registered on ALL active runs in this session\n      const eventCallback = async (event: import('@kb-labs/agent-contracts').AgentEvent) => {\n        // Resolve sessionId from event\n        const evtSessionId = (event as { sessionId?: string }).sessionId\n          || (event.metadata?.sessionId as string | undefined);\n\n        // Only forward events belonging to this session\n        if (evtSessionId && evtSessionId !== sessionId) {\n          return;\n        }\n\n        const targetSessionId = evtSessionId || sessionId;\n\n        try {\n          const turn = await resolveTurnForEvent(sessionManager, targetSessionId, event);\n          if (turn) {\n            const signature = getTurnSignature(turn);\n            const previousSignature = lastTurnSignatures.get(turn.id);\n            if (previousSignature !== signature) {\n              lastTurnSignatures.set(turn.id, signature);\n\n              await sender.send({\n                type: 'turn:snapshot',\n                payload: { sessionId: targetSessionId, turn, sequenceNumber: turn.sequence },\n                timestamp: Date.now(),\n              } satisfies TurnSnapshotMessage);\n            }\n          }\n        } catch (err) {\n          ctx.platform.logger.error(`[session-ws] Failed to process turn snapshot: ${err}`);\n        }\n\n        // Notify run completion\n        if (event.type === 'agent:end' && !event.parentAgentId) {\n          const runId = (event as { runId?: string }).runId || (event.metadata?.runId as string | undefined);\n          await sender.send({\n            type: 'run:completed',\n            payload: {\n              runId: runId ?? 'unknown',\n              success: event.data.success,\n              summary: event.data.summary,\n              durationMs: event.data.durationMs,\n            },\n            timestamp: Date.now(),\n          } satisfies RunCompletedMessage);\n        }\n      };\n\n      // Store callback for cleanup in a WeakMap keyed by ctx — no ctx mutation\n      connectionState.set(ctx as object, { sessionId, callback: eventCallback });\n\n      // Register on all currently active runs in this session\n      try {\n        RunManager.addSessionListener(sessionId, eventCallback);\n      } catch (err) {\n        ctx.platform.logger.error(`[session-ws] Failed to register session listener: ${String(err)}`);\n        throw err;\n      }\n      ctx.platform.logger.info(`[session-ws] Session listener registered for session ${sessionId}`);\n    },\n\n    async onMessage(ctx: PluginContextV3, message: ClientMessage, sender: TypedSender<ServerMessage>) {\n      // Handle ping or corrections here if needed\n      if (message.type === 'ping') {\n        const sessionId = connectionState.get(ctx as object)?.sessionId ?? '';\n        await sender.send({\n          type: 'connection:ready',\n          payload: { runId: sessionId ?? '', connectedAt: new Date().toISOString() },\n          timestamp: Date.now(),\n        });\n      }\n    },\n\n    async onDisconnect(ctx: PluginContextV3) {\n      const state = connectionState.get(ctx as object);\n      if (state) {\n        RunManager.removeSessionListener(state.sessionId, state.callback);\n        connectionState.delete(ctx as object);\n      }\n\n      ctx.platform.logger.info(`[session-ws] Client disconnected from session ${state?.sessionId}`);\n    },\n\n    async onError(ctx: PluginContextV3, error: Error, sender: TypedSender<ServerMessage>) {\n      ctx.platform.logger.error(`[session-ws] WebSocket error: ${error.message}`);\n      await sender.send({\n        type: 'error',\n        payload: { code: 'INTERNAL_ERROR', message: error.message },\n        timestamp: Date.now(),\n      } satisfies ErrorMessage);\n    },\n  },\n});\n"]}
+{"version":3,"sources":["../../src/rest/run-manager.ts","../../src/ws/session-stream-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;AChP5E,IAAM,eAAA,uBAAsB,OAAA,EAAwC;AAEpE,IAAO,iCAAQ,eAAA,CAAuD;AAAA,EACpE,IAAA,EAAM,qBAAA;AAAA,EACN,WAAA,EAAa,4CAAA;AAAA,EAEb,OAAA,EAAS;AAAA,IACP,MAAM,SAAA,CAAU,GAAA,EAAsB,MAAA,EAAoC;AACxE,MAAA,MAAM,cAAc,GAAA,CAAI,WAAA;AACxB,MAAA,MAAM,SAAA,GAAY,YAAY,MAAA,EAAQ,SAAA;AAMtC,MAAA,MAAM,QAAA,GAAW,YAAY,KAAA,EAAO,QAAA,GAAW,OAAO,WAAA,CAAY,KAAA,CAAM,QAAQ,CAAA,GAAI,MAAA;AAEpF,MAAA,IAAI,CAAC,SAAA,EAAW;AACd,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,OAAA;AAAA,UACN,OAAA,EAAS,EAAE,IAAA,EAAM,oBAAA,EAAsB,SAAS,wBAAA,EAAyB;AAAA,UACzE,SAAA,EAAW,KAAK,GAAA;AAAI,SACE,CAAA;AACxB,QAAA,MAAA,CAAO,KAAA,CAAM,KAAM,oBAAoB,CAAA;AACvC,QAAA;AAAA,MACF;AAEA,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,QAClB,QAAA,IAAY,OACR,CAAA,yCAAA,EAA4C,SAAS,sBAAsB,QAAQ,CAAA,CAAA,CAAA,GACnF,4CAA4C,SAAS,CAAA;AAAA,OAC3D;AAEA,MAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AAGjD,MAAA,IAAI;AACF,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,kBAAA;AAAA,UACN,OAAA,EAAS,EAAE,KAAA,EAAO,SAAA,EAAW,8BAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY,EAAE;AAAA,UACnE,SAAA,EAAW,KAAK,GAAA;AAAI,SACY,CAAA;AAAA,MACpC,SAAS,GAAA,EAAK;AACZ,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,iDAAiD,MAAA,CAAO,GAAG,CAAC,CAAA,CAAE,CAAA;AACxF,QAAA,MAAM,GAAA;AAAA,MACR;AAIA,MAAA,IAAI;AACF,QAAA,MAAM,EAAE,KAAA,EAAO,GAAA,KAAQ,MAAM,cAAA,CAAe,cAAc,SAAS,CAAA;AACnE,QAAA,MAAM,iBAAiB,KAAA,CAAM,MAAA,CAAO,CAAC,CAAA,KAAM,CAAA,CAAE,WAAW,WAAW,CAAA;AACnE,QAAA,MAAM,cAAc,KAAA,CAAM,MAAA,CAAO,CAAC,CAAA,KAAM,CAAA,CAAE,WAAW,WAAW,CAAA;AAChE,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,uBAAA;AAAA,UACN,OAAA,EAAS;AAAA,YACP,SAAA;AAAA,YACA,cAAA;AAAA,YACA,WAAA;AAAA,YACA,YAAY,KAAA,CAAM,MAAA;AAAA,YAClB,GAAA;AAAA,YACA,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,WACpC;AAAA,UACA,SAAA,EAAW,KAAK,GAAA;AAAI,SACiB,CAAA;AACvC,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA;AAAA,UAClB,+BAA+B,cAAA,CAAe,MAAM,gBAAgB,WAAA,CAAY,MAAM,wBAAwB,GAAG,CAAA;AAAA,SACnH;AAAA,MACF,SAAS,GAAA,EAAK;AACZ,QAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,KAAA,CAAM,CAAA,sCAAA,EAAyC,GAAG,CAAA,CAAE,CAAA;AAAA,MAC1E;AAKA,MAAA,MAAM,QAAA,GAAW,CAAC,OAAA,KAAqC;AACrD,QAAA,IAAI,OAAA,CAAQ,SAAS,QAAA,EAAU;AAC7B,UAAA,KAAA,MAAW,KAAA,IAAS,QAAQ,MAAA,EAAQ;AAClC,YAAA,KAAK,OAAO,IAAA,CAAK;AAAA,cACf,IAAA,EAAM,YAAA;AAAA,cACN,OAAA,EAAS,EAAE,SAAA,EAAW,KAAA,EAAM;AAAA,cAC5B,SAAA,EAAW,KAAK,GAAA;AAAI,aACM,CAAA,CAAE,KAAA,CAAM,CAAC,GAAA,KAAQ;AAC3C,cAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,KAAA,CAAM,CAAA,wCAAA,EAA2C,GAAG,CAAA,CAAE,CAAA;AAAA,YAC5E,CAAC,CAAA;AAAA,UACH;AACA,UAAA;AAAA,QACF;AAEA,QAAA,KAAK,OAAO,IAAA,CAAK;AAAA,UACf,IAAA,EAAM,eAAA;AAAA,UACN,OAAA,EAAS;AAAA,YACP,OAAO,OAAA,CAAQ,KAAA;AAAA,YACf,SAAS,OAAA,CAAQ,OAAA;AAAA,YACjB,SAAS,OAAA,CAAQ,OAAA;AAAA,YACjB,YAAY,OAAA,CAAQ,UAAA;AAAA,YACpB,KAAK,OAAA,CAAQ,GAAA;AAAA,YACb,MAAM,OAAA,CAAQ;AAAA,WAChB;AAAA,UACA,SAAA,EAAW,KAAK,GAAA;AAAI,SACS,CAAA,CAAE,KAAA,CAAM,CAAC,GAAA,KAAQ;AAC9C,UAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,KAAA,CAAM,CAAA,2CAAA,EAA8C,GAAG,CAAA,CAAE,CAAA;AAAA,QAC/E,CAAC,CAAA;AAAA,MACH,CAAA;AAGA,MAAA,eAAA,CAAgB,GAAA,CAAI,GAAA,EAAe,EAAE,SAAA,EAAW,UAAU,CAAA;AAE1D,MAAA,IAAI;AACF,QAAA,UAAA,CAAW,kBAAA,CAAmB,WAAW,QAAQ,CAAA;AAAA,MACnD,SAAS,GAAA,EAAK;AACZ,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,qDAAqD,MAAA,CAAO,GAAG,CAAC,CAAA,CAAE,CAAA;AAC5F,QAAA,MAAM,GAAA;AAAA,MACR;AACA,MAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,CAAA,qDAAA,EAAwD,SAAS,CAAA,CAAE,CAAA;AAAA,IAC9F,CAAA;AAAA,IAEA,MAAM,SAAA,CAAU,GAAA,EAAsB,OAAA,EAAwB,MAAA,EAAoC;AAEhG,MAAA,IAAI,OAAA,CAAQ,SAAS,MAAA,EAAQ;AAC3B,QAAA,MAAM,SAAA,GAAY,eAAA,CAAgB,GAAA,CAAI,GAAa,GAAG,SAAA,IAAa,EAAA;AACnE,QAAA,MAAM,OAAO,IAAA,CAAK;AAAA,UAChB,IAAA,EAAM,kBAAA;AAAA,UACN,OAAA,EAAS,EAAE,KAAA,EAAO,SAAA,IAAa,EAAA,EAAI,8BAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY,EAAE;AAAA,UACzE,SAAA,EAAW,KAAK,GAAA;AAAI,SACrB,CAAA;AAAA,MACH;AAAA,IACF,CAAA;AAAA,IAEA,MAAM,aAAa,GAAA,EAAsB;AACvC,MAAA,MAAM,KAAA,GAAQ,eAAA,CAAgB,GAAA,CAAI,GAAa,CAAA;AAC/C,MAAA,IAAI,KAAA,EAAO;AACT,QAAA,UAAA,CAAW,qBAAA,CAAsB,KAAA,CAAM,SAAA,EAAW,KAAA,CAAM,QAAQ,CAAA;AAChE,QAAA,eAAA,CAAgB,OAAO,GAAa,CAAA;AAAA,MACtC;AAEA,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA,CAAK,CAAA,8CAAA,EAAiD,KAAA,EAAO,SAAS,CAAA,CAAE,CAAA;AAAA,IAC9F,CAAA;AAAA,IAEA,MAAM,OAAA,CAAQ,GAAA,EAAsB,KAAA,EAAc,MAAA,EAAoC;AACpF,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,CAAA,8BAAA,EAAiC,KAAA,CAAM,OAAO,CAAA,CAAE,CAAA;AAC1E,MAAA,MAAM,OAAO,IAAA,CAAK;AAAA,QAChB,IAAA,EAAM,OAAA;AAAA,QACN,SAAS,EAAE,IAAA,EAAM,gBAAA,EAAkB,OAAA,EAAS,MAAM,OAAA,EAAQ;AAAA,QAC1D,SAAA,EAAW,KAAK,GAAA;AAAI,OACE,CAAA;AAAA,IAC1B;AAAA;AAEJ,CAAC","file":"session-stream-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 * WebSocket handler for session-level event streaming\n *\n * Path: /session/:sessionId\n * Single persistent connection per session — streams turn deltas for ALL runs in the session.\n * Unlike /events/:runId which closes when a run ends, this stays open for the entire session.\n *\n * Delta-based, not read-derived: the session listener registered here receives\n * the exact TurnDelta[] that SessionManager.processEventAndUpdateTurn computed\n * at persist time (via RunManager.broadcastSessionDeltas), not re-derived by\n * re-reading turns.json off disk on every event. That re-read (the previous\n * resolveTurnForEvent) was the actual root cause of the WS stream lagging one\n * event behind persisted state — see run-handler.ts's onEvent for the other\n * half of that fix (persist now happens before broadcast, not after).\n */\n\nimport {\n  defineWebSocket,\n  type PluginContextV3,\n  type TypedSender,\n} from '@kb-labs/sdk';\nimport type {\n  ServerMessage,\n  ClientMessage,\n  ConnectionReadyMessage,\n  RunCompletedMessage,\n  ErrorMessage,\n  TurnDeltaMessage,\n  ConversationSnapshotMessage,\n} from '@kb-labs/agent-contracts';\nimport { RunManager, type SessionBroadcastPayload } from '../rest/run-manager.js';\nimport { SessionManager } from '@kb-labs/agent-core';\n\ninterface SessionConnectionState {\n  sessionId: string;\n  callback: (payload: SessionBroadcastPayload) => void;\n}\n\n/** Per-connection state keyed by the opaque ctx object — avoids mutating ctx */\nconst connectionState = new WeakMap<object, SessionConnectionState>();\n\nexport default defineWebSocket<unknown, ClientMessage, ServerMessage>({\n  path: '/session/:sessionId',\n  description: 'Persistent session event stream (all runs)',\n\n  handler: {\n    async onConnect(ctx: PluginContextV3, sender: TypedSender<ServerMessage>) {\n      const hostContext = ctx.hostContext as { params?: { sessionId?: string }; query?: Record<string, string> };\n      const sessionId = hostContext.params?.sessionId;\n      // Present on resume/gap-recovery reconnects (see AgentsPage.tsx's turn-reducer\n      // 'gap' handling). Not used to filter the response — see getProjection's doc\n      // comment for why resume always re-sends the full projection rather than a\n      // replayed delta range — but accepted and logged so a resume is distinguishable\n      // from a cold connect in server logs.\n      const afterSeq = hostContext.query?.afterSeq ? Number(hostContext.query.afterSeq) : undefined;\n\n      if (!sessionId) {\n        await sender.send({\n          type: 'error',\n          payload: { code: 'MISSING_SESSION_ID', message: 'Session ID is required' },\n          timestamp: Date.now(),\n        } satisfies ErrorMessage);\n        sender.close(4000, 'Missing session ID');\n        return;\n      }\n\n      ctx.platform.logger.info(\n        afterSeq != null\n          ? `[session-ws] Client connected to session ${sessionId} (resume after seq ${afterSeq})`\n          : `[session-ws] Client connected to session ${sessionId}`\n      );\n\n      const sessionManager = new SessionManager(ctx.cwd);\n\n      // Send connection:ready immediately\n      try {\n        await sender.send({\n          type: 'connection:ready',\n          payload: { runId: sessionId, connectedAt: new Date().toISOString() },\n          timestamp: Date.now(),\n        } satisfies ConnectionReadyMessage);\n      } catch (err) {\n        ctx.platform.logger.error(`[session-ws] Failed to send connection:ready: ${String(err)}`);\n        throw err;\n      }\n\n      // Send conversation:snapshot (full current projection) as the baseline —\n      // both for a cold connect AND a resume/gap-recovery reconnect alike.\n      try {\n        const { turns, seq } = await sessionManager.getProjection(sessionId);\n        const completedTurns = turns.filter((t) => t.status !== 'streaming');\n        const activeTurns = turns.filter((t) => t.status === 'streaming');\n        await sender.send({\n          type: 'conversation:snapshot',\n          payload: {\n            sessionId,\n            completedTurns,\n            activeTurns,\n            totalTurns: turns.length,\n            seq,\n            timestamp: new Date().toISOString(),\n          },\n          timestamp: Date.now(),\n        } satisfies ConversationSnapshotMessage);\n        ctx.platform.logger.info(\n          `[session-ws] Sent snapshot: ${completedTurns.length} completed + ${activeTurns.length} active turns at seq ${seq}`\n        );\n      } catch (err) {\n        ctx.platform.logger.error(`[session-ws] Failed to send snapshot: ${err}`);\n      }\n\n      // Session-level listener — receives turn deltas and run:completed\n      // notifications from ALL runs in this session, already computed at\n      // persist time (see module doc comment above).\n      const callback = (payload: SessionBroadcastPayload) => {\n        if (payload.kind === 'deltas') {\n          for (const delta of payload.deltas) {\n            void sender.send({\n              type: 'turn:delta',\n              payload: { sessionId, delta },\n              timestamp: Date.now(),\n            } satisfies TurnDeltaMessage).catch((err) => {\n              ctx.platform.logger.error(`[session-ws] Failed to send turn:delta: ${err}`);\n            });\n          }\n          return;\n        }\n\n        void sender.send({\n          type: 'run:completed',\n          payload: {\n            runId: payload.runId,\n            success: payload.success,\n            summary: payload.summary,\n            durationMs: payload.durationMs,\n            seq: payload.seq,\n            turn: payload.turn,\n          },\n          timestamp: Date.now(),\n        } satisfies RunCompletedMessage).catch((err) => {\n          ctx.platform.logger.error(`[session-ws] Failed to send run:completed: ${err}`);\n        });\n      };\n\n      // Store callback for cleanup in a WeakMap keyed by ctx — no ctx mutation\n      connectionState.set(ctx as object, { sessionId, callback });\n\n      try {\n        RunManager.addSessionListener(sessionId, callback);\n      } catch (err) {\n        ctx.platform.logger.error(`[session-ws] Failed to register session listener: ${String(err)}`);\n        throw err;\n      }\n      ctx.platform.logger.info(`[session-ws] Session listener registered for session ${sessionId}`);\n    },\n\n    async onMessage(ctx: PluginContextV3, message: ClientMessage, sender: TypedSender<ServerMessage>) {\n      // Handle ping or corrections here if needed\n      if (message.type === 'ping') {\n        const sessionId = connectionState.get(ctx as object)?.sessionId ?? '';\n        await sender.send({\n          type: 'connection:ready',\n          payload: { runId: sessionId ?? '', connectedAt: new Date().toISOString() },\n          timestamp: Date.now(),\n        });\n      }\n    },\n\n    async onDisconnect(ctx: PluginContextV3) {\n      const state = connectionState.get(ctx as object);\n      if (state) {\n        RunManager.removeSessionListener(state.sessionId, state.callback);\n        connectionState.delete(ctx as object);\n      }\n\n      ctx.platform.logger.info(`[session-ws] Client disconnected from session ${state?.sessionId}`);\n    },\n\n    async onError(ctx: PluginContextV3, error: Error, sender: TypedSender<ServerMessage>) {\n      ctx.platform.logger.error(`[session-ws] WebSocket error: ${error.message}`);\n      await sender.send({\n        type: 'error',\n        payload: { code: 'INTERNAL_ERROR', message: error.message },\n        timestamp: Date.now(),\n      } satisfies ErrorMessage);\n    },\n  },\n});\n"]}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@kb-labs/agent-entry",
   "description": "CLI commands for KB Labs Agents V2. Provides agent:run-v2 command.",
-  "version": "2.118.2",
+  "version": "2.119.0-canary.0b1ec4151",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -23,16 +23,16 @@
   ],
   "dependencies": {
     "zod": "^3.25.76",
-    "@kb-labs/agent-contracts": "2.118.2",
-    "@kb-labs/agent-core": "2.118.2",
-    "@kb-labs/agent-history": "2.118.2",
-    "@kb-labs/agent-kernel": "2.118.2",
-    "@kb-labs/agent-sdk": "2.118.2",
-    "@kb-labs/agent-runtime": "2.118.2",
-    "@kb-labs/agent-store": "2.118.2",
-    "@kb-labs/agent-tools": "2.118.2",
-    "@kb-labs/agent-tracing": "2.118.2",
-    "@kb-labs/sdk": "2.115.3"
+    "@kb-labs/agent-contracts": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-kernel": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-history": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-runtime": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-core": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-sdk": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-tools": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-tracing": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/agent-store": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/sdk": "2.115.4"
   },
   "devDependencies": {
     "@rspack/cli": "latest",
@@ -46,9 +46,9 @@
     "tsup": "^8.5.0",
     "typescript": "^5.6.3",
     "vitest": "^3.2.6",
-    "@kb-labs/plugin-runtime": "2.118.2",
-    "@kb-labs/devkit": "2.118.2",
-    "@kb-labs/studio-plugin-tools": "2.118.2"
+    "@kb-labs/devkit": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/plugin-runtime": "2.119.0-canary.0b1ec4151",
+    "@kb-labs/studio-plugin-tools": "2.119.0-canary.0b1ec4151"
   },
   "engines": {
     "node": ">=22.0.0",