@kb-labs/agent-cli 0.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (136) hide show
  1. package/README.md +56 -0
  2. package/dist/cli/commands/diff.d.ts +17 -0
  3. package/dist/cli/commands/diff.js +182 -0
  4. package/dist/cli/commands/diff.js.map +1 -0
  5. package/dist/cli/commands/history.d.ts +16 -0
  6. package/dist/cli/commands/history.js +216 -0
  7. package/dist/cli/commands/history.js.map +1 -0
  8. package/dist/cli/commands/quality-report.d.ts +21 -0
  9. package/dist/cli/commands/quality-report.js +457 -0
  10. package/dist/cli/commands/quality-report.js.map +1 -0
  11. package/dist/cli/commands/rollback.d.ts +27 -0
  12. package/dist/cli/commands/rollback.js +109 -0
  13. package/dist/cli/commands/rollback.js.map +1 -0
  14. package/dist/cli/commands/run.d.ts +42 -0
  15. package/dist/cli/commands/run.js +923 -0
  16. package/dist/cli/commands/run.js.map +1 -0
  17. package/dist/cli/commands/trace-context.d.ts +22 -0
  18. package/dist/cli/commands/trace-context.js +131 -0
  19. package/dist/cli/commands/trace-context.js.map +1 -0
  20. package/dist/cli/commands/trace-diagnose.d.ts +20 -0
  21. package/dist/cli/commands/trace-diagnose.js +434 -0
  22. package/dist/cli/commands/trace-diagnose.js.map +1 -0
  23. package/dist/cli/commands/trace-event-normalizer.d.ts +13 -0
  24. package/dist/cli/commands/trace-event-normalizer.js +39 -0
  25. package/dist/cli/commands/trace-event-normalizer.js.map +1 -0
  26. package/dist/cli/commands/trace-filter.d.ts +19 -0
  27. package/dist/cli/commands/trace-filter.js +153 -0
  28. package/dist/cli/commands/trace-filter.js.map +1 -0
  29. package/dist/cli/commands/trace-iteration.d.ts +18 -0
  30. package/dist/cli/commands/trace-iteration.js +192 -0
  31. package/dist/cli/commands/trace-iteration.js.map +1 -0
  32. package/dist/cli/commands/trace-stats.d.ts +17 -0
  33. package/dist/cli/commands/trace-stats.js +247 -0
  34. package/dist/cli/commands/trace-stats.js.map +1 -0
  35. package/dist/index.d.ts +2 -0
  36. package/dist/index.js +473 -0
  37. package/dist/index.js.map +1 -0
  38. package/dist/manifest.d.ts +184 -0
  39. package/dist/manifest.js +473 -0
  40. package/dist/manifest.js.map +1 -0
  41. package/dist/rest/handlers/approve-handler.d.ts +15 -0
  42. package/dist/rest/handlers/approve-handler.js +60 -0
  43. package/dist/rest/handlers/approve-handler.js.map +1 -0
  44. package/dist/rest/handlers/approve-session-plan-handler.d.ts +10 -0
  45. package/dist/rest/handlers/approve-session-plan-handler.js +52 -0
  46. package/dist/rest/handlers/approve-session-plan-handler.js.map +1 -0
  47. package/dist/rest/handlers/correct-handler.d.ts +7 -0
  48. package/dist/rest/handlers/correct-handler.js +326 -0
  49. package/dist/rest/handlers/correct-handler.js.map +1 -0
  50. package/dist/rest/handlers/create-session-handler.d.ts +7 -0
  51. package/dist/rest/handlers/create-session-handler.js +25 -0
  52. package/dist/rest/handlers/create-session-handler.js.map +1 -0
  53. package/dist/rest/handlers/execute-session-plan-handler.d.ts +10 -0
  54. package/dist/rest/handlers/execute-session-plan-handler.js +635 -0
  55. package/dist/rest/handlers/execute-session-plan-handler.js.map +1 -0
  56. package/dist/rest/handlers/generate-spec-handler.d.ts +10 -0
  57. package/dist/rest/handlers/generate-spec-handler.js +389 -0
  58. package/dist/rest/handlers/generate-spec-handler.js.map +1 -0
  59. package/dist/rest/handlers/get-file-diff-handler.d.ts +24 -0
  60. package/dist/rest/handlers/get-file-diff-handler.js +44 -0
  61. package/dist/rest/handlers/get-file-diff-handler.js.map +1 -0
  62. package/dist/rest/handlers/get-session-handler.d.ts +10 -0
  63. package/dist/rest/handlers/get-session-handler.js +23 -0
  64. package/dist/rest/handlers/get-session-handler.js.map +1 -0
  65. package/dist/rest/handlers/get-session-plan-handler.d.ts +10 -0
  66. package/dist/rest/handlers/get-session-plan-handler.js +53 -0
  67. package/dist/rest/handlers/get-session-plan-handler.js.map +1 -0
  68. package/dist/rest/handlers/get-session-turns-handler.d.ts +16 -0
  69. package/dist/rest/handlers/get-session-turns-handler.js +35 -0
  70. package/dist/rest/handlers/get-session-turns-handler.js.map +1 -0
  71. package/dist/rest/handlers/get-spec-handler.d.ts +10 -0
  72. package/dist/rest/handlers/get-spec-handler.js +39 -0
  73. package/dist/rest/handlers/get-spec-handler.js.map +1 -0
  74. package/dist/rest/handlers/list-file-changes-handler.d.ts +13 -0
  75. package/dist/rest/handlers/list-file-changes-handler.js +34 -0
  76. package/dist/rest/handlers/list-file-changes-handler.js.map +1 -0
  77. package/dist/rest/handlers/list-sessions-handler.d.ts +7 -0
  78. package/dist/rest/handlers/list-sessions-handler.js +23 -0
  79. package/dist/rest/handlers/list-sessions-handler.js.map +1 -0
  80. package/dist/rest/handlers/rollback-handler.d.ts +22 -0
  81. package/dist/rest/handlers/rollback-handler.js +91 -0
  82. package/dist/rest/handlers/rollback-handler.js.map +1 -0
  83. package/dist/rest/handlers/run-handler.d.ts +7 -0
  84. package/dist/rest/handlers/run-handler.js +516 -0
  85. package/dist/rest/handlers/run-handler.js.map +1 -0
  86. package/dist/rest/handlers/sessions-handler.d.ts +18 -0
  87. package/dist/rest/handlers/sessions-handler.js +56 -0
  88. package/dist/rest/handlers/sessions-handler.js.map +1 -0
  89. package/dist/rest/handlers/status-handler.d.ts +7 -0
  90. package/dist/rest/handlers/status-handler.js +313 -0
  91. package/dist/rest/handlers/status-handler.js.map +1 -0
  92. package/dist/rest/handlers/stop-handler.d.ts +7 -0
  93. package/dist/rest/handlers/stop-handler.js +317 -0
  94. package/dist/rest/handlers/stop-handler.js.map +1 -0
  95. package/dist/widgets/220.js +446 -0
  96. package/dist/widgets/220.js.map +1 -0
  97. package/dist/widgets/331.js +2 -0
  98. package/dist/widgets/331.js.map +1 -0
  99. package/dist/widgets/403.js +2 -0
  100. package/dist/widgets/403.js.map +1 -0
  101. package/dist/widgets/406.js +35 -0
  102. package/dist/widgets/406.js.map +1 -0
  103. package/dist/widgets/455.js +2 -0
  104. package/dist/widgets/455.js.map +1 -0
  105. package/dist/widgets/482.js +2 -0
  106. package/dist/widgets/482.js.map +1 -0
  107. package/dist/widgets/485.js +2 -0
  108. package/dist/widgets/485.js.map +1 -0
  109. package/dist/widgets/527.js +2 -0
  110. package/dist/widgets/527.js.map +1 -0
  111. package/dist/widgets/628.js +2 -0
  112. package/dist/widgets/628.js.map +1 -0
  113. package/dist/widgets/694.js +2 -0
  114. package/dist/widgets/694.js.map +1 -0
  115. package/dist/widgets/712.js +2 -0
  116. package/dist/widgets/712.js.map +1 -0
  117. package/dist/widgets/866.js +2 -0
  118. package/dist/widgets/866.js.map +1 -0
  119. package/dist/widgets/915.js +39 -0
  120. package/dist/widgets/915.js.map +1 -0
  121. package/dist/widgets/957.js +10 -0
  122. package/dist/widgets/957.js.map +1 -0
  123. package/dist/widgets/983.js +2 -0
  124. package/dist/widgets/983.js.map +1 -0
  125. package/dist/widgets/@mf-types.d.ts +3 -0
  126. package/dist/widgets/@mf-types.zip +0 -0
  127. package/dist/widgets/__federation_expose_AgentsPage.js +2 -0
  128. package/dist/widgets/__federation_expose_AgentsPage.js.map +1 -0
  129. package/dist/widgets/mf-manifest.json +260 -0
  130. package/dist/widgets/mf-stats.json +305 -0
  131. package/dist/widgets/remoteEntry.js +7 -0
  132. package/dist/widgets/remoteEntry.js.map +1 -0
  133. package/dist/ws/session-stream-handler.d.ts +8 -0
  134. package/dist/ws/session-stream-handler.js +409 -0
  135. package/dist/ws/session-stream-handler.js.map +1 -0
  136. package/package.json +83 -0
@@ -0,0 +1,516 @@
1
+ import { defineHandler, useAnalytics, useCache, useConfig, usePlatform } from '@kb-labs/sdk';
2
+ import { bootstrapAgentSDK, SessionManager, createSessionMemoryBridge, createCoreToolPack } from '@kb-labs/agent-core';
3
+ import { createDefaultResponseRequirementsSelector } from '@kb-labs/agent-runtime';
4
+ import { IncrementalTraceWriter } from '@kb-labs/agent-tracing';
5
+ import { AgentSDK } from '@kb-labs/agent-sdk';
6
+ import { createToolRegistry } from '@kb-labs/agent-tools';
7
+ import path from 'path';
8
+ import fs from 'fs';
9
+ import { AGENT_ANALYTICS_EVENTS, AGENTS_WS_CHANNELS, AGENTS_WS_BASE_PATH } from '@kb-labs/agent-contracts';
10
+
11
+ // src/rest/handlers/run-handler.ts
12
+ var CACHE_PREFIX = "agent:run:";
13
+ var EVENT_TOPIC_PREFIX = "agent:events:";
14
+ var CACHE_TTL = 36e5;
15
+ var RunManagerImpl = class {
16
+ /** Live agents and listeners (not cacheable) */
17
+ activeRuns = /* @__PURE__ */ new Map();
18
+ /** Session-level listeners: sessionId → Set<callback> — receive events from ALL runs in session */
19
+ sessionListeners = /* @__PURE__ */ new Map();
20
+ /** sessionId per runId — set when run is registered */
21
+ runSessionMap = /* @__PURE__ */ new Map();
22
+ /**
23
+ * Generate unique run ID
24
+ */
25
+ generateRunId() {
26
+ return `run-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;
27
+ }
28
+ /**
29
+ * Register a new run
30
+ */
31
+ async register(runId, task, agent, sessionManager, sessionId) {
32
+ const now = (/* @__PURE__ */ new Date()).toISOString();
33
+ const run = {
34
+ runId,
35
+ task,
36
+ status: "pending",
37
+ agent,
38
+ sessionManager,
39
+ startedAt: now,
40
+ listeners: /* @__PURE__ */ new Set(),
41
+ lastSeq: 0,
42
+ eventBuffer: []
43
+ };
44
+ this.activeRuns.set(runId, run);
45
+ if (sessionId) {
46
+ this.runSessionMap.set(runId, sessionId);
47
+ }
48
+ await this.saveToCache(run);
49
+ return run;
50
+ }
51
+ /**
52
+ * Get run by ID (from memory first, then cache for state)
53
+ */
54
+ get(runId) {
55
+ return this.activeRuns.get(runId);
56
+ }
57
+ /**
58
+ * Check if run exists (in memory or cache)
59
+ */
60
+ async exists(runId) {
61
+ if (this.activeRuns.has(runId)) {
62
+ return true;
63
+ }
64
+ const state = await this.getState(runId);
65
+ return state !== null;
66
+ }
67
+ /**
68
+ * Get run state from cache (for completed runs or cross-process access)
69
+ */
70
+ async getState(runId) {
71
+ const cache = useCache();
72
+ if (!cache) {
73
+ return null;
74
+ }
75
+ return cache.get(`${CACHE_PREFIX}${runId}`);
76
+ }
77
+ /**
78
+ * Update run status
79
+ */
80
+ async updateStatus(runId, status, extra) {
81
+ const run = this.activeRuns.get(runId);
82
+ if (run) {
83
+ run.status = status;
84
+ if (extra) {
85
+ Object.assign(run, extra);
86
+ }
87
+ await this.saveToCache(run);
88
+ }
89
+ }
90
+ /**
91
+ * Save run state to cache
92
+ */
93
+ async saveToCache(run) {
94
+ const cache = useCache();
95
+ if (!cache) {
96
+ return;
97
+ }
98
+ const state = {
99
+ runId: run.runId,
100
+ task: run.task,
101
+ status: run.status,
102
+ startedAt: run.startedAt,
103
+ completedAt: run.completedAt,
104
+ durationMs: run.durationMs,
105
+ summary: run.summary,
106
+ error: run.error
107
+ };
108
+ await cache.set(`${CACHE_PREFIX}${run.runId}`, state, CACHE_TTL);
109
+ }
110
+ /** Track subscriptions for cleanup */
111
+ subscriptions = /* @__PURE__ */ new Map();
112
+ /**
113
+ * Add event listener to run (uses eventBus for cross-process)
114
+ */
115
+ addListener(runId, callback) {
116
+ const platform = usePlatform();
117
+ if (!platform?.eventBus) {
118
+ const run2 = this.activeRuns.get(runId);
119
+ if (run2) {
120
+ run2.listeners.add(callback);
121
+ return true;
122
+ }
123
+ return false;
124
+ }
125
+ const topic = `${EVENT_TOPIC_PREFIX}${runId}`;
126
+ const unsubscribe = platform.eventBus.subscribe(topic, async (event) => {
127
+ try {
128
+ callback(event);
129
+ } catch {
130
+ }
131
+ });
132
+ if (!this.subscriptions.has(runId)) {
133
+ this.subscriptions.set(runId, /* @__PURE__ */ new Map());
134
+ }
135
+ this.subscriptions.get(runId).set(callback, unsubscribe);
136
+ const run = this.activeRuns.get(runId);
137
+ if (run) {
138
+ run.listeners.add(callback);
139
+ }
140
+ return true;
141
+ }
142
+ /**
143
+ * Remove event listener from run
144
+ */
145
+ removeListener(runId, callback) {
146
+ const runSubs = this.subscriptions.get(runId);
147
+ if (runSubs) {
148
+ const unsubscribe = runSubs.get(callback);
149
+ if (unsubscribe) {
150
+ unsubscribe();
151
+ runSubs.delete(callback);
152
+ }
153
+ if (runSubs.size === 0) {
154
+ this.subscriptions.delete(runId);
155
+ }
156
+ }
157
+ const run = this.activeRuns.get(runId);
158
+ if (run) {
159
+ run.listeners.delete(callback);
160
+ }
161
+ }
162
+ /**
163
+ * Broadcast event to all listeners of a run (uses eventBus for cross-process)
164
+ * Assigns monotonic sequence number for reliable ordering
165
+ * @returns The event with seq assigned (for persistence)
166
+ */
167
+ broadcast(runId, event) {
168
+ const platform = usePlatform();
169
+ const run = this.activeRuns.get(runId);
170
+ let seqEvent = event;
171
+ if (run) {
172
+ run.lastSeq++;
173
+ seqEvent = { ...event, seq: run.lastSeq, runId };
174
+ run.eventBuffer.push(seqEvent);
175
+ }
176
+ if (platform?.eventBus) {
177
+ const topic = `${EVENT_TOPIC_PREFIX}${runId}`;
178
+ platform.eventBus.publish(topic, seqEvent).catch(() => {
179
+ });
180
+ }
181
+ if (run) {
182
+ for (const listener of run.listeners) {
183
+ try {
184
+ listener(seqEvent);
185
+ } catch {
186
+ }
187
+ }
188
+ }
189
+ const sessionId = this.runSessionMap.get(runId);
190
+ if (sessionId) {
191
+ const sListeners = this.sessionListeners.get(sessionId);
192
+ if (sListeners) {
193
+ for (const listener of sListeners) {
194
+ try {
195
+ listener(seqEvent);
196
+ } catch {
197
+ }
198
+ }
199
+ }
200
+ }
201
+ return seqEvent;
202
+ }
203
+ /**
204
+ * Add a session-level listener — receives events from ALL runs in this session.
205
+ */
206
+ addSessionListener(sessionId, callback) {
207
+ if (!this.sessionListeners.has(sessionId)) {
208
+ this.sessionListeners.set(sessionId, /* @__PURE__ */ new Set());
209
+ }
210
+ this.sessionListeners.get(sessionId).add(callback);
211
+ }
212
+ /**
213
+ * Remove a session-level listener.
214
+ */
215
+ removeSessionListener(sessionId, callback) {
216
+ const listeners = this.sessionListeners.get(sessionId);
217
+ if (listeners) {
218
+ listeners.delete(callback);
219
+ if (listeners.size === 0) {
220
+ this.sessionListeners.delete(sessionId);
221
+ }
222
+ }
223
+ }
224
+ /**
225
+ * Get replay buffer for a run (all events emitted since start).
226
+ * Used by WS handler to send missed events on late connection.
227
+ * Optionally filter by afterSeq to only get events the client hasn't seen.
228
+ */
229
+ getEventBuffer(runId, afterSeq) {
230
+ const run = this.activeRuns.get(runId);
231
+ if (!run) {
232
+ return [];
233
+ }
234
+ if (afterSeq != null) {
235
+ return run.eventBuffer.filter((e) => e.seq != null && e.seq > afterSeq);
236
+ }
237
+ return [...run.eventBuffer];
238
+ }
239
+ /**
240
+ * List all active runs
241
+ */
242
+ listActive() {
243
+ return Array.from(this.activeRuns.values()).map((r) => ({
244
+ runId: r.runId,
245
+ task: r.task,
246
+ status: r.status,
247
+ startedAt: r.startedAt
248
+ }));
249
+ }
250
+ /**
251
+ * Request graceful stop of a running agent (and its child agents via propagated AbortSignal).
252
+ * Agent finishes its current tool call then exits at the next iteration boundary.
253
+ */
254
+ requestStop(runId) {
255
+ const run = this.activeRuns.get(runId);
256
+ if (!run || run.status !== "running" || !run.agent) {
257
+ return false;
258
+ }
259
+ run.agent.requestStop();
260
+ return true;
261
+ }
262
+ /**
263
+ * Clean up completed runs from memory (cache handles its own TTL)
264
+ */
265
+ cleanup() {
266
+ for (const [runId, run] of this.activeRuns) {
267
+ if (run.status === "completed" || run.status === "failed" || run.status === "stopped") {
268
+ this.activeRuns.delete(runId);
269
+ }
270
+ }
271
+ }
272
+ };
273
+ var GLOBAL_KEY = "__kb_agent_run_manager__";
274
+ if (!globalThis[GLOBAL_KEY]) {
275
+ globalThis[GLOBAL_KEY] = new RunManagerImpl();
276
+ }
277
+ var RunManager = globalThis[GLOBAL_KEY];
278
+
279
+ // src/rest/handlers/run-handler.ts
280
+ bootstrapAgentSDK();
281
+ var FOLLOW_UP_SCOPE_RE = /\b(глубже|подробнее|детал|слишком поверхност|deeper|more depth|details?)\b/i;
282
+ function isLikelyFollowUpScopeTask(task) {
283
+ return FOLLOW_UP_SCOPE_RE.test(task);
284
+ }
285
+ function pathExists(dir) {
286
+ try {
287
+ return fs.existsSync(dir);
288
+ } catch {
289
+ return false;
290
+ }
291
+ }
292
+ function scoreRepoFromToolPath(rawPath, scores) {
293
+ const p = rawPath.replace(/\\/g, "/");
294
+ const explicit = p.match(/(^|\/)(kb-labs-[^/]+)\//);
295
+ if (explicit?.[2]) {
296
+ scores.set(explicit[2], (scores.get(explicit[2]) ?? 0) + 5);
297
+ }
298
+ if (p.startsWith("packages/")) {
299
+ if (/packages\/agent[-/]/.test(p) || /agent-core|agent-tools|agent-cli|agent-task-runner/.test(p)) {
300
+ scores.set("kb-labs-agents", (scores.get("kb-labs-agents") ?? 0) + 4);
301
+ }
302
+ if (/packages\/mind[-/]/.test(p) || /mind-engine|mind-core/.test(p)) {
303
+ scores.set("kb-labs-mind", (scores.get("kb-labs-mind") ?? 0) + 4);
304
+ }
305
+ }
306
+ }
307
+ async function inferFollowUpWorkingDir(sessionManager, sessionId, baseWorkingDir) {
308
+ const events = await sessionManager.getSessionEvents(sessionId);
309
+ if (!events.length) {
310
+ return null;
311
+ }
312
+ const completedRuns = /* @__PURE__ */ new Set();
313
+ for (const event of events) {
314
+ if (event.type === "agent:end" && event.runId) {
315
+ completedRuns.add(event.runId);
316
+ }
317
+ }
318
+ const lastRunId = Array.from(completedRuns).at(-1);
319
+ if (!lastRunId) {
320
+ return null;
321
+ }
322
+ const repoScores = /* @__PURE__ */ new Map();
323
+ for (const event of events) {
324
+ if (event.runId !== lastRunId) {
325
+ continue;
326
+ }
327
+ if (event.type === "tool:start") {
328
+ const input = event.data?.input ?? {};
329
+ const p1 = input.path;
330
+ const p2 = input.directory;
331
+ if (typeof p1 === "string") {
332
+ scoreRepoFromToolPath(p1, repoScores);
333
+ }
334
+ if (typeof p2 === "string") {
335
+ scoreRepoFromToolPath(p2, repoScores);
336
+ }
337
+ }
338
+ if (event.type === "tool:end") {
339
+ const metadata = event.data?.metadata ?? {};
340
+ const p = metadata.path;
341
+ if (typeof p === "string") {
342
+ scoreRepoFromToolPath(p, repoScores);
343
+ }
344
+ }
345
+ }
346
+ const ranked = Array.from(repoScores.entries()).sort((a, b) => b[1] - a[1]);
347
+ const top = ranked[0];
348
+ if (!top || top[1] < 4) {
349
+ return null;
350
+ }
351
+ const inferredDir = path.join(baseWorkingDir, top[0]);
352
+ return pathExists(inferredDir) ? inferredDir : null;
353
+ }
354
+ var run_handler_default = defineHandler({
355
+ async execute(ctx, input) {
356
+ const body = input.body;
357
+ if (!body?.task) {
358
+ throw new Error("Task is required");
359
+ }
360
+ const analytics = useAnalytics();
361
+ const runId = RunManager.generateRunId();
362
+ const startTime = Date.now();
363
+ let sessionId = body.sessionId;
364
+ let workingDir = body.workingDir || ctx.cwd;
365
+ let sessionManager = new SessionManager(workingDir);
366
+ if (!sessionId) {
367
+ const session = await sessionManager.createSession({
368
+ mode: "execute",
369
+ task: body.task,
370
+ agentId: body.agentId ?? "orchestrator"
371
+ });
372
+ sessionId = session.id;
373
+ ctx.platform.logger.info(`[run-handler] Created new session ${sessionId}`);
374
+ } else {
375
+ const existingSession = await sessionManager.loadSession(sessionId);
376
+ let resolvedSession = existingSession;
377
+ if (!resolvedSession && pathExists(ctx.cwd)) {
378
+ const fallbackManager = new SessionManager(ctx.cwd);
379
+ const fallbackSession = await fallbackManager.loadSession(sessionId);
380
+ if (fallbackSession) {
381
+ sessionManager = fallbackManager;
382
+ resolvedSession = fallbackSession;
383
+ }
384
+ }
385
+ if (!resolvedSession) {
386
+ throw new Error(`Session not found: ${sessionId} (cwd=${ctx.cwd})`);
387
+ }
388
+ workingDir = body.workingDir || resolvedSession.workingDir || workingDir;
389
+ if (!body.workingDir && isLikelyFollowUpScopeTask(body.task)) {
390
+ const inferred = await inferFollowUpWorkingDir(sessionManager, sessionId, resolvedSession.workingDir || ctx.cwd);
391
+ if (inferred && inferred !== workingDir) {
392
+ ctx.platform.logger.info(`[run-handler] Follow-up scope inferred: ${workingDir} -> ${inferred}`);
393
+ workingDir = inferred;
394
+ }
395
+ }
396
+ sessionManager = new SessionManager(workingDir);
397
+ ctx.platform.logger.info(`[run-handler] Continuing session ${sessionId} (workingDir=${workingDir})`);
398
+ }
399
+ ctx.platform.logger.info(`[run-handler] Starting run ${runId} for task: ${body.task}`);
400
+ await sessionManager.createUserTurn(sessionId, body.task, runId);
401
+ await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_STARTED, {
402
+ runId,
403
+ sessionId,
404
+ taskLength: body.task.length,
405
+ tier: body.tier ?? "medium",
406
+ enableEscalation: body.enableEscalation ?? true,
407
+ responseMode: body.responseMode ?? "auto",
408
+ verbose: body.verbose ?? false
409
+ });
410
+ const sessionMemory = createSessionMemoryBridge(workingDir, sessionId);
411
+ const responseRequirementsSelector = createDefaultResponseRequirementsSelector();
412
+ const toolRegistry = createToolRegistry({
413
+ workingDir,
414
+ currentTask: body.task,
415
+ sessionId,
416
+ verbose: body.verbose,
417
+ cache: useCache(),
418
+ sessionMemory,
419
+ responseRequirementsResolver: async ({ task, kernel }) => responseRequirementsSelector.select({
420
+ state: kernel,
421
+ messages: [],
422
+ task: task ?? body.task
423
+ })
424
+ });
425
+ const agentsConfig = await useConfig();
426
+ const finalSessionId = sessionId;
427
+ const traceDir = path.join(workingDir, ".kb", "traces", "incremental");
428
+ const traceWriter = new IncrementalTraceWriter(runId, {}, traceDir);
429
+ const agent = new AgentSDK().register(createCoreToolPack(toolRegistry)).createRunner({
430
+ sessionId: finalSessionId,
431
+ workingDir,
432
+ maxIterations: 50,
433
+ temperature: 0.1,
434
+ tier: body.tier ?? "medium",
435
+ tokenBudget: agentsConfig?.tokenBudget,
436
+ onEvent: (event) => {
437
+ const seqEvent = RunManager.broadcast(runId, event);
438
+ void sessionManager.addEvent(finalSessionId, {
439
+ ...seqEvent,
440
+ sessionId: finalSessionId,
441
+ runId,
442
+ metadata: {
443
+ ...seqEvent.metadata,
444
+ sessionId: finalSessionId,
445
+ runId,
446
+ workingDir
447
+ }
448
+ });
449
+ }
450
+ });
451
+ const run = await RunManager.register(runId, body.task, agent, sessionManager, finalSessionId);
452
+ await RunManager.updateStatus(runId, "running");
453
+ void (async () => {
454
+ try {
455
+ const result = await agent.execute(body.task);
456
+ const durationMs = Date.now() - startTime;
457
+ const detailedTrace = traceWriter.getEntries();
458
+ await traceWriter.finalize?.();
459
+ if (detailedTrace.length > 0) {
460
+ await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);
461
+ }
462
+ if (result.fileChanges && result.fileChanges.length > 0) {
463
+ await sessionManager.attachFileChangesToTurn(finalSessionId, runId, result.fileChanges);
464
+ }
465
+ await RunManager.updateStatus(runId, result.success ? "completed" : "failed", {
466
+ completedAt: (/* @__PURE__ */ new Date()).toISOString(),
467
+ durationMs,
468
+ summary: result.summary,
469
+ error: result.error
470
+ });
471
+ await analytics?.track(
472
+ result.success ? AGENT_ANALYTICS_EVENTS.RUN_COMPLETED : AGENT_ANALYTICS_EVENTS.RUN_FAILED,
473
+ {
474
+ runId,
475
+ durationMs,
476
+ success: result.success,
477
+ summary: result.summary?.slice(0, 200)
478
+ }
479
+ );
480
+ ctx.platform.logger.info(`[run-handler] Run ${runId} completed: ${result.success}`);
481
+ } catch (error) {
482
+ const errorMsg = error instanceof Error ? error.message : String(error);
483
+ const durationMs = Date.now() - startTime;
484
+ const detailedTrace = traceWriter.getEntries();
485
+ await traceWriter.finalize?.();
486
+ if (detailedTrace.length > 0) {
487
+ await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);
488
+ }
489
+ await RunManager.updateStatus(runId, "failed", {
490
+ completedAt: (/* @__PURE__ */ new Date()).toISOString(),
491
+ durationMs,
492
+ error: errorMsg
493
+ });
494
+ await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_FAILED, {
495
+ runId,
496
+ durationMs,
497
+ error: errorMsg.slice(0, 200)
498
+ });
499
+ ctx.platform.logger.error(`[run-handler] Run ${runId} failed: ${errorMsg}`);
500
+ }
501
+ })();
502
+ const wsPath = AGENTS_WS_CHANNELS.SESSION_STREAM.replace(":sessionId", finalSessionId);
503
+ const eventsPath = `${AGENTS_WS_BASE_PATH}${wsPath}`;
504
+ return {
505
+ runId,
506
+ sessionId: finalSessionId,
507
+ eventsPath,
508
+ status: "started",
509
+ startedAt: run.startedAt
510
+ };
511
+ }
512
+ });
513
+
514
+ export { run_handler_default as default };
515
+ //# sourceMappingURL=run-handler.js.map
516
+ //# sourceMappingURL=run-handler.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../../../src/rest/run-manager.ts","../../../src/rest/handlers/run-handler.ts"],"names":["run","useCache"],"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;;;ACjX5E,iBAAA,EAAkB;AAalB,IAAM,kBAAA,GAAqB,6EAAA;AAE3B,SAAS,0BAA0B,IAAA,EAAuB;AACxD,EAAA,OAAO,kBAAA,CAAmB,KAAK,IAAI,CAAA;AACrC;AAEA,SAAS,WAAW,GAAA,EAAsB;AACxC,EAAA,IAAI;AACF,IAAA,OAAO,EAAA,CAAG,WAAW,GAAG,CAAA;AAAA,EAC1B,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,KAAA;AAAA,EACT;AACF;AAEA,SAAS,qBAAA,CAAsB,SAAiB,MAAA,EAAmC;AACjF,EAAA,MAAM,CAAA,GAAI,OAAA,CAAQ,OAAA,CAAQ,KAAA,EAAO,GAAG,CAAA;AAGpC,EAAA,MAAM,QAAA,GAAW,CAAA,CAAE,KAAA,CAAM,yBAAyB,CAAA;AAClD,EAAA,IAAI,QAAA,GAAW,CAAC,CAAA,EAAG;AACjB,IAAA,MAAA,CAAO,GAAA,CAAI,QAAA,CAAS,CAAC,CAAA,EAAA,CAAI,MAAA,CAAO,GAAA,CAAI,QAAA,CAAS,CAAC,CAAC,CAAA,IAAK,CAAA,IAAK,CAAC,CAAA;AAAA,EAC5D;AAGA,EAAA,IAAI,CAAA,CAAE,UAAA,CAAW,WAAW,CAAA,EAAG;AAC7B,IAAA,IAAI,sBAAsB,IAAA,CAAK,CAAC,KAAK,oDAAA,CAAqD,IAAA,CAAK,CAAC,CAAA,EAAG;AACjG,MAAA,MAAA,CAAO,IAAI,gBAAA,EAAA,CAAmB,MAAA,CAAO,IAAI,gBAAgB,CAAA,IAAK,KAAK,CAAC,CAAA;AAAA,IACtE;AACA,IAAA,IAAI,qBAAqB,IAAA,CAAK,CAAC,KAAK,uBAAA,CAAwB,IAAA,CAAK,CAAC,CAAA,EAAG;AACnE,MAAA,MAAA,CAAO,IAAI,cAAA,EAAA,CAAiB,MAAA,CAAO,IAAI,cAAc,CAAA,IAAK,KAAK,CAAC,CAAA;AAAA,IAClE;AAAA,EACF;AACF;AAEA,eAAe,uBAAA,CACb,cAAA,EACA,SAAA,EACA,cAAA,EACwB;AACxB,EAAA,MAAM,MAAA,GAAS,MAAM,cAAA,CAAe,gBAAA,CAAiB,SAAS,CAAA;AAC9D,EAAA,IAAI,CAAC,OAAO,MAAA,EAAQ;AAClB,IAAA,OAAO,IAAA;AAAA,EACT;AAGA,EAAA,MAAM,aAAA,uBAAoB,GAAA,EAAY;AACtC,EAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,KAAA,EAAO;AAC7C,MAAA,aAAA,CAAc,GAAA,CAAI,MAAM,KAAK,CAAA;AAAA,IAC/B;AAAA,EACF;AACA,EAAA,MAAM,YAAY,KAAA,CAAM,IAAA,CAAK,aAAa,CAAA,CAAE,GAAG,EAAE,CAAA;AACjD,EAAA,IAAI,CAAC,SAAA,EAAW;AACd,IAAA,OAAO,IAAA;AAAA,EACT;AAEA,EAAA,MAAM,UAAA,uBAAiB,GAAA,EAAoB;AAC3C,EAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,IAAA,IAAI,KAAA,CAAM,UAAU,SAAA,EAAW;AAC7B,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,KAAA,CAAM,SAAS,YAAA,EAAc;AAC/B,MAAA,MAAM,KAAA,GAAS,KAAA,CAAM,IAAA,EAAM,KAAA,IAAiD,EAAC;AAC7E,MAAA,MAAM,KAAK,KAAA,CAAM,IAAA;AACjB,MAAA,MAAM,KAAK,KAAA,CAAM,SAAA;AACjB,MAAA,IAAI,OAAO,OAAO,QAAA,EAAU;AAAC,QAAA,qBAAA,CAAsB,IAAI,UAAU,CAAA;AAAA,MAAE;AACnE,MAAA,IAAI,OAAO,OAAO,QAAA,EAAU;AAAC,QAAA,qBAAA,CAAsB,IAAI,UAAU,CAAA;AAAA,MAAE;AAAA,IACrE;AAEA,IAAA,IAAI,KAAA,CAAM,SAAS,UAAA,EAAY;AAC7B,MAAA,MAAM,QAAA,GAAY,KAAA,CAAM,IAAA,EAAM,QAAA,IAAoD,EAAC;AACnF,MAAA,MAAM,IAAI,QAAA,CAAS,IAAA;AACnB,MAAA,IAAI,OAAO,MAAM,QAAA,EAAU;AAAC,QAAA,qBAAA,CAAsB,GAAG,UAAU,CAAA;AAAA,MAAE;AAAA,IACnE;AAAA,EACF;AAEA,EAAA,MAAM,SAAS,KAAA,CAAM,IAAA,CAAK,UAAA,CAAW,OAAA,EAAS,CAAA,CAAE,IAAA,CAAK,CAAC,CAAA,EAAG,MAAM,CAAA,CAAE,CAAC,CAAA,GAAI,CAAA,CAAE,CAAC,CAAC,CAAA;AAC1E,EAAA,MAAM,GAAA,GAAM,OAAO,CAAC,CAAA;AACpB,EAAA,IAAI,CAAC,GAAA,IAAO,GAAA,CAAI,CAAC,IAAI,CAAA,EAAG;AACtB,IAAA,OAAO,IAAA;AAAA,EACT;AAEA,EAAA,MAAM,cAAc,IAAA,CAAK,IAAA,CAAK,cAAA,EAAgB,GAAA,CAAI,CAAC,CAAC,CAAA;AACpD,EAAA,OAAO,UAAA,CAAW,WAAW,CAAA,GAAI,WAAA,GAAc,IAAA;AACjD;AAEA,IAAO,sBAAQ,aAAA,CAAc;AAAA,EAC3B,MAAM,OAAA,CACJ,GAAA,EACA,KAAA,EACsB;AACtB,IAAA,MAAM,OAAO,KAAA,CAAM,IAAA;AAEnB,IAAA,IAAI,CAAC,MAAM,IAAA,EAAM;AACf,MAAA,MAAM,IAAI,MAAM,kBAAkB,CAAA;AAAA,IACpC;AAEA,IAAA,MAAM,YAAY,YAAA,EAAa;AAC/B,IAAA,MAAM,KAAA,GAAQ,WAAW,aAAA,EAAc;AACvC,IAAA,MAAM,SAAA,GAAY,KAAK,GAAA,EAAI;AAG3B,IAAA,IAAI,YAAY,IAAA,CAAK,SAAA;AACrB,IAAA,IAAI,UAAA,GAAa,IAAA,CAAK,UAAA,IAAc,GAAA,CAAI,GAAA;AACxC,IAAA,IAAI,cAAA,GAAiB,IAAI,cAAA,CAAe,UAAU,CAAA;AAClD,IAAA,IAAI,CAAC,SAAA,EAAW;AAEd,MAAA,MAAM,OAAA,GAAU,MAAM,cAAA,CAAe,aAAA,CAAc;AAAA,QACjD,IAAA,EAAM,SAAA;AAAA,QACN,MAAM,IAAA,CAAK,IAAA;AAAA,QACX,OAAA,EAAS,KAAK,OAAA,IAAW;AAAA,OAC1B,CAAA;AACD,MAAA,SAAA,GAAY,OAAA,CAAQ,EAAA;AACpB,MAAA,GAAA,CAAI,QAAA,CAAS,MAAA,CAAO,IAAA,CAAK,CAAA,kCAAA,EAAqC,SAAS,CAAA,CAAE,CAAA;AAAA,IAC3E,CAAA,MAAO;AAEL,MAAA,MAAM,eAAA,GAAkB,MAAM,cAAA,CAAe,WAAA,CAAY,SAAS,CAAA;AAClE,MAAA,IAAI,eAAA,GAAkB,eAAA;AAGtB,MAAA,IAAI,CAAC,eAAA,IAAmB,UAAA,CAAW,GAAA,CAAI,GAAG,CAAA,EAAG;AAC3C,QAAA,MAAM,eAAA,GAAkB,IAAI,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AAClD,QAAA,MAAM,eAAA,GAAkB,MAAM,eAAA,CAAgB,WAAA,CAAY,SAAS,CAAA;AACnE,QAAA,IAAI,eAAA,EAAiB;AACnB,UAAA,cAAA,GAAiB,eAAA;AACjB,UAAA,eAAA,GAAkB,eAAA;AAAA,QACpB;AAAA,MACF;AAEA,MAAA,IAAI,CAAC,eAAA,EAAiB;AACpB,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,mBAAA,EAAsB,SAAS,CAAA,MAAA,EAAS,GAAA,CAAI,GAAG,CAAA,CAAA,CAAG,CAAA;AAAA,MACpE;AAGA,MAAA,UAAA,GAAa,IAAA,CAAK,UAAA,IAAc,eAAA,CAAgB,UAAA,IAAc,UAAA;AAG9D,MAAA,IAAI,CAAC,IAAA,CAAK,UAAA,IAAc,yBAAA,CAA0B,IAAA,CAAK,IAAI,CAAA,EAAG;AAC5D,QAAA,MAAM,QAAA,GAAW,MAAM,uBAAA,CAAwB,cAAA,EAAgB,WAAW,eAAA,CAAgB,UAAA,IAAc,IAAI,GAAG,CAAA;AAC/G,QAAA,IAAI,QAAA,IAAY,aAAa,UAAA,EAAY;AACvC,UAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA,CAAK,2CAA2C,UAAU,CAAA,IAAA,EAAO,QAAQ,CAAA,CAAE,CAAA;AAC/F,UAAA,UAAA,GAAa,QAAA;AAAA,QACf;AAAA,MACF;AAGA,MAAA,cAAA,GAAiB,IAAI,eAAe,UAAU,CAAA;AAC9C,MAAA,GAAA,CAAI,SAAS,MAAA,CAAO,IAAA,CAAK,oCAAoC,SAAS,CAAA,aAAA,EAAgB,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACrG;AAEA,IAAA,GAAA,CAAI,QAAA,CAAS,OAAO,IAAA,CAAK,CAAA,2BAAA,EAA8B,KAAK,CAAA,WAAA,EAAc,IAAA,CAAK,IAAI,CAAA,CAAE,CAAA;AAGrF,IAAA,MAAM,cAAA,CAAe,cAAA,CAAe,SAAA,EAAW,IAAA,CAAK,MAAM,KAAK,CAAA;AAG/D,IAAA,MAAM,SAAA,EAAW,KAAA,CAAM,sBAAA,CAAuB,WAAA,EAAa;AAAA,MACzD,KAAA;AAAA,MACA,SAAA;AAAA,MACA,UAAA,EAAY,KAAK,IAAA,CAAK,MAAA;AAAA,MACtB,IAAA,EAAM,KAAK,IAAA,IAAQ,QAAA;AAAA,MACnB,gBAAA,EAAkB,KAAK,gBAAA,IAAoB,IAAA;AAAA,MAC3C,YAAA,EAAc,KAAK,YAAA,IAAgB,MAAA;AAAA,MACnC,OAAA,EAAS,KAAK,OAAA,IAAW;AAAA,KAC1B,CAAA;AAGD,IAAA,MAAM,aAAA,GAAgB,yBAAA,CAA0B,UAAA,EAAY,SAAS,CAAA;AACrE,IAAA,MAAM,+BAA+B,yCAAA,EAA0C;AAC/E,IAAA,MAAM,eAAe,kBAAA,CAAmB;AAAA,MACtC,UAAA;AAAA,MACA,aAAa,IAAA,CAAK,IAAA;AAAA,MAClB,SAAA;AAAA,MACA,SAAS,IAAA,CAAK,OAAA;AAAA,MACd,OAAOC,QAAAA,EAAS;AAAA,MAChB,aAAA;AAAA,MACA,8BAA8B,OAAO,EAAE,MAAM,MAAA,EAAO,KAKlD,6BAA6B,MAAA,CAAO;AAAA,QAClC,KAAA,EAAO,MAAA;AAAA,QACP,UAAU,EAAC;AAAA,QACX,IAAA,EAAM,QAAQ,IAAA,CAAK;AAAA,OACpB;AAAA,KACJ,CAAA;AACD,IAAA,MAAM,YAAA,GAAe,MAAM,SAAA,EAA8B;AAEzD,IAAA,MAAM,cAAA,GAAiB,SAAA;AACvB,IAAA,MAAM,WAAW,IAAA,CAAK,IAAA,CAAK,UAAA,EAAY,KAAA,EAAO,UAAU,aAAa,CAAA;AACrE,IAAA,MAAM,cAAc,IAAI,sBAAA,CAAuB,KAAA,EAAO,IAAI,QAAQ,CAAA;AAGlE,IAAA,MAAM,KAAA,GAAQ,IAAI,QAAA,EAAS,CACxB,SAAS,kBAAA,CAAmB,YAAY,CAAC,CAAA,CACzC,YAAA,CAAa;AAAA,MACZ,SAAA,EAAW,cAAA;AAAA,MACX,UAAA;AAAA,MACA,aAAA,EAAe,EAAA;AAAA,MACf,WAAA,EAAa,GAAA;AAAA,MACb,IAAA,EAAM,KAAK,IAAA,IAAQ,QAAA;AAAA,MACnB,aAAa,YAAA,EAAc,WAAA;AAAA,MAC3B,OAAA,EAAS,CAAC,KAAA,KAAU;AAElB,QAAA,MAAM,QAAA,GAAW,UAAA,CAAW,SAAA,CAAU,KAAA,EAAO,KAAK,CAAA;AAGlD,QAAA,KAAK,cAAA,CAAe,SAAS,cAAA,EAAgB;AAAA,UAC3C,GAAG,QAAA;AAAA,UACH,SAAA,EAAW,cAAA;AAAA,UACX,KAAA;AAAA,UACA,QAAA,EAAU;AAAA,YACR,GAAG,QAAA,CAAS,QAAA;AAAA,YACZ,SAAA,EAAW,cAAA;AAAA,YACX,KAAA;AAAA,YACA;AAAA;AACF,SACD,CAAA;AAAA,MACH;AAAA,KACD,CAAA;AAGH,IAAA,MAAM,GAAA,GAAM,MAAM,UAAA,CAAW,QAAA,CAAS,OAAO,IAAA,CAAK,IAAA,EAAM,KAAA,EAAO,cAAA,EAAgB,cAAc,CAAA;AAC7F,IAAA,MAAM,UAAA,CAAW,YAAA,CAAa,KAAA,EAAO,SAAS,CAAA;AAG9C,IAAA,KAAA,CAAM,YAAY;AAChB,MAAA,IAAI;AACF,QAAA,MAAM,MAAA,GAAS,MAAM,KAAA,CAAM,OAAA,CAAQ,KAAK,IAAI,CAAA;AAC5C,QAAA,MAAM,UAAA,GAAa,IAAA,CAAK,GAAA,EAAI,GAAI,SAAA;AAChC,QAAA,MAAM,aAAA,GAAgB,YAAY,UAAA,EAAW;AAC7C,QAAA,MAAM,YAAY,QAAA,IAAW;AAE7B,QAAA,IAAI,aAAA,CAAc,SAAS,CAAA,EAAG;AAC5B,UAAA,MAAM,cAAA,CAAe,mBAAA,CAAoB,cAAA,EAAgB,KAAA,EAAO,aAAa,CAAA;AAAA,QAC/E;AAIA,QAAA,IAAI,MAAA,CAAO,WAAA,IAAe,MAAA,CAAO,WAAA,CAAY,SAAS,CAAA,EAAG;AACvD,UAAA,MAAM,cAAA,CAAe,uBAAA,CAAwB,cAAA,EAAgB,KAAA,EAAO,OAAO,WAAW,CAAA;AAAA,QACxF;AAEA,QAAA,MAAM,WAAW,YAAA,CAAa,KAAA,EAAO,MAAA,CAAO,OAAA,GAAU,cAAc,QAAA,EAAU;AAAA,UAC5E,WAAA,EAAA,iBAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,UACpC,UAAA;AAAA,UACA,SAAS,MAAA,CAAO,OAAA;AAAA,UAChB,OAAO,MAAA,CAAO;AAAA,SACf,CAAA;AAGD,QAAA,MAAM,SAAA,EAAW,KAAA;AAAA,UACf,MAAA,CAAO,OAAA,GAAU,sBAAA,CAAuB,aAAA,GAAgB,sBAAA,CAAuB,UAAA;AAAA,UAC/E;AAAA,YACE,KAAA;AAAA,YACA,UAAA;AAAA,YACA,SAAS,MAAA,CAAO,OAAA;AAAA,YAChB,OAAA,EAAS,MAAA,CAAO,OAAA,EAAS,KAAA,CAAM,GAAG,GAAG;AAAA;AACvC,SACF;AAEA,QAAA,GAAA,CAAI,QAAA,CAAS,OAAO,IAAA,CAAK,CAAA,kBAAA,EAAqB,KAAK,CAAA,YAAA,EAAe,MAAA,CAAO,OAAO,CAAA,CAAE,CAAA;AAAA,MACpF,SAAS,KAAA,EAAO;AACd,QAAA,MAAM,WAAW,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU,OAAO,KAAK,CAAA;AACtE,QAAA,MAAM,UAAA,GAAa,IAAA,CAAK,GAAA,EAAI,GAAI,SAAA;AAChC,QAAA,MAAM,aAAA,GAAgB,YAAY,UAAA,EAAW;AAC7C,QAAA,MAAM,YAAY,QAAA,IAAW;AAC7B,QAAA,IAAI,aAAA,CAAc,SAAS,CAAA,EAAG;AAC5B,UAAA,MAAM,cAAA,CAAe,mBAAA,CAAoB,cAAA,EAAgB,KAAA,EAAO,aAAa,CAAA;AAAA,QAC/E;AAEA,QAAA,MAAM,UAAA,CAAW,YAAA,CAAa,KAAA,EAAO,QAAA,EAAU;AAAA,UAC7C,WAAA,EAAA,iBAAa,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,UACpC,UAAA;AAAA,UACA,KAAA,EAAO;AAAA,SACR,CAAA;AAGD,QAAA,MAAM,SAAA,EAAW,KAAA,CAAM,sBAAA,CAAuB,UAAA,EAAY;AAAA,UACxD,KAAA;AAAA,UACA,UAAA;AAAA,UACA,KAAA,EAAO,QAAA,CAAS,KAAA,CAAM,CAAA,EAAG,GAAG;AAAA,SAC7B,CAAA;AAED,QAAA,GAAA,CAAI,SAAS,MAAA,CAAO,KAAA,CAAM,qBAAqB,KAAK,CAAA,SAAA,EAAY,QAAQ,CAAA,CAAE,CAAA;AAAA,MAC5E;AAAA,IACF,CAAA,GAAG;AAGH,IAAA,MAAM,MAAA,GAAS,kBAAA,CAAmB,cAAA,CAAe,OAAA,CAAQ,cAAc,cAAc,CAAA;AACrF,IAAA,MAAM,UAAA,GAAa,CAAA,EAAG,mBAAmB,CAAA,EAAG,MAAM,CAAA,CAAA;AAElD,IAAA,OAAO;AAAA,MACL,KAAA;AAAA,MACA,SAAA,EAAW,cAAA;AAAA,MACX,UAAA;AAAA,MACA,MAAA,EAAQ,SAAA;AAAA,MACR,WAAW,GAAA,CAAI;AAAA,KACjB;AAAA,EACF;AACF,CAAC","file":"run-handler.js","sourcesContent":["/**\n * Run Manager - tracks active agent runs for REST/WS API\n *\n * Uses platform cache for persistence and in-memory Map for active agents.\n * Cache stores serializable run state, Map stores live agent references.\n * Uses platform eventBus for cross-process event broadcasting.\n */\n\nimport type { IAgentRunner } from '@kb-labs/agent-sdk';\nimport type { SessionManager } from '@kb-labs/agent-core';\nimport type { AgentEvent, AgentEventCallback } from '@kb-labs/agent-contracts';\nimport { useCache, usePlatform } from '@kb-labs/sdk';\n\nconst CACHE_PREFIX = 'agent:run:';\nconst EVENT_TOPIC_PREFIX = 'agent:events:';\nconst CACHE_TTL = 3600000; // 1 hour\n\n/**\n * Run status\n */\nexport type RunStatus = 'pending' | 'running' | 'completed' | 'failed' | 'stopped';\n\n/**\n * Serializable run state (stored in cache)\n */\nexport interface RunState {\n runId: string;\n task: string;\n status: RunStatus;\n startedAt: string;\n completedAt?: string;\n durationMs?: number;\n summary?: string;\n error?: string;\n}\n\n/**\n * Active run with live agent (in-memory only)\n */\nexport interface ActiveRun extends RunState {\n agent?: IAgentRunner;\n sessionManager: SessionManager;\n listeners: Set<AgentEventCallback>;\n /** Monotonic sequence counter for event ordering */\n lastSeq: number;\n /** Replay buffer: all events emitted since run start (for late WS connections) */\n eventBuffer: AgentEvent[];\n}\n\n/**\n * Run Manager implementation\n */\nclass RunManagerImpl {\n /** Live agents and listeners (not cacheable) */\n private activeRuns: Map<string, ActiveRun> = new Map();\n\n /** Session-level listeners: sessionId → Set<callback> — receive events from ALL runs in session */\n private sessionListeners: Map<string, Set<AgentEventCallback>> = new Map();\n\n /** sessionId per runId — set when run is registered */\n private runSessionMap: Map<string, string> = new Map();\n\n /**\n * Generate unique run ID\n */\n generateRunId(): string {\n return `run-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`;\n }\n\n /**\n * Register a new run\n */\n async register(\n runId: string,\n task: string,\n agent: IAgentRunner | undefined,\n sessionManager: SessionManager,\n sessionId?: string\n ): Promise<ActiveRun> {\n const now = new Date().toISOString();\n\n const run: ActiveRun = {\n runId,\n task,\n status: 'pending',\n agent,\n sessionManager,\n startedAt: now,\n listeners: new Set(),\n lastSeq: 0,\n eventBuffer: [],\n };\n\n // Store in memory (for live agent)\n this.activeRuns.set(runId, run);\n\n // Track sessionId → runId mapping for session-level listeners\n if (sessionId) {\n this.runSessionMap.set(runId, sessionId);\n }\n\n // Store serializable state in cache\n await this.saveToCache(run);\n\n return run;\n }\n\n /**\n * Get run by ID (from memory first, then cache for state)\n */\n get(runId: string): ActiveRun | undefined {\n return this.activeRuns.get(runId);\n }\n\n /**\n * Check if run exists (in memory or cache)\n */\n async exists(runId: string): Promise<boolean> {\n // Check memory first\n if (this.activeRuns.has(runId)) {\n return true;\n }\n // Fallback to cache\n const state = await this.getState(runId);\n return state !== null;\n }\n\n /**\n * Get run state from cache (for completed runs or cross-process access)\n */\n async getState(runId: string): Promise<RunState | null> {\n const cache = useCache();\n if (!cache) {return null;}\n\n return cache.get<RunState>(`${CACHE_PREFIX}${runId}`);\n }\n\n /**\n * Update run status\n */\n async updateStatus(runId: string, status: RunStatus, extra?: Partial<RunState>): Promise<void> {\n const run = this.activeRuns.get(runId);\n if (run) {\n run.status = status;\n if (extra) {\n Object.assign(run, extra);\n }\n await this.saveToCache(run);\n }\n }\n\n /**\n * Save run state to cache\n */\n private async saveToCache(run: ActiveRun): Promise<void> {\n const cache = useCache();\n if (!cache) {return;}\n\n const state: RunState = {\n runId: run.runId,\n task: run.task,\n status: run.status,\n startedAt: run.startedAt,\n completedAt: run.completedAt,\n durationMs: run.durationMs,\n summary: run.summary,\n error: run.error,\n };\n\n await cache.set(`${CACHE_PREFIX}${run.runId}`, state, CACHE_TTL);\n }\n\n /** Track subscriptions for cleanup */\n private subscriptions: Map<string, Map<AgentEventCallback, () => void>> = new Map();\n\n /**\n * Add event listener to run (uses eventBus for cross-process)\n */\n addListener(runId: string, callback: AgentEventCallback): boolean {\n const platform = usePlatform();\n if (!platform?.eventBus) {\n // Fallback to in-memory only\n const run = this.activeRuns.get(runId);\n if (run) {\n run.listeners.add(callback);\n return true;\n }\n return false;\n }\n\n // Subscribe to eventBus topic for this run\n const topic = `${EVENT_TOPIC_PREFIX}${runId}`;\n const unsubscribe = platform.eventBus.subscribe<AgentEvent>(topic, async (event) => {\n try {\n callback(event);\n } catch {\n // Ignore callback errors\n }\n });\n\n // Track subscription for cleanup\n if (!this.subscriptions.has(runId)) {\n this.subscriptions.set(runId, new Map());\n }\n this.subscriptions.get(runId)!.set(callback, unsubscribe);\n\n // Also add to in-memory if run exists locally\n const run = this.activeRuns.get(runId);\n if (run) {\n run.listeners.add(callback);\n }\n\n return true;\n }\n\n /**\n * Remove event listener from run\n */\n removeListener(runId: string, callback: AgentEventCallback): void {\n // Unsubscribe from eventBus\n const runSubs = this.subscriptions.get(runId);\n if (runSubs) {\n const unsubscribe = runSubs.get(callback);\n if (unsubscribe) {\n unsubscribe();\n runSubs.delete(callback);\n }\n if (runSubs.size === 0) {\n this.subscriptions.delete(runId);\n }\n }\n\n // Also remove from in-memory\n const run = this.activeRuns.get(runId);\n if (run) {\n run.listeners.delete(callback);\n }\n }\n\n /**\n * Broadcast event to all listeners of a run (uses eventBus for cross-process)\n * Assigns monotonic sequence number for reliable ordering\n * @returns The event with seq assigned (for persistence)\n */\n broadcast(runId: string, event: AgentEvent): AgentEvent {\n const platform = usePlatform();\n const run = this.activeRuns.get(runId);\n\n // Assign sequence number for ordering\n let seqEvent = event;\n if (run) {\n run.lastSeq++;\n seqEvent = { ...event, seq: run.lastSeq, runId };\n\n // Store in replay buffer (for late WS connections)\n run.eventBuffer.push(seqEvent);\n }\n\n // Publish to eventBus for cross-process delivery\n if (platform?.eventBus) {\n const topic = `${EVENT_TOPIC_PREFIX}${runId}`;\n platform.eventBus.publish(topic, seqEvent).catch(() => {\n // Ignore publish errors\n });\n }\n\n // Also notify local in-memory listeners (for same-process)\n if (run) {\n for (const listener of run.listeners) {\n try {\n listener(seqEvent);\n } catch {\n // Ignore listener errors\n }\n }\n }\n\n // Notify session-level listeners (persistent connections)\n const sessionId = this.runSessionMap.get(runId);\n if (sessionId) {\n const sListeners = this.sessionListeners.get(sessionId);\n if (sListeners) {\n for (const listener of sListeners) {\n try {\n listener(seqEvent);\n } catch {\n // Ignore listener errors\n }\n }\n }\n }\n\n return seqEvent;\n }\n\n /**\n * Add a session-level listener — receives events from ALL runs in this session.\n */\n addSessionListener(sessionId: string, callback: AgentEventCallback): void {\n if (!this.sessionListeners.has(sessionId)) {\n this.sessionListeners.set(sessionId, new Set());\n }\n this.sessionListeners.get(sessionId)!.add(callback);\n }\n\n /**\n * Remove a session-level listener.\n */\n removeSessionListener(sessionId: string, callback: AgentEventCallback): void {\n const listeners = this.sessionListeners.get(sessionId);\n if (listeners) {\n listeners.delete(callback);\n if (listeners.size === 0) {\n this.sessionListeners.delete(sessionId);\n }\n }\n }\n\n /**\n * Get replay buffer for a run (all events emitted since start).\n * Used by WS handler to send missed events on late connection.\n * Optionally filter by afterSeq to only get events the client hasn't seen.\n */\n getEventBuffer(runId: string, afterSeq?: number): AgentEvent[] {\n const run = this.activeRuns.get(runId);\n if (!run) {return [];}\n\n if (afterSeq != null) {\n return run.eventBuffer.filter((e) => e.seq != null && e.seq > afterSeq);\n }\n return [...run.eventBuffer];\n }\n\n /**\n * List all active runs\n */\n listActive(): Array<{ runId: string; task: string; status: RunStatus; startedAt: string }> {\n return Array.from(this.activeRuns.values()).map(r => ({\n runId: r.runId,\n task: r.task,\n status: r.status,\n startedAt: r.startedAt,\n }));\n }\n\n /**\n * Request graceful stop of a running agent (and its child agents via propagated AbortSignal).\n * Agent finishes its current tool call then exits at the next iteration boundary.\n */\n requestStop(runId: string): boolean {\n const run = this.activeRuns.get(runId);\n if (!run || run.status !== 'running' || !run.agent) {\n return false;\n }\n run.agent.requestStop();\n return true;\n }\n\n /**\n * Clean up completed runs from memory (cache handles its own TTL)\n */\n cleanup(): void {\n for (const [runId, run] of this.activeRuns) {\n if (run.status === 'completed' || run.status === 'failed' || run.status === 'stopped') {\n // Remove from memory but keep in cache\n this.activeRuns.delete(runId);\n }\n }\n }\n}\n\n/**\n * Singleton instance — stored on globalThis so all bundled modules share one instance.\n * When tsup compiles multiple entry points, each gets its own module scope,\n * so a plain `export const RunManager = new RunManagerImpl()` creates separate instances.\n * Using globalThis ensures run-handler.js and session-stream-handler.js share one RunManager.\n */\nconst GLOBAL_KEY = '__kb_agent_run_manager__';\nif (!(globalThis as Record<string, unknown>)[GLOBAL_KEY]) {\n (globalThis as Record<string, unknown>)[GLOBAL_KEY] = new RunManagerImpl();\n}\nexport const RunManager = (globalThis as Record<string, unknown>)[GLOBAL_KEY] as RunManagerImpl;\n","/**\n * POST /run handler\n *\n * Starts a new agent run via Orchestrator\n */\n\nimport { defineHandler, useAnalytics, useCache, useConfig, type RestInput, type PluginContextV3 } from '@kb-labs/sdk';\nimport { SessionManager, createCoreToolPack, bootstrapAgentSDK, createSessionMemoryBridge } from '@kb-labs/agent-core';\nimport { createDefaultResponseRequirementsSelector } from '@kb-labs/agent-runtime';\nimport { IncrementalTraceWriter } from '@kb-labs/agent-tracing';\n\n// Register SDKAgentRunner as the RunnerFactory (idempotent — runs once per process)\nbootstrapAgentSDK();\nimport { AgentSDK } from '@kb-labs/agent-sdk';\nimport { createToolRegistry } from '@kb-labs/agent-tools';\nimport type { RunRequest, RunResponse, AgentsPluginConfig, KernelState } from '@kb-labs/agent-contracts';\nimport path from 'node:path';\nimport fs from 'node:fs';\nimport {\n AGENTS_WS_BASE_PATH,\n AGENTS_WS_CHANNELS,\n AGENT_ANALYTICS_EVENTS,\n} from '@kb-labs/agent-contracts';\nimport { RunManager } from '../run-manager.js';\n\nconst FOLLOW_UP_SCOPE_RE = /\\b(глубже|подробнее|детал|слишком поверхност|deeper|more depth|details?)\\b/i;\n\nfunction isLikelyFollowUpScopeTask(task: string): boolean {\n return FOLLOW_UP_SCOPE_RE.test(task);\n}\n\nfunction pathExists(dir: string): boolean {\n try {\n return fs.existsSync(dir);\n } catch {\n return false;\n }\n}\n\nfunction scoreRepoFromToolPath(rawPath: string, scores: Map<string, number>): void {\n const p = rawPath.replace(/\\\\/g, '/');\n\n // Explicit repo prefix: kb-labs-xxx/...\n const explicit = p.match(/(^|\\/)(kb-labs-[^/]+)\\//);\n if (explicit?.[2]) {\n scores.set(explicit[2], (scores.get(explicit[2]) ?? 0) + 5);\n }\n\n // Heuristic by package naming patterns\n if (p.startsWith('packages/')) {\n if (/packages\\/agent[-/]/.test(p) || /agent-core|agent-tools|agent-cli|agent-task-runner/.test(p)) {\n scores.set('kb-labs-agents', (scores.get('kb-labs-agents') ?? 0) + 4);\n }\n if (/packages\\/mind[-/]/.test(p) || /mind-engine|mind-core/.test(p)) {\n scores.set('kb-labs-mind', (scores.get('kb-labs-mind') ?? 0) + 4);\n }\n }\n}\n\nasync function inferFollowUpWorkingDir(\n sessionManager: SessionManager,\n sessionId: string,\n baseWorkingDir: string,\n): Promise<string | null> {\n const events = await sessionManager.getSessionEvents(sessionId);\n if (!events.length) {\n return null;\n }\n\n // Find the latest completed run in this session\n const completedRuns = new Set<string>();\n for (const event of events) {\n if (event.type === 'agent:end' && event.runId) {\n completedRuns.add(event.runId);\n }\n }\n const lastRunId = Array.from(completedRuns).at(-1);\n if (!lastRunId) {\n return null;\n }\n\n const repoScores = new Map<string, number>();\n for (const event of events) {\n if (event.runId !== lastRunId) {\n continue;\n }\n\n if (event.type === 'tool:start') {\n const input = (event.data?.input as Record<string, unknown> | undefined) ?? {};\n const p1 = input.path;\n const p2 = input.directory;\n if (typeof p1 === 'string') {scoreRepoFromToolPath(p1, repoScores);}\n if (typeof p2 === 'string') {scoreRepoFromToolPath(p2, repoScores);}\n }\n\n if (event.type === 'tool:end') {\n const metadata = (event.data?.metadata as Record<string, unknown> | undefined) ?? {};\n const p = metadata.path;\n if (typeof p === 'string') {scoreRepoFromToolPath(p, repoScores);}\n }\n }\n\n const ranked = Array.from(repoScores.entries()).sort((a, b) => b[1] - a[1]);\n const top = ranked[0];\n if (!top || top[1] < 4) {\n return null;\n }\n\n const inferredDir = path.join(baseWorkingDir, top[0]);\n return pathExists(inferredDir) ? inferredDir : null;\n}\n\nexport default defineHandler({\n async execute(\n ctx: PluginContextV3,\n input: RestInput<RunRequest>\n ): Promise<RunResponse> {\n const body = input.body as RunRequest | undefined;\n\n if (!body?.task) {\n throw new Error('Task is required');\n }\n\n const analytics = useAnalytics();\n const runId = RunManager.generateRunId();\n const startTime = Date.now();\n\n // Get or create session\n let sessionId = body.sessionId;\n let workingDir = body.workingDir || ctx.cwd;\n let sessionManager = new SessionManager(workingDir);\n if (!sessionId) {\n // Create new session\n const session = await sessionManager.createSession({\n mode: 'execute',\n task: body.task,\n agentId: body.agentId ?? 'orchestrator',\n });\n sessionId = session.id;\n ctx.platform.logger.info(`[run-handler] Created new session ${sessionId}`);\n } else {\n // Verify session exists (from current manager first)\n const existingSession = await sessionManager.loadSession(sessionId);\n let resolvedSession = existingSession;\n\n // Fallback: session may belong to a different root than current ctx.cwd\n if (!resolvedSession && pathExists(ctx.cwd)) {\n const fallbackManager = new SessionManager(ctx.cwd);\n const fallbackSession = await fallbackManager.loadSession(sessionId);\n if (fallbackSession) {\n sessionManager = fallbackManager;\n resolvedSession = fallbackSession;\n }\n }\n\n if (!resolvedSession) {\n throw new Error(`Session not found: ${sessionId} (cwd=${ctx.cwd})`);\n }\n\n // Session workingDir is primary source of truth for follow-up runs\n workingDir = body.workingDir || resolvedSession.workingDir || workingDir;\n\n // Smart follow-up anchoring: keep depth requests in the same repo/module as previous run\n if (!body.workingDir && isLikelyFollowUpScopeTask(body.task)) {\n const inferred = await inferFollowUpWorkingDir(sessionManager, sessionId, resolvedSession.workingDir || ctx.cwd);\n if (inferred && inferred !== workingDir) {\n ctx.platform.logger.info(`[run-handler] Follow-up scope inferred: ${workingDir} -> ${inferred}`);\n workingDir = inferred;\n }\n }\n\n // Rebind session manager to effective working directory\n sessionManager = new SessionManager(workingDir);\n ctx.platform.logger.info(`[run-handler] Continuing session ${sessionId} (workingDir=${workingDir})`);\n }\n\n ctx.platform.logger.info(`[run-handler] Starting run ${runId} for task: ${body.task}`);\n\n // Create user turn with task/question\n await sessionManager.createUserTurn(sessionId, body.task, runId);\n\n // Track run started\n await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_STARTED, {\n runId,\n sessionId,\n taskLength: body.task.length,\n tier: body.tier ?? 'medium',\n enableEscalation: body.enableEscalation ?? true,\n responseMode: body.responseMode ?? 'auto',\n verbose: body.verbose ?? false,\n });\n\n // Create tool registry with standard tools\n const sessionMemory = createSessionMemoryBridge(workingDir, sessionId);\n const responseRequirementsSelector = createDefaultResponseRequirementsSelector();\n const toolRegistry = createToolRegistry({\n workingDir,\n currentTask: body.task,\n sessionId,\n verbose: body.verbose,\n cache: useCache(),\n sessionMemory,\n responseRequirementsResolver: async ({ task, kernel }: {\n task?: string;\n answer: string;\n kernel: KernelState | null;\n }) =>\n responseRequirementsSelector.select({\n state: kernel,\n messages: [],\n task: task ?? body.task,\n }),\n });\n const agentsConfig = await useConfig<AgentsPluginConfig>();\n\n const finalSessionId = sessionId; // Capture for closure\n const traceDir = path.join(workingDir, '.kb', 'traces', 'incremental');\n const traceWriter = new IncrementalTraceWriter(runId, {}, traceDir);\n\n // Create agent with event broadcasting and session persistence\n const agent = new AgentSDK()\n .register(createCoreToolPack(toolRegistry))\n .createRunner({\n sessionId: finalSessionId,\n workingDir,\n maxIterations: 50,\n temperature: 0.1,\n tier: body.tier ?? 'medium',\n tokenBudget: agentsConfig?.tokenBudget,\n onEvent: (event) => {\n // Broadcast to all WebSocket listeners (assigns seq)\n const seqEvent = RunManager.broadcast(runId, event);\n\n // Persist event with seq + runId + sessionId in metadata to session storage\n void sessionManager.addEvent(finalSessionId, {\n ...seqEvent,\n sessionId: finalSessionId,\n runId,\n metadata: {\n ...seqEvent.metadata,\n sessionId: finalSessionId,\n runId,\n workingDir,\n },\n });\n },\n });\n\n // Register run (pass sessionManager and sessionId so session-level WS listeners receive events)\n const run = await RunManager.register(runId, body.task, agent, sessionManager, finalSessionId);\n await RunManager.updateStatus(runId, 'running');\n\n // Start execution in background (don't await)\n void (async () => {\n try {\n const result = await agent.execute(body.task);\n const durationMs = Date.now() - startTime;\n const detailedTrace = traceWriter.getEntries() as Array<Record<string, unknown>>;\n await traceWriter.finalize?.();\n\n if (detailedTrace.length > 0) {\n await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);\n }\n\n // Attach file change summaries to the turn so the UI can show rollback/approve panel\n // Populated by ChangeTrackingMiddleware via run.meta → TaskResult.fileChanges\n if (result.fileChanges && result.fileChanges.length > 0) {\n await sessionManager.attachFileChangesToTurn(finalSessionId, runId, result.fileChanges);\n }\n\n await RunManager.updateStatus(runId, result.success ? 'completed' : 'failed', {\n completedAt: new Date().toISOString(),\n durationMs,\n summary: result.summary,\n error: result.error,\n });\n\n // Track completion\n await analytics?.track(\n result.success ? AGENT_ANALYTICS_EVENTS.RUN_COMPLETED : AGENT_ANALYTICS_EVENTS.RUN_FAILED,\n {\n runId,\n durationMs,\n success: result.success,\n summary: result.summary?.slice(0, 200),\n }\n );\n\n ctx.platform.logger.info(`[run-handler] Run ${runId} completed: ${result.success}`);\n } catch (error) {\n const errorMsg = error instanceof Error ? error.message : String(error);\n const durationMs = Date.now() - startTime;\n const detailedTrace = traceWriter.getEntries() as Array<Record<string, unknown>>;\n await traceWriter.finalize?.();\n if (detailedTrace.length > 0) {\n await sessionManager.storeTraceArtifacts(finalSessionId, runId, detailedTrace);\n }\n\n await RunManager.updateStatus(runId, 'failed', {\n completedAt: new Date().toISOString(),\n durationMs,\n error: errorMsg,\n });\n\n // Track failure\n await analytics?.track(AGENT_ANALYTICS_EVENTS.RUN_FAILED, {\n runId,\n durationMs,\n error: errorMsg.slice(0, 200),\n });\n\n ctx.platform.logger.error(`[run-handler] Run ${runId} failed: ${errorMsg}`);\n }\n })();\n\n // Return relative WS path — clients construct the full URL from their own base URL.\n const wsPath = AGENTS_WS_CHANNELS.SESSION_STREAM.replace(':sessionId', finalSessionId);\n const eventsPath = `${AGENTS_WS_BASE_PATH}${wsPath}`;\n\n return {\n runId,\n sessionId: finalSessionId,\n eventsPath,\n status: 'started',\n startedAt: run.startedAt,\n };\n },\n});\n"]}
@@ -0,0 +1,18 @@
1
+ import * as _kb_labs_shared_command_kit from '@kb-labs/shared-command-kit';
2
+ import { RestInput } from '@kb-labs/sdk';
3
+ import { ListSessionsRequest, ListSessionsResponse, GetSessionRequest, GetSessionResponse, CreateSessionRequest, CreateSessionResponse } from '@kb-labs/agent-contracts';
4
+
5
+ /**
6
+ * GET /sessions - List all sessions
7
+ */
8
+ declare const listSessionsHandler: _kb_labs_shared_command_kit.Handler<unknown, RestInput<ListSessionsRequest, unknown, unknown>, ListSessionsResponse>;
9
+ /**
10
+ * GET /sessions/:sessionId - Get session details
11
+ */
12
+ declare const getSessionHandler: _kb_labs_shared_command_kit.Handler<unknown, RestInput<GetSessionRequest, unknown, unknown>, GetSessionResponse>;
13
+ /**
14
+ * POST /sessions - Create a new session
15
+ */
16
+ declare const createSessionHandler: _kb_labs_shared_command_kit.Handler<unknown, RestInput<CreateSessionRequest, unknown, unknown>, CreateSessionResponse>;
17
+
18
+ export { createSessionHandler, getSessionHandler, listSessionsHandler };
@@ -0,0 +1,56 @@
1
+ import { defineHandler } from '@kb-labs/sdk';
2
+ import { SessionManager } from '@kb-labs/agent-core';
3
+
4
+ // src/rest/handlers/sessions-handler.ts
5
+ var listSessionsHandler = defineHandler({
6
+ async execute(ctx, input) {
7
+ const query = input.query;
8
+ const sessionManager = new SessionManager(ctx.cwd);
9
+ const result = await sessionManager.listSessions({
10
+ agentId: query?.agentId,
11
+ status: query?.status,
12
+ limit: query?.limit ?? 50,
13
+ offset: query?.offset ?? 0
14
+ });
15
+ return {
16
+ sessions: result.sessions,
17
+ total: result.total
18
+ };
19
+ }
20
+ });
21
+ var getSessionHandler = defineHandler({
22
+ async execute(ctx, input) {
23
+ const params = input.params;
24
+ const sessionId = params?.sessionId;
25
+ if (!sessionId) {
26
+ throw new Error("Session ID is required");
27
+ }
28
+ const sessionManager = new SessionManager(ctx.cwd);
29
+ const session = await sessionManager.getSessionInfo(sessionId);
30
+ if (!session) {
31
+ throw new Error(`Session not found: ${sessionId}`);
32
+ }
33
+ return { session };
34
+ }
35
+ });
36
+ var createSessionHandler = defineHandler({
37
+ async execute(ctx, input) {
38
+ const body = input.body;
39
+ if (!body?.agentId) {
40
+ throw new Error("Agent ID is required");
41
+ }
42
+ const sessionManager = new SessionManager(ctx.cwd);
43
+ const session = await sessionManager.createSession({
44
+ mode: "execute",
45
+ task: body.task ?? "",
46
+ agentId: body.agentId,
47
+ name: body.name
48
+ });
49
+ ctx.platform.logger.info(`[sessions-handler] Created session ${session.id} for agent ${body.agentId}`);
50
+ return { session };
51
+ }
52
+ });
53
+
54
+ export { createSessionHandler, getSessionHandler, listSessionsHandler };
55
+ //# sourceMappingURL=sessions-handler.js.map
56
+ //# sourceMappingURL=sessions-handler.js.map