120 lines
3.8 KiB
TypeScript
120 lines
3.8 KiB
TypeScript
import type { AgentMessage } from "@mariozechner/pi-agent-core";
|
|
import { describe, expect, it } from "vitest";
|
|
import { compactMessagesIfNeeded } from "../../src/main/agent-runtime/context-compactor";
|
|
|
|
function textOf(message: AgentMessage): string {
|
|
const content = (message as { content?: unknown }).content;
|
|
if (!Array.isArray(content)) return "";
|
|
const first = content[0] as { text?: string } | undefined;
|
|
return first?.text ?? "";
|
|
}
|
|
|
|
function makeLongTranscript(olderFacts: AgentMessage[]): AgentMessage[] {
|
|
const messages: AgentMessage[] = [
|
|
{
|
|
role: "user",
|
|
content: [{ type: "text", text: `Ship the approval workflow.\n${"context ".repeat(12_000)}` }],
|
|
timestamp: 1,
|
|
} as AgentMessage,
|
|
...olderFacts,
|
|
];
|
|
|
|
for (let i = 0; i < 42; i += 1) {
|
|
messages.push({
|
|
role: i % 2 === 0 ? "user" : "assistant",
|
|
content: [{ type: "text", text: `recent turn ${i}\n${"tail ".repeat(4_000)}` }],
|
|
timestamp: 100 + i,
|
|
} as AgentMessage);
|
|
}
|
|
return messages;
|
|
}
|
|
|
|
describe("context compactor tool result facts", () => {
|
|
it("uses tokenizer counts so long CJK transcripts compact before provider overflow", () => {
|
|
const messages: AgentMessage[] = [
|
|
{
|
|
role: "user",
|
|
content: [{ type: "text", text: "Start the Chinese market launch review." }],
|
|
timestamp: 1,
|
|
} as AgentMessage,
|
|
{
|
|
role: "assistant",
|
|
content: [{ type: "text", text: "I will keep the plan concise." }],
|
|
timestamp: 2,
|
|
} as AgentMessage,
|
|
{
|
|
role: "user",
|
|
content: [{ type: "text", text: "请完成市场复盘，保留所有客户反馈与行动项。 ".repeat(8_000) }],
|
|
timestamp: 3,
|
|
} as AgentMessage,
|
|
];
|
|
for (let i = 0; i < 42; i += 1) {
|
|
messages.push({
|
|
role: i % 2 === 0 ? "user" : "assistant",
|
|
content: [{ type: "text", text: `recent ${i}` }],
|
|
timestamp: 100 + i,
|
|
} as AgentMessage);
|
|
}
|
|
|
|
const outcome = compactMessagesIfNeeded(messages);
|
|
|
|
expect(outcome.changed).toBe(true);
|
|
expect(outcome.telemetry?.estimatedTokens).toBeGreaterThanOrEqual(120_000);
|
|
});
|
|
|
|
it("preserves task, approval, message, status, decision, and error facts from tool results", () => {
|
|
const messages = makeLongTranscript([
|
|
{
|
|
role: "assistant",
|
|
content: [
|
|
{ type: "text", text: "I will request approval for the deliverable." },
|
|
{
|
|
type: "toolCall",
|
|
id: "call-approval",
|
|
name: "file_approval",
|
|
args: { taskId: "task_123" },
|
|
},
|
|
],
|
|
timestamp: 2,
|
|
} as AgentMessage,
|
|
{
|
|
role: "toolResult",
|
|
toolCallId: "call-approval",
|
|
content: [{
|
|
type: "text",
|
|
text: JSON.stringify({
|
|
ok: false,
|
|
taskId: "task_123",
|
|
approvalId: "approval_456",
|
|
messageId: "message_789",
|
|
status: "blocked",
|
|
decision: "requires_human_review",
|
|
error: "missing reviewer",
|
|
}),
|
|
}],
|
|
details: {
|
|
taskId: "task_123",
|
|
approvalId: "approval_456",
|
|
messageId: "message_789",
|
|
status: "blocked",
|
|
decision: "requires_human_review",
|
|
error: "missing reviewer",
|
|
},
|
|
timestamp: 3,
|
|
} as AgentMessage,
|
|
]);
|
|
|
|
const outcome = compactMessagesIfNeeded(messages);
|
|
expect(outcome.changed).toBe(true);
|
|
const summary = textOf(outcome.messages?.[0] as AgentMessage);
|
|
|
|
expect(summary).toContain("file_approval");
|
|
expect(summary).toContain("taskId=task_123");
|
|
expect(summary).toContain("approvalId=approval_456");
|
|
expect(summary).toContain("messageId=message_789");
|
|
expect(summary).toContain("status=blocked");
|
|
expect(summary).toContain("decision=requires_human_review");
|
|
expect(summary).toContain("error=missing reviewer");
|
|
});
|
|
});
|