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.
Files changed (224) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +129 -0
  3. package/dist/active-behavior/behavior-scheduler.d.ts +119 -0
  4. package/dist/active-behavior/behavior-scheduler.d.ts.map +1 -0
  5. package/dist/active-behavior/behavior-scheduler.js +297 -0
  6. package/dist/active-behavior/behavior-scheduler.js.map +1 -0
  7. package/dist/active-behavior/frequency-control.d.ts +72 -0
  8. package/dist/active-behavior/frequency-control.d.ts.map +1 -0
  9. package/dist/active-behavior/frequency-control.js +206 -0
  10. package/dist/active-behavior/frequency-control.js.map +1 -0
  11. package/dist/active-behavior/index.d.ts +8 -0
  12. package/dist/active-behavior/index.d.ts.map +1 -0
  13. package/dist/active-behavior/index.js +6 -0
  14. package/dist/active-behavior/index.js.map +1 -0
  15. package/dist/agent/agent-engine.d.ts +122 -0
  16. package/dist/agent/agent-engine.d.ts.map +1 -0
  17. package/dist/agent/agent-engine.js +380 -0
  18. package/dist/agent/agent-engine.js.map +1 -0
  19. package/dist/agent/context-window.d.ts +69 -0
  20. package/dist/agent/context-window.d.ts.map +1 -0
  21. package/dist/agent/context-window.js +210 -0
  22. package/dist/agent/context-window.js.map +1 -0
  23. package/dist/agent/llm-client.d.ts +37 -0
  24. package/dist/agent/llm-client.d.ts.map +1 -0
  25. package/dist/agent/llm-client.js +155 -0
  26. package/dist/agent/llm-client.js.map +1 -0
  27. package/dist/agent/personality.d.ts +71 -0
  28. package/dist/agent/personality.d.ts.map +1 -0
  29. package/dist/agent/personality.js +147 -0
  30. package/dist/agent/personality.js.map +1 -0
  31. package/dist/agent/session-manager.d.ts +111 -0
  32. package/dist/agent/session-manager.d.ts.map +1 -0
  33. package/dist/agent/session-manager.js +205 -0
  34. package/dist/agent/session-manager.js.map +1 -0
  35. package/dist/agent/tool-executor.d.ts +34 -0
  36. package/dist/agent/tool-executor.d.ts.map +1 -0
  37. package/dist/agent/tool-executor.js +145 -0
  38. package/dist/agent/tool-executor.js.map +1 -0
  39. package/dist/agent/tool-registry.d.ts +67 -0
  40. package/dist/agent/tool-registry.d.ts.map +1 -0
  41. package/dist/agent/tool-registry.js +272 -0
  42. package/dist/agent/tool-registry.js.map +1 -0
  43. package/dist/cli/configure.d.ts +16 -0
  44. package/dist/cli/configure.d.ts.map +1 -0
  45. package/dist/cli/configure.js +212 -0
  46. package/dist/cli/configure.js.map +1 -0
  47. package/dist/cli/index.d.ts +17 -0
  48. package/dist/cli/index.d.ts.map +1 -0
  49. package/dist/cli/index.js +256 -0
  50. package/dist/cli/index.js.map +1 -0
  51. package/dist/cli/prompter.d.ts +59 -0
  52. package/dist/cli/prompter.d.ts.map +1 -0
  53. package/dist/cli/prompter.js +120 -0
  54. package/dist/cli/prompter.js.map +1 -0
  55. package/dist/cli/sections/index.d.ts +9 -0
  56. package/dist/cli/sections/index.d.ts.map +1 -0
  57. package/dist/cli/sections/index.js +9 -0
  58. package/dist/cli/sections/index.js.map +1 -0
  59. package/dist/cli/sections/llm.d.ts +14 -0
  60. package/dist/cli/sections/llm.d.ts.map +1 -0
  61. package/dist/cli/sections/llm.js +155 -0
  62. package/dist/cli/sections/llm.js.map +1 -0
  63. package/dist/cli/sections/personality.d.ts +14 -0
  64. package/dist/cli/sections/personality.d.ts.map +1 -0
  65. package/dist/cli/sections/personality.js +90 -0
  66. package/dist/cli/sections/personality.js.map +1 -0
  67. package/dist/cli/sections/platform.d.ts +14 -0
  68. package/dist/cli/sections/platform.d.ts.map +1 -0
  69. package/dist/cli/sections/platform.js +179 -0
  70. package/dist/cli/sections/platform.js.map +1 -0
  71. package/dist/cli/sections/security.d.ts +14 -0
  72. package/dist/cli/sections/security.d.ts.map +1 -0
  73. package/dist/cli/sections/security.js +106 -0
  74. package/dist/cli/sections/security.js.map +1 -0
  75. package/dist/cli/sections/skills.d.ts +10 -0
  76. package/dist/cli/sections/skills.d.ts.map +1 -0
  77. package/dist/cli/sections/skills.js +149 -0
  78. package/dist/cli/sections/skills.js.map +1 -0
  79. package/dist/cli.d.ts +8 -0
  80. package/dist/cli.d.ts.map +1 -0
  81. package/dist/cli.js +116 -0
  82. package/dist/cli.js.map +1 -0
  83. package/dist/core/config.d.ts +175 -0
  84. package/dist/core/config.d.ts.map +1 -0
  85. package/dist/core/config.js +109 -0
  86. package/dist/core/config.js.map +1 -0
  87. package/dist/core/events.d.ts +211 -0
  88. package/dist/core/events.d.ts.map +1 -0
  89. package/dist/core/events.js +152 -0
  90. package/dist/core/events.js.map +1 -0
  91. package/dist/core/logger.d.ts +23 -0
  92. package/dist/core/logger.d.ts.map +1 -0
  93. package/dist/core/logger.js +53 -0
  94. package/dist/core/logger.js.map +1 -0
  95. package/dist/core/workerclaw.d.ts +83 -0
  96. package/dist/core/workerclaw.d.ts.map +1 -0
  97. package/dist/core/workerclaw.js +161 -0
  98. package/dist/core/workerclaw.js.map +1 -0
  99. package/dist/index.d.ts +27 -0
  100. package/dist/index.d.ts.map +1 -0
  101. package/dist/index.js +21 -0
  102. package/dist/index.js.map +1 -0
  103. package/dist/ingress/message-parser.d.ts +36 -0
  104. package/dist/ingress/message-parser.d.ts.map +1 -0
  105. package/dist/ingress/message-parser.js +107 -0
  106. package/dist/ingress/message-parser.js.map +1 -0
  107. package/dist/ingress/miniabc-client.d.ts +109 -0
  108. package/dist/ingress/miniabc-client.d.ts.map +1 -0
  109. package/dist/ingress/miniabc-client.js +345 -0
  110. package/dist/ingress/miniabc-client.js.map +1 -0
  111. package/dist/ingress/platform-api.d.ts +82 -0
  112. package/dist/ingress/platform-api.d.ts.map +1 -0
  113. package/dist/ingress/platform-api.js +209 -0
  114. package/dist/ingress/platform-api.js.map +1 -0
  115. package/dist/sandbox/command-sandbox.d.ts +46 -0
  116. package/dist/sandbox/command-sandbox.d.ts.map +1 -0
  117. package/dist/sandbox/command-sandbox.js +144 -0
  118. package/dist/sandbox/command-sandbox.js.map +1 -0
  119. package/dist/sandbox/fs-sandbox.d.ts +56 -0
  120. package/dist/sandbox/fs-sandbox.d.ts.map +1 -0
  121. package/dist/sandbox/fs-sandbox.js +119 -0
  122. package/dist/sandbox/fs-sandbox.js.map +1 -0
  123. package/dist/sandbox/network-sandbox.d.ts +41 -0
  124. package/dist/sandbox/network-sandbox.d.ts.map +1 -0
  125. package/dist/sandbox/network-sandbox.js +114 -0
  126. package/dist/sandbox/network-sandbox.js.map +1 -0
  127. package/dist/security/content-scanner.d.ts +59 -0
  128. package/dist/security/content-scanner.d.ts.map +1 -0
  129. package/dist/security/content-scanner.js +289 -0
  130. package/dist/security/content-scanner.js.map +1 -0
  131. package/dist/security/gate.d.ts +78 -0
  132. package/dist/security/gate.d.ts.map +1 -0
  133. package/dist/security/gate.js +150 -0
  134. package/dist/security/gate.js.map +1 -0
  135. package/dist/security/permission-level.d.ts +68 -0
  136. package/dist/security/permission-level.d.ts.map +1 -0
  137. package/dist/security/permission-level.js +191 -0
  138. package/dist/security/permission-level.js.map +1 -0
  139. package/dist/security/rate-limiter.d.ts +52 -0
  140. package/dist/security/rate-limiter.d.ts.map +1 -0
  141. package/dist/security/rate-limiter.js +133 -0
  142. package/dist/security/rate-limiter.js.map +1 -0
  143. package/dist/security/source-verifier.d.ts +33 -0
  144. package/dist/security/source-verifier.d.ts.map +1 -0
  145. package/dist/security/source-verifier.js +106 -0
  146. package/dist/security/source-verifier.js.map +1 -0
  147. package/dist/skills/builtin/code.d.ts +27 -0
  148. package/dist/skills/builtin/code.d.ts.map +1 -0
  149. package/dist/skills/builtin/code.js +132 -0
  150. package/dist/skills/builtin/code.js.map +1 -0
  151. package/dist/skills/builtin/index.d.ts +12 -0
  152. package/dist/skills/builtin/index.d.ts.map +1 -0
  153. package/dist/skills/builtin/index.js +16 -0
  154. package/dist/skills/builtin/index.js.map +1 -0
  155. package/dist/skills/builtin/search.d.ts +27 -0
  156. package/dist/skills/builtin/search.d.ts.map +1 -0
  157. package/dist/skills/builtin/search.js +105 -0
  158. package/dist/skills/builtin/search.js.map +1 -0
  159. package/dist/skills/builtin/writing.d.ts +22 -0
  160. package/dist/skills/builtin/writing.d.ts.map +1 -0
  161. package/dist/skills/builtin/writing.js +93 -0
  162. package/dist/skills/builtin/writing.js.map +1 -0
  163. package/dist/skills/index.d.ts +11 -0
  164. package/dist/skills/index.d.ts.map +1 -0
  165. package/dist/skills/index.js +10 -0
  166. package/dist/skills/index.js.map +1 -0
  167. package/dist/skills/pack-loader.d.ts +31 -0
  168. package/dist/skills/pack-loader.d.ts.map +1 -0
  169. package/dist/skills/pack-loader.js +167 -0
  170. package/dist/skills/pack-loader.js.map +1 -0
  171. package/dist/skills/pack-registry.d.ts +40 -0
  172. package/dist/skills/pack-registry.d.ts.map +1 -0
  173. package/dist/skills/pack-registry.js +80 -0
  174. package/dist/skills/pack-registry.js.map +1 -0
  175. package/dist/skills/pack-types.d.ts +64 -0
  176. package/dist/skills/pack-types.d.ts.map +1 -0
  177. package/dist/skills/pack-types.js +7 -0
  178. package/dist/skills/pack-types.js.map +1 -0
  179. package/dist/skills/skill-registry.d.ts +83 -0
  180. package/dist/skills/skill-registry.d.ts.map +1 -0
  181. package/dist/skills/skill-registry.js +206 -0
  182. package/dist/skills/skill-registry.js.map +1 -0
  183. package/dist/skills/skill-runner.d.ts +49 -0
  184. package/dist/skills/skill-runner.d.ts.map +1 -0
  185. package/dist/skills/skill-runner.js +128 -0
  186. package/dist/skills/skill-runner.js.map +1 -0
  187. package/dist/skills/types.d.ts +94 -0
  188. package/dist/skills/types.d.ts.map +1 -0
  189. package/dist/skills/types.js +7 -0
  190. package/dist/skills/types.js.map +1 -0
  191. package/dist/task/concurrency.d.ts +87 -0
  192. package/dist/task/concurrency.d.ts.map +1 -0
  193. package/dist/task/concurrency.js +269 -0
  194. package/dist/task/concurrency.js.map +1 -0
  195. package/dist/task/task-evaluator.d.ts +52 -0
  196. package/dist/task/task-evaluator.d.ts.map +1 -0
  197. package/dist/task/task-evaluator.js +141 -0
  198. package/dist/task/task-evaluator.js.map +1 -0
  199. package/dist/task/task-manager.d.ts +112 -0
  200. package/dist/task/task-manager.d.ts.map +1 -0
  201. package/dist/task/task-manager.js +382 -0
  202. package/dist/task/task-manager.js.map +1 -0
  203. package/dist/task/task-state-machine.d.ts +63 -0
  204. package/dist/task/task-state-machine.d.ts.map +1 -0
  205. package/dist/task/task-state-machine.js +149 -0
  206. package/dist/task/task-state-machine.js.map +1 -0
  207. package/dist/types/agent.d.ts +79 -0
  208. package/dist/types/agent.d.ts.map +1 -0
  209. package/dist/types/agent.js +7 -0
  210. package/dist/types/agent.js.map +1 -0
  211. package/dist/types/index.d.ts +7 -0
  212. package/dist/types/index.d.ts.map +1 -0
  213. package/dist/types/index.js +5 -0
  214. package/dist/types/index.js.map +1 -0
  215. package/dist/types/message.d.ts +87 -0
  216. package/dist/types/message.d.ts.map +1 -0
  217. package/dist/types/message.js +30 -0
  218. package/dist/types/message.js.map +1 -0
  219. package/dist/types/task.d.ts +176 -0
  220. package/dist/types/task.d.ts.map +1 -0
  221. package/dist/types/task.js +5 -0
  222. package/dist/types/task.js.map +1 -0
  223. package/package.json +57 -0
  224. package/workerclaw.config.example.json +78 -0
@@ -0,0 +1,380 @@
1
+ /**
2
+ * Agent 引擎 (Phase 4 完整版)
3
+ *
4
+ * WorkerClaw 的核心执行引擎
5
+ * 集成人格系统、会话管理、上下文窗口、技能系统和工具调用循环
6
+ *
7
+ * 执行流程:
8
+ * 1. 构建系统提示(人格 + 技能提示 + 权限信息)
9
+ * 2. 获取/创建会话
10
+ * 3. 添加用户消息
11
+ * 4. 上下文窗口适配
12
+ * 5. 调用 LLM
13
+ * 6. 处理工具调用(循环直到完成)
14
+ * 7. 返回结果
15
+ */
16
+ import { createLogger } from '../core/logger.js';
17
+ import { WorkerClawEvent } from '../core/events.js';
18
+ import { LLMClient } from './llm-client.js';
19
+ import { Personality } from './personality.js';
20
+ import { SessionManager } from './session-manager.js';
21
+ import { ToolRegistry } from './tool-registry.js';
22
+ import { ToolExecutor } from './tool-executor.js';
23
+ import { SkillRunner } from '../skills/skill-runner.js';
24
+ import { SkillRegistry } from '../skills/skill-registry.js';
25
+ export class AgentEngine {
26
+ logger;
27
+ config;
28
+ eventBus;
29
+ llm;
30
+ personality;
31
+ sessionManager;
32
+ toolExecutor;
33
+ skillRunner;
34
+ skillRegistry;
35
+ constructor(config, eventBus) {
36
+ this.config = config;
37
+ this.eventBus = eventBus;
38
+ this.llm = new LLMClient(config.llm);
39
+ this.personality = new Personality(config.personality);
40
+ // 会话管理
41
+ this.sessionManager = new SessionManager({
42
+ maxActiveSessions: config.session?.maxActiveSessions || 50,
43
+ sessionTTL: config.session?.sessionTTL || 30 * 60 * 1000,
44
+ contextWindow: {
45
+ maxTokens: config.session?.maxTokens || 8000,
46
+ },
47
+ });
48
+ // 工具系统
49
+ const toolRegistry = new ToolRegistry();
50
+ this.toolExecutor = new ToolExecutor(toolRegistry, { security: config.security }, eventBus);
51
+ // 技能系统
52
+ this.skillRegistry = new SkillRegistry();
53
+ this.skillRunner = new SkillRunner(this.skillRegistry, config.skillRunner, eventBus);
54
+ this.logger = createLogger('AgentEngine');
55
+ }
56
+ /**
57
+ * 执行任务(Phase 4 完整版)
58
+ */
59
+ async executeTask(task, context) {
60
+ const startTime = Date.now();
61
+ this.logger.info(`开始执行任务 [${task.taskId}]`, {
62
+ type: task.taskType,
63
+ title: task.title,
64
+ });
65
+ this.eventBus.emit(WorkerClawEvent.TASK_STARTED, { taskId: task.taskId });
66
+ // 创建会话
67
+ const systemPrompt = this.buildSystemPrompt(task, context);
68
+ this.sessionManager.createSession(task.taskId, systemPrompt);
69
+ try {
70
+ // 添加用户消息
71
+ const userMessage = this.buildUserMessage(task);
72
+ this.sessionManager.addMessage(task.taskId, {
73
+ role: 'user',
74
+ content: userMessage,
75
+ });
76
+ // 报告进度
77
+ this.eventBus.emit(WorkerClawEvent.TASK_PROGRESS, {
78
+ taskId: task.taskId,
79
+ progress: 20,
80
+ message: '正在调用 LLM...',
81
+ });
82
+ // LLM 调用循环(支持工具调用)
83
+ const finalResponse = await this.llmLoop(task, context);
84
+ // 保存最终回复
85
+ this.sessionManager.addMessage(task.taskId, {
86
+ role: 'assistant',
87
+ content: finalResponse.content,
88
+ });
89
+ // 构建结果
90
+ const durationMs = Date.now() - startTime;
91
+ const result = {
92
+ taskId: task.taskId,
93
+ status: 'completed',
94
+ content: finalResponse.content,
95
+ outputs: this.buildOutputs(finalResponse.content),
96
+ tokensUsed: finalResponse.usage ? {
97
+ prompt: finalResponse.usage.promptTokens,
98
+ completion: finalResponse.usage.completionTokens,
99
+ } : undefined,
100
+ durationMs,
101
+ };
102
+ this.logger.info(`任务完成 [${task.taskId}]`, {
103
+ durationMs,
104
+ tokens: result.tokensUsed,
105
+ toolCallRounds: finalResponse.toolCallRounds || 0,
106
+ contentLength: finalResponse.content.length,
107
+ });
108
+ this.eventBus.emit(WorkerClawEvent.TASK_COMPLETED, { taskId: task.taskId, result });
109
+ return result;
110
+ }
111
+ catch (err) {
112
+ const durationMs = Date.now() - startTime;
113
+ const error = err;
114
+ this.logger.error(`任务执行失败 [${task.taskId}]`, { error: error.message, durationMs });
115
+ this.eventBus.emit(WorkerClawEvent.LLM_ERROR, { taskId: task.taskId, error });
116
+ this.eventBus.emit(WorkerClawEvent.TASK_FAILED, { taskId: task.taskId, error });
117
+ return {
118
+ taskId: task.taskId,
119
+ status: 'failed',
120
+ error: error.message,
121
+ durationMs,
122
+ };
123
+ }
124
+ finally {
125
+ // 标记会话完成
126
+ this.sessionManager.completeSession(task.taskId);
127
+ }
128
+ }
129
+ /**
130
+ * LLM 调用循环(支持工具调用)
131
+ */
132
+ async llmLoop(task, context, maxRounds = 10) {
133
+ let round = 0;
134
+ while (round < maxRounds) {
135
+ round++;
136
+ // 获取适配后的消息
137
+ const { messages, stats } = this.sessionManager.getFittedMessages(task.taskId);
138
+ if (stats.isTruncated) {
139
+ this.logger.debug('上下文窗口已裁剪', { stats });
140
+ }
141
+ // 获取工具
142
+ const tools = this.getToolsForTask(task, context.permissionLevel);
143
+ // 调用 LLM
144
+ this.eventBus.emit(WorkerClawEvent.LLM_REQUEST, {
145
+ taskId: task.taskId,
146
+ model: this.config.llm.model,
147
+ });
148
+ const response = await this.llm.chat({
149
+ messages,
150
+ tools: tools.length > 0 ? tools : undefined,
151
+ maxTokens: context.maxOutputTokens,
152
+ });
153
+ this.eventBus.emit(WorkerClawEvent.LLM_RESPONSE, {
154
+ taskId: task.taskId,
155
+ tokens: response.usage ? {
156
+ prompt: response.usage.promptTokens,
157
+ completion: response.usage.completionTokens,
158
+ } : undefined,
159
+ });
160
+ // 如果没有工具调用,直接返回
161
+ if (!response.hasToolCalls || !response.toolCalls || response.toolCalls.length === 0) {
162
+ return { ...response, toolCallRounds: round - 1 };
163
+ }
164
+ // 处理工具调用
165
+ this.eventBus.emit(WorkerClawEvent.TASK_PROGRESS, {
166
+ taskId: task.taskId,
167
+ progress: Math.min(20 + round * 10, 90),
168
+ message: `处理工具调用 (轮次 ${round})...`,
169
+ });
170
+ // 保存 assistant 消息(包含 tool_calls)
171
+ this.sessionManager.addMessage(task.taskId, {
172
+ role: 'assistant',
173
+ content: response.content || '',
174
+ tool_calls: response.toolCalls,
175
+ });
176
+ // 执行每个工具调用
177
+ for (const toolCall of response.toolCalls) {
178
+ const toolResult = await this.handleToolCall(toolCall, task, context);
179
+ // 保存工具结果
180
+ this.sessionManager.addMessage(task.taskId, {
181
+ role: 'tool',
182
+ content: typeof toolResult.content === 'string'
183
+ ? toolResult.content
184
+ : JSON.stringify(toolResult.content),
185
+ tool_call_id: toolCall.id,
186
+ name: toolCall.name,
187
+ });
188
+ }
189
+ // 更新进度
190
+ this.eventBus.emit(WorkerClawEvent.TASK_PROGRESS, {
191
+ taskId: task.taskId,
192
+ progress: Math.min(20 + round * 15, 90),
193
+ message: `已完成 ${response.toolCalls.length} 个工具调用`,
194
+ });
195
+ }
196
+ // 超过最大轮次
197
+ this.logger.warn(`LLM 循环超过最大轮次 [${task.taskId}]`, { maxRounds });
198
+ // 获取最终消息
199
+ const { messages } = this.sessionManager.getFittedMessages(task.taskId);
200
+ const lastAssistant = [...messages].reverse().find(m => m.role === 'assistant');
201
+ return {
202
+ content: lastAssistant?.content || '任务执行完成(达到最大工具调用轮次)',
203
+ hasToolCalls: false,
204
+ toolCalls: [],
205
+ usage: undefined,
206
+ model: this.config.llm.model,
207
+ finishReason: 'max_rounds',
208
+ allMessages: messages,
209
+ toolCallRounds: maxRounds,
210
+ };
211
+ }
212
+ /**
213
+ * 处理单个工具调用
214
+ */
215
+ async handleToolCall(toolCall, task, context) {
216
+ const { id, name, arguments: argsStr } = toolCall;
217
+ this.eventBus.emit(WorkerClawEvent.TOOL_CALLED, {
218
+ taskId: task.taskId,
219
+ toolName: name,
220
+ toolCallId: id,
221
+ });
222
+ try {
223
+ // 解析参数
224
+ const params = JSON.parse(argsStr || '{}');
225
+ // 执行工具
226
+ const toolCallObj = {
227
+ id,
228
+ name,
229
+ arguments: argsStr,
230
+ };
231
+ const result = await this.toolExecutor.execute(toolCallObj, {
232
+ taskId: task.taskId,
233
+ permissionLevel: context.permissionLevel,
234
+ workDir: this.config.security.sandbox.workDir,
235
+ remainingMs: context.timeoutMs - (Date.now() - context.receivedAt),
236
+ toolCallCount: 0,
237
+ maxToolCalls: 20,
238
+ });
239
+ this.eventBus.emit(WorkerClawEvent.TOOL_COMPLETED, {
240
+ taskId: task.taskId,
241
+ toolName: name,
242
+ toolCallId: id,
243
+ success: result.success,
244
+ });
245
+ return {
246
+ content: result.success ? result.content : `错误: ${result.error}`,
247
+ success: result.success,
248
+ };
249
+ }
250
+ catch (err) {
251
+ const error = err;
252
+ this.eventBus.emit(WorkerClawEvent.TOOL_BLOCKED, {
253
+ taskId: task.taskId,
254
+ toolName: name,
255
+ reason: error.message,
256
+ });
257
+ return {
258
+ content: `工具调用失败: ${error.message}`,
259
+ success: false,
260
+ };
261
+ }
262
+ }
263
+ /**
264
+ * 获取任务可用的工具
265
+ */
266
+ getToolsForTask(task, permLevel) {
267
+ // 合并内置工具和技能工具
268
+ const builtinTools = this.toolExecutor.getRegistry();
269
+ const builtinForLLM = builtinTools.getToolsForLLM(permLevel);
270
+ const skillTools = this.skillRegistry.getToolsFromAllSkills(permLevel);
271
+ const skillToolsForLLM = skillTools.map(st => ({
272
+ type: 'function',
273
+ function: {
274
+ name: st.tool.name,
275
+ description: st.tool.description,
276
+ parameters: st.tool.parameters,
277
+ },
278
+ }));
279
+ return [...builtinForLLM, ...skillToolsForLLM];
280
+ }
281
+ /**
282
+ * 构建系统提示(人格 + 技能 + 权限)
283
+ */
284
+ buildSystemPrompt(task, context) {
285
+ const availableTools = this.toolExecutor.getRegistry().getToolNames();
286
+ const skillAddons = this.skillRegistry.getSystemPromptAddons(context.permissionLevel);
287
+ const prompt = this.personality.buildSystemPrompt({
288
+ permissionLevel: context.permissionLevel,
289
+ maxOutputTokens: context.maxOutputTokens,
290
+ timeoutMs: context.timeoutMs,
291
+ availableTools,
292
+ currentDate: new Date().toISOString().slice(0, 10),
293
+ });
294
+ // 附加技能提示
295
+ if (skillAddons.length > 0) {
296
+ return prompt + '\n\n' + skillAddons.join('\n\n');
297
+ }
298
+ return prompt;
299
+ }
300
+ /**
301
+ * 构建用户消息
302
+ */
303
+ buildUserMessage(task) {
304
+ const lines = [
305
+ `## 任务信息`,
306
+ `- 任务ID: ${task.taskId}`,
307
+ `- 任务类型: ${task.taskType}`,
308
+ `- 标题: ${task.title}`,
309
+ ``,
310
+ `## 任务描述`,
311
+ task.description,
312
+ ];
313
+ if (task.attachments && task.attachments.length > 0) {
314
+ lines.push('', '## 附件');
315
+ for (const att of task.attachments) {
316
+ lines.push(`- [${att.type}] ${att.name || att.url}`);
317
+ }
318
+ }
319
+ if (task.posterName) {
320
+ lines.push('', `---`, `发单人: ${task.posterName}`);
321
+ }
322
+ return lines.join('\n');
323
+ }
324
+ /**
325
+ * 构建任务输出
326
+ */
327
+ buildOutputs(content) {
328
+ return [{ type: 'text', content }];
329
+ }
330
+ /**
331
+ * 获取人格实例
332
+ */
333
+ getPersonality() {
334
+ return this.personality;
335
+ }
336
+ /**
337
+ * 获取会话管理器
338
+ */
339
+ getSessionManager() {
340
+ return this.sessionManager;
341
+ }
342
+ /**
343
+ * 获取技能注册表
344
+ */
345
+ getSkillRegistry() {
346
+ return this.skillRegistry;
347
+ }
348
+ /**
349
+ * 注册技能
350
+ */
351
+ registerSkill(skill) {
352
+ this.skillRegistry.register(skill);
353
+ }
354
+ /**
355
+ * 初始化所有技能
356
+ */
357
+ async initializeSkills() {
358
+ return this.skillRegistry.initializeAll();
359
+ }
360
+ /**
361
+ * 获取会话统计
362
+ */
363
+ getSessionStats() {
364
+ return this.sessionManager.getStats();
365
+ }
366
+ /**
367
+ * 获取技能统计
368
+ */
369
+ getSkillStats() {
370
+ return this.skillRegistry.getStats();
371
+ }
372
+ /**
373
+ * 清理资源
374
+ */
375
+ dispose() {
376
+ this.sessionManager.dispose();
377
+ this.skillRegistry.disposeAll().catch(() => { });
378
+ }
379
+ }
380
+ //# sourceMappingURL=agent-engine.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent-engine.js","sourceRoot":"","sources":["../../src/agent/agent-engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,YAAY,EAAe,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAY,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,WAAW,EAA0B,MAAM,2BAA2B,CAAC;AAChF,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAwB5D,MAAM,OAAO,WAAW;IACd,MAAM,CAAS;IACf,MAAM,CAAoB;IAC1B,QAAQ,CAAW;IAEnB,GAAG,CAAY;IACf,WAAW,CAAc;IACzB,cAAc,CAAiB;IAC/B,YAAY,CAAe;IAC3B,WAAW,CAAc;IACzB,aAAa,CAAgB;IAErC,YAAY,MAAyB,EAAE,QAAkB;QACvD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAEzB,IAAI,CAAC,GAAG,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAEvD,OAAO;QACP,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC;YACvC,iBAAiB,EAAE,MAAM,CAAC,OAAO,EAAE,iBAAiB,IAAI,EAAE;YAC1D,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,UAAU,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI;YACxD,aAAa,EAAE;gBACb,SAAS,EAAE,MAAM,CAAC,OAAO,EAAE,SAAS,IAAI,IAAI;aAC7C;SACF,CAAC,CAAC;QAEH,OAAO;QACP,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QACxC,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAClC,YAAY,EACZ,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,EAC7B,QAAQ,CACT,CAAC;QAEF,OAAO;QACP,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QACzC,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAChC,IAAI,CAAC,aAAa,EAClB,MAAM,CAAC,WAAW,EAClB,QAAQ,CACT,CAAC;QAEF,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,aAAa,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,IAAU,EAAE,OAA6B;QACzD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,EAAE;YAC1C,IAAI,EAAE,IAAI,CAAC,QAAQ;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAE1E,OAAO;QACP,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3D,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAE7D,IAAI,CAAC;YACH,SAAS;YACT,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAChD,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE;gBAC1C,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,WAAW;aACrB,CAAC,CAAC;YAEH,OAAO;YACP,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,EAAE;gBACZ,OAAO,EAAE,aAAa;aACvB,CAAC,CAAC;YAEH,mBAAmB;YACnB,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAExD,SAAS;YACT,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE;gBAC1C,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,aAAa,CAAC,OAAO;aAC/B,CAAC,CAAC;YAEH,OAAO;YACP,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,MAAM,MAAM,GAAe;gBACzB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,WAAW;gBACnB,OAAO,EAAE,aAAa,CAAC,OAAO;gBAC9B,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC;gBACjD,UAAU,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;oBAChC,MAAM,EAAE,aAAa,CAAC,KAAK,CAAC,YAAY;oBACxC,UAAU,EAAE,aAAa,CAAC,KAAK,CAAC,gBAAgB;iBACjD,CAAC,CAAC,CAAC,SAAS;gBACb,UAAU;aACX,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,GAAG,EAAE;gBACxC,UAAU;gBACV,MAAM,EAAE,MAAM,CAAC,UAAU;gBACzB,cAAc,EAAE,aAAa,CAAC,cAAc,IAAI,CAAC;gBACjD,aAAa,EAAE,aAAa,CAAC,OAAO,CAAC,MAAM;aAC5C,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAEpF,OAAO,MAAM,CAAC;QAEhB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,MAAM,KAAK,GAAG,GAAY,CAAC;YAE3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;YACnF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9E,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAEhF,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,OAAO;gBACpB,UAAU;aACX,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,SAAS;YACT,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO,CACnB,IAAU,EACV,OAA6B,EAC7B,SAAS,GAAG,EAAE;QAEd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,OAAO,KAAK,GAAG,SAAS,EAAE,CAAC;YACzB,KAAK,EAAE,CAAC;YAER,WAAW;YACX,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAE/E,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,OAAO;YACP,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;YAElE,SAAS;YACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE;gBAC9C,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK;aAC7B,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;gBACnC,QAAQ;gBACR,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBAC3C,SAAS,EAAE,OAAO,CAAC,eAAe;aACnC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE;gBAC/C,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;oBACvB,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACnC,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,gBAAgB;iBAC5C,CAAC,CAAC,CAAC,SAAS;aACd,CAAC,CAAC;YAEH,gBAAgB;YAChB,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrF,OAAO,EAAE,GAAG,QAAQ,EAAE,cAAc,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC;YACpD,CAAC;YAED,SAAS;YACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,KAAK,GAAG,EAAE,EAAE,EAAE,CAAC;gBACvC,OAAO,EAAE,cAAc,KAAK,MAAM;aACnC,CAAC,CAAC;YAEH,iCAAiC;YACjC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE;gBAC1C,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE;gBAC/B,UAAU,EAAE,QAAQ,CAAC,SAAS;aAC/B,CAAC,CAAC;YAEH,WAAW;YACX,KAAK,MAAM,QAAQ,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBAC1C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,cAAc,CAC1C,QAAQ,EACR,IAAI,EACJ,OAAO,CACR,CAAC;gBAEF,SAAS;gBACT,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE;oBAC1C,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO,UAAU,CAAC,OAAO,KAAK,QAAQ;wBAC7C,CAAC,CAAC,UAAU,CAAC,OAAO;wBACpB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC;oBACtC,YAAY,EAAE,QAAQ,CAAC,EAAE;oBACzB,IAAI,EAAE,QAAQ,CAAC,IAAI;iBACpB,CAAC,CAAC;YACL,CAAC;YAED,OAAO;YACP,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,KAAK,GAAG,EAAE,EAAE,EAAE,CAAC;gBACvC,OAAO,EAAE,OAAO,QAAQ,CAAC,SAAS,CAAC,MAAM,QAAQ;aAClD,CAAC,CAAC;QACL,CAAC;QAED,SAAS;QACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAEjE,SAAS;QACT,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,aAAa,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAEhF,OAAO;YACL,OAAO,EAAE,aAAa,EAAE,OAAO,IAAI,oBAAoB;YACvD,YAAY,EAAE,KAAK;YACnB,SAAS,EAAE,EAAE;YACb,KAAK,EAAE,SAAS;YAChB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK;YAC5B,YAAY,EAAE,YAAY;YAC1B,WAAW,EAAE,QAAQ;YACrB,cAAc,EAAE,SAAS;SAC1B,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAC1B,QAAkB,EAClB,IAAU,EACV,OAA6B;QAE7B,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;QAElD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE;YAC9C,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,QAAQ,EAAE,IAAI;YACd,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,OAAO;YACP,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC;YAE3C,OAAO;YACP,MAAM,WAAW,GAAa;gBAC5B,EAAE;gBACF,IAAI;gBACJ,SAAS,EAAE,OAAO;aACnB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAC5C,WAAW,EACX;gBACE,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,eAAe,EAAE,OAAO,CAAC,eAAe;gBACxC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO;gBAC7C,WAAW,EAAE,OAAO,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,UAAU,CAAC;gBAClE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,EAAE;aACjB,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE;gBACjD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,IAAI;gBACd,UAAU,EAAE,EAAE;gBACd,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,KAAK,EAAE;gBAChE,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC;QAEJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,GAAY,CAAC;YAE3B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE;gBAC/C,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,IAAI;gBACd,MAAM,EAAE,KAAK,CAAC,OAAO;aACtB,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,WAAW,KAAK,CAAC,OAAO,EAAE;gBACnC,OAAO,EAAE,KAAK;aACf,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,IAAU,EAAE,SAA0B;QAC5D,cAAc;QACd,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC;QACrD,MAAM,aAAa,GAAG,YAAY,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAE7D,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;QACvE,MAAM,gBAAgB,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7C,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI;gBAClB,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW;gBAChC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU;aAC/B;SACF,CAAC,CAAC,CAAC;QAEJ,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,gBAAgB,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,IAAU,EAAE,OAA6B;QACjE,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,YAAY,EAAE,CAAC;QACtE,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,qBAAqB,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEtF,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC;YAChD,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,cAAc;YACd,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;SACnD,CAAC,CAAC;QAEH,SAAS;QACT,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,OAAO,MAAM,GAAG,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,IAAU;QACjC,MAAM,KAAK,GAAG;YACZ,SAAS;YACT,WAAW,IAAI,CAAC,MAAM,EAAE;YACxB,WAAW,IAAI,CAAC,QAAQ,EAAE;YAC1B,SAAS,IAAI,CAAC,KAAK,EAAE;YACrB,EAAE;YACF,SAAS;YACT,IAAI,CAAC,WAAW;SACjB,CAAC;QAEF,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACxB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,QAAQ,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,OAAe;QAClC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,KAAyC;QACrD,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB;QACpB,OAAO,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;QAC9B,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAClD,CAAC;CACF"}
@@ -0,0 +1,69 @@
1
+ /**
2
+ * 上下文窗口管理
3
+ *
4
+ * 管理对话历史,控制 token 数量
5
+ * 当超过窗口限制时自动截断旧消息
6
+ */
7
+ import type { LLMMessage } from '../types/agent.js';
8
+ export interface ContextWindowConfig {
9
+ /** 上下文窗口最大 token 数 */
10
+ maxTokens: number;
11
+ /** 保留的系统提示 token 数(始终保留系统消息) */
12
+ systemReserveTokens: number;
13
+ /** 保留最近 N 条消息(即使超出窗口) */
14
+ keepRecentMessages: number;
15
+ /** token 估算因子(中文约 1.5 token/字,英文约 0.75 token/词) */
16
+ tokenEstimateFactor: number;
17
+ /** 截断策略 */
18
+ truncationStrategy: 'oldest' | 'middle' | 'summarize';
19
+ }
20
+ export declare const DEFAULT_CONTEXT_WINDOW_CONFIG: ContextWindowConfig;
21
+ export interface ContextWindowStats {
22
+ totalTokens: number;
23
+ systemTokens: number;
24
+ conversationTokens: number;
25
+ messageCount: number;
26
+ isTruncated: boolean;
27
+ }
28
+ export declare class ContextWindow {
29
+ private logger;
30
+ private config;
31
+ constructor(config?: Partial<ContextWindowConfig>);
32
+ /**
33
+ * 估算消息的 token 数
34
+ */
35
+ estimateTokens(message: LLMMessage): number;
36
+ /**
37
+ * 计算消息列表的总 token 数
38
+ */
39
+ estimateTotalTokens(messages: LLMMessage[]): number;
40
+ /**
41
+ * 裁剪消息列表以适应上下文窗口
42
+ * 始终保留系统消息和最近 N 条消息
43
+ */
44
+ fitToWindow(messages: LLMMessage[]): {
45
+ messages: LLMMessage[];
46
+ stats: ContextWindowStats;
47
+ };
48
+ /**
49
+ * 按策略截断消息列表
50
+ */
51
+ private truncateMessages;
52
+ /**
53
+ * 从最旧的消息开始删除
54
+ */
55
+ private truncateOldest;
56
+ /**
57
+ * 从中间删除,保留首尾
58
+ */
59
+ private truncateMiddle;
60
+ /**
61
+ * 保留首尾,中间替换为摘要占位
62
+ */
63
+ private truncateWithSummary;
64
+ /**
65
+ * 获取统计信息
66
+ */
67
+ getStats(messages: LLMMessage[]): ContextWindowStats;
68
+ }
69
+ //# sourceMappingURL=context-window.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"context-window.d.ts","sourceRoot":"","sources":["../../src/agent/context-window.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAIpD,MAAM,WAAW,mBAAmB;IAClC,sBAAsB;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,gCAAgC;IAChC,mBAAmB,EAAE,MAAM,CAAC;IAC5B,yBAAyB;IACzB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,mDAAmD;IACnD,mBAAmB,EAAE,MAAM,CAAC;IAC5B,WAAW;IACX,kBAAkB,EAAE,QAAQ,GAAG,QAAQ,GAAG,WAAW,CAAC;CACvD;AAED,eAAO,MAAM,6BAA6B,EAAE,mBAM3C,CAAC;AAIF,MAAM,WAAW,kBAAkB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,OAAO,CAAC;CACtB;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAsB;gBAExB,MAAM,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC;IAKjD;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,UAAU,GAAG,MAAM;IA6B3C;;OAEG;IACH,mBAAmB,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,MAAM;IAInD;;;OAGG;IACH,WAAW,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG;QACnC,QAAQ,EAAE,UAAU,EAAE,CAAC;QACvB,KAAK,EAAE,kBAAkB,CAAC;KAC3B;IAkED;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAgBxB;;OAEG;IACH,OAAO,CAAC,cAAc;IAetB;;OAEG;IACH,OAAO,CAAC,cAAc;IAqBtB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAyB3B;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,GAAG,kBAAkB;CAYrD"}
@@ -0,0 +1,210 @@
1
+ /**
2
+ * 上下文窗口管理
3
+ *
4
+ * 管理对话历史,控制 token 数量
5
+ * 当超过窗口限制时自动截断旧消息
6
+ */
7
+ import { createLogger } from '../core/logger.js';
8
+ export const DEFAULT_CONTEXT_WINDOW_CONFIG = {
9
+ maxTokens: 8000,
10
+ systemReserveTokens: 1000,
11
+ keepRecentMessages: 4,
12
+ tokenEstimateFactor: 1.5,
13
+ truncationStrategy: 'oldest',
14
+ };
15
+ export class ContextWindow {
16
+ logger;
17
+ config;
18
+ constructor(config) {
19
+ this.config = { ...DEFAULT_CONTEXT_WINDOW_CONFIG, ...config };
20
+ this.logger = createLogger('ContextWindow');
21
+ }
22
+ /**
23
+ * 估算消息的 token 数
24
+ */
25
+ estimateTokens(message) {
26
+ let text = '';
27
+ if (typeof message.content === 'string') {
28
+ text = message.content;
29
+ }
30
+ else if (Array.isArray(message.content)) {
31
+ // 多模态内容
32
+ for (const part of message.content) {
33
+ if (typeof part === 'string') {
34
+ text += part;
35
+ }
36
+ else if (part && typeof part === 'object' && 'text' in part) {
37
+ text += part.text;
38
+ }
39
+ else if (part && typeof part === 'object' && 'image_url' in part) {
40
+ // 图片按固定 token 数估算
41
+ text += ' [IMAGE]';
42
+ }
43
+ }
44
+ }
45
+ // 添加角色和工具调用等结构开销
46
+ const overhead = message.tool_calls?.length
47
+ ? message.tool_calls.length * 50
48
+ : message.tool_call_id
49
+ ? 20
50
+ : 10;
51
+ return Math.ceil(text.length * this.config.tokenEstimateFactor) + overhead;
52
+ }
53
+ /**
54
+ * 计算消息列表的总 token 数
55
+ */
56
+ estimateTotalTokens(messages) {
57
+ return messages.reduce((sum, msg) => sum + this.estimateTokens(msg), 0);
58
+ }
59
+ /**
60
+ * 裁剪消息列表以适应上下文窗口
61
+ * 始终保留系统消息和最近 N 条消息
62
+ */
63
+ fitToWindow(messages) {
64
+ const systemMessages = messages.filter(m => m.role === 'system');
65
+ const conversationMessages = messages.filter(m => m.role !== 'system');
66
+ const systemTokens = this.estimateTotalTokens(systemMessages);
67
+ const availableTokens = this.config.maxTokens - this.config.systemReserveTokens;
68
+ // 如果系统提示本身就超出限制
69
+ if (systemTokens > this.config.systemReserveTokens) {
70
+ this.logger.warn('系统提示过长,可能影响上下文窗口', {
71
+ systemTokens,
72
+ reserveTokens: this.config.systemReserveTokens,
73
+ });
74
+ }
75
+ let resultMessages = conversationMessages;
76
+ let isTruncated = false;
77
+ // 计算最近 N 条必须保留的消息
78
+ const recentMessages = conversationMessages.slice(-this.config.keepRecentMessages);
79
+ const recentTokens = this.estimateTotalTokens(recentMessages);
80
+ // 可用于历史消息的 token 预算
81
+ const historyBudget = Math.max(0, availableTokens - recentTokens);
82
+ if (recentTokens > availableTokens) {
83
+ // 即使只保留最近消息也超出限制,只能截断最近消息
84
+ this.logger.warn('最近消息也超出窗口限制,进行截断');
85
+ resultMessages = this.truncateMessages(recentMessages, availableTokens);
86
+ isTruncated = true;
87
+ }
88
+ else {
89
+ // 检查总历史 token 数
90
+ const historyMessages = conversationMessages.slice(0, -this.config.keepRecentMessages);
91
+ const historyTokens = this.estimateTotalTokens(historyMessages);
92
+ if (historyTokens > historyBudget) {
93
+ // 需要裁剪历史消息
94
+ isTruncated = true;
95
+ const fitted = this.truncateMessages(historyMessages, historyBudget);
96
+ resultMessages = [...fitted, ...recentMessages];
97
+ }
98
+ }
99
+ const finalMessages = [...systemMessages, ...resultMessages];
100
+ const conversationTokens = this.estimateTotalTokens(resultMessages);
101
+ if (isTruncated) {
102
+ this.logger.debug('上下文窗口已裁剪', {
103
+ originalCount: conversationMessages.length,
104
+ fittedCount: resultMessages.length,
105
+ conversationTokens,
106
+ });
107
+ }
108
+ return {
109
+ messages: finalMessages,
110
+ stats: {
111
+ totalTokens: systemTokens + conversationTokens,
112
+ systemTokens,
113
+ conversationTokens,
114
+ messageCount: finalMessages.length,
115
+ isTruncated,
116
+ },
117
+ };
118
+ }
119
+ /**
120
+ * 按策略截断消息列表
121
+ */
122
+ truncateMessages(messages, budget) {
123
+ if (messages.length === 0)
124
+ return [];
125
+ switch (this.config.truncationStrategy) {
126
+ case 'oldest':
127
+ return this.truncateOldest(messages, budget);
128
+ case 'middle':
129
+ return this.truncateMiddle(messages, budget);
130
+ case 'summarize':
131
+ // 简化版 summarize: 保留首尾,中间放占位
132
+ return this.truncateWithSummary(messages, budget);
133
+ default:
134
+ return this.truncateOldest(messages, budget);
135
+ }
136
+ }
137
+ /**
138
+ * 从最旧的消息开始删除
139
+ */
140
+ truncateOldest(messages, budget) {
141
+ const result = [];
142
+ let usedTokens = 0;
143
+ // 从最新开始保留
144
+ for (let i = messages.length - 1; i >= 0; i--) {
145
+ const tokens = this.estimateTokens(messages[i]);
146
+ if (usedTokens + tokens > budget)
147
+ break;
148
+ result.unshift(messages[i]);
149
+ usedTokens += tokens;
150
+ }
151
+ return result;
152
+ }
153
+ /**
154
+ * 从中间删除,保留首尾
155
+ */
156
+ truncateMiddle(messages, budget) {
157
+ if (messages.length <= 2) {
158
+ return this.truncateOldest(messages, budget);
159
+ }
160
+ const head = messages[0];
161
+ const tail = messages[messages.length - 1];
162
+ const headTokens = this.estimateTokens(head);
163
+ const tailTokens = this.estimateTokens(tail);
164
+ const middleBudget = budget - headTokens - tailTokens;
165
+ if (middleBudget <= 0) {
166
+ return this.truncateOldest(messages, budget);
167
+ }
168
+ const middle = messages.slice(1, -1);
169
+ const fittedMiddle = this.truncateOldest(middle, middleBudget);
170
+ return [head, ...fittedMiddle, tail];
171
+ }
172
+ /**
173
+ * 保留首尾,中间替换为摘要占位
174
+ */
175
+ truncateWithSummary(messages, budget) {
176
+ if (messages.length <= 2) {
177
+ return this.truncateOldest(messages, budget);
178
+ }
179
+ const head = messages.slice(0, 2);
180
+ const tail = messages.slice(-2);
181
+ const headTokens = this.estimateTotalTokens(head);
182
+ const tailTokens = this.estimateTotalTokens(tail);
183
+ const summaryTokenCost = 100; // 摘要占位 token 成本
184
+ const remainingBudget = budget - headTokens - tailTokens - summaryTokenCost;
185
+ if (remainingBudget < 0) {
186
+ return this.truncateOldest(messages, budget);
187
+ }
188
+ const omittedCount = messages.length - 4;
189
+ const summaryMessage = {
190
+ role: 'system',
191
+ content: `[已省略 ${omittedCount} 条历史消息以节省上下文空间]`,
192
+ };
193
+ return [...head, summaryMessage, ...tail];
194
+ }
195
+ /**
196
+ * 获取统计信息
197
+ */
198
+ getStats(messages) {
199
+ const systemMessages = messages.filter(m => m.role === 'system');
200
+ const conversationMessages = messages.filter(m => m.role !== 'system');
201
+ return {
202
+ totalTokens: this.estimateTotalTokens(messages),
203
+ systemTokens: this.estimateTotalTokens(systemMessages),
204
+ conversationTokens: this.estimateTotalTokens(conversationMessages),
205
+ messageCount: messages.length,
206
+ isTruncated: false,
207
+ };
208
+ }
209
+ }
210
+ //# sourceMappingURL=context-window.js.map