@diegopetrucci/pi-extensions 0.1.13 → 0.1.15

package/README.md

@@ -1,9 +1,10 @@
 # pi-extensions
 
-A collection of [pi](https://github.com/badlogic/pi-mono) agent extensions I made:
+A collection of [pi](https://github.com/earendil-works/pi-mono) agent extensions I made:
 
 - [`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 auto-compacts earlier; toggle temporarily with `/context-cap`.
 - [`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,13 +22,15 @@ 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.13
+pi install git:github.com/diegopetrucci/pi-extensions@v0.1.15
 ```
 
 Or a specific extension:
 
 ```bash
 pi install npm:@diegopetrucci/pi-oracle
+# or
+pi install npm:@diegopetrucci/pi-context-cap
 ```
 
 Then reload pi:

package/extensions/confirm-destructive/README.md

@@ -2,7 +2,7 @@
 
 A small pi extension that asks for confirmation before destructive session actions.
 
-This is adapted from the original `confirm-destructive.ts` example in [`badlogic/pi-mono`](https://github.com/badlogic/pi-mono/blob/main/packages/coding-agent/examples/extensions/confirm-destructive.ts) and kept basically the same.
+This is adapted from the original `confirm-destructive.ts` example in [`earendil-works/pi-mono`](https://github.com/earendil-works/pi-mono/blob/main/packages/coding-agent/examples/extensions/confirm-destructive.ts) and kept basically the same.
 
 ## What it checks
 

package/extensions/confirm-destructive/index.ts

@@ -5,7 +5,7 @@
  * Demonstrates how to cancel session events using the before_* events.
  */
 
-import type { ExtensionAPI, SessionBeforeSwitchEvent, SessionMessageEntry } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, SessionBeforeSwitchEvent, SessionMessageEntry } from "@earendil-works/pi-coding-agent";
 
 export default function (pi: ExtensionAPI) {
 	pi.on("session_before_switch", async (event: SessionBeforeSwitchEvent, ctx) => {

package/extensions/confirm-destructive/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-confirm-destructive",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "A pi extension that confirms destructive session actions.",
   "keywords": ["pi-package", "pi", "session", "safety"],
   "license": "MIT",
@@ -22,6 +22,6 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*"
+    "@earendil-works/pi-coding-agent": "*"
   }
 }

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.
+
+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/extensions/minimal-footer/index.ts

@@ -5,8 +5,8 @@ import {
 	getAgentDir,
 	type ExtensionAPI,
 	type ExtensionContext,
-} from "@mariozechner/pi-coding-agent";
-import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+} from "@earendil-works/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
 import {
 	fetchOpenAICodexUsage,
 	formatUsageSummary,

package/extensions/minimal-footer/openai-usage.ts

@@ -1,4 +1,4 @@
-import { AuthStorage } from "@mariozechner/pi-coding-agent";
+import { AuthStorage } from "@earendil-works/pi-coding-agent";
 
 const PROVIDER_ID = "openai-codex";
 

package/extensions/minimal-footer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-minimal-footer",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "A minimal custom footer for pi.",
   "keywords": ["pi-package", "pi", "terminal", "footer"],
   "license": "MIT",
@@ -25,7 +25,7 @@
     "image": "https://raw.githubusercontent.com/diegopetrucci/pi-extensions/main/assets/minimal-footer-preview.png"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*"
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
   }
 }

package/extensions/notify/README.md

@@ -2,7 +2,7 @@
 
 A pi extension that sends notifications when the agent finishes and is waiting for input.
 
-This started from the original `notify.ts` example in [`badlogic/pi-mono`](https://github.com/badlogic/pi-mono/blob/main/packages/coding-agent/examples/extensions/notify.ts), but now supports multiple notification channels and JSON configuration.
+This started from the original `notify.ts` example in [`earendil-works/pi-mono`](https://github.com/earendil-works/pi-mono/blob/main/packages/coding-agent/examples/extensions/notify.ts), but now supports multiple notification channels and JSON configuration.
 
 ## Supported notification channels
 

package/extensions/notify/index.ts

@@ -16,8 +16,8 @@
 import { execFile } from "node:child_process";
 import { existsSync, readFileSync } from "node:fs";
 import { join } from "node:path";
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { getAgentDir } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
 
 type TerminalBackend = "auto" | "osc777" | "osc99" | "none";
 type DesktopBackend = "auto" | "macos" | "linux" | "windows-toast" | "none";

package/extensions/notify/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-notify",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "A pi extension that sends a notification when the agent is ready for input.",
   "keywords": ["pi-package", "pi", "notification", "terminal"],
   "license": "MIT",
@@ -23,6 +23,6 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*"
+    "@earendil-works/pi-coding-agent": "*"
   }
 }

package/extensions/oracle/README.md

@@ -2,7 +2,7 @@
 
 ![oracle preview](https://raw.githubusercontent.com/diegopetrucci/pi-extensions/main/assets/oracle-preview.svg)
 
-An Amp-style oracle for [pi](https://github.com/badlogic/pi-mono).
+An Amp-style oracle for [pi](https://github.com/earendil-works/pi-mono).
 
 It adds an `oracle` tool that spins up a separate read-only pi subprocess and sends it to the strongest reasoning model available on the **same provider/subscription** the user is currently using.
 

package/extensions/oracle/index.ts

@@ -1,10 +1,10 @@
 import { spawn } from "node:child_process";
 import { existsSync } from "node:fs";
 import { basename } from "node:path";
-import { StringEnum } from "@mariozechner/pi-ai";
-import { getMarkdownTheme, type ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { Container, Markdown, Spacer, Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
+import { StringEnum } from "@earendil-works/pi-ai";
+import { getMarkdownTheme, type ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { Container, Markdown, Spacer, Text } from "@earendil-works/pi-tui";
+import { Type } from "typebox";
 
 type ThinkingLevel = "off" | "minimal" | "low" | "medium" | "high" | "xhigh";
 
@@ -171,14 +171,6 @@ const PROVIDER_MODEL_PREFERENCES: Record<string, string[]> = {
 		"gemini-2.5-flash-lite-preview",
 		"gemini-2.5-flash-lite",
 	],
-	"google-antigravity": [
-		"claude-opus-4-6-thinking",
-		"claude-sonnet-4-5-thinking",
-		"gemini-3.1-pro-low",
-		"gemini-3-flash",
-		"gemini-2.0-flash",
-	],
-	"google-gemini-cli": ["gemini-3-pro-preview", "gemini-2.5-pro", "gemini-1.5-flash"],
 	"google-vertex": [
 		"gemini-3.1-pro-preview-customtools",
 		"gemini-3.1-pro-preview",
@@ -335,6 +327,28 @@ const PROVIDER_MODEL_PREFERENCES: Record<string, string[]> = {
 		"grok-3-mini-fast",
 		"grok-3-latest",
 	],
+	xiaomi: ["mimo-v2.5-pro", "mimo-v2.5", "mimo-v2-pro", "mimo-v2-omni", "mimo-v2-flash"],
+	"xiaomi-token-plan-ams": [
+		"mimo-v2.5-pro",
+		"mimo-v2.5",
+		"mimo-v2-pro",
+		"mimo-v2-omni",
+		"mimo-v2-flash",
+	],
+	"xiaomi-token-plan-cn": [
+		"mimo-v2.5-pro",
+		"mimo-v2.5",
+		"mimo-v2-pro",
+		"mimo-v2-omni",
+		"mimo-v2-flash",
+	],
+	"xiaomi-token-plan-sgp": [
+		"mimo-v2.5-pro",
+		"mimo-v2.5",
+		"mimo-v2-pro",
+		"mimo-v2-omni",
+		"mimo-v2-flash",
+	],
 	zai: [
 		"glm-5.1",
 		"glm-5-turbo",
@@ -348,7 +362,6 @@ const PROVIDER_MODEL_PREFERENCES: Record<string, string[]> = {
 	],
 	moonshotai: ["kimi-k2.6", "kimi-k2-thinking-turbo", "kimi-k2-thinking", "kimi-k2.5"],
 	"moonshotai-cn": ["kimi-k2.6", "kimi-k2-thinking-turbo", "kimi-k2-thinking", "kimi-k2.5"],
-	"gemini-cli": ["gemini-3-pro-preview", "gemini-2.5-pro", "gemini-1.5-flash"],
 };
 
 const ORACLE_SYSTEM_PROMPT = [

package/extensions/oracle/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-oracle",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "An Amp-style oracle extension for pi that consults the strongest reasoning model on your current provider.",
   "keywords": ["pi-package", "pi", "oracle", "reasoning", "subagent"],
   "license": "MIT",
@@ -23,7 +23,9 @@
     "image": "https://raw.githubusercontent.com/diegopetrucci/pi-extensions/main/assets/oracle-preview.svg"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*"
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*",
+    "typebox": "*"
   }
 }

package/extensions/permission-gate/README.md

@@ -2,7 +2,7 @@
 
 A small pi extension that prompts for confirmation before running potentially dangerous bash commands.
 
-This is adapted from the original `permission-gate.ts` example in [`badlogic/pi-mono`](https://github.com/badlogic/pi-mono/blob/main/packages/coding-agent/examples/extensions/permission-gate.ts) and kept basically the same.
+This is adapted from the original `permission-gate.ts` example in [`earendil-works/pi-mono`](https://github.com/earendil-works/pi-mono/blob/main/packages/coding-agent/examples/extensions/permission-gate.ts) and kept basically the same.
 
 ## What it checks
 

package/extensions/permission-gate/index.ts

@@ -5,7 +5,7 @@
  * Patterns checked: rm -rf, sudo, chmod/chown 777
  */
 
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 
 export default function (pi: ExtensionAPI) {
 	const dangerousPatterns = [/\brm\s+(-rf?|--recursive)/i, /\bsudo\b/i, /\b(chmod|chown)\b.*777/i];

package/extensions/permission-gate/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-permission-gate",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "A pi extension that prompts before dangerous bash commands.",
   "keywords": ["pi-package", "pi", "security", "bash"],
   "license": "MIT",
@@ -22,6 +22,6 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*"
+    "@earendil-works/pi-coding-agent": "*"
   }
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@diegopetrucci/pi-extensions",
-  "version": "0.1.13",
-  "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.15",
+  "description": "A collection of pi extensions, including a minimal custom footer, an Amp-style oracle, a 200k context cap for auto-compaction, 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": {
@@ -21,13 +21,16 @@
     "access": "public"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*"
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*",
+    "typebox": "*"
   },
   "pi": {
     "extensions": [
       "./extensions/minimal-footer/index.ts",
       "./extensions/oracle/index.ts",
+      "./extensions/context-cap/index.ts",
       "./extensions/permission-gate/index.ts",
       "./extensions/confirm-destructive/index.ts",
       "./extensions/notify/index.ts"