chatccc 0.2.250 → 0.2.252
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/deepccc-agent/os-prompts/darwin.md +8 -8
- package/deepccc-agent/os-prompts/linux.md +8 -8
- package/deepccc-agent/os-prompts/win32.md +11 -11
- package/deepccc-agent/package-lock.json +2027 -2027
- package/deepccc-agent/package.json +65 -65
- package/deepccc-agent/src/__tests__/chat-session.test.ts +877 -852
- package/deepccc-agent/src/__tests__/context.test.ts +341 -341
- package/deepccc-agent/src/__tests__/skills.test.ts +284 -284
- package/deepccc-agent/src/context.ts +465 -465
- package/deepccc-agent/src/file-tools.ts +38 -38
- package/deepccc-agent/src/index.ts +72 -37
- package/deepccc-agent/src/skills.ts +205 -205
- package/package.json +74 -74
- package/src/__tests__/builtin-chat-session.test.ts +532 -532
- package/src/__tests__/builtin-context.test.ts +319 -319
- package/src/__tests__/builtin-skills.test.ts +284 -284
- package/src/__tests__/config-utils.test.ts +40 -40
- package/src/__tests__/session-ccc-config.test.ts +66 -66
- package/src/__tests__/web-ui.test.ts +438 -438
|
@@ -1,341 +1,341 @@
|
|
|
1
|
-
import { mkdir, mkdtemp, readFile, writeFile } from "node:fs/promises";
|
|
2
|
-
import { tmpdir } from "node:os";
|
|
3
|
-
import { join } from "node:path";
|
|
4
|
-
|
|
5
|
-
import { describe, expect, it } from "vitest";
|
|
6
|
-
|
|
7
|
-
import {
|
|
8
|
-
buildPersistedAssistantMessage,
|
|
9
|
-
buildSummaryPrompt,
|
|
10
|
-
BuiltinContextManager,
|
|
11
|
-
estimateBuiltinContextTokens,
|
|
12
|
-
listBuiltinContextSessions,
|
|
13
|
-
newBuiltinSessionId,
|
|
14
|
-
serializeMessagesForSummary,
|
|
15
|
-
} from "../context.js";
|
|
16
|
-
|
|
17
|
-
describe("BuiltinContextManager", () => {
|
|
18
|
-
it("defaults the compaction threshold to 128K tokens (one third of the 384K model window)", () => {
|
|
19
|
-
const context = new BuiltinContextManager();
|
|
20
|
-
|
|
21
|
-
expect(context.compactAtTokens).toBe(128_000);
|
|
22
|
-
});
|
|
23
|
-
|
|
24
|
-
it("caps existing summaries at 8K chars when rebuilding the compaction prompt", () => {
|
|
25
|
-
const context = new BuiltinContextManager({
|
|
26
|
-
compactAtTokens: 100,
|
|
27
|
-
keepRecentMessages: 1,
|
|
28
|
-
persist: false,
|
|
29
|
-
});
|
|
30
|
-
context.setSummary("previous ".repeat(30_000));
|
|
31
|
-
context.appendMessage({ role: "user", content: "earlier request" });
|
|
32
|
-
context.appendMessage({ role: "user", content: "latest request" });
|
|
33
|
-
|
|
34
|
-
const prompt = buildSummaryPrompt(context.planCompaction()!);
|
|
35
|
-
|
|
36
|
-
expect(prompt).toContain("
|
|
37
|
-
expect(prompt.length).toBeLessThan(40_000);
|
|
38
|
-
});
|
|
39
|
-
|
|
40
|
-
it("keeps recent messages within the token budget instead of a fixed count", () => {
|
|
41
|
-
const context = new BuiltinContextManager({
|
|
42
|
-
compactAtTokens: 300,
|
|
43
|
-
keepRecentMessages: 16,
|
|
44
|
-
persist: false,
|
|
45
|
-
});
|
|
46
|
-
for (let index = 0; index < 8; index += 1) {
|
|
47
|
-
context.appendMessage({
|
|
48
|
-
role: index % 2 === 0 ? "user" : "assistant",
|
|
49
|
-
content: `${index}:` + "x".repeat(400),
|
|
50
|
-
});
|
|
51
|
-
}
|
|
52
|
-
|
|
53
|
-
const plan = context.planCompaction();
|
|
54
|
-
|
|
55
|
-
expect(plan).not.toBeNull();
|
|
56
|
-
expect(plan!.recentMessages.length).toBeLessThan(8);
|
|
57
|
-
expect(plan!.recentMessages.at(-1)?.content).toContain("7:");
|
|
58
|
-
expect(estimateBuiltinContextTokens("", plan!.recentMessages)).toBeLessThanOrEqual(300);
|
|
59
|
-
});
|
|
60
|
-
|
|
61
|
-
it("bounds oversized source material sent to the compaction model", () => {
|
|
62
|
-
const context = new BuiltinContextManager({
|
|
63
|
-
compactAtTokens: 100,
|
|
64
|
-
keepRecentMessages: 1,
|
|
65
|
-
persist: false,
|
|
66
|
-
});
|
|
67
|
-
context.setSummary("previous ".repeat(30_000));
|
|
68
|
-
context.appendMessage({ role: "assistant", content: "a".repeat(200_000) });
|
|
69
|
-
context.appendMessage({ role: "user", content: "latest request" });
|
|
70
|
-
|
|
71
|
-
const prompt = buildSummaryPrompt(context.planCompaction()!);
|
|
72
|
-
|
|
73
|
-
expect(prompt.length).toBeLessThan(90_000);
|
|
74
|
-
expect(prompt).toContain("
|
|
75
|
-
});
|
|
76
|
-
|
|
77
|
-
it("persists and restores summary, messages, and total message count", async () => {
|
|
78
|
-
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-"));
|
|
79
|
-
|
|
80
|
-
const first = new BuiltinContextManager({
|
|
81
|
-
persist: true,
|
|
82
|
-
contextDir: dir,
|
|
83
|
-
sessionId: "persisted",
|
|
84
|
-
});
|
|
85
|
-
first.setSummary("## 用户目标\n- 保留这个摘要");
|
|
86
|
-
first.appendMessage({ role: "user", content: "第一轮" });
|
|
87
|
-
first.appendMessage({ role: "assistant", content: "第一轮回复" });
|
|
88
|
-
first.save();
|
|
89
|
-
|
|
90
|
-
const restored = new BuiltinContextManager({
|
|
91
|
-
persist: true,
|
|
92
|
-
contextDir: dir,
|
|
93
|
-
sessionId: "persisted",
|
|
94
|
-
});
|
|
95
|
-
|
|
96
|
-
expect(restored.summary).toContain("保留这个摘要");
|
|
97
|
-
expect(restored.messages).toEqual([
|
|
98
|
-
{ role: "user", content: "第一轮" },
|
|
99
|
-
{ role: "assistant", content: "第一轮回复" },
|
|
100
|
-
]);
|
|
101
|
-
expect(restored.totalMessages).toBe(2);
|
|
102
|
-
});
|
|
103
|
-
|
|
104
|
-
it("selects only older messages for compaction and keeps recent messages raw", () => {
|
|
105
|
-
const context = new BuiltinContextManager({
|
|
106
|
-
compactAtTokens: 100,
|
|
107
|
-
keepRecentMessages: 2,
|
|
108
|
-
persist: false,
|
|
109
|
-
});
|
|
110
|
-
context.setSummary("x".repeat(500));
|
|
111
|
-
context.appendMessage({ role: "user", content: "旧用户消息" });
|
|
112
|
-
context.appendMessage({ role: "assistant", content: "旧助手回复" });
|
|
113
|
-
context.appendMessage({ role: "user", content: "近期用户消息" });
|
|
114
|
-
context.appendMessage({ role: "assistant", content: "近期助手回复" });
|
|
115
|
-
|
|
116
|
-
const plan = context.planCompaction();
|
|
117
|
-
|
|
118
|
-
expect(plan).not.toBeNull();
|
|
119
|
-
expect(plan?.oldMessages).toEqual([
|
|
120
|
-
{ role: "user", content: "旧用户消息" },
|
|
121
|
-
{ role: "assistant", content: "旧助手回复" },
|
|
122
|
-
]);
|
|
123
|
-
expect(plan?.recentMessages).toEqual([
|
|
124
|
-
{ role: "user", content: "近期用户消息" },
|
|
125
|
-
{ role: "assistant", content: "近期助手回复" },
|
|
126
|
-
]);
|
|
127
|
-
});
|
|
128
|
-
|
|
129
|
-
it("applies a compacted summary and builds model messages with summary plus recent raw turns", () => {
|
|
130
|
-
const context = new BuiltinContextManager({
|
|
131
|
-
compactAtTokens: 1,
|
|
132
|
-
keepRecentMessages: 1,
|
|
133
|
-
persist: false,
|
|
134
|
-
});
|
|
135
|
-
context.appendMessage({ role: "user", content: "old" });
|
|
136
|
-
context.appendMessage({ role: "assistant", content: "recent" });
|
|
137
|
-
|
|
138
|
-
const plan = context.planCompaction();
|
|
139
|
-
expect(plan).not.toBeNull();
|
|
140
|
-
context.applyCompaction("## 当前任务状态\n- 已压缩旧上下文", plan!);
|
|
141
|
-
|
|
142
|
-
expect(context.summary).toContain("已压缩旧上下文");
|
|
143
|
-
expect(context.messages).toEqual([{ role: "assistant", content: "recent" }]);
|
|
144
|
-
expect(context.buildModelMessages()).toEqual([
|
|
145
|
-
{
|
|
146
|
-
role: "user",
|
|
147
|
-
content: expect.stringContaining("
|
|
148
|
-
},
|
|
149
|
-
{ role: "assistant", content: "recent" },
|
|
150
|
-
]);
|
|
151
|
-
});
|
|
152
|
-
|
|
153
|
-
it("persists structured tool calls and restores them", async () => {
|
|
154
|
-
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-tools-"));
|
|
155
|
-
|
|
156
|
-
const first = new BuiltinContextManager({
|
|
157
|
-
persist: true,
|
|
158
|
-
contextDir: dir,
|
|
159
|
-
sessionId: "tool-persisted",
|
|
160
|
-
});
|
|
161
|
-
first.appendMessage({
|
|
162
|
-
role: "assistant",
|
|
163
|
-
content: "回复\n\n[
|
|
164
|
-
toolCalls: [
|
|
165
|
-
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
166
|
-
{ name: "read_file", input: "{\"path\":\"a.ts\"}", output: "...", is_error: true },
|
|
167
|
-
],
|
|
168
|
-
});
|
|
169
|
-
first.save();
|
|
170
|
-
|
|
171
|
-
const restored = new BuiltinContextManager({
|
|
172
|
-
persist: true,
|
|
173
|
-
contextDir: dir,
|
|
174
|
-
sessionId: "tool-persisted",
|
|
175
|
-
});
|
|
176
|
-
|
|
177
|
-
expect(restored.messages[0].toolCalls).toEqual([
|
|
178
|
-
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
179
|
-
{ name: "read_file", input: "{\"path\":\"a.ts\"}", output: "...", is_error: true },
|
|
180
|
-
]);
|
|
181
|
-
});
|
|
182
|
-
|
|
183
|
-
it("loads legacy context files without toolCalls and filters malformed entries", async () => {
|
|
184
|
-
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-legacy-"));
|
|
185
|
-
const state = {
|
|
186
|
-
version: 1,
|
|
187
|
-
createdAt: 1,
|
|
188
|
-
updatedAt: 2,
|
|
189
|
-
sessionId: "legacy",
|
|
190
|
-
summary: "",
|
|
191
|
-
totalMessages: 2,
|
|
192
|
-
compactedMessages: 0,
|
|
193
|
-
messages: [
|
|
194
|
-
{ role: "assistant", content: "老消息" },
|
|
195
|
-
{
|
|
196
|
-
role: "assistant",
|
|
197
|
-
content: "带 toolCalls",
|
|
198
|
-
toolCalls: [
|
|
199
|
-
{ name: "ok", input: "{\"a\":1}" },
|
|
200
|
-
{ name: "" }, // 非法:空 name 应被过滤
|
|
201
|
-
{ input: "no-name" }, // 非法:缺 name 应被过滤
|
|
202
|
-
{ name: "bad-type", input: 42 }, // 非法:input 非 string 应被忽略字段
|
|
203
|
-
],
|
|
204
|
-
},
|
|
205
|
-
],
|
|
206
|
-
};
|
|
207
|
-
await mkdir(join(dir, "legacy"));
|
|
208
|
-
await writeFile(join(dir, "legacy", "context.json"), JSON.stringify(state), "utf8");
|
|
209
|
-
|
|
210
|
-
const restored = new BuiltinContextManager({
|
|
211
|
-
persist: true,
|
|
212
|
-
contextDir: dir,
|
|
213
|
-
sessionId: "legacy",
|
|
214
|
-
});
|
|
215
|
-
|
|
216
|
-
expect(restored.messages[0].toolCalls).toBeUndefined();
|
|
217
|
-
expect(restored.messages[1].toolCalls).toEqual([
|
|
218
|
-
{ name: "ok", input: "{\"a\":1}" },
|
|
219
|
-
{ name: "bad-type" },
|
|
220
|
-
]);
|
|
221
|
-
expect(restored.totalMessages).toBe(2);
|
|
222
|
-
});
|
|
223
|
-
|
|
224
|
-
it("builds a persisted assistant message with transcript text plus structured tool calls", () => {
|
|
225
|
-
const message = buildPersistedAssistantMessage({
|
|
226
|
-
fullText: "回复正文",
|
|
227
|
-
transcriptLines: [
|
|
228
|
-
"tool_call run_command: {\"command\":\"npm test\"}",
|
|
229
|
-
"tool_result run_command: {\"exitCode\":0}",
|
|
230
|
-
],
|
|
231
|
-
toolCalls: [
|
|
232
|
-
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
233
|
-
],
|
|
234
|
-
});
|
|
235
|
-
|
|
236
|
-
expect(message.role).toBe("assistant");
|
|
237
|
-
expect(message.content).toContain("回复正文");
|
|
238
|
-
expect(message.content).toContain("[
|
|
239
|
-
expect(message.content).toContain("tool_call run_command");
|
|
240
|
-
expect(message.toolCalls).toEqual([
|
|
241
|
-
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
242
|
-
]);
|
|
243
|
-
});
|
|
244
|
-
|
|
245
|
-
it("builds a plain assistant message without tool transcript when no tools ran", () => {
|
|
246
|
-
const message = buildPersistedAssistantMessage({
|
|
247
|
-
fullText: "纯文本回复",
|
|
248
|
-
transcriptLines: [],
|
|
249
|
-
});
|
|
250
|
-
|
|
251
|
-
expect(message.content).toBe("纯文本回复");
|
|
252
|
-
expect(message.content).not.toContain("[
|
|
253
|
-
expect(message.toolCalls).toBeUndefined();
|
|
254
|
-
});
|
|
255
|
-
|
|
256
|
-
it("caps both assistant text and tool transcript in persisted messages", () => {
|
|
257
|
-
const message = buildPersistedAssistantMessage({
|
|
258
|
-
fullText: "a".repeat(10_000),
|
|
259
|
-
transcriptLines: Array.from({ length: 12 }, () => "x".repeat(8_000)),
|
|
260
|
-
maxAssistantChars: 2_000,
|
|
261
|
-
maxTranscriptChars: 4_000,
|
|
262
|
-
});
|
|
263
|
-
|
|
264
|
-
expect(message.content.length).toBeLessThan(7_000);
|
|
265
|
-
expect(message.content).toContain("
|
|
266
|
-
expect(message.content).toContain("
|
|
267
|
-
});
|
|
268
|
-
|
|
269
|
-
it("reset clears memory and the persisted context file", async () => {
|
|
270
|
-
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-reset-"));
|
|
271
|
-
const context = new BuiltinContextManager({
|
|
272
|
-
persist: true,
|
|
273
|
-
contextDir: dir,
|
|
274
|
-
sessionId: "reset",
|
|
275
|
-
});
|
|
276
|
-
context.setSummary("summary");
|
|
277
|
-
context.appendMessage({ role: "user", content: "hello" });
|
|
278
|
-
context.save();
|
|
279
|
-
|
|
280
|
-
context.reset();
|
|
281
|
-
|
|
282
|
-
expect(context.summary).toBe("");
|
|
283
|
-
expect(context.messages).toEqual([]);
|
|
284
|
-
expect(context.totalMessages).toBe(0);
|
|
285
|
-
|
|
286
|
-
const raw = await readFile(join(dir, "reset", "context.json"), "utf8");
|
|
287
|
-
const persisted = JSON.parse(raw) as { summary: string; messages: unknown[]; totalMessages: number };
|
|
288
|
-
expect(persisted.summary).toBe("");
|
|
289
|
-
expect(persisted.messages).toEqual([]);
|
|
290
|
-
expect(persisted.totalMessages).toBe(0);
|
|
291
|
-
});
|
|
292
|
-
|
|
293
|
-
it("persists cwd metadata and lists saved sessions newest first", async () => {
|
|
294
|
-
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-list-"));
|
|
295
|
-
|
|
296
|
-
const first = new BuiltinContextManager({
|
|
297
|
-
persist: true,
|
|
298
|
-
contextDir: dir,
|
|
299
|
-
sessionId: "older",
|
|
300
|
-
cwd: "C:\\repo-a",
|
|
301
|
-
});
|
|
302
|
-
first.appendMessage({ role: "user", content: "old" });
|
|
303
|
-
|
|
304
|
-
const second = new BuiltinContextManager({
|
|
305
|
-
persist: true,
|
|
306
|
-
contextDir: dir,
|
|
307
|
-
sessionId: "newer",
|
|
308
|
-
cwd: "C:\\repo-b",
|
|
309
|
-
});
|
|
310
|
-
second.appendMessage({ role: "user", content: "new" });
|
|
311
|
-
|
|
312
|
-
const sessions = listBuiltinContextSessions(dir);
|
|
313
|
-
|
|
314
|
-
expect(sessions.map((s) => s.sessionId)).toEqual(["newer", "older"]);
|
|
315
|
-
expect(sessions[0]).toEqual(expect.objectContaining({
|
|
316
|
-
cwd: "C:\\repo-b",
|
|
317
|
-
totalMessages: 1,
|
|
318
|
-
hasSummary: false,
|
|
319
|
-
}));
|
|
320
|
-
});
|
|
321
|
-
});
|
|
322
|
-
|
|
323
|
-
describe("builtin context helpers", () => {
|
|
324
|
-
it("creates readable timestamp-based session ids", () => {
|
|
325
|
-
const id = newBuiltinSessionId(new Date(2026, 6, 2, 12, 15, 30), "a1b2c3");
|
|
326
|
-
|
|
327
|
-
expect(id).toBe("session-20260702-121530-a1b2c3");
|
|
328
|
-
});
|
|
329
|
-
|
|
330
|
-
it("estimates tokens from summary and messages", () => {
|
|
331
|
-
expect(estimateBuiltinContextTokens("abc", [{ role: "user", content: "abcdef" }]))
|
|
332
|
-
.toBeGreaterThanOrEqual(3);
|
|
333
|
-
});
|
|
334
|
-
|
|
335
|
-
it("serializes messages for summarization with stable role labels", () => {
|
|
336
|
-
expect(serializeMessagesForSummary([
|
|
337
|
-
{ role: "user", content: "你好" },
|
|
338
|
-
{ role: "assistant", content: "收到" },
|
|
339
|
-
])).toContain("user");
|
|
340
|
-
});
|
|
341
|
-
});
|
|
1
|
+
import { mkdir, mkdtemp, readFile, writeFile } from "node:fs/promises";
|
|
2
|
+
import { tmpdir } from "node:os";
|
|
3
|
+
import { join } from "node:path";
|
|
4
|
+
|
|
5
|
+
import { describe, expect, it } from "vitest";
|
|
6
|
+
|
|
7
|
+
import {
|
|
8
|
+
buildPersistedAssistantMessage,
|
|
9
|
+
buildSummaryPrompt,
|
|
10
|
+
BuiltinContextManager,
|
|
11
|
+
estimateBuiltinContextTokens,
|
|
12
|
+
listBuiltinContextSessions,
|
|
13
|
+
newBuiltinSessionId,
|
|
14
|
+
serializeMessagesForSummary,
|
|
15
|
+
} from "../context.js";
|
|
16
|
+
|
|
17
|
+
describe("BuiltinContextManager", () => {
|
|
18
|
+
it("defaults the compaction threshold to 128K tokens (one third of the 384K model window)", () => {
|
|
19
|
+
const context = new BuiltinContextManager();
|
|
20
|
+
|
|
21
|
+
expect(context.compactAtTokens).toBe(128_000);
|
|
22
|
+
});
|
|
23
|
+
|
|
24
|
+
it("caps existing summaries at 8K chars when rebuilding the compaction prompt", () => {
|
|
25
|
+
const context = new BuiltinContextManager({
|
|
26
|
+
compactAtTokens: 100,
|
|
27
|
+
keepRecentMessages: 1,
|
|
28
|
+
persist: false,
|
|
29
|
+
});
|
|
30
|
+
context.setSummary("previous ".repeat(30_000));
|
|
31
|
+
context.appendMessage({ role: "user", content: "earlier request" });
|
|
32
|
+
context.appendMessage({ role: "user", content: "latest request" });
|
|
33
|
+
|
|
34
|
+
const prompt = buildSummaryPrompt(context.planCompaction()!);
|
|
35
|
+
|
|
36
|
+
expect(prompt).toContain("已有摘要为压缩而截断");
|
|
37
|
+
expect(prompt.length).toBeLessThan(40_000);
|
|
38
|
+
});
|
|
39
|
+
|
|
40
|
+
it("keeps recent messages within the token budget instead of a fixed count", () => {
|
|
41
|
+
const context = new BuiltinContextManager({
|
|
42
|
+
compactAtTokens: 300,
|
|
43
|
+
keepRecentMessages: 16,
|
|
44
|
+
persist: false,
|
|
45
|
+
});
|
|
46
|
+
for (let index = 0; index < 8; index += 1) {
|
|
47
|
+
context.appendMessage({
|
|
48
|
+
role: index % 2 === 0 ? "user" : "assistant",
|
|
49
|
+
content: `${index}:` + "x".repeat(400),
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
const plan = context.planCompaction();
|
|
54
|
+
|
|
55
|
+
expect(plan).not.toBeNull();
|
|
56
|
+
expect(plan!.recentMessages.length).toBeLessThan(8);
|
|
57
|
+
expect(plan!.recentMessages.at(-1)?.content).toContain("7:");
|
|
58
|
+
expect(estimateBuiltinContextTokens("", plan!.recentMessages)).toBeLessThanOrEqual(300);
|
|
59
|
+
});
|
|
60
|
+
|
|
61
|
+
it("bounds oversized source material sent to the compaction model", () => {
|
|
62
|
+
const context = new BuiltinContextManager({
|
|
63
|
+
compactAtTokens: 100,
|
|
64
|
+
keepRecentMessages: 1,
|
|
65
|
+
persist: false,
|
|
66
|
+
});
|
|
67
|
+
context.setSummary("previous ".repeat(30_000));
|
|
68
|
+
context.appendMessage({ role: "assistant", content: "a".repeat(200_000) });
|
|
69
|
+
context.appendMessage({ role: "user", content: "latest request" });
|
|
70
|
+
|
|
71
|
+
const prompt = buildSummaryPrompt(context.planCompaction()!);
|
|
72
|
+
|
|
73
|
+
expect(prompt.length).toBeLessThan(90_000);
|
|
74
|
+
expect(prompt).toContain("为压缩而截断");
|
|
75
|
+
});
|
|
76
|
+
|
|
77
|
+
it("persists and restores summary, messages, and total message count", async () => {
|
|
78
|
+
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-"));
|
|
79
|
+
|
|
80
|
+
const first = new BuiltinContextManager({
|
|
81
|
+
persist: true,
|
|
82
|
+
contextDir: dir,
|
|
83
|
+
sessionId: "persisted",
|
|
84
|
+
});
|
|
85
|
+
first.setSummary("## 用户目标\n- 保留这个摘要");
|
|
86
|
+
first.appendMessage({ role: "user", content: "第一轮" });
|
|
87
|
+
first.appendMessage({ role: "assistant", content: "第一轮回复" });
|
|
88
|
+
first.save();
|
|
89
|
+
|
|
90
|
+
const restored = new BuiltinContextManager({
|
|
91
|
+
persist: true,
|
|
92
|
+
contextDir: dir,
|
|
93
|
+
sessionId: "persisted",
|
|
94
|
+
});
|
|
95
|
+
|
|
96
|
+
expect(restored.summary).toContain("保留这个摘要");
|
|
97
|
+
expect(restored.messages).toEqual([
|
|
98
|
+
{ role: "user", content: "第一轮" },
|
|
99
|
+
{ role: "assistant", content: "第一轮回复" },
|
|
100
|
+
]);
|
|
101
|
+
expect(restored.totalMessages).toBe(2);
|
|
102
|
+
});
|
|
103
|
+
|
|
104
|
+
it("selects only older messages for compaction and keeps recent messages raw", () => {
|
|
105
|
+
const context = new BuiltinContextManager({
|
|
106
|
+
compactAtTokens: 100,
|
|
107
|
+
keepRecentMessages: 2,
|
|
108
|
+
persist: false,
|
|
109
|
+
});
|
|
110
|
+
context.setSummary("x".repeat(500));
|
|
111
|
+
context.appendMessage({ role: "user", content: "旧用户消息" });
|
|
112
|
+
context.appendMessage({ role: "assistant", content: "旧助手回复" });
|
|
113
|
+
context.appendMessage({ role: "user", content: "近期用户消息" });
|
|
114
|
+
context.appendMessage({ role: "assistant", content: "近期助手回复" });
|
|
115
|
+
|
|
116
|
+
const plan = context.planCompaction();
|
|
117
|
+
|
|
118
|
+
expect(plan).not.toBeNull();
|
|
119
|
+
expect(plan?.oldMessages).toEqual([
|
|
120
|
+
{ role: "user", content: "旧用户消息" },
|
|
121
|
+
{ role: "assistant", content: "旧助手回复" },
|
|
122
|
+
]);
|
|
123
|
+
expect(plan?.recentMessages).toEqual([
|
|
124
|
+
{ role: "user", content: "近期用户消息" },
|
|
125
|
+
{ role: "assistant", content: "近期助手回复" },
|
|
126
|
+
]);
|
|
127
|
+
});
|
|
128
|
+
|
|
129
|
+
it("applies a compacted summary and builds model messages with summary plus recent raw turns", () => {
|
|
130
|
+
const context = new BuiltinContextManager({
|
|
131
|
+
compactAtTokens: 1,
|
|
132
|
+
keepRecentMessages: 1,
|
|
133
|
+
persist: false,
|
|
134
|
+
});
|
|
135
|
+
context.appendMessage({ role: "user", content: "old" });
|
|
136
|
+
context.appendMessage({ role: "assistant", content: "recent" });
|
|
137
|
+
|
|
138
|
+
const plan = context.planCompaction();
|
|
139
|
+
expect(plan).not.toBeNull();
|
|
140
|
+
context.applyCompaction("## 当前任务状态\n- 已压缩旧上下文", plan!);
|
|
141
|
+
|
|
142
|
+
expect(context.summary).toContain("已压缩旧上下文");
|
|
143
|
+
expect(context.messages).toEqual([{ role: "assistant", content: "recent" }]);
|
|
144
|
+
expect(context.buildModelMessages()).toEqual([
|
|
145
|
+
{
|
|
146
|
+
role: "user",
|
|
147
|
+
content: expect.stringContaining("以下是更早的对话摘要"),
|
|
148
|
+
},
|
|
149
|
+
{ role: "assistant", content: "recent" },
|
|
150
|
+
]);
|
|
151
|
+
});
|
|
152
|
+
|
|
153
|
+
it("persists structured tool calls and restores them", async () => {
|
|
154
|
+
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-tools-"));
|
|
155
|
+
|
|
156
|
+
const first = new BuiltinContextManager({
|
|
157
|
+
persist: true,
|
|
158
|
+
contextDir: dir,
|
|
159
|
+
sessionId: "tool-persisted",
|
|
160
|
+
});
|
|
161
|
+
first.appendMessage({
|
|
162
|
+
role: "assistant",
|
|
163
|
+
content: "回复\n\n[工具记录]\ntool_call run_command: {}\ntool_result run_command: {}",
|
|
164
|
+
toolCalls: [
|
|
165
|
+
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
166
|
+
{ name: "read_file", input: "{\"path\":\"a.ts\"}", output: "...", is_error: true },
|
|
167
|
+
],
|
|
168
|
+
});
|
|
169
|
+
first.save();
|
|
170
|
+
|
|
171
|
+
const restored = new BuiltinContextManager({
|
|
172
|
+
persist: true,
|
|
173
|
+
contextDir: dir,
|
|
174
|
+
sessionId: "tool-persisted",
|
|
175
|
+
});
|
|
176
|
+
|
|
177
|
+
expect(restored.messages[0].toolCalls).toEqual([
|
|
178
|
+
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
179
|
+
{ name: "read_file", input: "{\"path\":\"a.ts\"}", output: "...", is_error: true },
|
|
180
|
+
]);
|
|
181
|
+
});
|
|
182
|
+
|
|
183
|
+
it("loads legacy context files without toolCalls and filters malformed entries", async () => {
|
|
184
|
+
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-legacy-"));
|
|
185
|
+
const state = {
|
|
186
|
+
version: 1,
|
|
187
|
+
createdAt: 1,
|
|
188
|
+
updatedAt: 2,
|
|
189
|
+
sessionId: "legacy",
|
|
190
|
+
summary: "",
|
|
191
|
+
totalMessages: 2,
|
|
192
|
+
compactedMessages: 0,
|
|
193
|
+
messages: [
|
|
194
|
+
{ role: "assistant", content: "老消息" },
|
|
195
|
+
{
|
|
196
|
+
role: "assistant",
|
|
197
|
+
content: "带 toolCalls",
|
|
198
|
+
toolCalls: [
|
|
199
|
+
{ name: "ok", input: "{\"a\":1}" },
|
|
200
|
+
{ name: "" }, // 非法:空 name 应被过滤
|
|
201
|
+
{ input: "no-name" }, // 非法:缺 name 应被过滤
|
|
202
|
+
{ name: "bad-type", input: 42 }, // 非法:input 非 string 应被忽略字段
|
|
203
|
+
],
|
|
204
|
+
},
|
|
205
|
+
],
|
|
206
|
+
};
|
|
207
|
+
await mkdir(join(dir, "legacy"));
|
|
208
|
+
await writeFile(join(dir, "legacy", "context.json"), JSON.stringify(state), "utf8");
|
|
209
|
+
|
|
210
|
+
const restored = new BuiltinContextManager({
|
|
211
|
+
persist: true,
|
|
212
|
+
contextDir: dir,
|
|
213
|
+
sessionId: "legacy",
|
|
214
|
+
});
|
|
215
|
+
|
|
216
|
+
expect(restored.messages[0].toolCalls).toBeUndefined();
|
|
217
|
+
expect(restored.messages[1].toolCalls).toEqual([
|
|
218
|
+
{ name: "ok", input: "{\"a\":1}" },
|
|
219
|
+
{ name: "bad-type" },
|
|
220
|
+
]);
|
|
221
|
+
expect(restored.totalMessages).toBe(2);
|
|
222
|
+
});
|
|
223
|
+
|
|
224
|
+
it("builds a persisted assistant message with transcript text plus structured tool calls", () => {
|
|
225
|
+
const message = buildPersistedAssistantMessage({
|
|
226
|
+
fullText: "回复正文",
|
|
227
|
+
transcriptLines: [
|
|
228
|
+
"tool_call run_command: {\"command\":\"npm test\"}",
|
|
229
|
+
"tool_result run_command: {\"exitCode\":0}",
|
|
230
|
+
],
|
|
231
|
+
toolCalls: [
|
|
232
|
+
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
233
|
+
],
|
|
234
|
+
});
|
|
235
|
+
|
|
236
|
+
expect(message.role).toBe("assistant");
|
|
237
|
+
expect(message.content).toContain("回复正文");
|
|
238
|
+
expect(message.content).toContain("[工具记录]");
|
|
239
|
+
expect(message.content).toContain("tool_call run_command");
|
|
240
|
+
expect(message.toolCalls).toEqual([
|
|
241
|
+
{ name: "run_command", input: "{\"command\":\"npm test\"}", output: "{\"exitCode\":0}" },
|
|
242
|
+
]);
|
|
243
|
+
});
|
|
244
|
+
|
|
245
|
+
it("builds a plain assistant message without tool transcript when no tools ran", () => {
|
|
246
|
+
const message = buildPersistedAssistantMessage({
|
|
247
|
+
fullText: "纯文本回复",
|
|
248
|
+
transcriptLines: [],
|
|
249
|
+
});
|
|
250
|
+
|
|
251
|
+
expect(message.content).toBe("纯文本回复");
|
|
252
|
+
expect(message.content).not.toContain("[工具记录]");
|
|
253
|
+
expect(message.toolCalls).toBeUndefined();
|
|
254
|
+
});
|
|
255
|
+
|
|
256
|
+
it("caps both assistant text and tool transcript in persisted messages", () => {
|
|
257
|
+
const message = buildPersistedAssistantMessage({
|
|
258
|
+
fullText: "a".repeat(10_000),
|
|
259
|
+
transcriptLines: Array.from({ length: 12 }, () => "x".repeat(8_000)),
|
|
260
|
+
maxAssistantChars: 2_000,
|
|
261
|
+
maxTranscriptChars: 4_000,
|
|
262
|
+
});
|
|
263
|
+
|
|
264
|
+
expect(message.content.length).toBeLessThan(7_000);
|
|
265
|
+
expect(message.content).toContain("助手回复已在上下文中截断");
|
|
266
|
+
expect(message.content).toContain("工具记录已截断");
|
|
267
|
+
});
|
|
268
|
+
|
|
269
|
+
it("reset clears memory and the persisted context file", async () => {
|
|
270
|
+
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-reset-"));
|
|
271
|
+
const context = new BuiltinContextManager({
|
|
272
|
+
persist: true,
|
|
273
|
+
contextDir: dir,
|
|
274
|
+
sessionId: "reset",
|
|
275
|
+
});
|
|
276
|
+
context.setSummary("summary");
|
|
277
|
+
context.appendMessage({ role: "user", content: "hello" });
|
|
278
|
+
context.save();
|
|
279
|
+
|
|
280
|
+
context.reset();
|
|
281
|
+
|
|
282
|
+
expect(context.summary).toBe("");
|
|
283
|
+
expect(context.messages).toEqual([]);
|
|
284
|
+
expect(context.totalMessages).toBe(0);
|
|
285
|
+
|
|
286
|
+
const raw = await readFile(join(dir, "reset", "context.json"), "utf8");
|
|
287
|
+
const persisted = JSON.parse(raw) as { summary: string; messages: unknown[]; totalMessages: number };
|
|
288
|
+
expect(persisted.summary).toBe("");
|
|
289
|
+
expect(persisted.messages).toEqual([]);
|
|
290
|
+
expect(persisted.totalMessages).toBe(0);
|
|
291
|
+
});
|
|
292
|
+
|
|
293
|
+
it("persists cwd metadata and lists saved sessions newest first", async () => {
|
|
294
|
+
const dir = await mkdtemp(join(tmpdir(), "chatccc-builtin-context-list-"));
|
|
295
|
+
|
|
296
|
+
const first = new BuiltinContextManager({
|
|
297
|
+
persist: true,
|
|
298
|
+
contextDir: dir,
|
|
299
|
+
sessionId: "older",
|
|
300
|
+
cwd: "C:\\repo-a",
|
|
301
|
+
});
|
|
302
|
+
first.appendMessage({ role: "user", content: "old" });
|
|
303
|
+
|
|
304
|
+
const second = new BuiltinContextManager({
|
|
305
|
+
persist: true,
|
|
306
|
+
contextDir: dir,
|
|
307
|
+
sessionId: "newer",
|
|
308
|
+
cwd: "C:\\repo-b",
|
|
309
|
+
});
|
|
310
|
+
second.appendMessage({ role: "user", content: "new" });
|
|
311
|
+
|
|
312
|
+
const sessions = listBuiltinContextSessions(dir);
|
|
313
|
+
|
|
314
|
+
expect(sessions.map((s) => s.sessionId)).toEqual(["newer", "older"]);
|
|
315
|
+
expect(sessions[0]).toEqual(expect.objectContaining({
|
|
316
|
+
cwd: "C:\\repo-b",
|
|
317
|
+
totalMessages: 1,
|
|
318
|
+
hasSummary: false,
|
|
319
|
+
}));
|
|
320
|
+
});
|
|
321
|
+
});
|
|
322
|
+
|
|
323
|
+
describe("builtin context helpers", () => {
|
|
324
|
+
it("creates readable timestamp-based session ids", () => {
|
|
325
|
+
const id = newBuiltinSessionId(new Date(2026, 6, 2, 12, 15, 30), "a1b2c3");
|
|
326
|
+
|
|
327
|
+
expect(id).toBe("session-20260702-121530-a1b2c3");
|
|
328
|
+
});
|
|
329
|
+
|
|
330
|
+
it("estimates tokens from summary and messages", () => {
|
|
331
|
+
expect(estimateBuiltinContextTokens("abc", [{ role: "user", content: "abcdef" }]))
|
|
332
|
+
.toBeGreaterThanOrEqual(3);
|
|
333
|
+
});
|
|
334
|
+
|
|
335
|
+
it("serializes messages for summarization with stable role labels", () => {
|
|
336
|
+
expect(serializeMessagesForSummary([
|
|
337
|
+
{ role: "user", content: "你好" },
|
|
338
|
+
{ role: "assistant", content: "收到" },
|
|
339
|
+
])).toContain("user");
|
|
340
|
+
});
|
|
341
|
+
});
|