@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 @@
|
|
|
1
|
+
{"version":3,"file":"audit-log.d.ts","sourceRoot":"","sources":["../../../src/middleware/audit-log.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AA2BjE;;;;GAIG;AACH,qBAAa,kBAAmB,YAAW,UAAU;IACnD,IAAI,SAAe;IAEnB,OAAO,CAAC,MAAM,CAAwB;IAEtC;;;;OAIG;gBACS,MAAM,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,MAAM,KAAK,IAAI,CAAC,GAAG,SAAS,GAAG,MAAM;IAY3D,MAAM,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7C,KAAK,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5C,YAAY,CAAC,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAI1D,WAAW,CAAC,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;CAIhE"}
|
|
@@ -0,0 +1,104 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// ============================================================
|
|
3
|
+
// AuditLogMiddleware — 审计日志中间件
|
|
4
|
+
// ============================================================
|
|
5
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
6
|
+
if (k2 === undefined) k2 = k;
|
|
7
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
8
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
9
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
10
|
+
}
|
|
11
|
+
Object.defineProperty(o, k2, desc);
|
|
12
|
+
}) : (function(o, m, k, k2) {
|
|
13
|
+
if (k2 === undefined) k2 = k;
|
|
14
|
+
o[k2] = m[k];
|
|
15
|
+
}));
|
|
16
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
17
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
18
|
+
}) : function(o, v) {
|
|
19
|
+
o["default"] = v;
|
|
20
|
+
});
|
|
21
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
22
|
+
var ownKeys = function(o) {
|
|
23
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
24
|
+
var ar = [];
|
|
25
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
26
|
+
return ar;
|
|
27
|
+
};
|
|
28
|
+
return ownKeys(o);
|
|
29
|
+
};
|
|
30
|
+
return function (mod) {
|
|
31
|
+
if (mod && mod.__esModule) return mod;
|
|
32
|
+
var result = {};
|
|
33
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
34
|
+
__setModuleDefault(result, mod);
|
|
35
|
+
return result;
|
|
36
|
+
};
|
|
37
|
+
})();
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
exports.AuditLogMiddleware = void 0;
|
|
40
|
+
const fs = __importStar(require("fs"));
|
|
41
|
+
const path = __importStar(require("path"));
|
|
42
|
+
/**
|
|
43
|
+
* 文件审计日志器 — 以 JSONL 格式追加写入日志文件
|
|
44
|
+
* 默认路径:{logDir}/audit-YYYY-MM-DD.log
|
|
45
|
+
*/
|
|
46
|
+
function createFileAuditLogger(logDir) {
|
|
47
|
+
const dir = logDir || path.join(process.cwd(), 'data', 'logs');
|
|
48
|
+
if (!fs.existsSync(dir)) {
|
|
49
|
+
fs.mkdirSync(dir, { recursive: true });
|
|
50
|
+
}
|
|
51
|
+
return (msg) => {
|
|
52
|
+
const now = new Date();
|
|
53
|
+
const dateStr = now.toISOString().slice(0, 10);
|
|
54
|
+
const logFile = path.join(dir, `audit-${dateStr}.log`);
|
|
55
|
+
const entry = JSON.stringify({
|
|
56
|
+
ts: now.toISOString(),
|
|
57
|
+
msg,
|
|
58
|
+
}) + '\n';
|
|
59
|
+
fs.appendFileSync(logFile, entry, 'utf-8');
|
|
60
|
+
};
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* 记录每轮迭代和工具调用的审计日志
|
|
64
|
+
* - before/after: 记录迭代开始/结束 + 消息数量
|
|
65
|
+
* - onToolBefore/onToolAfter: 记录工具调用名称、参数、耗时
|
|
66
|
+
*/
|
|
67
|
+
class AuditLogMiddleware {
|
|
68
|
+
name = 'audit-log';
|
|
69
|
+
logger;
|
|
70
|
+
/**
|
|
71
|
+
* @param logger 自定义日志输出函数,默认写入文件 data/logs/audit-YYYY-MM-DD.log
|
|
72
|
+
* 传入 'console' 字符串则输出到 console.log
|
|
73
|
+
* 也可传入自定义函数
|
|
74
|
+
*/
|
|
75
|
+
constructor(logger) {
|
|
76
|
+
if (logger === 'console') {
|
|
77
|
+
this.logger = (msg) => console.log(`[Audit] ${msg}`);
|
|
78
|
+
}
|
|
79
|
+
else if (typeof logger === 'function') {
|
|
80
|
+
this.logger = logger;
|
|
81
|
+
}
|
|
82
|
+
else if (typeof logger === 'string') {
|
|
83
|
+
this.logger = createFileAuditLogger(logger);
|
|
84
|
+
}
|
|
85
|
+
else {
|
|
86
|
+
this.logger = createFileAuditLogger();
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
async before(ctx) {
|
|
90
|
+
this.logger(`Session=${ctx.sessionId} Iter=${ctx.iteration} START | messages=${ctx.messages.length} tools=${ctx.toolCount}`);
|
|
91
|
+
}
|
|
92
|
+
async after(ctx) {
|
|
93
|
+
this.logger(`Session=${ctx.sessionId} Iter=${ctx.iteration} END | messages=${ctx.messages.length}`);
|
|
94
|
+
}
|
|
95
|
+
async onToolBefore(ctx) {
|
|
96
|
+
this.logger(`Tool CALL: ${ctx.tool} | args=${JSON.stringify(ctx.args || {}).slice(0, 200)}`);
|
|
97
|
+
}
|
|
98
|
+
async onToolAfter(ctx) {
|
|
99
|
+
const output = ctx.result ? String(ctx.result).slice(0, 200) : '(empty)';
|
|
100
|
+
this.logger(`Tool DONE: ${ctx.tool} | output=${output}`);
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
exports.AuditLogMiddleware = AuditLogMiddleware;
|
|
104
|
+
//# sourceMappingURL=audit-log.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"audit-log.js","sourceRoot":"","sources":["../../../src/middleware/audit-log.ts"],"names":[],"mappings":";AAAA,+DAA+D;AAC/D,+BAA+B;AAC/B,+DAA+D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAI/D,uCAAyB;AACzB,2CAA6B;AAE7B;;;GAGG;AACH,SAAS,qBAAqB,CAAC,MAAe;IAC5C,MAAM,GAAG,GAAG,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IAE/D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,CAAC,GAAW,EAAE,EAAE;QACrB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,OAAO,MAAM,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;YAC3B,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE;YACrB,GAAG;SACJ,CAAC,GAAG,IAAI,CAAC;QACV,EAAE,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAa,kBAAkB;IAC7B,IAAI,GAAG,WAAW,CAAC;IAEX,MAAM,CAAwB;IAEtC;;;;OAIG;IACH,YAAY,MAAqD;QAC/D,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,EAAE,CAAC,CAAC;QACvD,CAAC;aAAM,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACvB,CAAC;aAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YACtC,IAAI,CAAC,MAAM,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,GAAG,qBAAqB,EAAE,CAAC;QACxC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAsB;QACjC,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,SAAS,SAAS,GAAG,CAAC,SAAS,qBAAqB,GAAG,CAAC,QAAQ,CAAC,MAAM,UAAU,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;IAC/H,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAsB;QAChC,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,SAAS,SAAS,GAAG,CAAC,SAAS,mBAAmB,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACtG,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAgB;QACjC,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,CAAC,IAAI,WAAW,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/F,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAgB;QAChC,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACzE,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,CAAC,IAAI,aAAa,MAAM,EAAE,CAAC,CAAC;IAC3D,CAAC;CACF;AAtCD,gDAsCC"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import type { Middleware, MiddlewareContext } from '../middleware/types.ts';
|
|
2
|
+
import type { HookContext, HookResult } from '../hooks/types.ts';
|
|
3
|
+
export interface TimingReport {
|
|
4
|
+
type: 'iteration' | 'tool';
|
|
5
|
+
sessionId?: string;
|
|
6
|
+
iteration?: number;
|
|
7
|
+
toolName?: string;
|
|
8
|
+
elapsedMs: number;
|
|
9
|
+
}
|
|
10
|
+
/**
|
|
11
|
+
* 统计每轮迭代和每次工具调用的耗时
|
|
12
|
+
* 可通过 onReport 回调获取汇总数据(如写入监控)
|
|
13
|
+
*/
|
|
14
|
+
export declare class TimingMiddleware implements Middleware {
|
|
15
|
+
name: string;
|
|
16
|
+
private iterStarts;
|
|
17
|
+
private toolTimings;
|
|
18
|
+
private onReport?;
|
|
19
|
+
constructor(onReport?: (report: TimingReport) => void);
|
|
20
|
+
before(ctx: MiddlewareContext): Promise<void>;
|
|
21
|
+
after(ctx: MiddlewareContext): Promise<void>;
|
|
22
|
+
onToolBefore(ctx: HookContext): Promise<HookResult | void>;
|
|
23
|
+
onToolAfter(ctx: HookContext): Promise<HookResult | void>;
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=timing.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"timing.d.ts","sourceRoot":"","sources":["../../../src/middleware/timing.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEjE,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,WAAW,GAAG,MAAM,CAAC;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAOD;;;GAGG;AACH,qBAAa,gBAAiB,YAAW,UAAU;IACjD,IAAI,SAAY;IAEhB,OAAO,CAAC,UAAU,CAA6B;IAC/C,OAAO,CAAC,WAAW,CAAiC;IACpD,OAAO,CAAC,QAAQ,CAAC,CAAiC;gBAEtC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,IAAI;IAI/C,MAAM,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7C,KAAK,CAAC,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAiB5C,YAAY,CAAC,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAM1D,WAAW,CAAC,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;CAgBhE"}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// ============================================================
|
|
3
|
+
// TimingMiddleware — 执行耗时统计中间件
|
|
4
|
+
// ============================================================
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.TimingMiddleware = void 0;
|
|
7
|
+
/**
|
|
8
|
+
* 统计每轮迭代和每次工具调用的耗时
|
|
9
|
+
* 可通过 onReport 回调获取汇总数据(如写入监控)
|
|
10
|
+
*/
|
|
11
|
+
class TimingMiddleware {
|
|
12
|
+
name = 'timing';
|
|
13
|
+
iterStarts = new Map();
|
|
14
|
+
toolTimings = new Map();
|
|
15
|
+
onReport;
|
|
16
|
+
constructor(onReport) {
|
|
17
|
+
this.onReport = onReport;
|
|
18
|
+
}
|
|
19
|
+
async before(ctx) {
|
|
20
|
+
this.iterStarts.set(ctx.sessionId, Date.now());
|
|
21
|
+
}
|
|
22
|
+
async after(ctx) {
|
|
23
|
+
const start = this.iterStarts.get(ctx.sessionId);
|
|
24
|
+
const elapsed = start ? Date.now() - start : -1;
|
|
25
|
+
this.iterStarts.delete(ctx.sessionId);
|
|
26
|
+
if (this.onReport) {
|
|
27
|
+
this.onReport({
|
|
28
|
+
type: 'iteration',
|
|
29
|
+
sessionId: ctx.sessionId,
|
|
30
|
+
iteration: ctx.iteration,
|
|
31
|
+
elapsedMs: elapsed,
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
else {
|
|
35
|
+
console.log(`[Timing] Session=${ctx.sessionId} Iter=${ctx.iteration} elapsed=${elapsed}ms`);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
async onToolBefore(ctx) {
|
|
39
|
+
const key = `${ctx.tool}_${Date.now()}`;
|
|
40
|
+
this.toolTimings.set(key, { name: ctx.tool || 'unknown', start: Date.now() });
|
|
41
|
+
ctx._timingKey = key;
|
|
42
|
+
}
|
|
43
|
+
async onToolAfter(ctx) {
|
|
44
|
+
const key = ctx._timingKey;
|
|
45
|
+
const timing = key ? this.toolTimings.get(key) : undefined;
|
|
46
|
+
const elapsed = timing ? Date.now() - timing.start : -1;
|
|
47
|
+
if (key)
|
|
48
|
+
this.toolTimings.delete(key);
|
|
49
|
+
if (this.onReport) {
|
|
50
|
+
this.onReport({
|
|
51
|
+
type: 'tool',
|
|
52
|
+
toolName: ctx.tool,
|
|
53
|
+
elapsedMs: elapsed,
|
|
54
|
+
});
|
|
55
|
+
}
|
|
56
|
+
else {
|
|
57
|
+
console.log(`[Timing] Tool=${ctx.tool} elapsed=${elapsed}ms`);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
exports.TimingMiddleware = TimingMiddleware;
|
|
62
|
+
//# 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,MAAa,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;AAtDD,4CAsDC"}
|
|
@@ -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,31 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// ============================================================
|
|
3
|
+
// ToolGuardMiddleware — 工具安全守卫中间件
|
|
4
|
+
// ============================================================
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.ToolGuardMiddleware = void 0;
|
|
7
|
+
/**
|
|
8
|
+
* 工具调用守卫:对指定工具进行拦截
|
|
9
|
+
* - 可配置需要拦截的工具名列表
|
|
10
|
+
* - 匹配时 onToolBefore 返回 block,阻止工具执行
|
|
11
|
+
*/
|
|
12
|
+
class ToolGuardMiddleware {
|
|
13
|
+
name = 'tool-guard';
|
|
14
|
+
blockedTools;
|
|
15
|
+
logger;
|
|
16
|
+
constructor(blockedTools, logger) {
|
|
17
|
+
this.blockedTools = new Set(blockedTools);
|
|
18
|
+
this.logger = logger || ((msg) => console.warn(`[ToolGuard] ${msg}`));
|
|
19
|
+
}
|
|
20
|
+
async onToolBefore(ctx) {
|
|
21
|
+
if (ctx.tool && this.blockedTools.has(ctx.tool)) {
|
|
22
|
+
this.logger(`BLOCKED tool call: ${ctx.tool}`);
|
|
23
|
+
return {
|
|
24
|
+
block: true,
|
|
25
|
+
reason: `Tool "${ctx.tool}" is blocked by ToolGuard middleware`,
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
exports.ToolGuardMiddleware = ToolGuardMiddleware;
|
|
31
|
+
//# 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,MAAa,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;AApBD,kDAoBC"}
|
|
@@ -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,35 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// ============================================================
|
|
3
|
+
// Observability — 结构化可观测性日志
|
|
4
|
+
// ============================================================
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.setLogTransport = setLogTransport;
|
|
7
|
+
exports.logEvent = logEvent;
|
|
8
|
+
/** 默认传输:console 输出 */
|
|
9
|
+
const defaultTransport = (payload) => {
|
|
10
|
+
const line = `[obs] ${JSON.stringify(payload)}`;
|
|
11
|
+
if (payload.level === 'error')
|
|
12
|
+
console.error(line);
|
|
13
|
+
else if (payload.level === 'warn')
|
|
14
|
+
console.warn(line);
|
|
15
|
+
else
|
|
16
|
+
console.log(line);
|
|
17
|
+
};
|
|
18
|
+
let _transport = defaultTransport;
|
|
19
|
+
/** 设置自定义日志传输层 */
|
|
20
|
+
function setLogTransport(transport) {
|
|
21
|
+
_transport = transport;
|
|
22
|
+
}
|
|
23
|
+
/** 记录可观测性事件 */
|
|
24
|
+
function logEvent(input) {
|
|
25
|
+
const payload = {
|
|
26
|
+
ts: input.ts || Date.now(),
|
|
27
|
+
level: input.level || 'info',
|
|
28
|
+
event: input.event,
|
|
29
|
+
traceId: input.traceId,
|
|
30
|
+
sessionId: input.sessionId,
|
|
31
|
+
data: input.data || {},
|
|
32
|
+
};
|
|
33
|
+
_transport(payload);
|
|
34
|
+
}
|
|
35
|
+
//# 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;;AA2B/D,0CAEC;AAGD,4BAUC;AA1BD,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,SAAgB,eAAe,CAAC,SAAuB;IACrD,UAAU,GAAG,SAAS,CAAC;AACzB,CAAC;AAED,eAAe;AACf,SAAgB,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,75 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
// ============================================================
|
|
3
|
+
// EnginePool — AgentEngine 实例复用池
|
|
4
|
+
// ============================================================
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.EnginePool = void 0;
|
|
7
|
+
/** AgentEngine 实例复用池 */
|
|
8
|
+
class EnginePool {
|
|
9
|
+
pool = new Map();
|
|
10
|
+
ttl;
|
|
11
|
+
maxSize;
|
|
12
|
+
constructor(ttlMs = 5 * 60_000, maxSize = 10) {
|
|
13
|
+
this.ttl = ttlMs;
|
|
14
|
+
this.maxSize = maxSize;
|
|
15
|
+
}
|
|
16
|
+
/** 生成缓存 key */
|
|
17
|
+
static key(profileId, scopeKey) {
|
|
18
|
+
return `${profileId ?? '_'}:${scopeKey || '_'}`;
|
|
19
|
+
}
|
|
20
|
+
/** 从池中获取 Engine(未命中或过期返回 null) */
|
|
21
|
+
get(profileId, scopeKey) {
|
|
22
|
+
const key = EnginePool.key(profileId, scopeKey);
|
|
23
|
+
const entry = this.pool.get(key);
|
|
24
|
+
if (!entry)
|
|
25
|
+
return null;
|
|
26
|
+
if (Date.now() - entry.createdAt > this.ttl) {
|
|
27
|
+
entry.engine.destroy();
|
|
28
|
+
this.pool.delete(key);
|
|
29
|
+
return null;
|
|
30
|
+
}
|
|
31
|
+
return entry.engine;
|
|
32
|
+
}
|
|
33
|
+
/** 将 Engine 放入池中 */
|
|
34
|
+
set(profileId, engine, scopeKey) {
|
|
35
|
+
const key = EnginePool.key(profileId, scopeKey);
|
|
36
|
+
const old = this.pool.get(key);
|
|
37
|
+
if (old)
|
|
38
|
+
old.engine.destroy();
|
|
39
|
+
if (this.pool.size >= this.maxSize && !this.pool.has(key)) {
|
|
40
|
+
this.evictOldest();
|
|
41
|
+
}
|
|
42
|
+
this.pool.set(key, { engine, createdAt: Date.now() });
|
|
43
|
+
}
|
|
44
|
+
/** 清除所有缓存(配置变更时调用) */
|
|
45
|
+
clear() {
|
|
46
|
+
for (const entry of this.pool.values()) {
|
|
47
|
+
entry.engine.destroy();
|
|
48
|
+
}
|
|
49
|
+
this.pool.clear();
|
|
50
|
+
}
|
|
51
|
+
/** 仅清除过期的实例 */
|
|
52
|
+
purgeExpired() {
|
|
53
|
+
const now = Date.now();
|
|
54
|
+
for (const [key, entry] of this.pool) {
|
|
55
|
+
if (now - entry.createdAt > this.ttl) {
|
|
56
|
+
entry.engine.destroy();
|
|
57
|
+
this.pool.delete(key);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
/** 池大小 */
|
|
62
|
+
get size() {
|
|
63
|
+
return this.pool.size;
|
|
64
|
+
}
|
|
65
|
+
evictOldest() {
|
|
66
|
+
const oldestKey = this.pool.keys().next().value;
|
|
67
|
+
if (oldestKey !== undefined) {
|
|
68
|
+
const entry = this.pool.get(oldestKey);
|
|
69
|
+
entry?.engine.destroy();
|
|
70
|
+
this.pool.delete(oldestKey);
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
exports.EnginePool = EnginePool;
|
|
75
|
+
//# 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,MAAa,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;AA5ED,gCA4EC"}
|
|
@@ -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"}
|