workerclaw 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +129 -0
- package/dist/active-behavior/behavior-scheduler.d.ts +119 -0
- package/dist/active-behavior/behavior-scheduler.d.ts.map +1 -0
- package/dist/active-behavior/behavior-scheduler.js +297 -0
- package/dist/active-behavior/behavior-scheduler.js.map +1 -0
- package/dist/active-behavior/frequency-control.d.ts +72 -0
- package/dist/active-behavior/frequency-control.d.ts.map +1 -0
- package/dist/active-behavior/frequency-control.js +206 -0
- package/dist/active-behavior/frequency-control.js.map +1 -0
- package/dist/active-behavior/index.d.ts +8 -0
- package/dist/active-behavior/index.d.ts.map +1 -0
- package/dist/active-behavior/index.js +6 -0
- package/dist/active-behavior/index.js.map +1 -0
- package/dist/agent/agent-engine.d.ts +122 -0
- package/dist/agent/agent-engine.d.ts.map +1 -0
- package/dist/agent/agent-engine.js +380 -0
- package/dist/agent/agent-engine.js.map +1 -0
- package/dist/agent/context-window.d.ts +69 -0
- package/dist/agent/context-window.d.ts.map +1 -0
- package/dist/agent/context-window.js +210 -0
- package/dist/agent/context-window.js.map +1 -0
- package/dist/agent/llm-client.d.ts +37 -0
- package/dist/agent/llm-client.d.ts.map +1 -0
- package/dist/agent/llm-client.js +155 -0
- package/dist/agent/llm-client.js.map +1 -0
- package/dist/agent/personality.d.ts +71 -0
- package/dist/agent/personality.d.ts.map +1 -0
- package/dist/agent/personality.js +147 -0
- package/dist/agent/personality.js.map +1 -0
- package/dist/agent/session-manager.d.ts +111 -0
- package/dist/agent/session-manager.d.ts.map +1 -0
- package/dist/agent/session-manager.js +205 -0
- package/dist/agent/session-manager.js.map +1 -0
- package/dist/agent/tool-executor.d.ts +34 -0
- package/dist/agent/tool-executor.d.ts.map +1 -0
- package/dist/agent/tool-executor.js +145 -0
- package/dist/agent/tool-executor.js.map +1 -0
- package/dist/agent/tool-registry.d.ts +67 -0
- package/dist/agent/tool-registry.d.ts.map +1 -0
- package/dist/agent/tool-registry.js +272 -0
- package/dist/agent/tool-registry.js.map +1 -0
- package/dist/cli/configure.d.ts +16 -0
- package/dist/cli/configure.d.ts.map +1 -0
- package/dist/cli/configure.js +212 -0
- package/dist/cli/configure.js.map +1 -0
- package/dist/cli/index.d.ts +17 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +256 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/prompter.d.ts +59 -0
- package/dist/cli/prompter.d.ts.map +1 -0
- package/dist/cli/prompter.js +120 -0
- package/dist/cli/prompter.js.map +1 -0
- package/dist/cli/sections/index.d.ts +9 -0
- package/dist/cli/sections/index.d.ts.map +1 -0
- package/dist/cli/sections/index.js +9 -0
- package/dist/cli/sections/index.js.map +1 -0
- package/dist/cli/sections/llm.d.ts +14 -0
- package/dist/cli/sections/llm.d.ts.map +1 -0
- package/dist/cli/sections/llm.js +155 -0
- package/dist/cli/sections/llm.js.map +1 -0
- package/dist/cli/sections/personality.d.ts +14 -0
- package/dist/cli/sections/personality.d.ts.map +1 -0
- package/dist/cli/sections/personality.js +90 -0
- package/dist/cli/sections/personality.js.map +1 -0
- package/dist/cli/sections/platform.d.ts +14 -0
- package/dist/cli/sections/platform.d.ts.map +1 -0
- package/dist/cli/sections/platform.js +179 -0
- package/dist/cli/sections/platform.js.map +1 -0
- package/dist/cli/sections/security.d.ts +14 -0
- package/dist/cli/sections/security.d.ts.map +1 -0
- package/dist/cli/sections/security.js +106 -0
- package/dist/cli/sections/security.js.map +1 -0
- package/dist/cli/sections/skills.d.ts +10 -0
- package/dist/cli/sections/skills.d.ts.map +1 -0
- package/dist/cli/sections/skills.js +149 -0
- package/dist/cli/sections/skills.js.map +1 -0
- package/dist/cli.d.ts +8 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +116 -0
- package/dist/cli.js.map +1 -0
- package/dist/core/config.d.ts +175 -0
- package/dist/core/config.d.ts.map +1 -0
- package/dist/core/config.js +109 -0
- package/dist/core/config.js.map +1 -0
- package/dist/core/events.d.ts +211 -0
- package/dist/core/events.d.ts.map +1 -0
- package/dist/core/events.js +152 -0
- package/dist/core/events.js.map +1 -0
- package/dist/core/logger.d.ts +23 -0
- package/dist/core/logger.d.ts.map +1 -0
- package/dist/core/logger.js +53 -0
- package/dist/core/logger.js.map +1 -0
- package/dist/core/workerclaw.d.ts +83 -0
- package/dist/core/workerclaw.d.ts.map +1 -0
- package/dist/core/workerclaw.js +161 -0
- package/dist/core/workerclaw.js.map +1 -0
- package/dist/index.d.ts +27 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +21 -0
- package/dist/index.js.map +1 -0
- package/dist/ingress/message-parser.d.ts +36 -0
- package/dist/ingress/message-parser.d.ts.map +1 -0
- package/dist/ingress/message-parser.js +107 -0
- package/dist/ingress/message-parser.js.map +1 -0
- package/dist/ingress/miniabc-client.d.ts +109 -0
- package/dist/ingress/miniabc-client.d.ts.map +1 -0
- package/dist/ingress/miniabc-client.js +345 -0
- package/dist/ingress/miniabc-client.js.map +1 -0
- package/dist/ingress/platform-api.d.ts +82 -0
- package/dist/ingress/platform-api.d.ts.map +1 -0
- package/dist/ingress/platform-api.js +209 -0
- package/dist/ingress/platform-api.js.map +1 -0
- package/dist/sandbox/command-sandbox.d.ts +46 -0
- package/dist/sandbox/command-sandbox.d.ts.map +1 -0
- package/dist/sandbox/command-sandbox.js +144 -0
- package/dist/sandbox/command-sandbox.js.map +1 -0
- package/dist/sandbox/fs-sandbox.d.ts +56 -0
- package/dist/sandbox/fs-sandbox.d.ts.map +1 -0
- package/dist/sandbox/fs-sandbox.js +119 -0
- package/dist/sandbox/fs-sandbox.js.map +1 -0
- package/dist/sandbox/network-sandbox.d.ts +41 -0
- package/dist/sandbox/network-sandbox.d.ts.map +1 -0
- package/dist/sandbox/network-sandbox.js +114 -0
- package/dist/sandbox/network-sandbox.js.map +1 -0
- package/dist/security/content-scanner.d.ts +59 -0
- package/dist/security/content-scanner.d.ts.map +1 -0
- package/dist/security/content-scanner.js +289 -0
- package/dist/security/content-scanner.js.map +1 -0
- package/dist/security/gate.d.ts +78 -0
- package/dist/security/gate.d.ts.map +1 -0
- package/dist/security/gate.js +150 -0
- package/dist/security/gate.js.map +1 -0
- package/dist/security/permission-level.d.ts +68 -0
- package/dist/security/permission-level.d.ts.map +1 -0
- package/dist/security/permission-level.js +191 -0
- package/dist/security/permission-level.js.map +1 -0
- package/dist/security/rate-limiter.d.ts +52 -0
- package/dist/security/rate-limiter.d.ts.map +1 -0
- package/dist/security/rate-limiter.js +133 -0
- package/dist/security/rate-limiter.js.map +1 -0
- package/dist/security/source-verifier.d.ts +33 -0
- package/dist/security/source-verifier.d.ts.map +1 -0
- package/dist/security/source-verifier.js +106 -0
- package/dist/security/source-verifier.js.map +1 -0
- package/dist/skills/builtin/code.d.ts +27 -0
- package/dist/skills/builtin/code.d.ts.map +1 -0
- package/dist/skills/builtin/code.js +132 -0
- package/dist/skills/builtin/code.js.map +1 -0
- package/dist/skills/builtin/index.d.ts +12 -0
- package/dist/skills/builtin/index.d.ts.map +1 -0
- package/dist/skills/builtin/index.js +16 -0
- package/dist/skills/builtin/index.js.map +1 -0
- package/dist/skills/builtin/search.d.ts +27 -0
- package/dist/skills/builtin/search.d.ts.map +1 -0
- package/dist/skills/builtin/search.js +105 -0
- package/dist/skills/builtin/search.js.map +1 -0
- package/dist/skills/builtin/writing.d.ts +22 -0
- package/dist/skills/builtin/writing.d.ts.map +1 -0
- package/dist/skills/builtin/writing.js +93 -0
- package/dist/skills/builtin/writing.js.map +1 -0
- package/dist/skills/index.d.ts +11 -0
- package/dist/skills/index.d.ts.map +1 -0
- package/dist/skills/index.js +10 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/pack-loader.d.ts +31 -0
- package/dist/skills/pack-loader.d.ts.map +1 -0
- package/dist/skills/pack-loader.js +167 -0
- package/dist/skills/pack-loader.js.map +1 -0
- package/dist/skills/pack-registry.d.ts +40 -0
- package/dist/skills/pack-registry.d.ts.map +1 -0
- package/dist/skills/pack-registry.js +80 -0
- package/dist/skills/pack-registry.js.map +1 -0
- package/dist/skills/pack-types.d.ts +64 -0
- package/dist/skills/pack-types.d.ts.map +1 -0
- package/dist/skills/pack-types.js +7 -0
- package/dist/skills/pack-types.js.map +1 -0
- package/dist/skills/skill-registry.d.ts +83 -0
- package/dist/skills/skill-registry.d.ts.map +1 -0
- package/dist/skills/skill-registry.js +206 -0
- package/dist/skills/skill-registry.js.map +1 -0
- package/dist/skills/skill-runner.d.ts +49 -0
- package/dist/skills/skill-runner.d.ts.map +1 -0
- package/dist/skills/skill-runner.js +128 -0
- package/dist/skills/skill-runner.js.map +1 -0
- package/dist/skills/types.d.ts +94 -0
- package/dist/skills/types.d.ts.map +1 -0
- package/dist/skills/types.js +7 -0
- package/dist/skills/types.js.map +1 -0
- package/dist/task/concurrency.d.ts +87 -0
- package/dist/task/concurrency.d.ts.map +1 -0
- package/dist/task/concurrency.js +269 -0
- package/dist/task/concurrency.js.map +1 -0
- package/dist/task/task-evaluator.d.ts +52 -0
- package/dist/task/task-evaluator.d.ts.map +1 -0
- package/dist/task/task-evaluator.js +141 -0
- package/dist/task/task-evaluator.js.map +1 -0
- package/dist/task/task-manager.d.ts +112 -0
- package/dist/task/task-manager.d.ts.map +1 -0
- package/dist/task/task-manager.js +382 -0
- package/dist/task/task-manager.js.map +1 -0
- package/dist/task/task-state-machine.d.ts +63 -0
- package/dist/task/task-state-machine.d.ts.map +1 -0
- package/dist/task/task-state-machine.js +149 -0
- package/dist/task/task-state-machine.js.map +1 -0
- package/dist/types/agent.d.ts +79 -0
- package/dist/types/agent.d.ts.map +1 -0
- package/dist/types/agent.js +7 -0
- package/dist/types/agent.js.map +1 -0
- package/dist/types/index.d.ts +7 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +5 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/message.d.ts +87 -0
- package/dist/types/message.d.ts.map +1 -0
- package/dist/types/message.js +30 -0
- package/dist/types/message.js.map +1 -0
- package/dist/types/task.d.ts +176 -0
- package/dist/types/task.d.ts.map +1 -0
- package/dist/types/task.js +5 -0
- package/dist/types/task.js.map +1 -0
- package/package.json +57 -0
- package/workerclaw.config.example.json +78 -0
|
@@ -0,0 +1,205 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 会话管理器
|
|
3
|
+
*
|
|
4
|
+
* 管理多轮会话的上下文累积
|
|
5
|
+
* 每个任务有独立会话,支持会话复用
|
|
6
|
+
*/
|
|
7
|
+
import { createLogger } from '../core/logger.js';
|
|
8
|
+
import { ContextWindow } from './context-window.js';
|
|
9
|
+
export const DEFAULT_SESSION_CONFIG = {
|
|
10
|
+
maxActiveSessions: 50,
|
|
11
|
+
sessionTTL: 30 * 60 * 1000, // 30 分钟
|
|
12
|
+
contextWindow: {
|
|
13
|
+
maxTokens: 8000,
|
|
14
|
+
keepRecentMessages: 4,
|
|
15
|
+
},
|
|
16
|
+
};
|
|
17
|
+
// ==================== SessionManager ====================
|
|
18
|
+
export class SessionManager {
|
|
19
|
+
logger;
|
|
20
|
+
config;
|
|
21
|
+
sessions = new Map();
|
|
22
|
+
contextWindow;
|
|
23
|
+
constructor(config) {
|
|
24
|
+
this.config = { ...DEFAULT_SESSION_CONFIG, ...config };
|
|
25
|
+
this.contextWindow = new ContextWindow(this.config.contextWindow);
|
|
26
|
+
this.logger = createLogger('SessionManager');
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* 创建新会话
|
|
30
|
+
*/
|
|
31
|
+
createSession(id, systemPrompt) {
|
|
32
|
+
if (this.sessions.has(id)) {
|
|
33
|
+
this.logger.warn(`会话 ${id} 已存在,将被重置`);
|
|
34
|
+
}
|
|
35
|
+
// 检查会话数量限制
|
|
36
|
+
if (this.sessions.size >= this.config.maxActiveSessions) {
|
|
37
|
+
this.evictExpiredSessions();
|
|
38
|
+
}
|
|
39
|
+
const session = {
|
|
40
|
+
id,
|
|
41
|
+
messages: systemPrompt ? [{ role: 'system', content: systemPrompt }] : [],
|
|
42
|
+
createdAt: Date.now(),
|
|
43
|
+
updatedAt: Date.now(),
|
|
44
|
+
turnCount: 0,
|
|
45
|
+
totalTokensUsed: 0,
|
|
46
|
+
completed: false,
|
|
47
|
+
};
|
|
48
|
+
this.sessions.set(id, session);
|
|
49
|
+
this.logger.debug(`创建会话 [${id}]`);
|
|
50
|
+
return session;
|
|
51
|
+
}
|
|
52
|
+
/**
|
|
53
|
+
* 获取会话
|
|
54
|
+
*/
|
|
55
|
+
getSession(id) {
|
|
56
|
+
return this.sessions.get(id);
|
|
57
|
+
}
|
|
58
|
+
/**
|
|
59
|
+
* 添加消息到会话
|
|
60
|
+
*/
|
|
61
|
+
addMessage(sessionId, message) {
|
|
62
|
+
const session = this.sessions.get(sessionId);
|
|
63
|
+
if (!session) {
|
|
64
|
+
this.logger.warn(`会话 ${sessionId} 不存在`);
|
|
65
|
+
return undefined;
|
|
66
|
+
}
|
|
67
|
+
session.messages.push(message);
|
|
68
|
+
session.updatedAt = Date.now();
|
|
69
|
+
session.totalTokensUsed += this.contextWindow.estimateTokens(message);
|
|
70
|
+
// 计算轮次(user 消息计为 1 轮)
|
|
71
|
+
if (message.role === 'user') {
|
|
72
|
+
session.turnCount++;
|
|
73
|
+
}
|
|
74
|
+
return session;
|
|
75
|
+
}
|
|
76
|
+
/**
|
|
77
|
+
* 获取会话的上下文窗口适配消息
|
|
78
|
+
*/
|
|
79
|
+
getFittedMessages(sessionId) {
|
|
80
|
+
const session = this.sessions.get(sessionId);
|
|
81
|
+
if (!session) {
|
|
82
|
+
return {
|
|
83
|
+
messages: [],
|
|
84
|
+
stats: {
|
|
85
|
+
totalTokens: 0,
|
|
86
|
+
systemTokens: 0,
|
|
87
|
+
conversationTokens: 0,
|
|
88
|
+
messageCount: 0,
|
|
89
|
+
isTruncated: false,
|
|
90
|
+
},
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
return this.contextWindow.fitToWindow(session.messages);
|
|
94
|
+
}
|
|
95
|
+
/**
|
|
96
|
+
* 标记会话完成
|
|
97
|
+
*/
|
|
98
|
+
completeSession(sessionId) {
|
|
99
|
+
const session = this.sessions.get(sessionId);
|
|
100
|
+
if (session) {
|
|
101
|
+
session.completed = true;
|
|
102
|
+
session.updatedAt = Date.now();
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
/**
|
|
106
|
+
* 删除会话
|
|
107
|
+
*/
|
|
108
|
+
deleteSession(sessionId) {
|
|
109
|
+
return this.sessions.delete(sessionId);
|
|
110
|
+
}
|
|
111
|
+
/**
|
|
112
|
+
* 检查会话是否存在
|
|
113
|
+
*/
|
|
114
|
+
hasSession(sessionId) {
|
|
115
|
+
return this.sessions.has(sessionId);
|
|
116
|
+
}
|
|
117
|
+
/**
|
|
118
|
+
* 获取所有活跃会话 ID
|
|
119
|
+
*/
|
|
120
|
+
getActiveSessionIds() {
|
|
121
|
+
return [...this.sessions.keys()].filter(id => !this.sessions.get(id).completed);
|
|
122
|
+
}
|
|
123
|
+
/**
|
|
124
|
+
* 获取会话统计
|
|
125
|
+
*/
|
|
126
|
+
getSessionStats(sessionId) {
|
|
127
|
+
const session = this.sessions.get(sessionId);
|
|
128
|
+
if (!session)
|
|
129
|
+
return null;
|
|
130
|
+
const contextStats = this.contextWindow.getStats(session.messages);
|
|
131
|
+
return {
|
|
132
|
+
messageCount: session.messages.length,
|
|
133
|
+
turnCount: session.turnCount,
|
|
134
|
+
totalTokensUsed: session.totalTokensUsed,
|
|
135
|
+
ageMs: Date.now() - session.createdAt,
|
|
136
|
+
contextStats,
|
|
137
|
+
};
|
|
138
|
+
}
|
|
139
|
+
/**
|
|
140
|
+
* 获取全局统计
|
|
141
|
+
*/
|
|
142
|
+
getStats() {
|
|
143
|
+
let active = 0;
|
|
144
|
+
let completed = 0;
|
|
145
|
+
let totalTokens = 0;
|
|
146
|
+
for (const session of this.sessions.values()) {
|
|
147
|
+
if (session.completed) {
|
|
148
|
+
completed++;
|
|
149
|
+
}
|
|
150
|
+
else {
|
|
151
|
+
active++;
|
|
152
|
+
}
|
|
153
|
+
totalTokens += session.totalTokensUsed;
|
|
154
|
+
}
|
|
155
|
+
return {
|
|
156
|
+
totalSessions: this.sessions.size,
|
|
157
|
+
activeSessions: active,
|
|
158
|
+
completedSessions: completed,
|
|
159
|
+
totalTokensUsed: totalTokens,
|
|
160
|
+
};
|
|
161
|
+
}
|
|
162
|
+
/**
|
|
163
|
+
* 清理过期会话
|
|
164
|
+
*/
|
|
165
|
+
evictExpiredSessions() {
|
|
166
|
+
const now = Date.now();
|
|
167
|
+
let evicted = 0;
|
|
168
|
+
for (const [id, session] of this.sessions) {
|
|
169
|
+
if (now - session.updatedAt > this.config.sessionTTL) {
|
|
170
|
+
this.sessions.delete(id);
|
|
171
|
+
evicted++;
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
if (evicted > 0) {
|
|
175
|
+
this.logger.info(`清理了 ${evicted} 个过期会话`);
|
|
176
|
+
}
|
|
177
|
+
return evicted;
|
|
178
|
+
}
|
|
179
|
+
/**
|
|
180
|
+
* 清理所有已完成会话
|
|
181
|
+
*/
|
|
182
|
+
evictCompletedSessions() {
|
|
183
|
+
let evicted = 0;
|
|
184
|
+
for (const [id, session] of this.sessions) {
|
|
185
|
+
if (session.completed) {
|
|
186
|
+
this.sessions.delete(id);
|
|
187
|
+
evicted++;
|
|
188
|
+
}
|
|
189
|
+
}
|
|
190
|
+
return evicted;
|
|
191
|
+
}
|
|
192
|
+
/**
|
|
193
|
+
* 获取上下文窗口实例(供外部使用)
|
|
194
|
+
*/
|
|
195
|
+
getContextWindow() {
|
|
196
|
+
return this.contextWindow;
|
|
197
|
+
}
|
|
198
|
+
/**
|
|
199
|
+
* 清理资源
|
|
200
|
+
*/
|
|
201
|
+
dispose() {
|
|
202
|
+
this.sessions.clear();
|
|
203
|
+
}
|
|
204
|
+
}
|
|
205
|
+
//# sourceMappingURL=session-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-manager.js","sourceRoot":"","sources":["../../src/agent/session-manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAqD,MAAM,qBAAqB,CAAC;AAiCvG,MAAM,CAAC,MAAM,sBAAsB,GAAyB;IAC1D,iBAAiB,EAAE,EAAE;IACrB,UAAU,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,QAAQ;IACpC,aAAa,EAAE;QACb,SAAS,EAAE,IAAI;QACf,kBAAkB,EAAE,CAAC;KACtB;CACF,CAAC;AAEF,2DAA2D;AAE3D,MAAM,OAAO,cAAc;IACjB,MAAM,CAAS;IACf,MAAM,CAAuB;IAC7B,QAAQ,GAAG,IAAI,GAAG,EAAmB,CAAC;IACtC,aAAa,CAAgB;IAErC,YAAY,MAAsC;QAChD,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,sBAAsB,EAAE,GAAG,MAAM,EAAE,CAAC;QACvD,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,gBAAgB,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,EAAU,EAAE,YAAqB;QAC7C,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACxC,CAAC;QAED,WAAW;QACX,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YACxD,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC;QAED,MAAM,OAAO,GAAY;YACvB,EAAE;YACF,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;YACzE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS,EAAE,CAAC;YACZ,eAAe,EAAE,CAAC;YAClB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAElC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,EAAU;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,SAAiB,EAAE,OAAmB;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,SAAS,MAAM,CAAC,CAAC;YACxC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,OAAO,CAAC,eAAe,IAAI,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAEtE,sBAAsB;QACtB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,SAAiB;QAIjC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;gBACL,QAAQ,EAAE,EAAE;gBACZ,KAAK,EAAE;oBACL,WAAW,EAAE,CAAC;oBACd,YAAY,EAAE,CAAC;oBACf,kBAAkB,EAAE,CAAC;oBACrB,YAAY,EAAE,CAAC;oBACf,WAAW,EAAE,KAAK;iBACnB;aACF,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,SAAiB;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;YACzB,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,SAAiB;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC,SAAS,CAAC,CAAC;IACnF,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,SAAiB;QAO/B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAE1B,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEnE,OAAO;YACL,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM;YACrC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,KAAK,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,SAAS;YACrC,YAAY;SACb,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,QAAQ;QAMN,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACtB,SAAS,EAAE,CAAC;YACd,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,CAAC;YACX,CAAC;YACD,WAAW,IAAI,OAAO,CAAC,eAAe,CAAC;QACzC,CAAC;QAED,OAAO;YACL,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI;YACjC,cAAc,EAAE,MAAM;YACtB,iBAAiB,EAAE,SAAS;YAC5B,eAAe,EAAE,WAAW;SAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC1C,IAAI,GAAG,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBACrD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,QAAQ,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,sBAAsB;QACpB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC1C,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACtB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具执行器
|
|
3
|
+
*
|
|
4
|
+
* 在安全沙箱内执行工具调用
|
|
5
|
+
* 每次执行前检查权限和资源限制
|
|
6
|
+
*/
|
|
7
|
+
import { EventBus } from '../core/events.js';
|
|
8
|
+
import { ToolRegistry } from './tool-registry.js';
|
|
9
|
+
import type { ToolCall, ToolResult, ToolExecutionContext } from '../types/agent.js';
|
|
10
|
+
import type { SecurityConfig } from '../core/config.js';
|
|
11
|
+
export interface ToolExecutorConfig {
|
|
12
|
+
/** 安全配置 */
|
|
13
|
+
security: SecurityConfig;
|
|
14
|
+
}
|
|
15
|
+
export declare class ToolExecutor {
|
|
16
|
+
private logger;
|
|
17
|
+
private eventBus;
|
|
18
|
+
private registry;
|
|
19
|
+
private config;
|
|
20
|
+
constructor(registry: ToolRegistry, config: ToolExecutorConfig, eventBus: EventBus);
|
|
21
|
+
/**
|
|
22
|
+
* 执行工具调用
|
|
23
|
+
*/
|
|
24
|
+
execute(toolCall: ToolCall, context: ToolExecutionContext): Promise<ToolResult>;
|
|
25
|
+
/**
|
|
26
|
+
* 带超时的执行
|
|
27
|
+
*/
|
|
28
|
+
private executeWithTimeout;
|
|
29
|
+
/**
|
|
30
|
+
* 获取工具注册表(用于查询)
|
|
31
|
+
*/
|
|
32
|
+
getRegistry(): ToolRegistry;
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=tool-executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-executor.d.ts","sourceRoot":"","sources":["../../src/agent/tool-executor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,QAAQ,EAAmB,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,oBAAoB,EAAmB,MAAM,mBAAmB,CAAC;AACrG,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExD,MAAM,WAAW,kBAAkB;IACjC,WAAW;IACX,QAAQ,EAAE,cAAc,CAAC;CAC1B;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,MAAM,CAAqB;gBAEvB,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,kBAAkB,EAAE,QAAQ,EAAE,QAAQ;IAMlF;;OAEG;IACG,OAAO,CACX,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,oBAAoB,GAC5B,OAAO,CAAC,UAAU,CAAC;IA2GtB;;OAEG;YACW,kBAAkB;IA2BhC;;OAEG;IACH,WAAW,IAAI,YAAY;CAG5B"}
|
|
@@ -0,0 +1,145 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具执行器
|
|
3
|
+
*
|
|
4
|
+
* 在安全沙箱内执行工具调用
|
|
5
|
+
* 每次执行前检查权限和资源限制
|
|
6
|
+
*/
|
|
7
|
+
import { createLogger } from '../core/logger.js';
|
|
8
|
+
import { WorkerClawEvent } from '../core/events.js';
|
|
9
|
+
export class ToolExecutor {
|
|
10
|
+
logger = createLogger('ToolExecutor');
|
|
11
|
+
eventBus;
|
|
12
|
+
registry;
|
|
13
|
+
config;
|
|
14
|
+
constructor(registry, config, eventBus) {
|
|
15
|
+
this.registry = registry;
|
|
16
|
+
this.config = config;
|
|
17
|
+
this.eventBus = eventBus;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* 执行工具调用
|
|
21
|
+
*/
|
|
22
|
+
async execute(toolCall, context) {
|
|
23
|
+
const { name, id: toolCallId, arguments: argsStr } = toolCall;
|
|
24
|
+
// 1. 检查工具是否存在
|
|
25
|
+
if (!this.registry.isToolAllowed(name, context.permissionLevel)) {
|
|
26
|
+
this.eventBus.emit(WorkerClawEvent.TOOL_BLOCKED, {
|
|
27
|
+
taskId: context.taskId,
|
|
28
|
+
toolName: name,
|
|
29
|
+
reason: `工具 "${name}" 在权限级别 "${context.permissionLevel}" 下不可用`,
|
|
30
|
+
});
|
|
31
|
+
return {
|
|
32
|
+
toolCallId,
|
|
33
|
+
success: false,
|
|
34
|
+
content: `工具 "${name}" 在当前权限级别下不可用`,
|
|
35
|
+
error: 'permission_denied',
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
// 2. 检查工具调用次数限制
|
|
39
|
+
if (context.toolCallCount >= context.maxToolCalls) {
|
|
40
|
+
this.eventBus.emit(WorkerClawEvent.TOOL_BLOCKED, {
|
|
41
|
+
taskId: context.taskId,
|
|
42
|
+
toolName: name,
|
|
43
|
+
reason: `已达到最大工具调用次数 ${context.maxToolCalls}`,
|
|
44
|
+
});
|
|
45
|
+
return {
|
|
46
|
+
toolCallId,
|
|
47
|
+
success: false,
|
|
48
|
+
content: `已达到最大工具调用次数 (${context.maxToolCalls})`,
|
|
49
|
+
error: 'tool_call_limit',
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
// 3. 解析参数
|
|
53
|
+
let params;
|
|
54
|
+
try {
|
|
55
|
+
params = typeof argsStr === 'string' ? JSON.parse(argsStr) : argsStr;
|
|
56
|
+
}
|
|
57
|
+
catch (err) {
|
|
58
|
+
return {
|
|
59
|
+
toolCallId,
|
|
60
|
+
success: false,
|
|
61
|
+
content: `工具参数解析失败: ${argsStr}`,
|
|
62
|
+
error: 'invalid_arguments',
|
|
63
|
+
};
|
|
64
|
+
}
|
|
65
|
+
// 4. 获取执行器
|
|
66
|
+
const executor = this.registry.getExecutor(name);
|
|
67
|
+
this.eventBus.emit(WorkerClawEvent.TOOL_CALLED, {
|
|
68
|
+
taskId: context.taskId,
|
|
69
|
+
toolName: name,
|
|
70
|
+
toolCallId,
|
|
71
|
+
});
|
|
72
|
+
// 5. 执行
|
|
73
|
+
try {
|
|
74
|
+
if (!executor) {
|
|
75
|
+
// 无执行器:返回工具描述(说明可用但未实现)
|
|
76
|
+
const tool = this.registry.getTool(name);
|
|
77
|
+
return {
|
|
78
|
+
toolCallId,
|
|
79
|
+
success: true,
|
|
80
|
+
content: JSON.stringify({
|
|
81
|
+
status: 'not_implemented',
|
|
82
|
+
tool: name,
|
|
83
|
+
description: tool?.description || '未知工具',
|
|
84
|
+
hint: '该工具已注册但执行器未绑定。在 Phase 4 技能系统中将实现具体执行逻辑。',
|
|
85
|
+
}),
|
|
86
|
+
};
|
|
87
|
+
}
|
|
88
|
+
// 在超时限制内执行
|
|
89
|
+
const result = await this.executeWithTimeout(() => executor(params, context), context.remainingMs);
|
|
90
|
+
this.eventBus.emit(WorkerClawEvent.TOOL_COMPLETED, {
|
|
91
|
+
taskId: context.taskId,
|
|
92
|
+
toolName: name,
|
|
93
|
+
toolCallId,
|
|
94
|
+
success: result.success,
|
|
95
|
+
});
|
|
96
|
+
return result;
|
|
97
|
+
}
|
|
98
|
+
catch (err) {
|
|
99
|
+
const error = err;
|
|
100
|
+
this.logger.error(`工具执行失败: ${name}`, { taskId: context.taskId, error: error.message });
|
|
101
|
+
this.eventBus.emit(WorkerClawEvent.TOOL_COMPLETED, {
|
|
102
|
+
taskId: context.taskId,
|
|
103
|
+
toolName: name,
|
|
104
|
+
toolCallId,
|
|
105
|
+
success: false,
|
|
106
|
+
});
|
|
107
|
+
return {
|
|
108
|
+
toolCallId,
|
|
109
|
+
success: false,
|
|
110
|
+
content: `工具执行异常: ${error.message}`,
|
|
111
|
+
error: error.message,
|
|
112
|
+
};
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
/**
|
|
116
|
+
* 带超时的执行
|
|
117
|
+
*/
|
|
118
|
+
async executeWithTimeout(fn, timeoutMs) {
|
|
119
|
+
if (timeoutMs <= 0) {
|
|
120
|
+
throw new Error('执行超时:无剩余时间');
|
|
121
|
+
}
|
|
122
|
+
const effectiveTimeout = Math.min(timeoutMs, this.config.security.sandbox.commandTimeoutMs);
|
|
123
|
+
return new Promise((resolve, reject) => {
|
|
124
|
+
const timer = setTimeout(() => {
|
|
125
|
+
reject(new Error(`工具执行超时 (${effectiveTimeout}ms)`));
|
|
126
|
+
}, effectiveTimeout);
|
|
127
|
+
fn()
|
|
128
|
+
.then(result => {
|
|
129
|
+
clearTimeout(timer);
|
|
130
|
+
resolve(result);
|
|
131
|
+
})
|
|
132
|
+
.catch(err => {
|
|
133
|
+
clearTimeout(timer);
|
|
134
|
+
reject(err);
|
|
135
|
+
});
|
|
136
|
+
});
|
|
137
|
+
}
|
|
138
|
+
/**
|
|
139
|
+
* 获取工具注册表(用于查询)
|
|
140
|
+
*/
|
|
141
|
+
getRegistry() {
|
|
142
|
+
return this.registry;
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
//# sourceMappingURL=tool-executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-executor.js","sourceRoot":"","sources":["../../src/agent/tool-executor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAY,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAU9D,MAAM,OAAO,YAAY;IACf,MAAM,GAAG,YAAY,CAAC,cAAc,CAAC,CAAC;IACtC,QAAQ,CAAW;IACnB,QAAQ,CAAe;IACvB,MAAM,CAAqB;IAEnC,YAAY,QAAsB,EAAE,MAA0B,EAAE,QAAkB;QAChF,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAkB,EAClB,OAA6B;QAE7B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;QAE9D,cAAc;QACd,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;YAChE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE;gBAC/C,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,IAAI;gBACd,MAAM,EAAE,OAAO,IAAI,YAAY,OAAO,CAAC,eAAe,QAAQ;aAC/D,CAAC,CAAC;YACH,OAAO;gBACL,UAAU;gBACV,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,OAAO,IAAI,eAAe;gBACnC,KAAK,EAAE,mBAAmB;aAC3B,CAAC;QACJ,CAAC;QAED,gBAAgB;QAChB,IAAI,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YAClD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE;gBAC/C,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,IAAI;gBACd,MAAM,EAAE,eAAe,OAAO,CAAC,YAAY,EAAE;aAC9C,CAAC,CAAC;YACH,OAAO;gBACL,UAAU;gBACV,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,gBAAgB,OAAO,CAAC,YAAY,GAAG;gBAChD,KAAK,EAAE,iBAAiB;aACzB,CAAC;QACJ,CAAC;QAED,UAAU;QACV,IAAI,MAAW,CAAC;QAChB,IAAI,CAAC;YACH,MAAM,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACvE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,UAAU;gBACV,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,aAAa,OAAO,EAAE;gBAC/B,KAAK,EAAE,mBAAmB;aAC3B,CAAC;QACJ,CAAC;QAED,WAAW;QACX,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAEjD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE;YAC9C,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,IAAI;YACd,UAAU;SACX,CAAC,CAAC;QAEH,QAAQ;QACR,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,wBAAwB;gBACxB,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACzC,OAAO;oBACL,UAAU;oBACV,OAAO,EAAE,IAAI;oBACb,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;wBACtB,MAAM,EAAE,iBAAiB;wBACzB,IAAI,EAAE,IAAI;wBACV,WAAW,EAAE,IAAI,EAAE,WAAW,IAAI,MAAM;wBACxC,IAAI,EAAE,yCAAyC;qBAChD,CAAC;iBACH,CAAC;YACJ,CAAC;YAED,WAAW;YACX,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAC1C,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,OAAO,CAAC,WAAW,CACpB,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE;gBACjD,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,IAAI;gBACd,UAAU;gBACV,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAEhB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,GAAY,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAEvF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE;gBACjD,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,IAAI;gBACd,UAAU;gBACV,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YAEH,OAAO;gBACL,UAAU;gBACV,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,WAAW,KAAK,CAAC,OAAO,EAAE;gBACnC,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAC9B,EAAoB,EACpB,SAAiB;QAEjB,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAE5F,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,gBAAgB,KAAK,CAAC,CAAC,CAAC;YACtD,CAAC,EAAE,gBAAgB,CAAC,CAAC;YAErB,EAAE,EAAE;iBACD,IAAI,CAAC,MAAM,CAAC,EAAE;gBACb,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC,CAAC;iBACD,KAAK,CAAC,GAAG,CAAC,EAAE;gBACX,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;CACF"}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具注册表
|
|
3
|
+
*
|
|
4
|
+
* 管理所有可用工具,按权限级别过滤
|
|
5
|
+
*
|
|
6
|
+
* 内置工具:
|
|
7
|
+
* read_only: llm_query
|
|
8
|
+
* limited: web_search, read_file
|
|
9
|
+
* standard: write_file, image_generate, data_analyze
|
|
10
|
+
* elevated: run_code, system_command
|
|
11
|
+
*/
|
|
12
|
+
import type { ToolDefinition, PermissionLevel, ToolExecutorFn } from '../types/agent.js';
|
|
13
|
+
export declare class ToolRegistry {
|
|
14
|
+
private logger;
|
|
15
|
+
private tools;
|
|
16
|
+
/**
|
|
17
|
+
* 注册工具
|
|
18
|
+
*/
|
|
19
|
+
register(tool: ToolDefinition): void;
|
|
20
|
+
/**
|
|
21
|
+
* 注销工具
|
|
22
|
+
*/
|
|
23
|
+
unregister(name: string): boolean;
|
|
24
|
+
/**
|
|
25
|
+
* 获取指定权限级别允许的工具列表
|
|
26
|
+
*/
|
|
27
|
+
getTools(level: PermissionLevel): ToolDefinition[];
|
|
28
|
+
/**
|
|
29
|
+
* 转换为 OpenAI function calling 格式
|
|
30
|
+
*/
|
|
31
|
+
getToolsForLLM(level: PermissionLevel): Array<{
|
|
32
|
+
type: 'function';
|
|
33
|
+
function: {
|
|
34
|
+
name: string;
|
|
35
|
+
description: string;
|
|
36
|
+
parameters: Record<string, any>;
|
|
37
|
+
};
|
|
38
|
+
}>;
|
|
39
|
+
/**
|
|
40
|
+
* 检查工具是否允许在指定权限级别下使用
|
|
41
|
+
*/
|
|
42
|
+
isToolAllowed(toolName: string, level: PermissionLevel): boolean;
|
|
43
|
+
/**
|
|
44
|
+
* 获取工具的执行器
|
|
45
|
+
*/
|
|
46
|
+
getExecutor(toolName: string): ToolExecutorFn | undefined;
|
|
47
|
+
/**
|
|
48
|
+
* 获取工具定义
|
|
49
|
+
*/
|
|
50
|
+
getTool(name: string): ToolDefinition | undefined;
|
|
51
|
+
/**
|
|
52
|
+
* 获取所有已注册的工具名
|
|
53
|
+
*/
|
|
54
|
+
getToolNames(): string[];
|
|
55
|
+
/**
|
|
56
|
+
* 获取统计信息
|
|
57
|
+
*/
|
|
58
|
+
getStats(): {
|
|
59
|
+
total: number;
|
|
60
|
+
byLevel: Record<PermissionLevel, number>;
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
/**
|
|
64
|
+
* 创建内置工具注册表
|
|
65
|
+
*/
|
|
66
|
+
export declare function createBuiltinToolRegistry(): ToolRegistry;
|
|
67
|
+
//# sourceMappingURL=tool-registry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-registry.d.ts","sourceRoot":"","sources":["../../src/agent/tool-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AASzF,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,KAAK,CAAqC;IAElD;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,cAAc,GAAG,IAAI;IAQpC;;OAEG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIjC;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,eAAe,GAAG,cAAc,EAAE;IAOlD;;OAEG;IACH,cAAc,CAAC,KAAK,EAAE,eAAe,GAAG,KAAK,CAAC;QAC5C,IAAI,EAAE,UAAU,CAAC;QACjB,QAAQ,EAAE;YACR,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,EAAE,MAAM,CAAC;YACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SACjC,CAAC;KACH,CAAC;IAWF;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO;IAMhE;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAIzD;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAIjD;;OAEG;IACH,YAAY,IAAI,MAAM,EAAE;IAIxB;;OAEG;IACH,QAAQ,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC,eAAe,EAAE,MAAM,CAAC,CAAA;KAAE;CAYxE;AAED;;GAEG;AACH,wBAAgB,yBAAyB,IAAI,YAAY,CA8KxD"}
|