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,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
|