@kb-labs/agent-cli 0.5.0

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/rest/handlers/sessions-handler.ts"],"names":[],"mappings":";;;;AAkBO,IAAM,sBAAsB,aAAA,CAAc;AAAA,EAC/C,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EAC+B;AAC/B,IAAA,MAAM,QAAQ,KAAA,CAAM,KAAA;AACpB,IAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AAEjD,IAAA,MAAM,MAAA,GAAS,MAAM,cAAA,CAAe,YAAA,CAAa;AAAA,MAC/C,SAAS,KAAA,EAAO,OAAA;AAAA,MAChB,QAAQ,KAAA,EAAO,MAAA;AAAA,MACf,KAAA,EAAO,OAAO,KAAA,IAAS,EAAA;AAAA,MACvB,MAAA,EAAQ,OAAO,MAAA,IAAU;AAAA,KAC1B,CAAA;AAED,IAAA,OAAO;AAAA,MACL,UAAU,MAAA,CAAO,QAAA;AAAA,MACjB,OAAO,MAAA,CAAO;AAAA,KAChB;AAAA,EACF;AACF,CAAC;AAKM,IAAM,oBAAoB,aAAA,CAAc;AAAA,EAC7C,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EAC6B;AAC7B,IAAA,MAAM,SAAS,KAAA,CAAM,MAAA;AACrB,IAAA,MAAM,YAAY,MAAA,EAAQ,SAAA;AAE1B,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,IAC1C;AAEA,IAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACjD,IAAA,MAAM,OAAA,GAAU,MAAM,cAAA,CAAe,cAAA,CAAe,SAAS,CAAA;AAE7D,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,mBAAA,EAAsB,SAAS,CAAA,CAAE,CAAA;AAAA,IACnD;AAEA,IAAA,OAAO,EAAE,OAAA,EAAQ;AAAA,EACnB;AACF,CAAC;AAKM,IAAM,uBAAuB,aAAA,CAAc;AAAA,EAChD,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EACgC;AAChC,IAAA,MAAM,OAAO,KAAA,CAAM,IAAA;AAEnB,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAClB,MAAA,MAAM,IAAI,MAAM,sBAAsB,CAAA;AAAA,IACxC;AAEA,IAAA,MAAM,cAAA,GAAiB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AAEjD,IAAA,MAAM,OAAA,GAAU,MAAM,cAAA,CAAe,aAAA,CAAc;AAAA,MACjD,IAAA,EAAM,SAAA;AAAA,MACN,IAAA,EAAM,KAAK,IAAA,IAAQ,EAAA;AAAA,MACnB,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,MAAM,IAAA,CAAK;AAAA,KACZ,CAAA;AAED,IAAA,GAAA,CAAI,QAAA,CAAS,OAAO,IAAA,CAAK,CAAA,mCAAA,EAAsC,QAAQ,EAAE,CAAA,WAAA,EAAc,IAAA,CAAK,OAAO,CAAA,CAAE,CAAA;AAErG,IAAA,OAAO,EAAE,OAAA,EAAQ;AAAA,EACnB;AACF,CAAC","file":"sessions-handler.js","sourcesContent":["/**\n * Session management REST handlers\n */\n\nimport { defineHandler, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport { SessionManager } from '@kb-labs/agent-core';\nimport type {\n  ListSessionsRequest,\n  ListSessionsResponse,\n  GetSessionRequest,\n  GetSessionResponse,\n  CreateSessionRequest,\n  CreateSessionResponse,\n} from '@kb-labs/agent-contracts';\n\n/**\n * GET /sessions - List all sessions\n */\nexport const listSessionsHandler = defineHandler({\n  async execute(\n    ctx: PluginContextV3,\n    input: RestInput<ListSessionsRequest>\n  ): Promise<ListSessionsResponse> {\n    const query = input.query as ListSessionsRequest | undefined;\n    const sessionManager = new SessionManager(ctx.cwd);\n\n    const result = await sessionManager.listSessions({\n      agentId: query?.agentId,\n      status: query?.status,\n      limit: query?.limit ?? 50,\n      offset: query?.offset ?? 0,\n    });\n\n    return {\n      sessions: result.sessions,\n      total: result.total,\n    };\n  },\n});\n\n/**\n * GET /sessions/:sessionId - Get session details\n */\nexport const getSessionHandler = defineHandler({\n  async execute(\n    ctx: PluginContextV3,\n    input: RestInput<GetSessionRequest>\n  ): Promise<GetSessionResponse> {\n    const params = input.params as Record<string, string> | undefined;\n    const sessionId = params?.sessionId;\n\n    if (!sessionId) {\n      throw new Error('Session ID is required');\n    }\n\n    const sessionManager = new SessionManager(ctx.cwd);\n    const session = await sessionManager.getSessionInfo(sessionId);\n\n    if (!session) {\n      throw new Error(`Session not found: ${sessionId}`);\n    }\n\n    return { session };\n  },\n});\n\n/**\n * POST /sessions - Create a new session\n */\nexport const createSessionHandler = defineHandler({\n  async execute(\n    ctx: PluginContextV3,\n    input: RestInput<CreateSessionRequest>\n  ): Promise<CreateSessionResponse> {\n    const body = input.body as CreateSessionRequest | undefined;\n\n    if (!body?.agentId) {\n      throw new Error('Agent ID is required');\n    }\n\n    const sessionManager = new SessionManager(ctx.cwd);\n\n    const session = await sessionManager.createSession({\n      mode: 'execute',\n      task: body.task ?? '',\n      agentId: body.agentId,\n      name: body.name,\n    });\n\n    ctx.platform.logger.info(`[sessions-handler] Created session ${session.id} for agent ${body.agentId}`);\n\n    return { session };\n  },\n});\n"]}

package/dist/rest/handlers/status-handler.d.ts

@@ -0,0 +1,7 @@
+import * as _kb_labs_shared_command_kit from '@kb-labs/shared-command-kit';
+import { RestInput } from '@kb-labs/sdk';
+import { RunStatusResponse } from '@kb-labs/agent-contracts';
+
+declare const _default: _kb_labs_shared_command_kit.Handler<unknown, RestInput<void, unknown, unknown>, RunStatusResponse>;
+
+export { _default as default };

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

@@ -0,0 +1,313 @@
+import { defineHandler, useCache, usePlatform } from '@kb-labs/sdk';
+
+// src/rest/handlers/status-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];
+
+// src/rest/handlers/status-handler.ts
+var status_handler_default = defineHandler({
+  async execute(ctx, input) {
+    const { runId } = input.params;
+    const run = RunManager.get(runId);
+    if (!run) {
+      const state = await RunManager.getState(runId);
+      if (!state) {
+        return {
+          runId,
+          status: "failed",
+          task: "",
+          startedAt: "",
+          error: `Run ${runId} not found`
+        };
+      }
+      return {
+        runId: state.runId,
+        status: state.status,
+        task: state.task,
+        startedAt: state.startedAt,
+        completedAt: state.completedAt,
+        durationMs: state.durationMs,
+        summary: state.summary,
+        error: state.error
+      };
+    }
+    return {
+      runId: run.runId,
+      status: run.status,
+      task: run.task,
+      startedAt: run.startedAt,
+      completedAt: run.completedAt,
+      durationMs: run.durationMs,
+      summary: run.summary,
+      error: run.error
+    };
+  }
+});
+
+export { status_handler_default as default };
+//# sourceMappingURL=status-handler.js.map
+//# sourceMappingURL=status-handler.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/rest/run-manager.ts","../../../src/rest/handlers/status-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;;;ACnX5E,IAAO,yBAAQ,aAAA,CAAc;AAAA,EAC3B,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EAC4B;AAC5B,IAAA,MAAM,EAAE,KAAA,EAAM,GAAI,KAAA,CAAM,MAAA;AAGxB,IAAA,MAAM,GAAA,GAAM,UAAA,CAAW,GAAA,CAAI,KAAK,CAAA;AAEhC,IAAA,IAAI,CAAC,GAAA,EAAK;AAER,MAAA,MAAM,KAAA,GAAQ,MAAM,UAAA,CAAW,QAAA,CAAS,KAAK,CAAA;AAE7C,MAAA,IAAI,CAAC,KAAA,EAAO;AAEV,QAAA,OAAO;AAAA,UACL,KAAA;AAAA,UACA,MAAA,EAAQ,QAAA;AAAA,UACR,IAAA,EAAM,EAAA;AAAA,UACN,SAAA,EAAW,EAAA;AAAA,UACX,KAAA,EAAO,OAAO,KAAK,CAAA,UAAA;AAAA,SACrB;AAAA,MACF;AAGA,MAAA,OAAO;AAAA,QACL,OAAO,KAAA,CAAM,KAAA;AAAA,QACb,QAAQ,KAAA,CAAM,MAAA;AAAA,QACd,MAAM,KAAA,CAAM,IAAA;AAAA,QACZ,WAAW,KAAA,CAAM,SAAA;AAAA,QACjB,aAAa,KAAA,CAAM,WAAA;AAAA,QACnB,YAAY,KAAA,CAAM,UAAA;AAAA,QAClB,SAAS,KAAA,CAAM,OAAA;AAAA,QACf,OAAO,KAAA,CAAM;AAAA,OACf;AAAA,IACF;AAEA,IAAA,OAAO;AAAA,MACL,OAAO,GAAA,CAAI,KAAA;AAAA,MACX,QAAQ,GAAA,CAAI,MAAA;AAAA,MACZ,MAAM,GAAA,CAAI,IAAA;AAAA,MACV,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;AAAA,EACF;AACF,CAAC","file":"status-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 * GET /run/:runId handler\n *\n * Returns current status of a run\n */\n\nimport { defineHandler, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport type { RunStatusResponse } from '@kb-labs/agent-contracts';\nimport { RunManager } from '../run-manager.js';\n\nexport default defineHandler({\n  async execute(\n    ctx: PluginContextV3,\n    input: RestInput<void>\n  ): Promise<RunStatusResponse> {\n    const { runId } = input.params as { runId: string };\n\n    // First try to get active run from memory\n    const run = RunManager.get(runId);\n\n    if (!run) {\n      // If not in memory, check cache for completed runs\n      const state = await RunManager.getState(runId);\n\n      if (!state) {\n        // Return 404-like response\n        return {\n          runId,\n          status: 'failed',\n          task: '',\n          startedAt: '',\n          error: `Run ${runId} not found`,\n        };\n      }\n\n      // Return cached state (no active orchestrator)\n      return {\n        runId: state.runId,\n        status: state.status,\n        task: state.task,\n        startedAt: state.startedAt,\n        completedAt: state.completedAt,\n        durationMs: state.durationMs,\n        summary: state.summary,\n        error: state.error,\n      };\n    }\n\n    return {\n      runId: run.runId,\n      status: run.status,\n      task: run.task,\n      startedAt: run.startedAt,\n      completedAt: run.completedAt,\n      durationMs: run.durationMs,\n      summary: run.summary,\n      error: run.error,\n    };\n  },\n});\n"]}

package/dist/rest/handlers/stop-handler.d.ts

@@ -0,0 +1,7 @@
+import * as _kb_labs_shared_command_kit from '@kb-labs/shared-command-kit';
+import { RestInput } from '@kb-labs/sdk';
+import { StopRequest, StopResponse } from '@kb-labs/agent-contracts';
+
+declare const _default: _kb_labs_shared_command_kit.Handler<unknown, RestInput<StopRequest, unknown, unknown>, StopResponse>;
+
+export { _default as default };