@cicctencent/agent-core 0.1.6 → 0.1.8

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 (142) hide show
  1. package/README.md +4 -3
  2. package/dist/cjs/a2a/client.d.ts +47 -0
  3. package/dist/cjs/a2a/client.d.ts.map +1 -0
  4. package/dist/cjs/a2a/client.js +178 -0
  5. package/dist/cjs/a2a/client.js.map +1 -0
  6. package/dist/cjs/a2a/index.d.ts +3 -0
  7. package/dist/cjs/a2a/index.d.ts.map +1 -0
  8. package/dist/cjs/a2a/index.js +9 -0
  9. package/dist/cjs/a2a/index.js.map +1 -0
  10. package/dist/cjs/a2a/types.d.ts +149 -0
  11. package/dist/cjs/a2a/types.d.ts.map +1 -0
  12. package/dist/cjs/a2a/types.js +8 -0
  13. package/dist/cjs/a2a/types.js.map +1 -0
  14. package/dist/cjs/index.d.ts +22 -0
  15. package/dist/cjs/index.d.ts.map +1 -1
  16. package/dist/cjs/index.js +52 -2
  17. package/dist/cjs/index.js.map +1 -1
  18. package/dist/cjs/middleware/audit-log.d.ts +22 -0
  19. package/dist/cjs/middleware/audit-log.d.ts.map +1 -0
  20. package/dist/cjs/middleware/audit-log.js +104 -0
  21. package/dist/cjs/middleware/audit-log.js.map +1 -0
  22. package/dist/cjs/middleware/timing.d.ts +25 -0
  23. package/dist/cjs/middleware/timing.d.ts.map +1 -0
  24. package/dist/cjs/middleware/timing.js +62 -0
  25. package/dist/cjs/middleware/timing.js.map +1 -0
  26. package/dist/cjs/middleware/tool-guard.d.ts +15 -0
  27. package/dist/cjs/middleware/tool-guard.d.ts.map +1 -0
  28. package/dist/cjs/middleware/tool-guard.js +31 -0
  29. package/dist/cjs/middleware/tool-guard.js.map +1 -0
  30. package/dist/cjs/observability/logger.d.ts +16 -0
  31. package/dist/cjs/observability/logger.d.ts.map +1 -0
  32. package/dist/cjs/observability/logger.js +35 -0
  33. package/dist/cjs/observability/logger.js.map +1 -0
  34. package/dist/cjs/pool/engine-pool.d.ts +22 -0
  35. package/dist/cjs/pool/engine-pool.d.ts.map +1 -0
  36. package/dist/cjs/pool/engine-pool.js +75 -0
  37. package/dist/cjs/pool/engine-pool.js.map +1 -0
  38. package/dist/cjs/runtime/run-registry.d.ts +104 -0
  39. package/dist/cjs/runtime/run-registry.d.ts.map +1 -0
  40. package/dist/cjs/runtime/run-registry.js +261 -0
  41. package/dist/cjs/runtime/run-registry.js.map +1 -0
  42. package/dist/cjs/sandbox/types.d.ts +1 -1
  43. package/dist/cjs/sandbox/types.d.ts.map +1 -1
  44. package/dist/cjs/security/guard.js +2 -2
  45. package/dist/cjs/security/guard.js.map +1 -1
  46. package/dist/cjs/security/permission.d.ts +39 -0
  47. package/dist/cjs/security/permission.d.ts.map +1 -0
  48. package/dist/cjs/security/permission.js +111 -0
  49. package/dist/cjs/security/permission.js.map +1 -0
  50. package/dist/cjs/security/risk.d.ts +28 -0
  51. package/dist/cjs/security/risk.d.ts.map +1 -0
  52. package/dist/cjs/security/risk.js +148 -0
  53. package/dist/cjs/security/risk.js.map +1 -0
  54. package/dist/cjs/security/types.d.ts +1 -1
  55. package/dist/cjs/security/types.d.ts.map +1 -1
  56. package/dist/cjs/store/file-tool-result-store.d.ts +27 -0
  57. package/dist/cjs/store/file-tool-result-store.d.ts.map +1 -0
  58. package/dist/cjs/store/file-tool-result-store.js +166 -0
  59. package/dist/cjs/store/file-tool-result-store.js.map +1 -0
  60. package/dist/cjs/store/json-store.d.ts +54 -0
  61. package/dist/cjs/store/json-store.d.ts.map +1 -0
  62. package/dist/cjs/store/json-store.js +274 -0
  63. package/dist/cjs/store/json-store.js.map +1 -0
  64. package/dist/cjs/utils/data-dir.d.ts +27 -0
  65. package/dist/cjs/utils/data-dir.d.ts.map +1 -0
  66. package/dist/cjs/utils/data-dir.js +122 -0
  67. package/dist/cjs/utils/data-dir.js.map +1 -0
  68. package/dist/cjs/utils/event-collector.d.ts +13 -0
  69. package/dist/cjs/utils/event-collector.d.ts.map +1 -0
  70. package/dist/cjs/utils/event-collector.js +33 -0
  71. package/dist/cjs/utils/event-collector.js.map +1 -0
  72. package/dist/esm/a2a/client.d.ts +47 -0
  73. package/dist/esm/a2a/client.d.ts.map +1 -0
  74. package/dist/esm/a2a/client.js +174 -0
  75. package/dist/esm/a2a/client.js.map +1 -0
  76. package/dist/esm/a2a/index.d.ts +3 -0
  77. package/dist/esm/a2a/index.d.ts.map +1 -0
  78. package/dist/esm/a2a/index.js +5 -0
  79. package/dist/esm/a2a/index.js.map +1 -0
  80. package/dist/esm/a2a/types.d.ts +149 -0
  81. package/dist/esm/a2a/types.d.ts.map +1 -0
  82. package/dist/esm/a2a/types.js +7 -0
  83. package/dist/esm/a2a/types.js.map +1 -0
  84. package/dist/esm/index.d.ts +22 -0
  85. package/dist/esm/index.d.ts.map +1 -1
  86. package/dist/esm/index.js +20 -0
  87. package/dist/esm/index.js.map +1 -1
  88. package/dist/esm/middleware/audit-log.d.ts +22 -0
  89. package/dist/esm/middleware/audit-log.d.ts.map +1 -0
  90. package/dist/esm/middleware/audit-log.js +67 -0
  91. package/dist/esm/middleware/audit-log.js.map +1 -0
  92. package/dist/esm/middleware/timing.d.ts +25 -0
  93. package/dist/esm/middleware/timing.d.ts.map +1 -0
  94. package/dist/esm/middleware/timing.js +58 -0
  95. package/dist/esm/middleware/timing.js.map +1 -0
  96. package/dist/esm/middleware/tool-guard.d.ts +15 -0
  97. package/dist/esm/middleware/tool-guard.d.ts.map +1 -0
  98. package/dist/esm/middleware/tool-guard.js +27 -0
  99. package/dist/esm/middleware/tool-guard.js.map +1 -0
  100. package/dist/esm/observability/logger.d.ts +16 -0
  101. package/dist/esm/observability/logger.d.ts.map +1 -0
  102. package/dist/esm/observability/logger.js +31 -0
  103. package/dist/esm/observability/logger.js.map +1 -0
  104. package/dist/esm/pool/engine-pool.d.ts +22 -0
  105. package/dist/esm/pool/engine-pool.d.ts.map +1 -0
  106. package/dist/esm/pool/engine-pool.js +71 -0
  107. package/dist/esm/pool/engine-pool.js.map +1 -0
  108. package/dist/esm/runtime/run-registry.d.ts +104 -0
  109. package/dist/esm/runtime/run-registry.d.ts.map +1 -0
  110. package/dist/esm/runtime/run-registry.js +256 -0
  111. package/dist/esm/runtime/run-registry.js.map +1 -0
  112. package/dist/esm/sandbox/types.d.ts +1 -1
  113. package/dist/esm/sandbox/types.d.ts.map +1 -1
  114. package/dist/esm/security/guard.js +2 -2
  115. package/dist/esm/security/guard.js.map +1 -1
  116. package/dist/esm/security/permission.d.ts +39 -0
  117. package/dist/esm/security/permission.d.ts.map +1 -0
  118. package/dist/esm/security/permission.js +102 -0
  119. package/dist/esm/security/permission.js.map +1 -0
  120. package/dist/esm/security/risk.d.ts +28 -0
  121. package/dist/esm/security/risk.d.ts.map +1 -0
  122. package/dist/esm/security/risk.js +142 -0
  123. package/dist/esm/security/risk.js.map +1 -0
  124. package/dist/esm/security/types.d.ts +1 -1
  125. package/dist/esm/security/types.d.ts.map +1 -1
  126. package/dist/esm/store/file-tool-result-store.d.ts +27 -0
  127. package/dist/esm/store/file-tool-result-store.d.ts.map +1 -0
  128. package/dist/esm/store/file-tool-result-store.js +129 -0
  129. package/dist/esm/store/file-tool-result-store.js.map +1 -0
  130. package/dist/esm/store/json-store.d.ts +54 -0
  131. package/dist/esm/store/json-store.d.ts.map +1 -0
  132. package/dist/esm/store/json-store.js +236 -0
  133. package/dist/esm/store/json-store.js.map +1 -0
  134. package/dist/esm/utils/data-dir.d.ts +27 -0
  135. package/dist/esm/utils/data-dir.d.ts.map +1 -0
  136. package/dist/esm/utils/data-dir.js +80 -0
  137. package/dist/esm/utils/data-dir.js.map +1 -0
  138. package/dist/esm/utils/event-collector.d.ts +13 -0
  139. package/dist/esm/utils/event-collector.d.ts.map +1 -0
  140. package/dist/esm/utils/event-collector.js +30 -0
  141. package/dist/esm/utils/event-collector.js.map +1 -0
  142. package/package.json +1 -1
@@ -0,0 +1,58 @@
1
+ // ============================================================
2
+ // TimingMiddleware — 执行耗时统计中间件
3
+ // ============================================================
4
+ /**
5
+ * 统计每轮迭代和每次工具调用的耗时
6
+ * 可通过 onReport 回调获取汇总数据(如写入监控)
7
+ */
8
+ export class TimingMiddleware {
9
+ name = 'timing';
10
+ iterStarts = new Map();
11
+ toolTimings = new Map();
12
+ onReport;
13
+ constructor(onReport) {
14
+ this.onReport = onReport;
15
+ }
16
+ async before(ctx) {
17
+ this.iterStarts.set(ctx.sessionId, Date.now());
18
+ }
19
+ async after(ctx) {
20
+ const start = this.iterStarts.get(ctx.sessionId);
21
+ const elapsed = start ? Date.now() - start : -1;
22
+ this.iterStarts.delete(ctx.sessionId);
23
+ if (this.onReport) {
24
+ this.onReport({
25
+ type: 'iteration',
26
+ sessionId: ctx.sessionId,
27
+ iteration: ctx.iteration,
28
+ elapsedMs: elapsed,
29
+ });
30
+ }
31
+ else {
32
+ console.log(`[Timing] Session=${ctx.sessionId} Iter=${ctx.iteration} elapsed=${elapsed}ms`);
33
+ }
34
+ }
35
+ async onToolBefore(ctx) {
36
+ const key = `${ctx.tool}_${Date.now()}`;
37
+ this.toolTimings.set(key, { name: ctx.tool || 'unknown', start: Date.now() });
38
+ ctx._timingKey = key;
39
+ }
40
+ async onToolAfter(ctx) {
41
+ const key = ctx._timingKey;
42
+ const timing = key ? this.toolTimings.get(key) : undefined;
43
+ const elapsed = timing ? Date.now() - timing.start : -1;
44
+ if (key)
45
+ this.toolTimings.delete(key);
46
+ if (this.onReport) {
47
+ this.onReport({
48
+ type: 'tool',
49
+ toolName: ctx.tool,
50
+ elapsedMs: elapsed,
51
+ });
52
+ }
53
+ else {
54
+ console.log(`[Timing] Tool=${ctx.tool} elapsed=${elapsed}ms`);
55
+ }
56
+ }
57
+ }
58
+ //# sourceMappingURL=timing.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"timing.js","sourceRoot":"","sources":["../../../src/middleware/timing.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,+BAA+B;AAC/B,+DAA+D;AAkB/D;;;GAGG;AACH,MAAM,OAAO,gBAAgB;IAC3B,IAAI,GAAG,QAAQ,CAAC;IAER,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IACvC,WAAW,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC5C,QAAQ,CAAkC;IAElD,YAAY,QAAyC;QACnD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAsB;QACjC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAsB;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAEtC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,QAAQ,CAAC;gBACZ,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,SAAS,EAAE,OAAO;aACnB,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,CAAC,SAAS,SAAS,GAAG,CAAC,SAAS,YAAY,OAAO,IAAI,CAAC,CAAC;QAC9F,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAgB;QACjC,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC7E,GAAW,CAAC,UAAU,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAgB;QAChC,MAAM,GAAG,GAAI,GAAW,CAAC,UAAoB,CAAC;QAC9C,MAAM,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3D,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,IAAI,GAAG;YAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAEtC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,QAAQ,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,GAAG,CAAC,IAAI;gBAClB,SAAS,EAAE,OAAO;aACnB,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,CAAC,IAAI,YAAY,OAAO,IAAI,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,15 @@
1
+ import type { Middleware } from '../middleware/types.ts';
2
+ import type { HookContext, HookResult } from '../hooks/types.ts';
3
+ /**
4
+ * 工具调用守卫:对指定工具进行拦截
5
+ * - 可配置需要拦截的工具名列表
6
+ * - 匹配时 onToolBefore 返回 block,阻止工具执行
7
+ */
8
+ export declare class ToolGuardMiddleware implements Middleware {
9
+ name: string;
10
+ private blockedTools;
11
+ private logger;
12
+ constructor(blockedTools: string[], logger?: (msg: string) => void);
13
+ onToolBefore(ctx: HookContext): Promise<HookResult | void>;
14
+ }
15
+ //# sourceMappingURL=tool-guard.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tool-guard.d.ts","sourceRoot":"","sources":["../../../src/middleware/tool-guard.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEjE;;;;GAIG;AACH,qBAAa,mBAAoB,YAAW,UAAU;IACpD,IAAI,SAAgB;IAEpB,OAAO,CAAC,YAAY,CAAc;IAClC,OAAO,CAAC,MAAM,CAAwB;gBAE1B,YAAY,EAAE,MAAM,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI;IAK5D,YAAY,CAAC,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;CASjE"}
@@ -0,0 +1,27 @@
1
+ // ============================================================
2
+ // ToolGuardMiddleware — 工具安全守卫中间件
3
+ // ============================================================
4
+ /**
5
+ * 工具调用守卫:对指定工具进行拦截
6
+ * - 可配置需要拦截的工具名列表
7
+ * - 匹配时 onToolBefore 返回 block,阻止工具执行
8
+ */
9
+ export class ToolGuardMiddleware {
10
+ name = 'tool-guard';
11
+ blockedTools;
12
+ logger;
13
+ constructor(blockedTools, logger) {
14
+ this.blockedTools = new Set(blockedTools);
15
+ this.logger = logger || ((msg) => console.warn(`[ToolGuard] ${msg}`));
16
+ }
17
+ async onToolBefore(ctx) {
18
+ if (ctx.tool && this.blockedTools.has(ctx.tool)) {
19
+ this.logger(`BLOCKED tool call: ${ctx.tool}`);
20
+ return {
21
+ block: true,
22
+ reason: `Tool "${ctx.tool}" is blocked by ToolGuard middleware`,
23
+ };
24
+ }
25
+ }
26
+ }
27
+ //# sourceMappingURL=tool-guard.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tool-guard.js","sourceRoot":"","sources":["../../../src/middleware/tool-guard.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,kCAAkC;AAClC,+DAA+D;AAK/D;;;;GAIG;AACH,MAAM,OAAO,mBAAmB;IAC9B,IAAI,GAAG,YAAY,CAAC;IAEZ,YAAY,CAAc;IAC1B,MAAM,CAAwB;IAEtC,YAAY,YAAsB,EAAE,MAA8B;QAChE,IAAI,CAAC,YAAY,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,GAAG,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAgB;QACjC,IAAI,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,sBAAsB,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;YAC9C,OAAO;gBACL,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,SAAS,GAAG,CAAC,IAAI,sCAAsC;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,16 @@
1
+ export type LogLevel = 'debug' | 'info' | 'warn' | 'error';
2
+ export interface ObservabilityEvent {
3
+ ts?: number;
4
+ level?: LogLevel;
5
+ event: string;
6
+ traceId?: string;
7
+ sessionId?: string;
8
+ data?: Record<string, unknown>;
9
+ }
10
+ /** 日志传输层接口 */
11
+ export type LogTransport = (payload: Required<Pick<ObservabilityEvent, 'ts' | 'level'>> & ObservabilityEvent) => void;
12
+ /** 设置自定义日志传输层 */
13
+ export declare function setLogTransport(transport: LogTransport): void;
14
+ /** 记录可观测性事件 */
15
+ export declare function logEvent(input: ObservabilityEvent): void;
16
+ //# sourceMappingURL=logger.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/observability/logger.ts"],"names":[],"mappings":"AAIA,MAAM,MAAM,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;AAE3D,MAAM,WAAW,kBAAkB;IACjC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,QAAQ,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,cAAc;AACd,MAAM,MAAM,YAAY,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,GAAG,OAAO,CAAC,CAAC,GAAG,kBAAkB,KAAK,IAAI,CAAC;AAYtH,iBAAiB;AACjB,wBAAgB,eAAe,CAAC,SAAS,EAAE,YAAY,GAAG,IAAI,CAE7D;AAED,eAAe;AACf,wBAAgB,QAAQ,CAAC,KAAK,EAAE,kBAAkB,GAAG,IAAI,CAUxD"}
@@ -0,0 +1,31 @@
1
+ // ============================================================
2
+ // Observability — 结构化可观测性日志
3
+ // ============================================================
4
+ /** 默认传输:console 输出 */
5
+ const defaultTransport = (payload) => {
6
+ const line = `[obs] ${JSON.stringify(payload)}`;
7
+ if (payload.level === 'error')
8
+ console.error(line);
9
+ else if (payload.level === 'warn')
10
+ console.warn(line);
11
+ else
12
+ console.log(line);
13
+ };
14
+ let _transport = defaultTransport;
15
+ /** 设置自定义日志传输层 */
16
+ export function setLogTransport(transport) {
17
+ _transport = transport;
18
+ }
19
+ /** 记录可观测性事件 */
20
+ export function logEvent(input) {
21
+ const payload = {
22
+ ts: input.ts || Date.now(),
23
+ level: input.level || 'info',
24
+ event: input.event,
25
+ traceId: input.traceId,
26
+ sessionId: input.sessionId,
27
+ data: input.data || {},
28
+ };
29
+ _transport(payload);
30
+ }
31
+ //# sourceMappingURL=logger.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"logger.js","sourceRoot":"","sources":["../../../src/observability/logger.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,4BAA4B;AAC5B,+DAA+D;AAgB/D,sBAAsB;AACtB,MAAM,gBAAgB,GAAiB,CAAC,OAAO,EAAE,EAAE;IACjD,MAAM,IAAI,GAAG,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,IAAI,OAAO,CAAC,KAAK,KAAK,OAAO;QAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;SAC9C,IAAI,OAAO,CAAC,KAAK,KAAK,MAAM;QAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;;QACjD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACzB,CAAC,CAAC;AAEF,IAAI,UAAU,GAAiB,gBAAgB,CAAC;AAEhD,iBAAiB;AACjB,MAAM,UAAU,eAAe,CAAC,SAAuB;IACrD,UAAU,GAAG,SAAS,CAAC;AACzB,CAAC;AAED,eAAe;AACf,MAAM,UAAU,QAAQ,CAAC,KAAyB;IAChD,MAAM,OAAO,GAAG;QACd,EAAE,EAAE,KAAK,CAAC,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE;QAC1B,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,MAAM;QAC5B,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;KACvB,CAAC;IACF,UAAU,CAAC,OAAO,CAAC,CAAC;AACtB,CAAC"}
@@ -0,0 +1,22 @@
1
+ import type { AgentEngine } from '../agent/engine.ts';
2
+ /** AgentEngine 实例复用池 */
3
+ export declare class EnginePool {
4
+ private pool;
5
+ private readonly ttl;
6
+ private readonly maxSize;
7
+ constructor(ttlMs?: number, maxSize?: number);
8
+ /** 生成缓存 key */
9
+ static key(profileId?: number | string, scopeKey?: string): string;
10
+ /** 从池中获取 Engine(未命中或过期返回 null) */
11
+ get(profileId?: number | string, scopeKey?: string): AgentEngine | null;
12
+ /** 将 Engine 放入池中 */
13
+ set(profileId: number | string | undefined, engine: AgentEngine, scopeKey?: string): void;
14
+ /** 清除所有缓存(配置变更时调用) */
15
+ clear(): void;
16
+ /** 仅清除过期的实例 */
17
+ purgeExpired(): void;
18
+ /** 池大小 */
19
+ get size(): number;
20
+ private evictOldest;
21
+ }
22
+ //# sourceMappingURL=engine-pool.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"engine-pool.d.ts","sourceRoot":"","sources":["../../../src/pool/engine-pool.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAQtD,wBAAwB;AACxB,qBAAa,UAAU;IACrB,OAAO,CAAC,IAAI,CAAmC;IAC/C,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAErB,KAAK,GAAE,MAAmB,EAAE,OAAO,GAAE,MAAW;IAK5D,eAAe;IACf,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,MAAM;IAIlE,kCAAkC;IAClC,GAAG,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,WAAW,GAAG,IAAI;IAcvE,oBAAoB;IACpB,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI;IAazF,sBAAsB;IACtB,KAAK,IAAI,IAAI;IAOb,eAAe;IACf,YAAY,IAAI,IAAI;IAUpB,UAAU;IACV,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,OAAO,CAAC,WAAW;CAQpB"}
@@ -0,0 +1,71 @@
1
+ // ============================================================
2
+ // EnginePool — AgentEngine 实例复用池
3
+ // ============================================================
4
+ /** AgentEngine 实例复用池 */
5
+ export class EnginePool {
6
+ pool = new Map();
7
+ ttl;
8
+ maxSize;
9
+ constructor(ttlMs = 5 * 60_000, maxSize = 10) {
10
+ this.ttl = ttlMs;
11
+ this.maxSize = maxSize;
12
+ }
13
+ /** 生成缓存 key */
14
+ static key(profileId, scopeKey) {
15
+ return `${profileId ?? '_'}:${scopeKey || '_'}`;
16
+ }
17
+ /** 从池中获取 Engine(未命中或过期返回 null) */
18
+ get(profileId, scopeKey) {
19
+ const key = EnginePool.key(profileId, scopeKey);
20
+ const entry = this.pool.get(key);
21
+ if (!entry)
22
+ return null;
23
+ if (Date.now() - entry.createdAt > this.ttl) {
24
+ entry.engine.destroy();
25
+ this.pool.delete(key);
26
+ return null;
27
+ }
28
+ return entry.engine;
29
+ }
30
+ /** 将 Engine 放入池中 */
31
+ set(profileId, engine, scopeKey) {
32
+ const key = EnginePool.key(profileId, scopeKey);
33
+ const old = this.pool.get(key);
34
+ if (old)
35
+ old.engine.destroy();
36
+ if (this.pool.size >= this.maxSize && !this.pool.has(key)) {
37
+ this.evictOldest();
38
+ }
39
+ this.pool.set(key, { engine, createdAt: Date.now() });
40
+ }
41
+ /** 清除所有缓存(配置变更时调用) */
42
+ clear() {
43
+ for (const entry of this.pool.values()) {
44
+ entry.engine.destroy();
45
+ }
46
+ this.pool.clear();
47
+ }
48
+ /** 仅清除过期的实例 */
49
+ purgeExpired() {
50
+ const now = Date.now();
51
+ for (const [key, entry] of this.pool) {
52
+ if (now - entry.createdAt > this.ttl) {
53
+ entry.engine.destroy();
54
+ this.pool.delete(key);
55
+ }
56
+ }
57
+ }
58
+ /** 池大小 */
59
+ get size() {
60
+ return this.pool.size;
61
+ }
62
+ evictOldest() {
63
+ const oldestKey = this.pool.keys().next().value;
64
+ if (oldestKey !== undefined) {
65
+ const entry = this.pool.get(oldestKey);
66
+ entry?.engine.destroy();
67
+ this.pool.delete(oldestKey);
68
+ }
69
+ }
70
+ }
71
+ //# sourceMappingURL=engine-pool.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"engine-pool.js","sourceRoot":"","sources":["../../../src/pool/engine-pool.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,iCAAiC;AACjC,+DAA+D;AAU/D,wBAAwB;AACxB,MAAM,OAAO,UAAU;IACb,IAAI,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC9B,GAAG,CAAS;IACZ,OAAO,CAAS;IAEjC,YAAY,QAAgB,CAAC,GAAG,MAAM,EAAE,UAAkB,EAAE;QAC1D,IAAI,CAAC,GAAG,GAAG,KAAK,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,eAAe;IACf,MAAM,CAAC,GAAG,CAAC,SAA2B,EAAE,QAAiB;QACvD,OAAO,GAAG,SAAS,IAAI,GAAG,IAAI,QAAQ,IAAI,GAAG,EAAE,CAAC;IAClD,CAAC;IAED,kCAAkC;IAClC,GAAG,CAAC,SAA2B,EAAE,QAAiB;QAChD,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC5C,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,CAAC;IACtB,CAAC;IAED,oBAAoB;IACpB,GAAG,CAAC,SAAsC,EAAE,MAAmB,EAAE,QAAiB;QAChF,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAEhD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC/B,IAAI,GAAG;YAAE,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QAE9B,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1D,IAAI,CAAC,WAAW,EAAE,CAAC;QACrB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,sBAAsB;IACtB,KAAK;QACH,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;IAED,eAAe;IACf,YAAY;QACV,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,GAAG,GAAG,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACrC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;IACH,CAAC;IAED,UAAU;IACV,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACxB,CAAC;IAEO,WAAW;QACjB,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QAChD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvC,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,104 @@
1
+ import type { WorkflowCheckpoint } from '../agent/workflow.ts';
2
+ /** 单条事件 */
3
+ export interface BufferedEvent {
4
+ event: string;
5
+ data: any;
6
+ }
7
+ /** 工具调用记录(用于重连时恢复 UI 状态) */
8
+ export interface RunToolCallRecord {
9
+ toolCallId: string;
10
+ toolName: string;
11
+ displayName?: string;
12
+ input?: Record<string, any>;
13
+ output?: any;
14
+ error?: string;
15
+ duration?: number;
16
+ status: 'running' | 'success' | 'error';
17
+ }
18
+ /** 事件订阅回调 */
19
+ export type EventSubscriber = (evt: BufferedEvent) => void;
20
+ /** Agent 运行状态 */
21
+ export type RunStatus = 'running' | 'done' | 'error';
22
+ /** 活跃运行记录 */
23
+ export interface ActiveRun {
24
+ threadId: number | string;
25
+ sessionId: string;
26
+ traceId?: string;
27
+ status: RunStatus;
28
+ eventBuffer: BufferedEvent[];
29
+ fullContent: string;
30
+ reasoningText: string;
31
+ toolCalls: RunToolCallRecord[];
32
+ widgets: Array<{
33
+ widgetId: string;
34
+ html: string;
35
+ title: string;
36
+ height?: number;
37
+ }>;
38
+ artifacts: Array<{
39
+ artifactId: string;
40
+ filename: string;
41
+ mimeType: string;
42
+ size: number;
43
+ url: string;
44
+ description?: string;
45
+ }>;
46
+ checkpoints: WorkflowCheckpoint[];
47
+ stepCount: number;
48
+ toolCallCount: number;
49
+ error?: string;
50
+ messageId?: number | string;
51
+ subscribers: Set<EventSubscriber>;
52
+ createdAt: number;
53
+ lastFlushAt: number;
54
+ lastFailedStage?: string;
55
+ /** 扩展字段(上层可附加任意数据) */
56
+ meta?: Record<string, unknown>;
57
+ }
58
+ /** 可跨进程读取的运行态快照(不包含实时订阅者) */
59
+ export type RunSnapshot = Omit<ActiveRun, 'subscribers'> & {
60
+ updatedAt: number;
61
+ };
62
+ /** 运行态持久化存储接口(上层实现) */
63
+ export interface RunStateStore {
64
+ list(): RunSnapshot[];
65
+ upsert(snapshot: RunSnapshot): void;
66
+ remove(threadId: number | string): void;
67
+ }
68
+ /** 内存实现(默认) */
69
+ export declare class InMemoryRunStateStore implements RunStateStore {
70
+ private snapshots;
71
+ list(): RunSnapshot[];
72
+ upsert(snapshot: RunSnapshot): void;
73
+ remove(threadId: number | string): void;
74
+ }
75
+ export declare class RunRegistry {
76
+ private runs;
77
+ private runStateStore;
78
+ private cleanupTimer;
79
+ constructor(store?: RunStateStore);
80
+ /** 注册一个新的运行 */
81
+ startRun(threadId: number | string, sessionId: string, traceId?: string): ActiveRun;
82
+ getRun(threadId: number | string): ActiveRun | null;
83
+ hasActiveRun(threadId: number | string): boolean;
84
+ /** 添加事件到缓冲区并通知所有订阅者 */
85
+ addEvent(threadId: number | string, event: string, data: any): void;
86
+ finishRun(threadId: number | string, status: 'done' | 'error', error?: string): void;
87
+ setStepCount(threadId: number | string, count: number): void;
88
+ setMessageId(threadId: number | string, messageId: number | string): void;
89
+ subscribe(threadId: number | string, callback: EventSubscriber): () => void;
90
+ getBufferedEvents(threadId: number | string): BufferedEvent[];
91
+ getRunSnapshot(threadId: number | string): RunSnapshot | null;
92
+ listRunSnapshots(): RunSnapshot[];
93
+ getFailedCheckpoint(threadId: number | string): WorkflowCheckpoint | null;
94
+ setLastFailedStage(threadId: number | string, stageId: string): void;
95
+ removeRun(threadId: number | string): void;
96
+ getActiveThreadIds(): (number | string)[];
97
+ destroy(): void;
98
+ private upsertCheckpoint;
99
+ private notifySubscribers;
100
+ private toSnapshot;
101
+ private persistRun;
102
+ private startCleanup;
103
+ }
104
+ //# sourceMappingURL=run-registry.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"run-registry.d.ts","sourceRoot":"","sources":["../../../src/runtime/run-registry.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE/D,WAAW;AACX,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,GAAG,CAAC;CACX;AAED,4BAA4B;AAC5B,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC5B,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,SAAS,GAAG,SAAS,GAAG,OAAO,CAAC;CACzC;AAED,aAAa;AACb,MAAM,MAAM,eAAe,GAAG,CAAC,GAAG,EAAE,aAAa,KAAK,IAAI,CAAC;AAE3D,iBAAiB;AACjB,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,MAAM,GAAG,OAAO,CAAC;AAErD,aAAa;AACb,MAAM,WAAW,SAAS;IACxB,QAAQ,EAAE,MAAM,GAAG,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,SAAS,CAAC;IAClB,WAAW,EAAE,aAAa,EAAE,CAAC;IAC7B,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,iBAAiB,EAAE,CAAC;IAC/B,OAAO,EAAE,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnF,SAAS,EAAE,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9H,WAAW,EAAE,kBAAkB,EAAE,CAAC;IAClC,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC5B,WAAW,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;IAClC,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,sBAAsB;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,6BAA6B;AAC7B,MAAM,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,EAAE,aAAa,CAAC,GAAG;IACzD,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,uBAAuB;AACvB,MAAM,WAAW,aAAa;IAC5B,IAAI,IAAI,WAAW,EAAE,CAAC;IACtB,MAAM,CAAC,QAAQ,EAAE,WAAW,GAAG,IAAI,CAAC;IACpC,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;CACzC;AAED,eAAe;AACf,qBAAa,qBAAsB,YAAW,aAAa;IACzD,OAAO,CAAC,SAAS,CAAqB;IAEtC,IAAI,IAAI,WAAW,EAAE;IAIrB,MAAM,CAAC,QAAQ,EAAE,WAAW,GAAG,IAAI;IAQnC,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;CAGxC;AAOD,qBAAa,WAAW;IACtB,OAAO,CAAC,IAAI,CAAyC;IACrD,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,YAAY,CAA+C;gBAEvD,KAAK,CAAC,EAAE,aAAa;IAKjC,eAAe;IACf,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS;IAkCnF,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,GAAG,IAAI;IAInD,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO;IAKhD,uBAAuB;IACvB,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,IAAI;IAgDnE,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAoBpF,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAQ5D,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAQzE,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,eAAe,GAAG,MAAM,IAAI;IAO3E,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,aAAa,EAAE;IAK7D,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,WAAW,GAAG,IAAI;IAM7D,gBAAgB,IAAI,WAAW,EAAE;IAOjC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,kBAAkB,GAAG,IAAI;IAOzE,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAQpE,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAS1C,kBAAkB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE;IAQzC,OAAO,IAAI,IAAI;IAYf,OAAO,CAAC,gBAAgB;IASxB,OAAO,CAAC,iBAAiB;IAMzB,OAAO,CAAC,UAAU;IAKlB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,YAAY;CAcrB"}
@@ -0,0 +1,256 @@
1
+ // ============================================================
2
+ // RunRegistry — Agent 运行会话注册表
3
+ // 解耦 Engine 执行与传输层,支持事件缓冲、订阅和断线重连
4
+ // ============================================================
5
+ import { logEvent } from "../observability/logger.js";
6
+ /** 内存实现(默认) */
7
+ export class InMemoryRunStateStore {
8
+ snapshots = [];
9
+ list() {
10
+ return [...this.snapshots];
11
+ }
12
+ upsert(snapshot) {
13
+ this.snapshots = this.snapshots
14
+ .filter(item => item.threadId !== snapshot.threadId)
15
+ .concat(snapshot)
16
+ .sort((a, b) => b.updatedAt - a.updatedAt)
17
+ .slice(0, 200);
18
+ }
19
+ remove(threadId) {
20
+ this.snapshots = this.snapshots.filter(item => item.threadId !== threadId);
21
+ }
22
+ }
23
+ // ---- 配置 ----
24
+ const CLEANUP_MAX_AGE = 30 * 60 * 1000;
25
+ const CLEANUP_INTERVAL = 5 * 60 * 1000;
26
+ export class RunRegistry {
27
+ runs = new Map();
28
+ runStateStore;
29
+ cleanupTimer = null;
30
+ constructor(store) {
31
+ this.runStateStore = store || new InMemoryRunStateStore();
32
+ this.startCleanup();
33
+ }
34
+ /** 注册一个新的运行 */
35
+ startRun(threadId, sessionId, traceId) {
36
+ const existing = this.runs.get(threadId);
37
+ if (existing && existing.status === 'running') {
38
+ existing.status = 'error';
39
+ existing.error = 'Superseded by new run';
40
+ this.notifySubscribers(existing, { event: 'error', data: { error: '被新的对话覆盖' } });
41
+ this.persistRun(existing);
42
+ }
43
+ const run = {
44
+ threadId,
45
+ sessionId,
46
+ traceId,
47
+ status: 'running',
48
+ eventBuffer: [],
49
+ fullContent: '',
50
+ reasoningText: '',
51
+ toolCalls: [],
52
+ widgets: [],
53
+ artifacts: [],
54
+ checkpoints: [],
55
+ stepCount: 0,
56
+ toolCallCount: 0,
57
+ subscribers: new Set(),
58
+ createdAt: Date.now(),
59
+ lastFlushAt: Date.now(),
60
+ };
61
+ this.runs.set(threadId, run);
62
+ this.persistRun(run);
63
+ logEvent({ event: 'run.start', sessionId, data: { threadId } });
64
+ return run;
65
+ }
66
+ getRun(threadId) {
67
+ return this.runs.get(threadId) || null;
68
+ }
69
+ hasActiveRun(threadId) {
70
+ const run = this.runs.get(threadId);
71
+ return !!run && run.status === 'running';
72
+ }
73
+ /** 添加事件到缓冲区并通知所有订阅者 */
74
+ addEvent(threadId, event, data) {
75
+ const run = this.runs.get(threadId);
76
+ if (!run)
77
+ return;
78
+ const buffered = { event, data };
79
+ run.eventBuffer.push(buffered);
80
+ switch (event) {
81
+ case 'content':
82
+ run.fullContent += (typeof data === 'string' ? data : '');
83
+ break;
84
+ case 'reasoning':
85
+ run.reasoningText += (typeof data === 'string' ? data : '');
86
+ break;
87
+ case 'tool-call':
88
+ run.toolCallCount++;
89
+ run.toolCalls.push({
90
+ toolCallId: data.toolCallId,
91
+ toolName: data.toolName,
92
+ displayName: data.displayName,
93
+ input: data.input,
94
+ status: 'running',
95
+ });
96
+ break;
97
+ case 'tool-result': {
98
+ const tc = run.toolCalls.find(t => t.toolCallId === data.toolCallId);
99
+ if (tc) {
100
+ tc.status = data.error ? 'error' : 'success';
101
+ tc.output = data.output;
102
+ tc.error = data.error;
103
+ tc.duration = data.duration;
104
+ }
105
+ break;
106
+ }
107
+ case 'sub-agent':
108
+ if (data?.eventType === 'checkpoint' && data?.data) {
109
+ this.upsertCheckpoint(run, data.data);
110
+ if (data.data.status === 'failed') {
111
+ run.lastFailedStage = data.data.id;
112
+ }
113
+ }
114
+ break;
115
+ }
116
+ this.persistRun(run);
117
+ this.notifySubscribers(run, buffered);
118
+ }
119
+ finishRun(threadId, status, error) {
120
+ const run = this.runs.get(threadId);
121
+ if (!run)
122
+ return;
123
+ run.status = status;
124
+ if (error)
125
+ run.error = error;
126
+ this.persistRun(run);
127
+ logEvent({
128
+ event: 'run.finish',
129
+ level: status === 'error' ? 'error' : 'info',
130
+ traceId: run.traceId,
131
+ sessionId: run.sessionId,
132
+ data: { threadId, status, error },
133
+ });
134
+ if (status === 'error' && error) {
135
+ this.notifySubscribers(run, { event: 'error', data: { error } });
136
+ }
137
+ }
138
+ setStepCount(threadId, count) {
139
+ const run = this.runs.get(threadId);
140
+ if (run) {
141
+ run.stepCount = count;
142
+ this.persistRun(run);
143
+ }
144
+ }
145
+ setMessageId(threadId, messageId) {
146
+ const run = this.runs.get(threadId);
147
+ if (run) {
148
+ run.messageId = messageId;
149
+ this.persistRun(run);
150
+ }
151
+ }
152
+ subscribe(threadId, callback) {
153
+ const run = this.runs.get(threadId);
154
+ if (!run)
155
+ return () => { };
156
+ run.subscribers.add(callback);
157
+ return () => { run.subscribers.delete(callback); };
158
+ }
159
+ getBufferedEvents(threadId) {
160
+ const run = this.runs.get(threadId);
161
+ return run ? [...run.eventBuffer] : [];
162
+ }
163
+ getRunSnapshot(threadId) {
164
+ const run = this.runs.get(threadId);
165
+ if (run)
166
+ return this.toSnapshot(run);
167
+ return this.runStateStore.list().find(item => item.threadId === threadId) || null;
168
+ }
169
+ listRunSnapshots() {
170
+ const merged = new Map();
171
+ for (const snapshot of this.runStateStore.list())
172
+ merged.set(snapshot.threadId, snapshot);
173
+ for (const [threadId, run] of this.runs)
174
+ merged.set(threadId, this.toSnapshot(run));
175
+ return [...merged.values()].sort((a, b) => b.updatedAt - a.updatedAt);
176
+ }
177
+ getFailedCheckpoint(threadId) {
178
+ const run = this.runs.get(threadId);
179
+ const checkpoints = run?.checkpoints || this.getRunSnapshot(threadId)?.checkpoints || [];
180
+ const failed = checkpoints.filter(c => c.status === 'failed');
181
+ return failed.length > 0 ? failed[failed.length - 1] : null;
182
+ }
183
+ setLastFailedStage(threadId, stageId) {
184
+ const run = this.runs.get(threadId);
185
+ if (run) {
186
+ run.lastFailedStage = stageId;
187
+ this.persistRun(run);
188
+ }
189
+ }
190
+ removeRun(threadId) {
191
+ const run = this.runs.get(threadId);
192
+ if (run) {
193
+ run.subscribers.clear();
194
+ this.runs.delete(threadId);
195
+ }
196
+ this.runStateStore.remove(threadId);
197
+ }
198
+ getActiveThreadIds() {
199
+ const ids = [];
200
+ for (const [threadId, run] of this.runs) {
201
+ if (run.status === 'running')
202
+ ids.push(threadId);
203
+ }
204
+ return ids;
205
+ }
206
+ destroy() {
207
+ if (this.cleanupTimer) {
208
+ clearInterval(this.cleanupTimer);
209
+ this.cleanupTimer = null;
210
+ }
211
+ for (const [threadId] of this.runs) {
212
+ this.removeRun(threadId);
213
+ }
214
+ }
215
+ // ---- 内部方法 ----
216
+ upsertCheckpoint(run, checkpoint) {
217
+ const idx = run.checkpoints.findIndex(item => item.id === checkpoint.id);
218
+ if (idx >= 0) {
219
+ run.checkpoints[idx] = checkpoint;
220
+ }
221
+ else {
222
+ run.checkpoints.push(checkpoint);
223
+ }
224
+ }
225
+ notifySubscribers(run, evt) {
226
+ for (const sub of run.subscribers) {
227
+ try {
228
+ sub(evt);
229
+ }
230
+ catch { /* 订阅者异常不影响其他订阅者 */ }
231
+ }
232
+ }
233
+ toSnapshot(run) {
234
+ const { subscribers: _subscribers, ...serializable } = run;
235
+ return { ...serializable, updatedAt: Date.now() };
236
+ }
237
+ persistRun(run) {
238
+ this.runStateStore.upsert(this.toSnapshot(run));
239
+ }
240
+ startCleanup() {
241
+ if (this.cleanupTimer)
242
+ return;
243
+ this.cleanupTimer = setInterval(() => {
244
+ const now = Date.now();
245
+ for (const [threadId, run] of this.runs) {
246
+ if (run.status !== 'running' && (now - run.createdAt) > CLEANUP_MAX_AGE) {
247
+ this.removeRun(threadId);
248
+ }
249
+ }
250
+ }, CLEANUP_INTERVAL);
251
+ if (this.cleanupTimer && typeof this.cleanupTimer === 'object' && 'unref' in this.cleanupTimer) {
252
+ this.cleanupTimer.unref();
253
+ }
254
+ }
255
+ }
256
+ //# sourceMappingURL=run-registry.js.map