@diegopetrucci/pi-extensions 0.1.14 → 0.1.16

package/README.md

@@ -4,6 +4,8 @@ A collection of [pi](https://github.com/earendil-works/pi-mono) agent extensions
 
 - [`minimal-footer`](./extensions/minimal-footer): Replaces pi's built-in footer with a minimal configurable two-line layout: branch/repo on the first line, context/model on the second, optional `DUMB ZONE`, plus OpenAI Codex 5-hour and 7-day usage when available.
 - [`oracle`](./extensions/oracle): Adds an Amp-style read-only oracle tool that auto-selects the strongest reasoning model on the current provider/subscription, covers pi’s built-in providers with hardcoded rankings, sets reasoning to xhigh by default, and shows live status while running.
+- [`context-cap`](./extensions/context-cap): Caps effective model context windows at 200k tokens by default so pi avoids the `dumb zone`; toggle temporarily with `/context-cap`.
+- [`compact-bash`](./extensions/compact-bash): Renders collapsed assistant `bash` tool output as a one-line preview; toggle temporarily with `/compact-bash`.
 - [`permission-gate`](./extensions/permission-gate): Prompts for confirmation before dangerous bash commands like `rm -rf`, `sudo`, and `chmod 777`.
 - [`confirm-destructive`](./extensions/confirm-destructive): Confirms before destructive session actions like clear, switch, and fork.
 - [`notify`](./extensions/notify): Sends configurable terminal, desktop, bell, and sound notifications when pi finishes and is ready for input.
@@ -21,7 +23,7 @@ pi install npm:@diegopetrucci/pi-extensions
 Or pin the GitHub package to this release:
 
 ```bash
-pi install git:github.com/diegopetrucci/pi-extensions@v0.1.14
+pi install git:github.com/diegopetrucci/pi-extensions@v0.1.16
 ```
 
 Or a specific extension:

package/extensions/compact-bash/README.md

@@ -0,0 +1,49 @@
+# compact-bash
+
+A pi extension that makes collapsed assistant `bash` tool output much quieter.
+
+When enabled, collapsed bash output renders as one output line plus an inline hidden-line count and `Ctrl+O` expand hint. Expanding with `Ctrl+O` still shows the full rendered output.
+
+## Commands
+
+```text
+/compact-bash status
+/compact-bash off
+/compact-bash on
+/compact-bash toggle
+```
+
+The extension starts enabled by default. Disabling is temporary for the current extension runtime/session; after `/reload`, `/new`, `/resume`, or `/fork`, it starts enabled again.
+
+## Install
+
+### Standalone npm package
+
+```bash
+pi install npm:@diegopetrucci/pi-compact-bash
+```
+
+### Collection package
+
+```bash
+pi install npm:@diegopetrucci/pi-extensions
+```
+
+### GitHub package
+
+```bash
+pi install git:github.com/diegopetrucci/pi-extensions
+```
+
+Then reload pi:
+
+```text
+/reload
+```
+
+## Notes
+
+- This extension overrides pi's built-in `bash` tool so it can customize only the TUI renderer.
+- It reuses pi's built-in bash implementation and preserves `shellPath`/`shellCommandPrefix` settings when they are available from settings files.
+- It does not truncate or rewrite the actual tool result sent to the model.
+- It affects assistant-invoked `bash` tool rows. User `!`/`!!` bash commands are rendered by a separate pi component and keep pi's default preview behavior.

package/extensions/compact-bash/index.ts

@@ -0,0 +1,241 @@
+import {
+	createBashToolDefinition,
+	DEFAULT_MAX_BYTES,
+	formatSize,
+	keyHint,
+	SettingsManager,
+	type ExtensionAPI,
+} from "@earendil-works/pi-coding-agent";
+import { Container, Text, truncateToWidth } from "@earendil-works/pi-tui";
+
+const COLLAPSED_PREVIEW_LINES = 1;
+
+type BashToolDefinition = ReturnType<typeof createBashToolDefinition>;
+type BashRenderResult = NonNullable<BashToolDefinition["renderResult"]>;
+type BashRenderResultParams = Parameters<BashRenderResult>;
+type BashRenderState = {
+	endedAt?: number;
+	interval?: ReturnType<typeof setInterval>;
+};
+
+function getTextOutput(result: BashRenderResultParams[0]): string {
+	return result.content
+		.filter((content) => content.type === "text")
+		.map((content) => content.text ?? "")
+		.join("\n")
+		.trim();
+}
+
+function formatDuration(ms: number): string {
+	return `${(ms / 1000).toFixed(1)}s`;
+}
+
+function visibleLength(text: string): number {
+	return text.replace(/\x1b\[[0-?]*[ -/]*[@-~]/g, "").length;
+}
+
+function renderCollapsedLine(
+	line: string,
+	hiddenLines: number,
+	theme: BashRenderResultParams[2],
+	width: number,
+): string {
+	if (width <= 0) return "";
+
+	const styledLine = theme.fg("toolOutput", line);
+	if (hiddenLines <= 0) {
+		return truncateToWidth(styledLine, width, "...");
+	}
+
+	const hint = theme.fg(
+		"muted",
+		`  ... ${hiddenLines} more line${hiddenLines === 1 ? "" : "s"} (${keyHint("app.tools.expand", "to expand")})`,
+	);
+	const hintWidth = visibleLength(hint);
+	if (hintWidth + 8 > width) {
+		return truncateToWidth(`${styledLine}${hint}`, width, "...");
+	}
+
+	return `${truncateToWidth(styledLine, width - hintWidth, "...")}${hint}`;
+}
+
+function getTruncationWarnings(result: BashRenderResultParams[0]): string[] {
+	const details = result.details as
+		| {
+				truncation?: {
+					truncated?: boolean;
+					truncatedBy?: "lines" | "bytes";
+					outputLines?: number;
+					totalLines?: number;
+					maxBytes?: number;
+				};
+				fullOutputPath?: string;
+		  }
+		| undefined;
+	const truncation = details?.truncation;
+	const warnings: string[] = [];
+
+	if (details?.fullOutputPath) {
+		warnings.push(`Full output: ${details.fullOutputPath}`);
+	}
+
+	if (truncation?.truncated) {
+		if (truncation.truncatedBy === "lines") {
+			warnings.push(`Truncated: showing ${truncation.outputLines} of ${truncation.totalLines} lines`);
+		} else {
+			warnings.push(
+				`Truncated: ${truncation.outputLines} lines shown (${formatSize(truncation.maxBytes ?? DEFAULT_MAX_BYTES)} limit)`,
+			);
+		}
+	}
+
+	return warnings;
+}
+
+function syncElapsedTimer(options: BashRenderResultParams[1], context: BashRenderResultParams[3]): BashRenderState {
+	const state = context.state as BashRenderState & { startedAt?: number };
+
+	if (state.startedAt !== undefined && options.isPartial && !state.interval) {
+		state.interval = setInterval(() => context.invalidate(), 1000);
+	}
+
+	if (!options.isPartial || context.isError) {
+		state.endedAt ??= Date.now();
+		if (state.interval) {
+			clearInterval(state.interval);
+			state.interval = undefined;
+		}
+	}
+
+	return state;
+}
+
+const renderCompactBashResult: BashRenderResult = (result, options, theme, context) => {
+	const state = syncElapsedTimer(options, context);
+	const component = (context.lastComponent as Container | undefined) ?? new Container();
+	component.clear();
+
+	const output = getTextOutput(result);
+	const outputLines = output ? output.split("\n") : [];
+
+	if (outputLines.length > 0) {
+		if (options.expanded) {
+			const displayText = outputLines.map((line) => theme.fg("toolOutput", line)).join("\n");
+			component.addChild(new Text(displayText, 0, 0));
+		} else {
+			const firstLine = outputLines[0] ?? "";
+			const hiddenLines = Math.max(0, outputLines.length - COLLAPSED_PREVIEW_LINES);
+			component.addChild({
+				render: (width) => [renderCollapsedLine(firstLine, hiddenLines, theme, width)],
+				invalidate: () => undefined,
+			});
+		}
+	} else if (options.isPartial) {
+		component.addChild(new Text(theme.fg("muted", "Running..."), 0, 0));
+	}
+
+	const warnings = getTruncationWarnings(result);
+	if (warnings.length > 0) {
+		component.addChild(new Text(theme.fg("warning", `[${warnings.join(". ")}]`), 0, 0));
+	}
+
+	if (options.expanded) {
+		const statusParts: string[] = [];
+		if (outputLines.length > COLLAPSED_PREVIEW_LINES) {
+			statusParts.push(`(${keyHint("app.tools.expand", "to collapse")})`);
+		}
+		if (state.startedAt !== undefined) {
+			const endTime = state.endedAt ?? Date.now();
+			statusParts.push(theme.fg("muted", `${options.isPartial ? "Elapsed" : "Took"} ${formatDuration(endTime - state.startedAt)}`));
+		}
+		if (statusParts.length > 0) {
+			component.addChild(new Text(statusParts.join("\n"), 0, 0));
+		}
+	}
+
+	component.invalidate();
+	return component;
+};
+
+function createBaseBashToolDefinition(cwd: string): BashToolDefinition {
+	try {
+		const settings = SettingsManager.create(cwd);
+		return createBashToolDefinition(cwd, {
+			commandPrefix: settings.getShellCommandPrefix(),
+			shellPath: settings.getShellPath(),
+		});
+	} catch {
+		return createBashToolDefinition(cwd);
+	}
+}
+
+function createCompactBashToolDefinition(cwd: string, enabled: boolean): BashToolDefinition {
+	const base = createBaseBashToolDefinition(cwd);
+	if (!enabled) return base;
+
+	return {
+		...base,
+		renderResult: renderCompactBashResult,
+	};
+}
+
+export default function compactBashExtension(pi: ExtensionAPI) {
+	let enabled = true;
+
+	function registerBashTool(cwd: string): void {
+		pi.registerTool(createCompactBashToolDefinition(cwd, enabled));
+	}
+
+	pi.on("session_start", async (_event, ctx) => {
+		registerBashTool(ctx.cwd);
+	});
+
+	pi.registerCommand("compact-bash", {
+		description: "Toggle one-line collapsed previews for bash tool output",
+		getArgumentCompletions: (prefix) => {
+			const commands = ["on", "off", "toggle", "status"];
+			const query = prefix.trim().toLowerCase();
+			const matches = commands.filter((command) => command.startsWith(query));
+			return matches.length > 0 ? matches.map((value) => ({ value, label: value })) : null;
+		},
+		handler: async (args, ctx) => {
+			const action = args.trim().toLowerCase() || "toggle";
+
+			if (action === "on" || action === "enable") {
+				enabled = true;
+				registerBashTool(ctx.cwd);
+				ctx.ui.notify("Compact bash previews enabled: collapsed bash output shows one line.", "info");
+				return;
+			}
+
+			if (action === "off" || action === "disable") {
+				enabled = false;
+				registerBashTool(ctx.cwd);
+				ctx.ui.notify("Compact bash previews disabled: restored pi's standard bash renderer.", "info");
+				return;
+			}
+
+			if (action === "toggle") {
+				enabled = !enabled;
+				registerBashTool(ctx.cwd);
+				ctx.ui.notify(
+					enabled
+						? "Compact bash previews enabled: collapsed bash output shows one line."
+						: "Compact bash previews disabled: restored pi's standard bash renderer.",
+					"info",
+				);
+				return;
+			}
+
+			if (action === "status") {
+				ctx.ui.notify(
+					`Compact bash previews are ${enabled ? "enabled" : "disabled"}. Collapsed preview lines: ${enabled ? COLLAPSED_PREVIEW_LINES : "pi default"}.`,
+					"info",
+				);
+				return;
+			}
+
+			ctx.ui.notify("Usage: /compact-bash on | off | toggle | status", "warning");
+		},
+	});
+}

package/extensions/compact-bash/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "@diegopetrucci/pi-compact-bash",
+  "version": "0.1.0",
+  "description": "A pi extension that renders collapsed bash tool output as a one-line preview.",
+  "keywords": ["pi-package", "pi", "bash", "terminal"],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/diegopetrucci/pi-extensions.git",
+    "directory": "extensions/compact-bash"
+  },
+  "files": [
+    "index.ts",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "pi": {
+    "extensions": [
+      "index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
+  }
+}

package/extensions/context-cap/README.md

@@ -0,0 +1,60 @@
+# context-cap
+
+A pi extension that treats large-context models as having an effective 200k-token context window, so pi's built-in auto-compaction starts earlier, and avoids the dumb zone.
+
+By default, pi auto-compacts when:
+
+```text
+contextTokens > model.contextWindow - reserveTokens
+```
+
+This extension changes the active model's in-memory `contextWindow` to:
+
+```text
+min(originalContextWindow, 200000)
+```
+
+With pi's default `reserveTokens` of 16,384, models larger than 200k will proactively compact around 183,616 tokens.
+
+## Commands
+
+```text
+/context-cap status
+/context-cap off
+/context-cap on
+/context-cap toggle
+```
+
+The extension starts enabled by default. Disabling is temporary for the current extension runtime/session; after `/reload`, `/new`, `/resume`, or `/fork`, the extension starts enabled again.
+
+## Install
+
+### Standalone npm package
+
+```bash
+pi install npm:@diegopetrucci/pi-context-cap
+```
+
+### Collection package
+
+```bash
+pi install npm:@diegopetrucci/pi-extensions
+```
+
+### GitHub package
+
+```bash
+pi install git:github.com/diegopetrucci/pi-extensions
+```
+
+Then reload pi:
+
+```text
+/reload
+```
+
+## Notes
+
+- This extension mutates pi's in-memory model metadata only. It does not edit `models.json`.
+- The cap affects pi logic that reads `model.contextWindow`, including auto-compaction thresholding and UI context-window display.
+- Because pi also uses `model.contextWindow` for some overflow detection, a request that succeeds above 200k tokens on a larger model may be treated as overflow and retried after compaction. Use `/context-cap off` if you need the full model window temporarily.

package/extensions/context-cap/index.ts

@@ -0,0 +1,179 @@
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { Model, Api } from "@earendil-works/pi-ai";
+
+const DEFAULT_MAX_CONTEXT_WINDOW = 200_000;
+type AnyModel = Model<Api> | Model<any>;
+
+const originalContextWindows = new WeakMap<AnyModel, number>();
+const touchedModels = new Set<AnyModel>();
+
+type ApplyResult = {
+	changed: boolean;
+	key: string;
+	original: number;
+	effective: number;
+};
+
+function modelKey(model: AnyModel): string {
+	return `${model.provider}/${model.id}`;
+}
+
+function getOriginalContextWindow(model: AnyModel): number {
+	const existing = originalContextWindows.get(model);
+	if (typeof existing === "number") return existing;
+
+	originalContextWindows.set(model, model.contextWindow);
+	touchedModels.add(model);
+	return model.contextWindow;
+}
+
+function getEffectiveContextWindow(model: AnyModel): number {
+	return Math.min(getOriginalContextWindow(model), DEFAULT_MAX_CONTEXT_WINDOW);
+}
+
+function applyContextCap(model: AnyModel | undefined): ApplyResult | undefined {
+	if (!model) return undefined;
+
+	const key = modelKey(model);
+	const original = getOriginalContextWindow(model);
+	const effective = Math.min(original, DEFAULT_MAX_CONTEXT_WINDOW);
+	const changed = model.contextWindow !== effective;
+	model.contextWindow = effective;
+
+	return { changed, key, original, effective };
+}
+
+function restoreContextWindow(model: AnyModel | undefined): boolean {
+	if (!model) return false;
+
+	const original = originalContextWindows.get(model);
+	if (typeof original !== "number" || model.contextWindow === original) return false;
+
+	model.contextWindow = original;
+	return true;
+}
+
+function forEachRegistryModel(ctx: ExtensionContext, callback: (model: AnyModel) => void): void {
+	try {
+		for (const model of ctx.modelRegistry.getAll()) {
+			callback(model);
+		}
+	} catch {
+		// Best effort only. The active ctx.model is handled separately by callers.
+	}
+}
+
+function applyContextCapToSession(ctx: ExtensionContext): number {
+	let changed = 0;
+
+	forEachRegistryModel(ctx, (model) => {
+		if (applyContextCap(model)?.changed) changed++;
+	});
+
+	if (applyContextCap(ctx.model)?.changed) changed++;
+	return changed;
+}
+
+function restoreContextCapForSession(ctx: ExtensionContext): number {
+	let changed = 0;
+
+	forEachRegistryModel(ctx, (model) => {
+		if (restoreContextWindow(model)) changed++;
+	});
+
+	if (restoreContextWindow(ctx.model)) changed++;
+	return changed;
+}
+
+function formatTokens(tokens: number): string {
+	return tokens >= 1000 ? `${Math.round(tokens / 1000)}k` : String(tokens);
+}
+
+export default function contextCapExtension(pi: ExtensionAPI) {
+	let enabled = true;
+
+	pi.on("session_start", async (_event, ctx) => {
+		if (enabled) applyContextCapToSession(ctx);
+	});
+
+	pi.on("model_select", async (event, ctx) => {
+		if (!enabled) return;
+
+		const result = applyContextCap(event.model);
+		if (!result || !ctx.hasUI) return;
+
+		ctx.ui.setStatus(
+			"context-cap",
+			result.original > DEFAULT_MAX_CONTEXT_WINDOW
+				? `ctx cap ${formatTokens(result.effective)}/${formatTokens(result.original)}`
+				: undefined,
+		);
+	});
+
+	pi.on("session_shutdown", async (_event, ctx) => {
+		restoreContextCapForSession(ctx);
+		for (const model of touchedModels) restoreContextWindow(model);
+	});
+
+	pi.registerCommand("context-cap", {
+		description: "Toggle the 200k effective context-window cap for auto-compaction",
+		getArgumentCompletions: (prefix) => {
+			const commands = ["on", "off", "toggle", "status"];
+			const matches = commands.filter((command) => command.startsWith(prefix.trim()));
+			return matches.length > 0 ? matches.map((value) => ({ value, label: value })) : null;
+		},
+		handler: async (args, ctx) => {
+			const action = args.trim().toLowerCase() || "toggle";
+
+			if (action === "on" || action === "enable") {
+				enabled = true;
+				const changed = applyContextCapToSession(ctx);
+				ctx.ui.setStatus("context-cap", `ctx cap ${formatTokens(DEFAULT_MAX_CONTEXT_WINDOW)}`);
+				ctx.ui.notify(`Context cap enabled (${changed} model window(s) capped/restored).`, "info");
+				return;
+			}
+
+			if (action === "off" || action === "disable") {
+				enabled = false;
+				const changed = restoreContextCapForSession(ctx);
+				ctx.ui.setStatus("context-cap", undefined);
+				ctx.ui.notify(`Context cap disabled for this extension session (${changed} model window(s) restored).`, "info");
+				return;
+			}
+
+			if (action === "toggle") {
+				if (enabled) {
+					enabled = false;
+					const changed = restoreContextCapForSession(ctx);
+					ctx.ui.setStatus("context-cap", undefined);
+					ctx.ui.notify(`Context cap disabled for this extension session (${changed} model window(s) restored).`, "info");
+				} else {
+					enabled = true;
+					const changed = applyContextCapToSession(ctx);
+					ctx.ui.setStatus("context-cap", `ctx cap ${formatTokens(DEFAULT_MAX_CONTEXT_WINDOW)}`);
+					ctx.ui.notify(`Context cap enabled (${changed} model window(s) capped/restored).`, "info");
+				}
+				return;
+			}
+
+			if (action === "status") {
+				const model = ctx.model;
+				const status = enabled ? "enabled" : "disabled";
+				if (!model) {
+					ctx.ui.notify(`Context cap is ${status}. No model selected.`, "info");
+					return;
+				}
+
+				const original = getOriginalContextWindow(model);
+				const effective = enabled ? getEffectiveContextWindow(model) : model.contextWindow;
+				ctx.ui.notify(
+					`Context cap is ${status}. Current model: ${modelKey(model)} (${formatTokens(effective)}/${formatTokens(original)} effective/original).`,
+					"info",
+				);
+				return;
+			}
+
+			ctx.ui.notify("Usage: /context-cap on | off | toggle | status", "warning");
+		},
+	});
+}

package/extensions/context-cap/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "@diegopetrucci/pi-context-cap",
+  "version": "0.1.0",
+  "description": "A pi extension that caps effective model context windows at 200k tokens for earlier auto-compaction.",
+  "keywords": ["pi-package", "pi", "context", "compaction"],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/diegopetrucci/pi-extensions.git",
+    "directory": "extensions/context-cap"
+  },
+  "files": [
+    "index.ts",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "pi": {
+    "extensions": [
+      "index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*"
+  }
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@diegopetrucci/pi-extensions",
-  "version": "0.1.14",
-  "description": "A collection of pi extensions, including a minimal custom footer, an Amp-style oracle, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
+  "version": "0.1.16",
+  "description": "A collection of pi extensions, including a minimal custom footer, an Amp-style oracle, a 200k context cap for auto-compaction, one-line collapsed bash previews, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
   "keywords": ["pi-package", "pi", "terminal", "agent"],
   "license": "MIT",
   "repository": {
@@ -30,6 +30,8 @@
     "extensions": [
       "./extensions/minimal-footer/index.ts",
       "./extensions/oracle/index.ts",
+      "./extensions/context-cap/index.ts",
+      "./extensions/compact-bash/index.ts",
       "./extensions/permission-gate/index.ts",
       "./extensions/confirm-destructive/index.ts",
       "./extensions/notify/index.ts"