wassname/pi-auto-compressor
1
0
Fork 0
mirror of https://github.com/wassname/pi-auto-compressor.git synced 2026-06-27 15:16:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: adjust tail budget for forced compression and fix missing timestamps

Browse Source
This commit is contained in:
wassname
2026-04-23 15:36:03 +08:00
parent 9c400d0e0a
commit 5a4e708f4b
2 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
package.json
+1
View File
@@ -25,6 +25,7 @@
},
"peerDependencies": {
"@mariozechner/pi-coding-agent": "*",
"@mariozechner/pi-ai": "*",
"@mariozechner/pi-tui": "*",
"@sinclair/typebox": "*"
},
pruner.ts
+12 -3
View File
@@ -372,8 +372,13 @@ export async function applyPruning(
state.forceCompressNext ||
(totalTokens >= thresholdTokens && msgs.length > AUTO_COMPRESS_CONFIG.protectFirstN + 4)
) {
let tailBudget = Math.floor(thresholdTokens * AUTO_COMPRESS_CONFIG.summaryTargetRatio);
if (state.forceCompressNext) {
// Force compression: reduce tail budget to ensure we have something to compress
tailBudget = Math.min(tailBudget, Math.floor(totalTokens * 0.3));
}
state.forceCompressNext = false;
const tailBudget = Math.floor(thresholdTokens * AUTO_COMPRESS_CONFIG.summaryTargetRatio);
const compressStart = alignBoundaryForward(msgs, AUTO_COMPRESS_CONFIG.protectFirstN);
const compressEnd = findTailCutByTokens(msgs, compressStart, tailBudget);
@@ -397,7 +402,7 @@ export async function applyPruning(
if (flipped !== lastHeadRole) {
summaryRole = flipped;
} else {
const tailMsg = { ...msgs[compressEnd] };
const tailMsg = { ...msgs[compressEnd], timestamp: msgs[compressEnd].timestamp || Date.now() };
const originalContent = tailMsg.content || "";
tailMsg.content =
"## Goal\n" + summary + "\n\n--- END OF CONTEXT SUMMARY ---\n\n" +
@@ -422,6 +427,7 @@ export async function applyPruning(
compressed.push({
role: summaryRole,
content: prefix + "\n\n" + summary,
timestamp: Date.now(),
});
for (let i = compressEnd; i < msgs.length; i++) {
@@ -431,9 +437,12 @@ export async function applyPruning(
state.previousSummary = summary;
state.compressionCount++;
state.tokensSaved += totalTokens - estimateMessagesTokens(compressed);
return sanitizeToolPairs(compressed);
}
} else {
if (config.debug) {
console.log(`[ACP] Compression skipped: conversation too short (start: ${compressStart}, end: ${compressEnd})`);
}
}
}