@ryan_nookpi/pi-extension-codex-large-context 0.1.0 → 0.1.1

package/README.md

@@ -10,10 +10,23 @@ It is intended for `gpt-5.4` and `gpt-5.5` model IDs when pi reports a smaller c
 pi install npm:@ryan_nookpi/pi-extension-codex-large-context
 ```
 
+## Usage
+
+Large context mode is ON by default.
+
+```text
+/codex-large-context on
+/codex-large-context off
+/codex-large-context status
+```
+
+Running `/codex-large-context` with no argument shows the current status.
+
 ## What it does
 
-- Watches `session_start` and `model_select` events.
+- Watches `session_start` and `model_select` events while enabled.
 - If the active model ID starts with `gpt-5.4` or `gpt-5.5`, sets its context window to `922000` tokens.
+- Stores the on/off setting locally so it persists across sessions.
 - Shows a small notification when the context window is updated in the interactive UI.
 
 ## Notes
@@ -21,3 +34,4 @@ pi install npm:@ryan_nookpi/pi-extension-codex-large-context
 - This only changes pi's local model metadata for the active session/model selection.
 - It does not change API limits on the provider side.
 - If pi already reports a context window of `922000` tokens or more, it leaves the model unchanged.
+- Turning the feature off prevents future context-window adjustments; it does not restore a model that was already adjusted earlier in the session.

package/index.ts

@@ -1,7 +1,45 @@
+import { mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { dirname, join } from "node:path";
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 
+const STATE_FILE = join(homedir(), ".pi", "agent", "state", "codex-large-context.json");
 export const TARGET_CONTEXT_WINDOW = 922_000;
 export const TARGET_MODEL_PREFIXES = ["gpt-5.4", "gpt-5.5"] as const;
+const TARGET_MODEL_LABEL = TARGET_MODEL_PREFIXES.join(" or ");
+
+type LargeContextState = {
+	enabled: boolean;
+};
+
+export function loadCodexLargeContextState(): LargeContextState {
+	try {
+		const raw = readFileSync(STATE_FILE, "utf8");
+		const parsed = JSON.parse(raw);
+		if (typeof parsed?.enabled === "boolean") {
+			return { enabled: parsed.enabled };
+		}
+	} catch {
+		// Ignore missing/corrupt state and fall back to the default-enabled behavior.
+	}
+
+	return { enabled: true };
+}
+
+export function isCodexLargeContextEnabled(): boolean {
+	return loadCodexLargeContextState().enabled;
+}
+
+function saveState(state: LargeContextState): void {
+	mkdirSync(dirname(STATE_FILE), { recursive: true });
+	writeFileSync(STATE_FILE, `${JSON.stringify(state, null, 2)}\n`, "utf8");
+}
+
+function parseCommandArg(args: string): "on" | "off" | "status" | "help" {
+	const arg = args.trim().toLowerCase();
+	if (arg === "on" || arg === "off" || arg === "status") return arg;
+	return arg ? "help" : "status";
+}
 
 export function isTargetModelId(modelId: string | undefined): boolean {
 	if (!modelId) return false;
@@ -9,6 +47,8 @@ export function isTargetModelId(modelId: string | undefined): boolean {
 }
 
 export async function applyLargeContext(pi: ExtensionAPI, ctx: ExtensionContext): Promise<boolean> {
+	if (!isCodexLargeContextEnabled()) return false;
+
 	const model = ctx.model;
 	if (!model) return false;
 	if (!isTargetModelId(model.id)) return false;
@@ -36,4 +76,49 @@ export default function codexLargeContext(pi: ExtensionAPI) {
 	pi.on("session_start", async (_event, ctx) => {
 		await applyLargeContext(pi, ctx);
 	});
+
+	pi.registerCommand("codex-large-context", {
+		description: "Toggle Codex Large Context for openai-codex/gpt-5.4 or gpt-5.5",
+		getArgumentCompletions: (prefix) => {
+			const options = ["on", "off", "status"];
+			const filtered = options.filter((option) => option.startsWith(prefix.trim().toLowerCase()));
+			return filtered.length > 0 ? filtered.map((option) => ({ value: option, label: option })) : null;
+		},
+		handler: async (args, ctx) => {
+			const action = parseCommandArg(args);
+
+			if (action === "help") {
+				if (ctx.hasUI) {
+					ctx.ui.notify("Usage: /codex-large-context [on|off|status]", "info");
+				}
+				return;
+			}
+
+			if (action === "status") {
+				if (ctx.hasUI) {
+					ctx.ui.notify(
+						`Codex Large Context: ${isCodexLargeContextEnabled() ? "ON" : "OFF"} (${TARGET_MODEL_LABEL} → ${(TARGET_CONTEXT_WINDOW / 1000).toFixed(0)}K)`,
+						"info",
+					);
+				}
+				return;
+			}
+
+			const nextState = { enabled: action === "on" };
+			saveState(nextState);
+
+			if (ctx.hasUI) {
+				ctx.ui.notify(
+					nextState.enabled
+						? `Codex Large Context enabled (${TARGET_MODEL_LABEL} → ${(TARGET_CONTEXT_WINDOW / 1000).toFixed(0)}K)`
+						: "Codex Large Context disabled",
+					"info",
+				);
+			}
+
+			if (nextState.enabled) {
+				await applyLargeContext(pi, ctx);
+			}
+		},
+	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ryan_nookpi/pi-extension-codex-large-context",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Codex large context window extension for pi.",
   "license": "MIT",
   "repository": {