@cicctencent/agent-core 0.1.16 → 0.1.18
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 +1 -0
- package/dist/cjs/a2a/index.d.ts +2 -0
- package/dist/cjs/a2a/index.d.ts.map +1 -1
- package/dist/cjs/a2a/index.js +7 -1
- package/dist/cjs/a2a/index.js.map +1 -1
- package/dist/cjs/a2a/server.d.ts +50 -0
- package/dist/cjs/a2a/server.d.ts.map +1 -0
- package/dist/cjs/a2a/server.js +173 -0
- package/dist/cjs/a2a/server.js.map +1 -0
- package/dist/cjs/agent/runtime-builder.d.ts +83 -0
- package/dist/cjs/agent/runtime-builder.d.ts.map +1 -0
- package/dist/cjs/agent/runtime-builder.js +127 -0
- package/dist/cjs/agent/runtime-builder.js.map +1 -0
- package/dist/cjs/index.d.ts +10 -2
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/index.js +26 -2
- package/dist/cjs/index.js.map +1 -1
- package/dist/cjs/runtime/run-controller.d.ts +40 -0
- package/dist/cjs/runtime/run-controller.d.ts.map +1 -0
- package/dist/cjs/runtime/run-controller.js +93 -0
- package/dist/cjs/runtime/run-controller.js.map +1 -0
- package/dist/cjs/sandbox/local-sandbox.d.ts +59 -0
- package/dist/cjs/sandbox/local-sandbox.d.ts.map +1 -0
- package/dist/cjs/sandbox/local-sandbox.js +228 -0
- package/dist/cjs/sandbox/local-sandbox.js.map +1 -0
- package/dist/cjs/tool/builtin-tools.d.ts +22 -0
- package/dist/cjs/tool/builtin-tools.d.ts.map +1 -0
- package/dist/cjs/tool/builtin-tools.js +221 -0
- package/dist/cjs/tool/builtin-tools.js.map +1 -0
- package/dist/cjs/tool/registry.d.ts +25 -0
- package/dist/cjs/tool/registry.d.ts.map +1 -1
- package/dist/cjs/tool/registry.js +53 -0
- package/dist/cjs/tool/registry.js.map +1 -1
- package/dist/cjs/utils/event-adapter.d.ts +25 -0
- package/dist/cjs/utils/event-adapter.d.ts.map +1 -0
- package/dist/cjs/utils/event-adapter.js +93 -0
- package/dist/cjs/utils/event-adapter.js.map +1 -0
- package/dist/esm/a2a/index.d.ts +2 -0
- package/dist/esm/a2a/index.d.ts.map +1 -1
- package/dist/esm/a2a/index.js +1 -0
- package/dist/esm/a2a/index.js.map +1 -1
- package/dist/esm/a2a/server.d.ts +50 -0
- package/dist/esm/a2a/server.d.ts.map +1 -0
- package/dist/esm/a2a/server.js +165 -0
- package/dist/esm/a2a/server.js.map +1 -0
- package/dist/esm/agent/runtime-builder.d.ts +83 -0
- package/dist/esm/agent/runtime-builder.d.ts.map +1 -0
- package/dist/esm/agent/runtime-builder.js +120 -0
- package/dist/esm/agent/runtime-builder.js.map +1 -0
- package/dist/esm/index.d.ts +10 -2
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +5 -1
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/runtime/run-controller.d.ts +40 -0
- package/dist/esm/runtime/run-controller.d.ts.map +1 -0
- package/dist/esm/runtime/run-controller.js +89 -0
- package/dist/esm/runtime/run-controller.js.map +1 -0
- package/dist/esm/sandbox/local-sandbox.d.ts +59 -0
- package/dist/esm/sandbox/local-sandbox.d.ts.map +1 -0
- package/dist/esm/sandbox/local-sandbox.js +188 -0
- package/dist/esm/sandbox/local-sandbox.js.map +1 -0
- package/dist/esm/tool/builtin-tools.d.ts +22 -0
- package/dist/esm/tool/builtin-tools.d.ts.map +1 -0
- package/dist/esm/tool/builtin-tools.js +214 -0
- package/dist/esm/tool/builtin-tools.js.map +1 -0
- package/dist/esm/tool/registry.d.ts +25 -0
- package/dist/esm/tool/registry.d.ts.map +1 -1
- package/dist/esm/tool/registry.js +53 -0
- package/dist/esm/tool/registry.js.map +1 -1
- package/dist/esm/utils/event-adapter.d.ts +25 -0
- package/dist/esm/utils/event-adapter.d.ts.map +1 -0
- package/dist/esm/utils/event-adapter.js +87 -0
- package/dist/esm/utils/event-adapter.js.map +1 -0
- package/package.json +1 -1
|
@@ -0,0 +1,89 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// RunController — Engine 执行、事件映射、RunRegistry、持久化 Sink 编排
|
|
3
|
+
// ============================================================
|
|
4
|
+
import { RunRegistry } from "./run-registry.js";
|
|
5
|
+
import { createAgentEventCollectorState, mapAgentEvent } from "../utils/event-adapter.js";
|
|
6
|
+
export class RunController {
|
|
7
|
+
registry;
|
|
8
|
+
persistence;
|
|
9
|
+
eventAdapter;
|
|
10
|
+
flushIntervalMs;
|
|
11
|
+
flushChars;
|
|
12
|
+
constructor(options = {}) {
|
|
13
|
+
this.registry = options.registry || new RunRegistry();
|
|
14
|
+
this.persistence = options.persistence;
|
|
15
|
+
this.eventAdapter = options.eventAdapter || {};
|
|
16
|
+
this.flushIntervalMs = options.flushIntervalMs ?? 5000;
|
|
17
|
+
this.flushChars = options.flushChars ?? 500;
|
|
18
|
+
}
|
|
19
|
+
subscribe(threadId, callback) {
|
|
20
|
+
return this.registry.subscribe(threadId, callback);
|
|
21
|
+
}
|
|
22
|
+
getBufferedEvents(threadId) {
|
|
23
|
+
return this.registry.getBufferedEvents(threadId);
|
|
24
|
+
}
|
|
25
|
+
async run(engine, options) {
|
|
26
|
+
const run = this.registry.startRun(options.threadId, options.sessionId, options.traceId);
|
|
27
|
+
const emitted = [];
|
|
28
|
+
let lastFlushedLen = 0;
|
|
29
|
+
let flushTimer = null;
|
|
30
|
+
let terminalStatus = 'done';
|
|
31
|
+
let terminalError;
|
|
32
|
+
const isCurrentRun = () => this.registry.getRun(options.threadId) === run;
|
|
33
|
+
await this.persistence?.onRunStart?.(run);
|
|
34
|
+
const flush = async () => {
|
|
35
|
+
if (!isCurrentRun())
|
|
36
|
+
return;
|
|
37
|
+
if (run.fullContent.length === lastFlushedLen && run.status === 'running')
|
|
38
|
+
return;
|
|
39
|
+
lastFlushedLen = run.fullContent.length;
|
|
40
|
+
run.lastFlushAt = Date.now();
|
|
41
|
+
await this.persistence?.onFlush?.(run);
|
|
42
|
+
};
|
|
43
|
+
flushTimer = setInterval(() => { flush().catch(() => undefined); }, this.flushIntervalMs);
|
|
44
|
+
if (typeof flushTimer === 'object' && 'unref' in flushTimer)
|
|
45
|
+
flushTimer.unref();
|
|
46
|
+
const state = createAgentEventCollectorState();
|
|
47
|
+
const adapterOptions = { ...this.eventAdapter, toolRegistry: options.toolRegistry || this.eventAdapter.toolRegistry };
|
|
48
|
+
try {
|
|
49
|
+
for await (const event of engine.run(options)) {
|
|
50
|
+
if (!isCurrentRun())
|
|
51
|
+
break;
|
|
52
|
+
const mapped = mapAgentEvent(event, state, adapterOptions);
|
|
53
|
+
if (!mapped)
|
|
54
|
+
continue;
|
|
55
|
+
emitted.push(mapped);
|
|
56
|
+
this.registry.addEvent(options.threadId, mapped.event, mapped.data);
|
|
57
|
+
const buffered = { event: mapped.event, data: mapped.data };
|
|
58
|
+
await this.persistence?.onEvent?.(run, buffered, event);
|
|
59
|
+
if (event.type === 'error') {
|
|
60
|
+
terminalStatus = 'error';
|
|
61
|
+
terminalError = event.error || 'Agent execution failed';
|
|
62
|
+
}
|
|
63
|
+
if (run.fullContent.length - lastFlushedLen >= this.flushChars) {
|
|
64
|
+
await flush();
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
await flush();
|
|
68
|
+
if (isCurrentRun()) {
|
|
69
|
+
this.registry.finishRun(options.threadId, terminalStatus, terminalError);
|
|
70
|
+
await this.persistence?.onRunFinish?.(run, terminalStatus, terminalError);
|
|
71
|
+
}
|
|
72
|
+
return { run, events: emitted };
|
|
73
|
+
}
|
|
74
|
+
catch (err) {
|
|
75
|
+
const message = err?.message || String(err);
|
|
76
|
+
if (isCurrentRun()) {
|
|
77
|
+
this.registry.addEvent(options.threadId, 'error', { error: message });
|
|
78
|
+
this.registry.finishRun(options.threadId, 'error', message);
|
|
79
|
+
await this.persistence?.onRunFinish?.(run, 'error', message);
|
|
80
|
+
}
|
|
81
|
+
return { run, events: emitted };
|
|
82
|
+
}
|
|
83
|
+
finally {
|
|
84
|
+
if (flushTimer)
|
|
85
|
+
clearInterval(flushTimer);
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
//# sourceMappingURL=run-controller.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"run-controller.js","sourceRoot":"","sources":["../../../src/runtime/run-controller.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,yDAAyD;AACzD,+DAA+D;AAM/D,OAAO,EAAE,WAAW,EAAsC,MAAM,mBAAmB,CAAC;AACpF,OAAO,EAAE,8BAA8B,EAAE,aAAa,EAAwD,MAAM,2BAA2B,CAAC;AA4BhJ,MAAM,OAAO,aAAa;IACf,QAAQ,CAAc;IACvB,WAAW,CAAyB;IACpC,YAAY,CAA2B;IACvC,eAAe,CAAS;IACxB,UAAU,CAAS;IAE3B,YAAY,UAAgC,EAAE;QAC5C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,WAAW,EAAE,CAAC;QACtD,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,IAAI,CAAC;QACvD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,GAAG,CAAC;IAC9C,CAAC;IAED,SAAS,CAAC,QAAyB,EAAE,QAAwC;QAC3E,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED,iBAAiB,CAAC,QAAyB;QACzC,OAAO,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAmB,EAAE,OAAyB;QACtD,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACzF,MAAM,OAAO,GAAuB,EAAE,CAAC;QACvC,IAAI,cAAc,GAAG,CAAC,CAAC;QACvB,IAAI,UAAU,GAA0C,IAAI,CAAC;QAC7D,IAAI,cAAc,GAAqB,MAAM,CAAC;QAC9C,IAAI,aAAiC,CAAC;QAEtC,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC;QAE1E,MAAM,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC;QAE1C,MAAM,KAAK,GAAG,KAAK,IAAI,EAAE;YACvB,IAAI,CAAC,YAAY,EAAE;gBAAE,OAAO;YAC5B,IAAI,GAAG,CAAC,WAAW,CAAC,MAAM,KAAK,cAAc,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS;gBAAE,OAAO;YAClF,cAAc,GAAG,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC;YACxC,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;QACzC,CAAC,CAAC;QAEF,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1F,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,OAAO,IAAI,UAAU;YAAE,UAAU,CAAC,KAAK,EAAE,CAAC;QAEhF,MAAM,KAAK,GAAG,8BAA8B,EAAE,CAAC;QAC/C,MAAM,cAAc,GAAG,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,CAAC;QAEtH,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC9C,IAAI,CAAC,YAAY,EAAE;oBAAE,MAAM;gBAC3B,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC;gBAC3D,IAAI,CAAC,MAAM;oBAAE,SAAS;gBACtB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpE,MAAM,QAAQ,GAAkB,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC3E,MAAM,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,CAAC,GAAG,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;gBAExD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC3B,cAAc,GAAG,OAAO,CAAC;oBACzB,aAAa,GAAG,KAAK,CAAC,KAAK,IAAI,wBAAwB,CAAC;gBAC1D,CAAC;gBAED,IAAI,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,cAAc,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBAC/D,MAAM,KAAK,EAAE,CAAC;gBAChB,CAAC;YACH,CAAC;YACD,MAAM,KAAK,EAAE,CAAC;YACd,IAAI,YAAY,EAAE,EAAE,CAAC;gBACnB,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,EAAE,aAAa,CAAC,CAAC;gBACzE,MAAM,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,CAAC,GAAG,EAAE,cAAc,EAAE,aAAa,CAAC,CAAC;YAC5E,CAAC;YACD,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAClC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,OAAO,GAAG,GAAG,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC;YAC5C,IAAI,YAAY,EAAE,EAAE,CAAC;gBACnB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;gBACtE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC5D,MAAM,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC/D,CAAC;YACD,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAClC,CAAC;gBAAS,CAAC;YACT,IAAI,UAAU;gBAAE,aAAa,CAAC,UAAU,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
import type { ShellExecutor, FileSystemExecutor } from '../tool/types.js';
|
|
2
|
+
import type { Sandbox, SandboxConfig, SandboxManager, ScriptLanguage } from './types.js';
|
|
3
|
+
export interface LocalSandboxOptions extends Partial<SandboxConfig> {
|
|
4
|
+
allowedReadDirs?: string[];
|
|
5
|
+
env?: Record<string, string | undefined>;
|
|
6
|
+
cleanupOnDestroy?: boolean;
|
|
7
|
+
strictPath?: boolean;
|
|
8
|
+
}
|
|
9
|
+
export declare class LocalSandboxFileSystemExecutor implements FileSystemExecutor {
|
|
10
|
+
private readonly rootDir;
|
|
11
|
+
private readonly maxFileSize;
|
|
12
|
+
private readonly allowedReadDirs;
|
|
13
|
+
private readonly strictPath;
|
|
14
|
+
constructor(rootDir: string, maxFileSize?: number, allowedReadDirs?: string[], strictPath?: boolean);
|
|
15
|
+
readFile(filePath: string): Promise<string>;
|
|
16
|
+
readFileBuffer(filePath: string): Promise<Buffer>;
|
|
17
|
+
writeFile(filePath: string, content: string): Promise<void>;
|
|
18
|
+
appendFile(filePath: string, content: string): Promise<void>;
|
|
19
|
+
listDir(dirPath: string): Promise<string[]>;
|
|
20
|
+
private isInside;
|
|
21
|
+
private safeInRoot;
|
|
22
|
+
private resolveWritePath;
|
|
23
|
+
private resolveReadPath;
|
|
24
|
+
}
|
|
25
|
+
export declare class LocalSandboxShellExecutor implements ShellExecutor {
|
|
26
|
+
private readonly rootDir;
|
|
27
|
+
private readonly commandRestriction;
|
|
28
|
+
private readonly maxBuffer;
|
|
29
|
+
private guard;
|
|
30
|
+
private outputBuffer;
|
|
31
|
+
private env;
|
|
32
|
+
constructor(rootDir: string, commandRestriction?: boolean, env?: Record<string, string | undefined>, maxBuffer?: number);
|
|
33
|
+
exec(command: string, timeout?: number): Promise<string>;
|
|
34
|
+
write(_data: string): void;
|
|
35
|
+
read(lines?: number): string;
|
|
36
|
+
}
|
|
37
|
+
export declare class LocalSandbox implements Sandbox {
|
|
38
|
+
readonly id: string;
|
|
39
|
+
readonly rootDir: string;
|
|
40
|
+
private fsExecutor;
|
|
41
|
+
private shellExecutor;
|
|
42
|
+
private cleanupOnDestroy;
|
|
43
|
+
constructor(id: string, options: LocalSandboxOptions);
|
|
44
|
+
init(): Promise<void>;
|
|
45
|
+
getShellExecutor(): ShellExecutor;
|
|
46
|
+
getFileSystemExecutor(): FileSystemExecutor;
|
|
47
|
+
runScript(language: ScriptLanguage, code: string, timeout?: number): Promise<string>;
|
|
48
|
+
destroy(): Promise<void>;
|
|
49
|
+
}
|
|
50
|
+
export declare class LocalSandboxManager implements SandboxManager {
|
|
51
|
+
private readonly options;
|
|
52
|
+
private sandboxes;
|
|
53
|
+
constructor(options?: LocalSandboxOptions);
|
|
54
|
+
create(id: string, config?: Partial<SandboxConfig>): Promise<Sandbox>;
|
|
55
|
+
get(id: string): Sandbox | undefined;
|
|
56
|
+
destroy(id: string): Promise<void>;
|
|
57
|
+
destroyAll(): Promise<void>;
|
|
58
|
+
}
|
|
59
|
+
//# sourceMappingURL=local-sandbox.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"local-sandbox.d.ts","sourceRoot":"","sources":["../../../src/sandbox/local-sandbox.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAC1E,OAAO,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAGzF,MAAM,WAAW,mBAAoB,SAAQ,OAAO,CAAC,aAAa,CAAC;IACjE,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IACzC,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAgBD,qBAAa,8BAA+B,YAAW,kBAAkB;IAErE,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,UAAU;gBAHV,OAAO,EAAE,MAAM,EACf,WAAW,SAAsC,EACjD,eAAe,GAAE,MAAM,EAAO,EAC9B,UAAU,UAAQ;IAG/B,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAI3C,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAIjD,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ3D,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAM5D,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAKjD,OAAO,CAAC,QAAQ;IAKhB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,eAAe;CAMxB;AAED,qBAAa,yBAA0B,YAAW,aAAa;IAM3D,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IAEnC,OAAO,CAAC,QAAQ,CAAC,SAAS;IAR5B,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,GAAG,CAAyB;gBAGjB,OAAO,EAAE,MAAM,EACf,kBAAkB,UAAO,EAC1C,GAAG,GAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAe,EACpC,SAAS,SAAM;IAM5B,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,SAAsC,GAAG,OAAO,CAAC,MAAM,CAAC;IA4B3F,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAE1B,IAAI,CAAC,KAAK,SAAK,GAAG,MAAM;CAGzB;AAED,qBAAa,YAAa,YAAW,OAAO;IAC1C,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,OAAO,CAAC,UAAU,CAAiC;IACnD,OAAO,CAAC,aAAa,CAA4B;IACjD,OAAO,CAAC,gBAAgB,CAAU;gBAEtB,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB;IAU9C,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,gBAAgB,IAAI,aAAa;IACjC,qBAAqB,IAAI,kBAAkB;IAErC,SAAS,CAAC,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOpF,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAG/B;AAED,qBAAa,mBAAoB,YAAW,cAAc;IAG5C,OAAO,CAAC,QAAQ,CAAC,OAAO;IAFpC,OAAO,CAAC,SAAS,CAAmC;gBAEvB,OAAO,GAAE,mBAAwB;IAExD,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;IAS3E,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS;IAE9B,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOlC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAGlC"}
|
|
@@ -0,0 +1,188 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// LocalSandbox — 基于本地进程的默认沙箱实现
|
|
3
|
+
// ============================================================
|
|
4
|
+
import { exec as cpExec } from 'child_process';
|
|
5
|
+
import * as fs from 'fs/promises';
|
|
6
|
+
import * as path from 'path';
|
|
7
|
+
import * as crypto from 'crypto';
|
|
8
|
+
import { existsSync } from 'fs';
|
|
9
|
+
import { getProjectDataPath } from "../utils/data-dir.js";
|
|
10
|
+
import { SecurityGuard } from "../security/guard.js";
|
|
11
|
+
import { DEFAULT_SANDBOX_CONFIG, SANDBOX_BLOCKED_COMMANDS, SANDBOX_SAFE_ENV_KEYS } from "./types.js";
|
|
12
|
+
function filterEnv(env) {
|
|
13
|
+
const filtered = {};
|
|
14
|
+
for (const key of SANDBOX_SAFE_ENV_KEYS)
|
|
15
|
+
if (env[key])
|
|
16
|
+
filtered[key] = env[key];
|
|
17
|
+
filtered.PYTHONIOENCODING = 'utf-8';
|
|
18
|
+
filtered.PYTHONUTF8 = '1';
|
|
19
|
+
return filtered;
|
|
20
|
+
}
|
|
21
|
+
const LANGUAGE_MAP = {
|
|
22
|
+
python: { ext: '.py', command: 'python3' },
|
|
23
|
+
nodejs: { ext: '.cjs', command: 'node' },
|
|
24
|
+
shell: { ext: '.sh', command: process.platform === 'win32' ? 'bash' : 'bash' },
|
|
25
|
+
};
|
|
26
|
+
export class LocalSandboxFileSystemExecutor {
|
|
27
|
+
rootDir;
|
|
28
|
+
maxFileSize;
|
|
29
|
+
allowedReadDirs;
|
|
30
|
+
strictPath;
|
|
31
|
+
constructor(rootDir, maxFileSize = DEFAULT_SANDBOX_CONFIG.maxFileSize, allowedReadDirs = [], strictPath = false) {
|
|
32
|
+
this.rootDir = rootDir;
|
|
33
|
+
this.maxFileSize = maxFileSize;
|
|
34
|
+
this.allowedReadDirs = allowedReadDirs;
|
|
35
|
+
this.strictPath = strictPath;
|
|
36
|
+
}
|
|
37
|
+
async readFile(filePath) {
|
|
38
|
+
return fs.readFile(this.resolveReadPath(filePath), 'utf-8');
|
|
39
|
+
}
|
|
40
|
+
async readFileBuffer(filePath) {
|
|
41
|
+
return fs.readFile(this.resolveReadPath(filePath));
|
|
42
|
+
}
|
|
43
|
+
async writeFile(filePath, content) {
|
|
44
|
+
const resolved = this.resolveWritePath(filePath);
|
|
45
|
+
const size = Buffer.byteLength(content, 'utf-8');
|
|
46
|
+
if (size > this.maxFileSize)
|
|
47
|
+
throw new Error(`Sandbox: file size ${size} bytes exceeds limit ${this.maxFileSize} bytes`);
|
|
48
|
+
await fs.mkdir(path.dirname(resolved), { recursive: true });
|
|
49
|
+
await fs.writeFile(resolved, content, 'utf-8');
|
|
50
|
+
}
|
|
51
|
+
async appendFile(filePath, content) {
|
|
52
|
+
const resolved = this.resolveWritePath(filePath);
|
|
53
|
+
await fs.mkdir(path.dirname(resolved), { recursive: true });
|
|
54
|
+
await fs.appendFile(resolved, content, 'utf-8');
|
|
55
|
+
}
|
|
56
|
+
async listDir(dirPath) {
|
|
57
|
+
const entries = await fs.readdir(this.resolveReadPath(dirPath), { withFileTypes: true });
|
|
58
|
+
return entries.map(e => e.isDirectory() ? `${e.name}/` : e.name);
|
|
59
|
+
}
|
|
60
|
+
isInside(target, base) {
|
|
61
|
+
const rel = path.relative(path.resolve(base), path.resolve(target));
|
|
62
|
+
return !rel.startsWith('..') && !path.isAbsolute(rel);
|
|
63
|
+
}
|
|
64
|
+
safeInRoot(filePath) {
|
|
65
|
+
return path.resolve(this.rootDir, path.basename(filePath));
|
|
66
|
+
}
|
|
67
|
+
resolveWritePath(filePath) {
|
|
68
|
+
const resolved = path.isAbsolute(filePath) ? path.resolve(filePath) : path.resolve(this.rootDir, filePath);
|
|
69
|
+
return this.isInside(resolved, this.rootDir) ? resolved : this.safeInRoot(filePath);
|
|
70
|
+
}
|
|
71
|
+
resolveReadPath(filePath) {
|
|
72
|
+
const resolved = path.isAbsolute(filePath) ? path.resolve(filePath) : path.resolve(this.rootDir, filePath);
|
|
73
|
+
if (this.isInside(resolved, this.rootDir))
|
|
74
|
+
return resolved;
|
|
75
|
+
for (const dir of this.allowedReadDirs)
|
|
76
|
+
if (this.isInside(resolved, dir))
|
|
77
|
+
return resolved;
|
|
78
|
+
return this.safeInRoot(filePath);
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
export class LocalSandboxShellExecutor {
|
|
82
|
+
rootDir;
|
|
83
|
+
commandRestriction;
|
|
84
|
+
maxBuffer;
|
|
85
|
+
guard;
|
|
86
|
+
outputBuffer = [];
|
|
87
|
+
env;
|
|
88
|
+
constructor(rootDir, commandRestriction = true, env = process.env, maxBuffer = 500) {
|
|
89
|
+
this.rootDir = rootDir;
|
|
90
|
+
this.commandRestriction = commandRestriction;
|
|
91
|
+
this.maxBuffer = maxBuffer;
|
|
92
|
+
this.guard = new SecurityGuard({ deleteProtection: true, blacklistEnabled: true, rules: SANDBOX_BLOCKED_COMMANDS });
|
|
93
|
+
this.env = filterEnv(env);
|
|
94
|
+
}
|
|
95
|
+
async exec(command, timeout = DEFAULT_SANDBOX_CONFIG.execTimeout) {
|
|
96
|
+
if (this.commandRestriction) {
|
|
97
|
+
const sanitized = this.guard.sanitize(command.trim());
|
|
98
|
+
if (!sanitized.safe)
|
|
99
|
+
return `[Sandbox blocked] ${sanitized.reason}`;
|
|
100
|
+
if (sanitized.rewritten)
|
|
101
|
+
command = sanitized.rewritten;
|
|
102
|
+
}
|
|
103
|
+
const shellCommand = process.platform === 'win32' ? `chcp 65001 >nul 2>&1 & ${command}` : command;
|
|
104
|
+
return new Promise(resolve => {
|
|
105
|
+
cpExec(shellCommand, {
|
|
106
|
+
timeout,
|
|
107
|
+
maxBuffer: 1024 * 1024 * 5,
|
|
108
|
+
windowsHide: true,
|
|
109
|
+
encoding: 'buffer',
|
|
110
|
+
cwd: this.rootDir,
|
|
111
|
+
env: this.env,
|
|
112
|
+
}, (error, stdoutBuf, stderrBuf) => {
|
|
113
|
+
const stdout = stdoutBuf ? stdoutBuf.toString('utf-8') : '';
|
|
114
|
+
const stderr = stderrBuf ? stderrBuf.toString('utf-8') : '';
|
|
115
|
+
let output = [stdout, stderr].filter(Boolean).join('\n');
|
|
116
|
+
if (error && !output.trim())
|
|
117
|
+
output = error.message;
|
|
118
|
+
this.outputBuffer.push(...output.split('\n'));
|
|
119
|
+
if (this.outputBuffer.length > this.maxBuffer)
|
|
120
|
+
this.outputBuffer = this.outputBuffer.slice(-this.maxBuffer);
|
|
121
|
+
resolve(output || '(no output)');
|
|
122
|
+
});
|
|
123
|
+
});
|
|
124
|
+
}
|
|
125
|
+
write(_data) { }
|
|
126
|
+
read(lines = 50) {
|
|
127
|
+
return this.outputBuffer.slice(Math.max(0, this.outputBuffer.length - lines)).join('\n') || '(no terminal output)';
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
export class LocalSandbox {
|
|
131
|
+
id;
|
|
132
|
+
rootDir;
|
|
133
|
+
fsExecutor;
|
|
134
|
+
shellExecutor;
|
|
135
|
+
cleanupOnDestroy;
|
|
136
|
+
constructor(id, options) {
|
|
137
|
+
this.id = id;
|
|
138
|
+
this.rootDir = options.rootDir || path.join(getProjectDataPath(), 'sandboxes', id);
|
|
139
|
+
this.cleanupOnDestroy = options.cleanupOnDestroy ?? false;
|
|
140
|
+
const maxFileSize = options.maxFileSize ?? DEFAULT_SANDBOX_CONFIG.maxFileSize;
|
|
141
|
+
const allowedReadDirs = options.allowedReadDirs?.map(d => path.resolve(d)) || [];
|
|
142
|
+
this.fsExecutor = new LocalSandboxFileSystemExecutor(this.rootDir, maxFileSize, allowedReadDirs, options.strictPath ?? false);
|
|
143
|
+
this.shellExecutor = new LocalSandboxShellExecutor(this.rootDir, options.commandRestriction ?? DEFAULT_SANDBOX_CONFIG.commandRestriction, options.env);
|
|
144
|
+
}
|
|
145
|
+
async init() {
|
|
146
|
+
await fs.mkdir(this.rootDir, { recursive: true });
|
|
147
|
+
}
|
|
148
|
+
getShellExecutor() { return this.shellExecutor; }
|
|
149
|
+
getFileSystemExecutor() { return this.fsExecutor; }
|
|
150
|
+
async runScript(language, code, timeout) {
|
|
151
|
+
const spec = LANGUAGE_MAP[language];
|
|
152
|
+
const filename = `script_${Date.now()}_${crypto.randomBytes(4).toString('hex')}${spec.ext}`;
|
|
153
|
+
await this.fsExecutor.writeFile(filename, code);
|
|
154
|
+
return this.shellExecutor.exec(`${spec.command} ${filename}`, timeout);
|
|
155
|
+
}
|
|
156
|
+
async destroy() {
|
|
157
|
+
if (this.cleanupOnDestroy && existsSync(this.rootDir))
|
|
158
|
+
await fs.rm(this.rootDir, { recursive: true, force: true });
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
export class LocalSandboxManager {
|
|
162
|
+
options;
|
|
163
|
+
sandboxes = new Map();
|
|
164
|
+
constructor(options = {}) {
|
|
165
|
+
this.options = options;
|
|
166
|
+
}
|
|
167
|
+
async create(id, config) {
|
|
168
|
+
const existing = this.sandboxes.get(id);
|
|
169
|
+
if (existing)
|
|
170
|
+
return existing;
|
|
171
|
+
const sandbox = new LocalSandbox(id, { ...this.options, ...config });
|
|
172
|
+
await sandbox.init();
|
|
173
|
+
this.sandboxes.set(id, sandbox);
|
|
174
|
+
return sandbox;
|
|
175
|
+
}
|
|
176
|
+
get(id) { return this.sandboxes.get(id); }
|
|
177
|
+
async destroy(id) {
|
|
178
|
+
const sandbox = this.sandboxes.get(id);
|
|
179
|
+
if (!sandbox)
|
|
180
|
+
return;
|
|
181
|
+
await sandbox.destroy();
|
|
182
|
+
this.sandboxes.delete(id);
|
|
183
|
+
}
|
|
184
|
+
async destroyAll() {
|
|
185
|
+
await Promise.all([...this.sandboxes.keys()].map(id => this.destroy(id)));
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
//# sourceMappingURL=local-sandbox.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"local-sandbox.js","sourceRoot":"","sources":["../../../src/sandbox/local-sandbox.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,+BAA+B;AAC/B,+DAA+D;AAE/D,OAAO,EAAE,IAAI,IAAI,MAAM,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAChC,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAGrD,OAAO,EAAE,sBAAsB,EAAE,wBAAwB,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AASrG,SAAS,SAAS,CAAC,GAAuC;IACxD,MAAM,QAAQ,GAA2B,EAAE,CAAC;IAC5C,KAAK,MAAM,GAAG,IAAI,qBAAqB;QAAE,IAAI,GAAG,CAAC,GAAG,CAAC;YAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,CAAE,CAAC;IACjF,QAAQ,CAAC,gBAAgB,GAAG,OAAO,CAAC;IACpC,QAAQ,CAAC,UAAU,GAAG,GAAG,CAAC;IAC1B,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,YAAY,GAA6D;IAC7E,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE;IAC1C,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;IACxC,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE;CAC/E,CAAC;AAEF,MAAM,OAAO,8BAA8B;IAEtB;IACA;IACA;IACA;IAJnB,YACmB,OAAe,EACf,cAAc,sBAAsB,CAAC,WAAY,EACjD,kBAA4B,EAAE,EAC9B,aAAa,KAAK;QAHlB,YAAO,GAAP,OAAO,CAAQ;QACf,gBAAW,GAAX,WAAW,CAAsC;QACjD,oBAAe,GAAf,eAAe,CAAe;QAC9B,eAAU,GAAV,UAAU,CAAQ;IAClC,CAAC;IAEJ,KAAK,CAAC,QAAQ,CAAC,QAAgB;QAC7B,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB;QACnC,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAAgB,EAAE,OAAe;QAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACjD,IAAI,IAAI,GAAG,IAAI,CAAC,WAAW;YAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,wBAAwB,IAAI,CAAC,WAAW,QAAQ,CAAC,CAAC;QACzH,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,OAAe;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAe;QAC3B,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QACzF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACnE,CAAC;IAEO,QAAQ,CAAC,MAAc,EAAE,IAAY;QAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QACpE,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IACxD,CAAC;IAEO,UAAU,CAAC,QAAgB;QACjC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,gBAAgB,CAAC,QAAgB;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC3G,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACtF,CAAC;IAEO,eAAe,CAAC,QAAgB;QACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC3G,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC3D,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,eAAe;YAAE,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC;gBAAE,OAAO,QAAQ,CAAC;QAC1F,OAAO,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;CACF;AAED,MAAM,OAAO,yBAAyB;IAMjB;IACA;IAEA;IARX,KAAK,CAAgB;IACrB,YAAY,GAAa,EAAE,CAAC;IAC5B,GAAG,CAAyB;IAEpC,YACmB,OAAe,EACf,qBAAqB,IAAI,EAC1C,MAA0C,OAAO,CAAC,GAAG,EACpC,YAAY,GAAG;QAHf,YAAO,GAAP,OAAO,CAAQ;QACf,uBAAkB,GAAlB,kBAAkB,CAAO;QAEzB,cAAS,GAAT,SAAS,CAAM;QAEhC,IAAI,CAAC,KAAK,GAAG,IAAI,aAAa,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC,CAAC;QACpH,IAAI,CAAC,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAe,EAAE,UAAU,sBAAsB,CAAC,WAAY;QACvE,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YACtD,IAAI,CAAC,SAAS,CAAC,IAAI;gBAAE,OAAO,qBAAqB,SAAS,CAAC,MAAM,EAAE,CAAC;YACpE,IAAI,SAAS,CAAC,SAAS;gBAAE,OAAO,GAAG,SAAS,CAAC,SAAS,CAAC;QACzD,CAAC;QAED,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,0BAA0B,OAAO,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;QAClG,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YAC3B,MAAM,CAAC,YAAY,EAAE;gBACnB,OAAO;gBACP,SAAS,EAAE,IAAI,GAAG,IAAI,GAAG,CAAC;gBAC1B,WAAW,EAAE,IAAI;gBACjB,QAAQ,EAAE,QAAQ;gBAClB,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,GAAG,EAAE,IAAI,CAAC,GAAG;aACd,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE;gBACjC,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAE,SAA+B,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnF,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAE,SAA+B,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnF,IAAI,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACzD,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;oBAAE,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC;gBACpD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC9C,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS;oBAAE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC5G,OAAO,CAAC,MAAM,IAAI,aAAa,CAAC,CAAC;YACnC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAa,IAAS,CAAC;IAE7B,IAAI,CAAC,KAAK,GAAG,EAAE;QACb,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,sBAAsB,CAAC;IACrH,CAAC;CACF;AAED,MAAM,OAAO,YAAY;IACd,EAAE,CAAS;IACX,OAAO,CAAS;IACjB,UAAU,CAAiC;IAC3C,aAAa,CAA4B;IACzC,gBAAgB,CAAU;IAElC,YAAY,EAAU,EAAE,OAA4B;QAClD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;QACnF,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,KAAK,CAAC;QAC1D,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,sBAAsB,CAAC,WAAY,CAAC;QAC/E,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACjF,IAAI,CAAC,UAAU,GAAG,IAAI,8BAA8B,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,CAAC,UAAU,IAAI,KAAK,CAAC,CAAC;QAC9H,IAAI,CAAC,aAAa,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,kBAAkB,IAAI,sBAAsB,CAAC,kBAAmB,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;IAC1J,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IAED,gBAAgB,KAAoB,OAAO,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;IAChE,qBAAqB,KAAyB,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAEvE,KAAK,CAAC,SAAS,CAAC,QAAwB,EAAE,IAAY,EAAE,OAAgB;QACtE,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,UAAU,IAAI,CAAC,GAAG,EAAE,IAAI,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5F,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,IAAI,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,gBAAgB,IAAI,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC;YAAE,MAAM,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACrH,CAAC;CACF;AAED,MAAM,OAAO,mBAAmB;IAGD;IAFrB,SAAS,GAAG,IAAI,GAAG,EAAwB,CAAC;IAEpD,YAA6B,UAA+B,EAAE;QAAjC,YAAO,GAAP,OAAO,CAA0B;IAAG,CAAC;IAElE,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,MAA+B;QACtD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QACrE,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAChC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,GAAG,CAAC,EAAU,IAAyB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEvE,KAAK,CAAC,OAAO,CAAC,EAAU;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC5E,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { RegisteredTool } from './types.js';
|
|
2
|
+
import type { ToolRegistry } from './registry.js';
|
|
3
|
+
import type { SkillManager } from '../skill/manager.js';
|
|
4
|
+
export interface BuiltinToolOptions {
|
|
5
|
+
includeShell?: boolean;
|
|
6
|
+
includeFileSystem?: boolean;
|
|
7
|
+
includeToolDoc?: boolean;
|
|
8
|
+
includeSkillLoader?: boolean;
|
|
9
|
+
toolRegistry?: ToolRegistry;
|
|
10
|
+
skillManager?: SkillManager;
|
|
11
|
+
toolResultStore?: ToolResultStoreLike;
|
|
12
|
+
}
|
|
13
|
+
export interface ToolResultStoreLike {
|
|
14
|
+
read(callId: string): string | null;
|
|
15
|
+
has?(callId: string): boolean;
|
|
16
|
+
}
|
|
17
|
+
export declare function createShellTools(): RegisteredTool[];
|
|
18
|
+
export declare function createFileSystemTools(): RegisteredTool[];
|
|
19
|
+
export declare function createToolDocTool(registry?: ToolRegistry): RegisteredTool;
|
|
20
|
+
export declare function createSkillLoaderTool(skillManager?: SkillManager): RegisteredTool;
|
|
21
|
+
export declare function createBuiltinTools(options?: BuiltinToolOptions): RegisteredTool[];
|
|
22
|
+
//# sourceMappingURL=builtin-tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"builtin-tools.d.ts","sourceRoot":"","sources":["../../../src/tool/builtin-tools.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD,MAAM,WAAW,kBAAkB;IACjC,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,eAAe,CAAC,EAAE,mBAAmB,CAAC;CACvC;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IACpC,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;CAC/B;AAED,wBAAgB,gBAAgB,IAAI,cAAc,EAAE,CA2CnD;AAED,wBAAgB,qBAAqB,IAAI,cAAc,EAAE,CAkFxD;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,CAAC,EAAE,YAAY,GAAG,cAAc,CA0BzE;AAED,wBAAgB,qBAAqB,CAAC,YAAY,CAAC,EAAE,YAAY,GAAG,cAAc,CAyBjF;AAED,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,kBAAuB,GAAG,cAAc,EAAE,CAOrF"}
|
|
@@ -0,0 +1,214 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// Builtin Tools — 可选通用工具工厂
|
|
3
|
+
// ============================================================
|
|
4
|
+
export function createShellTools() {
|
|
5
|
+
return [
|
|
6
|
+
{
|
|
7
|
+
definition: {
|
|
8
|
+
type: 'function',
|
|
9
|
+
function: {
|
|
10
|
+
name: 'execute_command',
|
|
11
|
+
description: 'Execute a shell command through the injected ToolContext.shell executor.',
|
|
12
|
+
parameters: {
|
|
13
|
+
type: 'object',
|
|
14
|
+
properties: {
|
|
15
|
+
command: { type: 'string', description: 'Shell command to execute' },
|
|
16
|
+
timeout: { type: 'number', description: 'Timeout in milliseconds' },
|
|
17
|
+
},
|
|
18
|
+
required: ['command'],
|
|
19
|
+
},
|
|
20
|
+
},
|
|
21
|
+
},
|
|
22
|
+
executor: async (args, ctx) => {
|
|
23
|
+
const command = String(args.command || '').trim();
|
|
24
|
+
if (!command)
|
|
25
|
+
return 'ERROR: execute_command requires non-empty command.';
|
|
26
|
+
return ctx.shell.exec(command, Number(args.timeout) || undefined);
|
|
27
|
+
},
|
|
28
|
+
category: 'builtin',
|
|
29
|
+
concurrent: false,
|
|
30
|
+
},
|
|
31
|
+
{
|
|
32
|
+
definition: {
|
|
33
|
+
type: 'function',
|
|
34
|
+
function: {
|
|
35
|
+
name: 'read_terminal',
|
|
36
|
+
description: 'Read recent terminal output from the injected ToolContext.shell executor.',
|
|
37
|
+
parameters: {
|
|
38
|
+
type: 'object',
|
|
39
|
+
properties: { lines: { type: 'number', description: 'Number of lines to read' } },
|
|
40
|
+
},
|
|
41
|
+
},
|
|
42
|
+
},
|
|
43
|
+
executor: async (args, ctx) => ctx.shell.read(Number(args.lines) || 50),
|
|
44
|
+
category: 'builtin',
|
|
45
|
+
concurrent: true,
|
|
46
|
+
},
|
|
47
|
+
];
|
|
48
|
+
}
|
|
49
|
+
export function createFileSystemTools() {
|
|
50
|
+
return [
|
|
51
|
+
{
|
|
52
|
+
definition: {
|
|
53
|
+
type: 'function',
|
|
54
|
+
function: {
|
|
55
|
+
name: 'read_file',
|
|
56
|
+
description: 'Read a text file through the injected ToolContext.fs executor.',
|
|
57
|
+
parameters: {
|
|
58
|
+
type: 'object',
|
|
59
|
+
properties: { path: { type: 'string', description: 'File path to read' } },
|
|
60
|
+
required: ['path'],
|
|
61
|
+
},
|
|
62
|
+
},
|
|
63
|
+
},
|
|
64
|
+
executor: async (args, ctx) => {
|
|
65
|
+
if (!ctx.fs)
|
|
66
|
+
return 'Error: FileSystem executor not available';
|
|
67
|
+
const filePath = String(args.path || '').trim();
|
|
68
|
+
if (!filePath)
|
|
69
|
+
return 'Error: path is required.';
|
|
70
|
+
return ctx.fs.readFile(filePath);
|
|
71
|
+
},
|
|
72
|
+
category: 'builtin',
|
|
73
|
+
concurrent: true,
|
|
74
|
+
},
|
|
75
|
+
{
|
|
76
|
+
definition: {
|
|
77
|
+
type: 'function',
|
|
78
|
+
function: {
|
|
79
|
+
name: 'write_file',
|
|
80
|
+
description: 'Write text content to a file through the injected ToolContext.fs executor.',
|
|
81
|
+
parameters: {
|
|
82
|
+
type: 'object',
|
|
83
|
+
properties: {
|
|
84
|
+
path: { type: 'string', description: 'File path to write' },
|
|
85
|
+
content: { type: 'string', description: 'Text content' },
|
|
86
|
+
},
|
|
87
|
+
required: ['path', 'content'],
|
|
88
|
+
},
|
|
89
|
+
},
|
|
90
|
+
},
|
|
91
|
+
executor: async (args, ctx) => {
|
|
92
|
+
if (!ctx.fs)
|
|
93
|
+
return 'Error: FileSystem executor not available';
|
|
94
|
+
const filePath = String(args.path || '').trim();
|
|
95
|
+
if (!filePath)
|
|
96
|
+
return 'Error: path is required.';
|
|
97
|
+
await ctx.fs.writeFile(filePath, String(args.content ?? ''));
|
|
98
|
+
return `File written: ${filePath}`;
|
|
99
|
+
},
|
|
100
|
+
category: 'builtin',
|
|
101
|
+
concurrent: false,
|
|
102
|
+
},
|
|
103
|
+
{
|
|
104
|
+
definition: {
|
|
105
|
+
type: 'function',
|
|
106
|
+
function: {
|
|
107
|
+
name: 'append_file',
|
|
108
|
+
description: 'Append text content to a file through the injected ToolContext.fs executor.',
|
|
109
|
+
parameters: {
|
|
110
|
+
type: 'object',
|
|
111
|
+
properties: {
|
|
112
|
+
path: { type: 'string', description: 'File path to append' },
|
|
113
|
+
content: { type: 'string', description: 'Text content to append' },
|
|
114
|
+
},
|
|
115
|
+
required: ['path', 'content'],
|
|
116
|
+
},
|
|
117
|
+
},
|
|
118
|
+
},
|
|
119
|
+
executor: async (args, ctx) => {
|
|
120
|
+
if (!ctx.fs)
|
|
121
|
+
return 'Error: FileSystem executor not available';
|
|
122
|
+
const filePath = String(args.path || '').trim();
|
|
123
|
+
if (!filePath)
|
|
124
|
+
return 'Error: path is required.';
|
|
125
|
+
if (ctx.fs.appendFile)
|
|
126
|
+
await ctx.fs.appendFile(filePath, String(args.content ?? ''));
|
|
127
|
+
else {
|
|
128
|
+
let existing = '';
|
|
129
|
+
try {
|
|
130
|
+
existing = await ctx.fs.readFile(filePath);
|
|
131
|
+
}
|
|
132
|
+
catch {
|
|
133
|
+
existing = '';
|
|
134
|
+
}
|
|
135
|
+
await ctx.fs.writeFile(filePath, existing + String(args.content ?? ''));
|
|
136
|
+
}
|
|
137
|
+
return `Content appended to: ${filePath}`;
|
|
138
|
+
},
|
|
139
|
+
category: 'builtin',
|
|
140
|
+
concurrent: false,
|
|
141
|
+
},
|
|
142
|
+
];
|
|
143
|
+
}
|
|
144
|
+
export function createToolDocTool(registry) {
|
|
145
|
+
return {
|
|
146
|
+
definition: {
|
|
147
|
+
type: 'function',
|
|
148
|
+
function: {
|
|
149
|
+
name: 'read_tool_doc',
|
|
150
|
+
description: 'Read full schema/documentation for a registered tool. Use tool_name="list" to list names.',
|
|
151
|
+
parameters: {
|
|
152
|
+
type: 'object',
|
|
153
|
+
properties: { tool_name: { type: 'string', description: 'Tool name, or "list"' } },
|
|
154
|
+
required: ['tool_name'],
|
|
155
|
+
},
|
|
156
|
+
},
|
|
157
|
+
},
|
|
158
|
+
executor: async (args, ctx) => {
|
|
159
|
+
const toolRegistry = ctx.custom?.toolRegistry || registry;
|
|
160
|
+
if (!toolRegistry)
|
|
161
|
+
return 'Error: ToolRegistry not available';
|
|
162
|
+
const name = String(args.tool_name || '').trim();
|
|
163
|
+
if (name === 'list')
|
|
164
|
+
return toolRegistry.listNames().join('\n');
|
|
165
|
+
const tool = toolRegistry.get(name);
|
|
166
|
+
if (!tool)
|
|
167
|
+
return `Tool not found: ${name}`;
|
|
168
|
+
return JSON.stringify(tool.definition.function, null, 2);
|
|
169
|
+
},
|
|
170
|
+
category: 'builtin',
|
|
171
|
+
concurrent: true,
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
export function createSkillLoaderTool(skillManager) {
|
|
175
|
+
return {
|
|
176
|
+
definition: {
|
|
177
|
+
type: 'function',
|
|
178
|
+
function: {
|
|
179
|
+
name: 'load_skill',
|
|
180
|
+
description: 'Load full instructions for an available skill by skill_id.',
|
|
181
|
+
parameters: {
|
|
182
|
+
type: 'object',
|
|
183
|
+
properties: { skill_id: { type: 'string', description: 'Skill id' } },
|
|
184
|
+
required: ['skill_id'],
|
|
185
|
+
},
|
|
186
|
+
},
|
|
187
|
+
},
|
|
188
|
+
executor: async (args, ctx) => {
|
|
189
|
+
const manager = ctx.custom?.skillManager || skillManager;
|
|
190
|
+
if (!manager)
|
|
191
|
+
return 'Error: SkillManager not available';
|
|
192
|
+
const id = String(args.skill_id || '').trim();
|
|
193
|
+
const skill = manager.get(id);
|
|
194
|
+
if (!skill)
|
|
195
|
+
return `Skill not found: ${id}`;
|
|
196
|
+
return skill.content || '';
|
|
197
|
+
},
|
|
198
|
+
category: 'builtin',
|
|
199
|
+
concurrent: true,
|
|
200
|
+
};
|
|
201
|
+
}
|
|
202
|
+
export function createBuiltinTools(options = {}) {
|
|
203
|
+
const tools = [];
|
|
204
|
+
if (options.includeShell ?? true)
|
|
205
|
+
tools.push(...createShellTools());
|
|
206
|
+
if (options.includeFileSystem ?? true)
|
|
207
|
+
tools.push(...createFileSystemTools());
|
|
208
|
+
if (options.includeToolDoc ?? true)
|
|
209
|
+
tools.push(createToolDocTool(options.toolRegistry));
|
|
210
|
+
if (options.includeSkillLoader ?? true)
|
|
211
|
+
tools.push(createSkillLoaderTool(options.skillManager));
|
|
212
|
+
return tools;
|
|
213
|
+
}
|
|
214
|
+
//# sourceMappingURL=builtin-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"builtin-tools.js","sourceRoot":"","sources":["../../../src/tool/builtin-tools.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,2BAA2B;AAC3B,+DAA+D;AAqB/D,MAAM,UAAU,gBAAgB;IAC9B,OAAO;QACL;YACE,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,iBAAiB;oBACvB,WAAW,EAAE,0EAA0E;oBACvF,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0BAA0B,EAAE;4BACpE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yBAAyB,EAAE;yBACpE;wBACD,QAAQ,EAAE,CAAC,SAAS,CAAC;qBACtB;iBACF;aACF;YACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAClD,IAAI,CAAC,OAAO;oBAAE,OAAO,oDAAoD,CAAC;gBAC1E,OAAO,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,CAAC;YACpE,CAAC;YACD,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,KAAK;SAClB;QACD;YACE,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,eAAe;oBACrB,WAAW,EAAE,2EAA2E;oBACxF,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yBAAyB,EAAE,EAAE;qBAClF;iBACF;aACF;YACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACvE,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,IAAI;SACjB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB;IACnC,OAAO;QACL;YACE,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,WAAW;oBACjB,WAAW,EAAE,gEAAgE;oBAC7E,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mBAAmB,EAAE,EAAE;wBAC1E,QAAQ,EAAE,CAAC,MAAM,CAAC;qBACnB;iBACF;aACF;YACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBAC5B,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,OAAO,0CAA0C,CAAC;gBAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAChD,IAAI,CAAC,QAAQ;oBAAE,OAAO,0BAA0B,CAAC;gBACjD,OAAO,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACnC,CAAC;YACD,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,IAAI;SACjB;QACD;YACE,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,YAAY;oBAClB,WAAW,EAAE,4EAA4E;oBACzF,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE;4BAC3D,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,cAAc,EAAE;yBACzD;wBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;qBAC9B;iBACF;aACF;YACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBAC5B,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,OAAO,0CAA0C,CAAC;gBAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAChD,IAAI,CAAC,QAAQ;oBAAE,OAAO,0BAA0B,CAAC;gBACjD,MAAM,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC7D,OAAO,iBAAiB,QAAQ,EAAE,CAAC;YACrC,CAAC;YACD,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,KAAK;SAClB;QACD;YACE,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,aAAa;oBACnB,WAAW,EAAE,6EAA6E;oBAC1F,UAAU,EAAE;wBACV,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,qBAAqB,EAAE;4BAC5D,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wBAAwB,EAAE;yBACnE;wBACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;qBAC9B;iBACF;aACF;YACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBAC5B,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,OAAO,0CAA0C,CAAC;gBAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAChD,IAAI,CAAC,QAAQ;oBAAE,OAAO,0BAA0B,CAAC;gBACjD,IAAI,GAAG,CAAC,EAAE,CAAC,UAAU;oBAAE,MAAM,GAAG,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;qBAChF,CAAC;oBACJ,IAAI,QAAQ,GAAG,EAAE,CAAC;oBAClB,IAAI,CAAC;wBAAC,QAAQ,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC;wBAAC,QAAQ,GAAG,EAAE,CAAC;oBAAC,CAAC;oBAC5E,MAAM,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC1E,CAAC;gBACD,OAAO,wBAAwB,QAAQ,EAAE,CAAC;YAC5C,CAAC;YACD,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,KAAK;SAClB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAAuB;IACvD,OAAO;QACL,UAAU,EAAE;YACV,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,2FAA2F;gBACxG,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,sBAAsB,EAAE,EAAE;oBAClF,QAAQ,EAAE,CAAC,WAAW,CAAC;iBACxB;aACF;SACF;QACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;YAC5B,MAAM,YAAY,GAAI,GAAG,CAAC,MAAM,EAAE,YAAyC,IAAI,QAAQ,CAAC;YACxF,IAAI,CAAC,YAAY;gBAAE,OAAO,mCAAmC,CAAC;YAC9D,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACjD,IAAI,IAAI,KAAK,MAAM;gBAAE,OAAO,YAAY,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,CAAC,IAAI;gBAAE,OAAO,mBAAmB,IAAI,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC3D,CAAC;QACD,QAAQ,EAAE,SAAS;QACnB,UAAU,EAAE,IAAI;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,YAA2B;IAC/D,OAAO;QACL,UAAU,EAAE;YACV,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,4DAA4D;gBACzE,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE;oBACrE,QAAQ,EAAE,CAAC,UAAU,CAAC;iBACvB;aACF;SACF;QACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;YAC5B,MAAM,OAAO,GAAI,GAAG,CAAC,MAAM,EAAE,YAAyC,IAAI,YAAY,CAAC;YACvF,IAAI,CAAC,OAAO;gBAAE,OAAO,mCAAmC,CAAC;YACzD,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9C,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC9B,IAAI,CAAC,KAAK;gBAAE,OAAO,oBAAoB,EAAE,EAAE,CAAC;YAC5C,OAAO,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;QAC7B,CAAC;QACD,QAAQ,EAAE,SAAS;QACnB,UAAU,EAAE,IAAI;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,UAA8B,EAAE;IACjE,MAAM,KAAK,GAAqB,EAAE,CAAC;IACnC,IAAI,OAAO,CAAC,YAAY,IAAI,IAAI;QAAE,KAAK,CAAC,IAAI,CAAC,GAAG,gBAAgB,EAAE,CAAC,CAAC;IACpE,IAAI,OAAO,CAAC,iBAAiB,IAAI,IAAI;QAAE,KAAK,CAAC,IAAI,CAAC,GAAG,qBAAqB,EAAE,CAAC,CAAC;IAC9E,IAAI,OAAO,CAAC,cAAc,IAAI,IAAI;QAAE,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC;IACxF,IAAI,OAAO,CAAC,kBAAkB,IAAI,IAAI;QAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC;IAChG,OAAO,KAAK,CAAC;AACf,CAAC"}
|
|
@@ -82,6 +82,31 @@ export declare class ToolRegistry {
|
|
|
82
82
|
registerAll(tools: RegisteredTool[]): void;
|
|
83
83
|
/** 注销工具 */
|
|
84
84
|
unregister(name: string): void;
|
|
85
|
+
/** 克隆当前注册表(浅拷贝工具定义与 executor 引用) */
|
|
86
|
+
clone(filter?: {
|
|
87
|
+
category?: ToolCategory;
|
|
88
|
+
names?: string[];
|
|
89
|
+
predicate?: (tool: RegisteredTool) => boolean;
|
|
90
|
+
}): ToolRegistry;
|
|
91
|
+
/** 按分类生成新注册表 */
|
|
92
|
+
filterByCategory(category: ToolCategory): ToolRegistry;
|
|
93
|
+
/** 按 metadata 条件生成新注册表 */
|
|
94
|
+
filterByMetadata(predicate: (metadata: Record<string, unknown>, tool: RegisteredTool) => boolean): ToolRegistry;
|
|
95
|
+
/** 导出扁平工具定义(用于使用方适配旧 API 或 UI 展示) */
|
|
96
|
+
toFlatTools(): Array<Omit<FlatToolOptions, 'execute'> & {
|
|
97
|
+
execute: ToolExecutor;
|
|
98
|
+
}>;
|
|
99
|
+
/** 转为 OpenAI function calling tools 格式 */
|
|
100
|
+
toOpenAITools(filter?: {
|
|
101
|
+
category?: ToolCategory;
|
|
102
|
+
names?: string[];
|
|
103
|
+
}): ToolDefinition[];
|
|
104
|
+
/** 生成工具摘要,适合注入 prompt 或 UI 列表 */
|
|
105
|
+
summary(options?: {
|
|
106
|
+
maxDescriptionLength?: number;
|
|
107
|
+
category?: ToolCategory;
|
|
108
|
+
names?: string[];
|
|
109
|
+
}): string;
|
|
85
110
|
/** 获取指定工具 */
|
|
86
111
|
get(name: string): RegisteredTool | undefined;
|
|
87
112
|
/** 获取所有工具定义(传给 LLM) */
|