@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.
- package/README.md +4 -3
- package/dist/cjs/a2a/client.d.ts +47 -0
- package/dist/cjs/a2a/client.d.ts.map +1 -0
- package/dist/cjs/a2a/client.js +178 -0
- package/dist/cjs/a2a/client.js.map +1 -0
- package/dist/cjs/a2a/index.d.ts +3 -0
- package/dist/cjs/a2a/index.d.ts.map +1 -0
- package/dist/cjs/a2a/index.js +9 -0
- package/dist/cjs/a2a/index.js.map +1 -0
- package/dist/cjs/a2a/types.d.ts +149 -0
- package/dist/cjs/a2a/types.d.ts.map +1 -0
- package/dist/cjs/a2a/types.js +8 -0
- package/dist/cjs/a2a/types.js.map +1 -0
- package/dist/cjs/index.d.ts +22 -0
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/index.js +52 -2
- package/dist/cjs/index.js.map +1 -1
- package/dist/cjs/middleware/audit-log.d.ts +22 -0
- package/dist/cjs/middleware/audit-log.d.ts.map +1 -0
- package/dist/cjs/middleware/audit-log.js +104 -0
- package/dist/cjs/middleware/audit-log.js.map +1 -0
- package/dist/cjs/middleware/timing.d.ts +25 -0
- package/dist/cjs/middleware/timing.d.ts.map +1 -0
- package/dist/cjs/middleware/timing.js +62 -0
- package/dist/cjs/middleware/timing.js.map +1 -0
- package/dist/cjs/middleware/tool-guard.d.ts +15 -0
- package/dist/cjs/middleware/tool-guard.d.ts.map +1 -0
- package/dist/cjs/middleware/tool-guard.js +31 -0
- package/dist/cjs/middleware/tool-guard.js.map +1 -0
- package/dist/cjs/observability/logger.d.ts +16 -0
- package/dist/cjs/observability/logger.d.ts.map +1 -0
- package/dist/cjs/observability/logger.js +35 -0
- package/dist/cjs/observability/logger.js.map +1 -0
- package/dist/cjs/pool/engine-pool.d.ts +22 -0
- package/dist/cjs/pool/engine-pool.d.ts.map +1 -0
- package/dist/cjs/pool/engine-pool.js +75 -0
- package/dist/cjs/pool/engine-pool.js.map +1 -0
- package/dist/cjs/runtime/run-registry.d.ts +104 -0
- package/dist/cjs/runtime/run-registry.d.ts.map +1 -0
- package/dist/cjs/runtime/run-registry.js +261 -0
- package/dist/cjs/runtime/run-registry.js.map +1 -0
- package/dist/cjs/sandbox/types.d.ts +1 -1
- package/dist/cjs/sandbox/types.d.ts.map +1 -1
- package/dist/cjs/security/guard.js +2 -2
- package/dist/cjs/security/guard.js.map +1 -1
- package/dist/cjs/security/permission.d.ts +39 -0
- package/dist/cjs/security/permission.d.ts.map +1 -0
- package/dist/cjs/security/permission.js +111 -0
- package/dist/cjs/security/permission.js.map +1 -0
- package/dist/cjs/security/risk.d.ts +28 -0
- package/dist/cjs/security/risk.d.ts.map +1 -0
- package/dist/cjs/security/risk.js +148 -0
- package/dist/cjs/security/risk.js.map +1 -0
- package/dist/cjs/security/types.d.ts +1 -1
- package/dist/cjs/security/types.d.ts.map +1 -1
- package/dist/cjs/store/file-tool-result-store.d.ts +27 -0
- package/dist/cjs/store/file-tool-result-store.d.ts.map +1 -0
- package/dist/cjs/store/file-tool-result-store.js +166 -0
- package/dist/cjs/store/file-tool-result-store.js.map +1 -0
- package/dist/cjs/store/json-store.d.ts +54 -0
- package/dist/cjs/store/json-store.d.ts.map +1 -0
- package/dist/cjs/store/json-store.js +274 -0
- package/dist/cjs/store/json-store.js.map +1 -0
- package/dist/cjs/utils/data-dir.d.ts +27 -0
- package/dist/cjs/utils/data-dir.d.ts.map +1 -0
- package/dist/cjs/utils/data-dir.js +122 -0
- package/dist/cjs/utils/data-dir.js.map +1 -0
- package/dist/cjs/utils/event-collector.d.ts +13 -0
- package/dist/cjs/utils/event-collector.d.ts.map +1 -0
- package/dist/cjs/utils/event-collector.js +33 -0
- package/dist/cjs/utils/event-collector.js.map +1 -0
- package/dist/esm/a2a/client.d.ts +47 -0
- package/dist/esm/a2a/client.d.ts.map +1 -0
- package/dist/esm/a2a/client.js +174 -0
- package/dist/esm/a2a/client.js.map +1 -0
- package/dist/esm/a2a/index.d.ts +3 -0
- package/dist/esm/a2a/index.d.ts.map +1 -0
- package/dist/esm/a2a/index.js +5 -0
- package/dist/esm/a2a/index.js.map +1 -0
- package/dist/esm/a2a/types.d.ts +149 -0
- package/dist/esm/a2a/types.d.ts.map +1 -0
- package/dist/esm/a2a/types.js +7 -0
- package/dist/esm/a2a/types.js.map +1 -0
- package/dist/esm/index.d.ts +22 -0
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +20 -0
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/middleware/audit-log.d.ts +22 -0
- package/dist/esm/middleware/audit-log.d.ts.map +1 -0
- package/dist/esm/middleware/audit-log.js +67 -0
- package/dist/esm/middleware/audit-log.js.map +1 -0
- package/dist/esm/middleware/timing.d.ts +25 -0
- package/dist/esm/middleware/timing.d.ts.map +1 -0
- package/dist/esm/middleware/timing.js +58 -0
- package/dist/esm/middleware/timing.js.map +1 -0
- package/dist/esm/middleware/tool-guard.d.ts +15 -0
- package/dist/esm/middleware/tool-guard.d.ts.map +1 -0
- package/dist/esm/middleware/tool-guard.js +27 -0
- package/dist/esm/middleware/tool-guard.js.map +1 -0
- package/dist/esm/observability/logger.d.ts +16 -0
- package/dist/esm/observability/logger.d.ts.map +1 -0
- package/dist/esm/observability/logger.js +31 -0
- package/dist/esm/observability/logger.js.map +1 -0
- package/dist/esm/pool/engine-pool.d.ts +22 -0
- package/dist/esm/pool/engine-pool.d.ts.map +1 -0
- package/dist/esm/pool/engine-pool.js +71 -0
- package/dist/esm/pool/engine-pool.js.map +1 -0
- package/dist/esm/runtime/run-registry.d.ts +104 -0
- package/dist/esm/runtime/run-registry.d.ts.map +1 -0
- package/dist/esm/runtime/run-registry.js +256 -0
- package/dist/esm/runtime/run-registry.js.map +1 -0
- package/dist/esm/sandbox/types.d.ts +1 -1
- package/dist/esm/sandbox/types.d.ts.map +1 -1
- package/dist/esm/security/guard.js +2 -2
- package/dist/esm/security/guard.js.map +1 -1
- package/dist/esm/security/permission.d.ts +39 -0
- package/dist/esm/security/permission.d.ts.map +1 -0
- package/dist/esm/security/permission.js +102 -0
- package/dist/esm/security/permission.js.map +1 -0
- package/dist/esm/security/risk.d.ts +28 -0
- package/dist/esm/security/risk.d.ts.map +1 -0
- package/dist/esm/security/risk.js +142 -0
- package/dist/esm/security/risk.js.map +1 -0
- package/dist/esm/security/types.d.ts +1 -1
- package/dist/esm/security/types.d.ts.map +1 -1
- package/dist/esm/store/file-tool-result-store.d.ts +27 -0
- package/dist/esm/store/file-tool-result-store.d.ts.map +1 -0
- package/dist/esm/store/file-tool-result-store.js +129 -0
- package/dist/esm/store/file-tool-result-store.js.map +1 -0
- package/dist/esm/store/json-store.d.ts +54 -0
- package/dist/esm/store/json-store.d.ts.map +1 -0
- package/dist/esm/store/json-store.js +236 -0
- package/dist/esm/store/json-store.js.map +1 -0
- package/dist/esm/utils/data-dir.d.ts +27 -0
- package/dist/esm/utils/data-dir.d.ts.map +1 -0
- package/dist/esm/utils/data-dir.js +80 -0
- package/dist/esm/utils/data-dir.js.map +1 -0
- package/dist/esm/utils/event-collector.d.ts +13 -0
- package/dist/esm/utils/event-collector.d.ts.map +1 -0
- package/dist/esm/utils/event-collector.js +30 -0
- package/dist/esm/utils/event-collector.js.map +1 -0
- 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
|