pi-squeeze 0.2.1 → 0.3.0

package/README.md

@@ -86,9 +86,22 @@ Safety rules:
 | `/squeeze min <chars>` | smallest output worth compacting (default 400) |
 
 Settings are remembered in `<pi agent dir>/squeeze.json` (`PI_CODING_AGENT_DIR`, else `~/.pi/agent`); it also holds
-`lineChars`, `perGroup` and `total`, the limits of a compacted result. `PI_SQUEEZE=off` disables it at start.
+`lineChars`, `perGroup` and `total`, the limits of a compacted result, and `logSavings` (below). `PI_SQUEEZE=off` disables it at start.
 The status line shows the tokens saved.
 
+## Savings log
+
+Each compacted output adds one line to `<pi agent dir>/squeeze-savings.jsonl`, so the savings can be tracked
+over time; the model never sees it:
+
+```json
+{"ts":"2026-01-02T09:15:00.000Z","session":"…","cwd":"/work/app","tokens":1840,"chars":7360,"kind":"build"}
+```
+
+[tokenburn](https://github.com/sorinirimies/tokenburn) and [pi-tokenburn](https://github.com/sorinirimies/pi-tokenburn)
+read it as the "pi squeeze" saving. When rtk ran the command, rtk logs its own part; this log holds only what
+pi-squeeze cut after it. `"logSavings": false` in `squeeze.json` turns it off.
+
 ## Development
 
 ```bash

package/extensions/core/config.ts

@@ -14,6 +14,8 @@ export interface Config {
 	enabled: boolean;
 	/** Show the savings in the status line. */
 	showStatus: boolean;
+	/** Append each saving to <agent dir>/squeeze-savings.jsonl for usage trackers. */
+	logSavings: boolean;
 	/** Which compactors run. */
 	kinds: Record<Kind, boolean>;
 	/** Rewrite commands to their `rtk` form (git, cargo, docker, …) when the rtk binary is installed. */
@@ -31,6 +33,7 @@ export interface Config {
 export const DEFAULT_CONFIG: Config = {
 	enabled: true,
 	showStatus: true,
+	logSavings: true,
 	rtk: true,
 	kinds: Object.fromEntries(KINDS.map((k) => [k, true])) as Record<Kind, boolean>,
 	minChars: 400,
@@ -51,6 +54,7 @@ export function sanitizeConfig(raw: unknown): Config {
 	return {
 		enabled: bool(r.enabled, d.enabled),
 		showStatus: bool(r.showStatus, d.showStatus),
+		logSavings: bool(r.logSavings, d.logSavings),
 		rtk: bool(r.rtk, d.rtk),
 		kinds: Object.fromEntries(KINDS.map((kind) => [kind, bool(k[kind], d.kinds[kind])])) as Record<Kind, boolean>,
 		minChars: clampInt(r.minChars, 100, 1_000_000, d.minChars),

package/extensions/core/savings.ts

@@ -0,0 +1,42 @@
+/**
+ * The savings log: one JSON line per compacted output in <agent dir>/squeeze-savings.jsonl, so usage
+ * trackers (tokenburn, pi-tokenburn) can count what pi-squeeze saved over time. The model never sees
+ * it. When rtk ran the command, rtk logs its own part; this logs only what pi-squeeze cut after it.
+ */
+import { appendFile, mkdir } from "node:fs/promises";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
+import type { Kind } from "./compact.ts";
+import { resolveAgentDir } from "./config.ts";
+
+export const SAVINGS_LOG = "squeeze-savings.jsonl";
+
+export const savingsLogPath = (env: Record<string, string | undefined> = process.env, home: string = homedir()): string =>
+	join(resolveAgentDir(env, home), SAVINGS_LOG);
+
+/** One line of the log. `tokens` is estimated from `chars` (4 characters per token). */
+export interface SavingRecord {
+	/** When the output was compacted (ISO 8601, UTC). */
+	ts: string;
+	session: string;
+	cwd: string;
+	/** Tokens kept out of the context. */
+	tokens: number;
+	/** Characters kept out of the context. */
+	chars: number;
+	/** The compactor that did it. */
+	kind: Kind;
+}
+
+export const savingLine = (r: SavingRecord): string => JSON.stringify(r) + "\n";
+
+/** Appends one record. Never throws: a log that cannot be written must not affect the session. */
+export async function appendSaving(record: SavingRecord, path: string = savingsLogPath()): Promise<boolean> {
+	try {
+		await mkdir(dirname(path), { recursive: true });
+		await appendFile(path, savingLine(record), "utf8");
+		return true;
+	} catch {
+		return false;
+	}
+}

package/extensions/squeeze.ts

@@ -11,6 +11,7 @@ import { completions, parseCommand } from "./command/parse.ts";
 import { compact, KINDS } from "./core/compact.ts";
 import { type Config, DEFAULT_CONFIG, optionsOf, readConfig, saveConfig } from "./core/config.ts";
 import { rewrite } from "./core/rtk.ts";
+import { appendSaving } from "./core/savings.ts";
 import { fileIn, readFullOutput } from "./core/files.ts";
 import { saveFull } from "./core/save.ts";
 import { emptyStats, record, saved, summary, tokens } from "./core/stats.ts";
@@ -68,6 +69,17 @@ export default function (pi: ExtensionAPI) {
 		// Never more than pi would have shown.
 		if (!r || r.text.length >= shown.length) return;
 		record(stats, { ...r, before: shown.length });
+		if (config.logSavings) {
+			const chars = shown.length - r.after;
+			void appendSaving({
+				ts: new Date(Date.now()).toISOString(),
+				session: ctx.sessionManager?.getSessionId?.() ?? "",
+				cwd: ctx.cwd ?? "",
+				tokens: tokens(chars),
+				chars,
+				kind: r.kind,
+			});
+		}
 		showStatus(ctx);
 		const where = fullPath ?? (r.truncated ? saveFull(original) : undefined);
 		const text = where ? `${r.text}\n[squeeze: output shortened; full output: ${where}]` : r.text;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-squeeze",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Shrinks bash output before it reaches the agent's context: native compactors for git, search, listings, tests, builds (cargo, gradle, Maven, javac, Xcode, tsc, docker), Android (adb, lint, detekt), linters, package installs and logs, plus rtk when installed. Full output stays one temp file away.",
   "keywords": ["pi-package", "pi-extension", "pi", "tokens", "context", "compaction", "rtk"],
   "license": "MIT",