ton/pi_harness
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: complete extension usage accounting

Browse Source 
closes #6509
This commit is contained in:
Mario Zechner
2026-07-20 17:04:12 +02:00
parent 2fd3868401
commit f8b74a4507
20 changed files with 267 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/agent/test/harness/compaction.test.ts
+10 -1
View File
@@ -20,6 +20,7 @@ import {
findCutPoint,
findTurnStartIndex,
generateSummary,
generateSummaryWithUsage,
getLastAssistantUsage,
prepareCompaction,
serializeConversation,
@@ -510,7 +511,7 @@ describe("harness compaction", () => {
]);
const summary = getOrThrow(
await generateSummary(messages, models, model, 2000, undefined, "focus", "old summary"),
await generateSummaryWithUsage(messages, models, model, 2000, undefined, "focus", "old summary"),
);
expect(summary.text).toContain("Test summary");
@@ -523,6 +524,14 @@ describe("harness compaction", () => {
expect(promptText).toContain("Additional focus: focus");
});
it("preserves the string result from generateSummary", async () => {
const messages: AgentMessage[] = [createUserMessage("Summarize this.")];
const { faux, model } = createFauxModel(false);
faux.setResponses([fauxAssistantMessage("## Goal\nTest summary")]);
expect(getOrThrow(await generateSummary(messages, models, model, 2000))).toBe("## Goal\nTest summary");
});
it("returns error results for failed or aborted summary generations", async () => {
const messages: AgentMessage[] = [createUserMessage("Summarize this.")];
const { faux: errorFaux, model: errorModel } = createFauxModel(false);