Something went wrong. Try again.
more up to date at codeberg https://codeberg.org/cowie/tau.git
Something went wrong. Try again.
1.4 kB · 50 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
const COMPACT_THRESHOLD_TOKENS = 100_000;
export default function (pi: ExtensionAPI) { let previousTokens: number | null | undefined;
const triggerCompaction = (ctx: ExtensionContext, customInstructions?: string) => { if (ctx.hasUI) { ctx.ui.notify("Compaction started", "info"); } ctx.compact({ customInstructions, onComplete: () => { if (ctx.hasUI) { ctx.ui.notify("Compaction completed", "info"); } }, onError: (error) => { if (ctx.hasUI) { ctx.ui.notify(`Compaction failed: ${error.message}`, "error"); } }, }); };
pi.on("turn_end", (_event, ctx) => { const usage = ctx.getContextUsage(); const currentTokens = usage?.tokens ?? null; if (currentTokens === null) { return; }
const crossedThreshold = previousTokens !== undefined && previousTokens !== null && previousTokens <= COMPACT_THRESHOLD_TOKENS; previousTokens = currentTokens; if (!crossedThreshold || currentTokens <= COMPACT_THRESHOLD_TOKENS) { return; } triggerCompaction(ctx); });
pi.registerCommand("trigger-compact", { description: "Trigger compaction immediately", handler: async (args, ctx) => { const instructions = args.trim() || undefined; triggerCompaction(ctx, instructions); }, });}