@kb-labs/agent-cli 0.5.0

package/dist/ws/session-stream-handler.d.ts

@@ -0,0 +1,8 @@
+import * as _kb_labs_plugin_contracts from '@kb-labs/plugin-contracts';
+import { PluginContextV3 } from '@kb-labs/sdk';
+
+declare const _default: {
+    execute(context: PluginContextV3<unknown>, input: _kb_labs_plugin_contracts.WSInput): Promise<_kb_labs_plugin_contracts.CommandResult | void>;
+};
+
+export { _default as default };

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

@@ -0,0 +1,409 @@
+import { defineWebSocket, useCache, usePlatform } 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) */
+  activeRuns = /* @__PURE__ */ new Map();
+  /** Session-level listeners: sessionId → Set<callback> — receive events 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
+   */
+  generateRunId() {
+    return `run-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;
+  }
+  /**
+   * Register a new run
+   */
+  async register(runId, task, agent, sessionManager, sessionId) {
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const run = {
+      runId,
+      task,
+      status: "pending",
+      agent,
+      sessionManager,
+      startedAt: now,
+      listeners: /* @__PURE__ */ new Set(),
+      lastSeq: 0,
+      eventBuffer: []
+    };
+    this.activeRuns.set(runId, run);
+    if (sessionId) {
+      this.runSessionMap.set(runId, sessionId);
+    }
+    await this.saveToCache(run);
+    return run;
+  }
+  /**
+   * Get run by ID (from memory first, then cache for state)
+   */
+  get(runId) {
+    return this.activeRuns.get(runId);
+  }
+  /**
+   * Check if run exists (in memory or cache)
+   */
+  async exists(runId) {
+    if (this.activeRuns.has(runId)) {
+      return true;
+    }
+    const state = await this.getState(runId);
+    return state !== null;
+  }
+  /**
+   * Get run state from cache (for completed runs or cross-process access)
+   */
+  async getState(runId) {
+    const cache = useCache();
+    if (!cache) {
+      return null;
+    }
+    return cache.get(`${CACHE_PREFIX}${runId}`);
+  }
+  /**
+   * Update run status
+   */
+  async updateStatus(runId, status, extra) {
+    const run = this.activeRuns.get(runId);
+    if (run) {
+      run.status = status;
+      if (extra) {
+        Object.assign(run, extra);
+      }
+      await this.saveToCache(run);
+    }
+  }
+  /**
+   * Save run state to cache
+   */
+  async saveToCache(run) {
+    const cache = useCache();
+    if (!cache) {
+      return;
+    }
+    const state = {
+      runId: run.runId,
+      task: run.task,
+      status: run.status,
+      startedAt: run.startedAt,
+      completedAt: run.completedAt,
+      durationMs: run.durationMs,
+      summary: run.summary,
+      error: run.error
+    };
+    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)
+   */
+  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);
+    }
+    return true;
+  }
+  /**
+   * Remove event listener from run
+   */
+  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);
+    }
+  }
+  /**
+   * 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 {
+        }
+      }
+    }
+    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 {
+          }
+        }
+      }
+    }
+    return seqEvent;
+  }
+  /**
+   * Add a session-level listener — receives events from ALL runs in this session.
+   */
+  addSessionListener(sessionId, callback) {
+    if (!this.sessionListeners.has(sessionId)) {
+      this.sessionListeners.set(sessionId, /* @__PURE__ */ new Set());
+    }
+    this.sessionListeners.get(sessionId).add(callback);
+  }
+  /**
+   * Remove a session-level listener.
+   */
+  removeSessionListener(sessionId, callback) {
+    const listeners = this.sessionListeners.get(sessionId);
+    if (listeners) {
+      listeners.delete(callback);
+      if (listeners.size === 0) {
+        this.sessionListeners.delete(sessionId);
+      }
+    }
+  }
+  /**
+   * 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
+   */
+  listActive() {
+    return Array.from(this.activeRuns.values()).map((r) => ({
+      runId: r.runId,
+      task: r.task,
+      status: r.status,
+      startedAt: r.startedAt
+    }));
+  }
+  /**
+   * Request graceful stop of a running agent (and its child agents via propagated AbortSignal).
+   * Agent finishes its current tool call then exits at the next iteration boundary.
+   */
+  requestStop(runId) {
+    const run = this.activeRuns.get(runId);
+    if (!run || run.status !== "running" || !run.agent) {
+      return false;
+    }
+    run.agent.requestStop();
+    return true;
+  }
+  /**
+   * Clean up completed runs from memory (cache handles its own TTL)
+   */
+  cleanup() {
+    for (const [runId, run] of this.activeRuns) {
+      if (run.status === "completed" || run.status === "failed" || run.status === "stopped") {
+        this.activeRuns.delete(runId);
+      }
+    }
+  }
+};
+var GLOBAL_KEY = "__kb_agent_run_manager__";
+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;
+      if (!sessionId) {
+        await sender.send({
+          type: "error",
+          payload: { code: "MISSING_SESSION_ID", message: "Session ID is required" },
+          timestamp: Date.now()
+        });
+        sender.close(4e3, "Missing session ID");
+        return;
+      }
+      ctx.platform.logger.info(`[session-ws] Client connected to session ${sessionId}`);
+      const sessionManager = new SessionManager(ctx.cwd);
+      const lastTurnSignatures = /* @__PURE__ */ new Map();
+      await sender.send({
+        type: "connection:ready",
+        payload: { runId: sessionId, connectedAt: (/* @__PURE__ */ new Date()).toISOString() },
+        timestamp: Date.now()
+      });
+      try {
+        const snapshot = await sessionManager.getConversationSnapshot(sessionId);
+        await sender.send({
+          type: "conversation:snapshot",
+          payload: {
+            sessionId,
+            completedTurns: snapshot.completedTurns,
+            activeTurns: snapshot.activeTurns,
+            totalTurns: snapshot.totalTurns,
+            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`
+        );
+      } 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()
+              });
+            }
+          }
+        } 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()
+          });
+        }
+      };
+      connectionState.set(ctx, { sessionId, callback: eventCallback });
+      RunManager.addSessionListener(sessionId, eventCallback);
+    },
+    async onMessage(ctx, message, sender) {
+      if (message.type === "ping") {
+        const sessionId = connectionState.get(ctx)?.sessionId ?? "";
+        await sender.send({
+          type: "connection:ready",
+          payload: { runId: sessionId ?? "", connectedAt: (/* @__PURE__ */ new Date()).toISOString() },
+          timestamp: Date.now()
+        });
+      }
+    },
+    async onDisconnect(ctx) {
+      const state = connectionState.get(ctx);
+      if (state) {
+        RunManager.removeSessionListener(state.sessionId, state.callback);
+        connectionState.delete(ctx);
+      }
+      ctx.platform.logger.info(`[session-ws] Client disconnected from session ${state?.sessionId}`);
+    },
+    async onError(ctx, error, sender) {
+      ctx.platform.logger.error(`[session-ws] WebSocket error: ${error.message}`);
+      await sender.send({
+        type: "error",
+        payload: { code: "INTERNAL_ERROR", message: error.message },
+        timestamp: Date.now()
+      });
+    }
+  }
+});
+
+export { session_stream_handler_default as default };
+//# sourceMappingURL=session-stream-handler.js.map
+//# sourceMappingURL=session-stream-handler.js.map

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

@@ -0,0 +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,MAAM,OAAO,IAAA,CAAK;AAAA,QAChB,IAAA,EAAM,kBAAA;AAAA,QACN,OAAA,EAAS,EAAE,KAAA,EAAO,SAAA,EAAW,8BAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY,EAAE;AAAA,QACnE,SAAA,EAAW,KAAK,GAAA;AAAI,OACY,CAAA;AAGlC,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,UAAA,CAAW,kBAAA,CAAmB,WAAW,aAAa,CAAA;AAAA,IACxD,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      await sender.send({\n        type: 'connection:ready',\n        payload: { runId: sessionId, connectedAt: new Date().toISOString() },\n        timestamp: Date.now(),\n      } satisfies ConnectionReadyMessage);\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      RunManager.addSessionListener(sessionId, eventCallback);\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

@@ -0,0 +1,83 @@
+{
+  "name": "@kb-labs/agent-cli",
+  "version": "0.5.0",
+  "type": "module",
+  "description": "CLI commands for KB Labs Agents V2. Provides agent:run-v2 command.",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    },
+    "./plugin-manifest": {
+      "import": "./dist/manifest.js",
+      "types": "./dist/manifest.d.ts"
+    },
+    "./dist/*": "./dist/*"
+  },
+  "kb": {
+    "manifest": "./dist/manifest.js"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "sideEffects": false,
+  "scripts": {
+    "pretype-check": "pnpm --filter @kb-labs/agent-cli build",
+    "clean": "rimraf dist",
+    "build": "tsup --config tsup.config.ts && pnpm run build:studio",
+    "build:cli": "tsup --config tsup.config.ts",
+    "build:studio": "rspack build --config rspack.studio.config.mjs",
+    "dev": "tsup --config tsup.config.ts --watch",
+    "dev:studio": "rspack serve --config rspack.studio.config.mjs",
+    "lint": "eslint src --ext .ts,.tsx",
+    "lint:fix": "eslint . --fix",
+    "type-check": "tsc --noEmit",
+    "test": "vitest run --passWithNoTests",
+    "test:watch": "vitest"
+  },
+  "dependencies": {
+    "@kb-labs/agent-contracts": "^0.1.0",
+    "@kb-labs/agent-core": "^0.1.0",
+    "@kb-labs/agent-history": "^0.5.0",
+    "@kb-labs/agent-kernel": "^0.1.0",
+    "@kb-labs/agent-runtime": "^0.1.0",
+    "@kb-labs/agent-sdk": "^0.1.0",
+    "@kb-labs/agent-store": "^0.1.0",
+    "@kb-labs/agent-tools": "^0.5.0",
+    "@kb-labs/agent-tracing": "^0.5.0",
+    "@kb-labs/sdk": "^1.4.0",
+    "zod": "^3.23.8"
+  },
+  "peerDependencies": {
+    "@ant-design/icons": "^5.0.0",
+    "@tanstack/react-query": "^5.0.0",
+    "antd": "^5.0.0",
+    "react": ">=18.0.0",
+    "react-dom": ">=18.0.0",
+    "react-router-dom": ">=6.0.0"
+  },
+  "devDependencies": {
+    "@kb-labs/devkit": "link:../../../../infra/kb-labs-devkit",
+    "@kb-labs/plugin-runtime": "link:../../../../infra/kb-labs-plugin/packages/plugin-runtime",
+    "@kb-labs/studio-plugin-tools": "link:../../../../platform/kb-labs-studio/packages/studio-plugin-tools",
+    "@rspack/cli": "latest",
+    "@rspack/core": "latest",
+    "@types/node": "^24.3.3",
+    "@types/react": "^18.3.18",
+    "@types/react-dom": "^18.3.5",
+    "react": "^18.3.1",
+    "react-dom": "^18.3.1",
+    "rimraf": "^6.0.1",
+    "tsup": "^8.5.0",
+    "typescript": "^5.6.3",
+    "vitest": "^3.2.4"
+  },
+  "engines": {
+    "node": ">=20.0.0",
+    "pnpm": ">=9.0.0"
+  }
+}