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

package/README.md

@@ -0,0 +1,23 @@
+# @ryan_nookpi/pi-extension-codex-large-context
+
+This extension raises the pi context window metadata for newer OpenAI Codex models.
+
+It is intended for `gpt-5.4` and `gpt-5.5` model IDs when pi reports a smaller context window than the model can actually handle.
+
+## Install
+
+```bash
+pi install npm:@ryan_nookpi/pi-extension-codex-large-context
+```
+
+## What it does
+
+- Watches `session_start` and `model_select` events.
+- If the active model ID starts with `gpt-5.4` or `gpt-5.5`, sets its context window to `922000` tokens.
+- Shows a small notification when the context window is updated in the interactive UI.
+
+## Notes
+
+- 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.

package/index.ts

@@ -0,0 +1,39 @@
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+
+export const TARGET_CONTEXT_WINDOW = 922_000;
+export const TARGET_MODEL_PREFIXES = ["gpt-5.4", "gpt-5.5"] as const;
+
+export function isTargetModelId(modelId: string | undefined): boolean {
+	if (!modelId) return false;
+	return TARGET_MODEL_PREFIXES.some((prefix) => modelId.startsWith(prefix));
+}
+
+export async function applyLargeContext(pi: ExtensionAPI, ctx: ExtensionContext): Promise<boolean> {
+	const model = ctx.model;
+	if (!model) return false;
+	if (!isTargetModelId(model.id)) return false;
+	if (model.contextWindow >= TARGET_CONTEXT_WINDOW) return false;
+
+	const previousContextWindow = model.contextWindow;
+	const updated = { ...model, contextWindow: TARGET_CONTEXT_WINDOW };
+	const ok = await pi.setModel(updated);
+
+	if (ok && ctx.hasUI) {
+		ctx.ui.notify(
+			`${model.id} context window: ${(previousContextWindow / 1000).toFixed(0)}K → ${(TARGET_CONTEXT_WINDOW / 1000).toFixed(0)}K`,
+			"info",
+		);
+	}
+
+	return ok;
+}
+
+export default function codexLargeContext(pi: ExtensionAPI) {
+	pi.on("model_select", async (_event, ctx) => {
+		await applyLargeContext(pi, ctx);
+	});
+
+	pi.on("session_start", async (_event, ctx) => {
+		await applyLargeContext(pi, ctx);
+	});
+}

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "@ryan_nookpi/pi-extension-codex-large-context",
+  "version": "0.1.0",
+  "description": "Codex large context window extension for pi.",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/Jonghakseo/pi-extension.git",
+    "directory": "packages/codex-large-context"
+  },
+  "bugs": {
+    "url": "https://github.com/Jonghakseo/pi-extension/issues"
+  },
+  "homepage": "https://github.com/Jonghakseo/pi-extension/tree/main/packages/codex-large-context#readme",
+  "type": "module",
+  "keywords": [
+    "pi-package"
+  ],
+  "files": [
+    "index.ts",
+    "README.md"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": "*"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}