@diegopetrucci/pi-openai-fast 0.1.0

package/README.md

@@ -0,0 +1,95 @@
+# openai-fast
+
+A pi extension that enables OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4 and GPT-5.5.
+
+When active, the extension injects this into eligible OpenAI Codex request payloads:
+
+```json
+{
+  "service_tier": "priority"
+}
+```
+
+The user-facing feature is OpenAI Codex **Fast mode**. The wire value is `priority` because current Codex clients map Fast mode to the OpenAI priority service tier.
+
+## Eligibility
+
+Fast mode is only injected when all of these are true:
+
+- The current provider is `openai-codex`.
+- The current API is `openai-codex-responses`.
+- The current model is `gpt-5.4` or `gpt-5.5`.
+- The provider is using ChatGPT OAuth/subscription auth, not API-key auth.
+- The request payload does not already include `service_tier`.
+
+## Commands
+
+```text
+/fast status
+/fast on
+/fast off
+/fast auto
+/fast toggle
+```
+
+`/fast on` and `/fast off` are temporary session/runtime overrides. Use `/fast auto` to reload and follow config defaults again.
+
+The extension defaults to off so installing the full collection does not accidentally spend Fast-mode credits.
+
+## Config
+
+Optional global config:
+
+```text
+~/.pi/agent/extensions/openai-fast.json
+```
+
+Optional project config:
+
+```text
+.pi/openai-fast.json
+```
+
+Project config overrides global config.
+
+```json
+{
+  "enabled": false,
+  "showStatus": true
+}
+```
+
+- `enabled`: default Fast-mode state when there is no session override.
+- `showStatus`: show a compact `fast` status when Fast mode is active for the current model.
+
+## Install
+
+### Standalone npm package
+
+```bash
+pi install npm:@diegopetrucci/pi-openai-fast
+```
+
+### 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 intentionally does not affect API-key OpenAI models.
+- Pi may only account Fast-mode cost correctly when the backend reports `service_tier: "priority"` in the streamed response. The extension does not patch usage totals to avoid double-counting.
+- If pi adds first-class service-tier support later, this extension skips payloads that already contain `service_tier`.

package/index.ts

@@ -0,0 +1,292 @@
+import { existsSync, readFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import {
+	getAgentDir,
+	type ExtensionAPI,
+	type ExtensionContext,
+} from "@earendil-works/pi-coding-agent";
+
+const EXTENSION_ID = "openai-fast";
+const PROVIDER_ID = "openai-codex";
+const API_ID = "openai-codex-responses";
+const FAST_SERVICE_TIER = "priority";
+const SUPPORTED_MODELS = new Set(["gpt-5.4", "gpt-5.5"]);
+const COMMANDS = ["status", "on", "off", "auto", "toggle"];
+
+const DEFAULT_CONFIG: OpenAIFastConfig = {
+	enabled: false,
+	showStatus: true,
+};
+
+type FastOverride = "auto" | "on" | "off";
+
+type OpenAIFastConfig = {
+	/** Default Fast-mode state when there is no session override. */
+	enabled: boolean;
+	/** Show a compact `fast` status when Fast mode is active for the current model. */
+	showStatus: boolean;
+};
+
+type SessionState = {
+	config: OpenAIFastConfig;
+	override: FastOverride;
+	lastInjectedAt?: number;
+	lastInjectedModel?: string;
+};
+
+type RecursivePartial<T> = {
+	[P in keyof T]?: T[P] extends object ? RecursivePartial<T[P]> : T[P];
+};
+
+type PayloadRecord = Record<string, unknown>;
+
+type Eligibility = {
+	eligible: boolean;
+	modelKey: string;
+	reason?: string;
+};
+
+function readConfigFile(path: string): RecursivePartial<OpenAIFastConfig> {
+	if (!existsSync(path)) return {};
+
+	try {
+		const parsed = JSON.parse(readFileSync(path, "utf-8"));
+		return isPayloadRecord(parsed) ? (parsed as RecursivePartial<OpenAIFastConfig>) : {};
+	} catch (error) {
+		console.error(`Warning: Could not parse ${path}: ${error}`);
+		return {};
+	}
+}
+
+function normalizeBoolean(value: unknown, fallback: boolean): boolean {
+	return typeof value === "boolean" ? value : fallback;
+}
+
+function mergeConfig(
+	base: OpenAIFastConfig,
+	overrides: RecursivePartial<OpenAIFastConfig>,
+): OpenAIFastConfig {
+	return {
+		enabled: normalizeBoolean(overrides.enabled, base.enabled),
+		showStatus: normalizeBoolean(overrides.showStatus, base.showStatus),
+	};
+}
+
+function findProjectConfigPath(cwd: string): string {
+	let current = cwd;
+	while (true) {
+		const candidate = join(current, ".pi", "openai-fast.json");
+		if (existsSync(candidate)) return candidate;
+
+		const parent = dirname(current);
+		if (parent === current) return join(cwd, ".pi", "openai-fast.json");
+		current = parent;
+	}
+}
+
+function loadConfig(cwd: string): OpenAIFastConfig {
+	const globalConfig = readConfigFile(join(getAgentDir(), "extensions", "openai-fast.json"));
+	const projectConfig = readConfigFile(findProjectConfigPath(cwd));
+	return mergeConfig(mergeConfig(DEFAULT_CONFIG, globalConfig), projectConfig);
+}
+
+function isPayloadRecord(payload: unknown): payload is PayloadRecord {
+	return typeof payload === "object" && payload !== null && !Array.isArray(payload);
+}
+
+function modelKey(ctx: ExtensionContext): string {
+	const model = ctx.model;
+	return model ? `${model.provider}/${model.id}` : "no-model";
+}
+
+function isFastEnabled(state: SessionState): boolean {
+	if (state.override === "on") return true;
+	if (state.override === "off") return false;
+	return state.config.enabled;
+}
+
+function describeMode(state: SessionState): string {
+	if (state.override === "on") return "on (session override)";
+	if (state.override === "off") return "off (session override)";
+	return state.config.enabled ? "on (config default)" : "off (config default)";
+}
+
+function getEligibility(ctx: ExtensionContext): Eligibility {
+	const model = ctx.model;
+	if (!model) {
+		return { eligible: false, modelKey: "no-model", reason: "no model is selected" };
+	}
+
+	const key = `${model.provider}/${model.id}`;
+	if (model.provider !== PROVIDER_ID) {
+		return {
+			eligible: false,
+			modelKey: key,
+			reason: `current provider is ${model.provider}, not ${PROVIDER_ID}`,
+		};
+	}
+
+	if (model.api !== API_ID) {
+		return {
+			eligible: false,
+			modelKey: key,
+			reason: `current API is ${model.api}, not ${API_ID}`,
+		};
+	}
+
+	if (!SUPPORTED_MODELS.has(model.id)) {
+		return {
+			eligible: false,
+			modelKey: key,
+			reason: "Fast mode is only enabled for gpt-5.4 and gpt-5.5",
+		};
+	}
+
+	if (!ctx.modelRegistry.isUsingOAuth(model)) {
+		return {
+			eligible: false,
+			modelKey: key,
+			reason: "ChatGPT OAuth auth is required; API-key auth is intentionally not used",
+		};
+	}
+
+	return { eligible: true, modelKey: key };
+}
+
+function updateStatus(ctx: ExtensionContext, state: SessionState): void {
+	if (!ctx.hasUI) return;
+	if (!state.config.showStatus) {
+		ctx.ui.setStatus(EXTENSION_ID, undefined);
+		return;
+	}
+
+	const eligibility = getEligibility(ctx);
+	ctx.ui.setStatus(
+		EXTENSION_ID,
+		isFastEnabled(state) && eligibility.eligible ? "fast" : undefined,
+	);
+}
+
+function getStatusMessage(ctx: ExtensionContext, state: SessionState): string {
+	const enabled = isFastEnabled(state);
+	const eligibility = getEligibility(ctx);
+	const active = enabled && eligibility.eligible;
+	const injected = state.lastInjectedAt
+		? ` Last injected for ${state.lastInjectedModel ?? "unknown model"} ${Math.max(0, Math.round((Date.now() - state.lastInjectedAt) / 1000))}s ago.`
+		: "";
+
+	if (active) {
+		return `OpenAI Fast mode is ${describeMode(state)} and active for ${eligibility.modelKey}; requests will use service_tier=${FAST_SERVICE_TIER}.${injected}`;
+	}
+
+	if (enabled) {
+		return `OpenAI Fast mode is ${describeMode(state)}, but inactive for ${eligibility.modelKey}: ${eligibility.reason}.${injected}`;
+	}
+
+	return `OpenAI Fast mode is ${describeMode(state)}. Current model: ${eligibility.modelKey}.${injected}`;
+}
+
+function injectFastServiceTier(
+	payload: unknown,
+	ctx: ExtensionContext,
+	state: SessionState,
+): PayloadRecord | undefined {
+	if (!isFastEnabled(state)) return undefined;
+	if (!getEligibility(ctx).eligible) return undefined;
+	if (!isPayloadRecord(payload)) return undefined;
+	if (payload.model !== ctx.model?.id) return undefined;
+	if ("service_tier" in payload) return undefined;
+
+	state.lastInjectedAt = Date.now();
+	state.lastInjectedModel = modelKey(ctx);
+	return {
+		...payload,
+		service_tier: FAST_SERVICE_TIER,
+	};
+}
+
+export default function openAIFastExtension(pi: ExtensionAPI) {
+	const states = new WeakMap<object, SessionState>();
+
+	function getState(ctx: ExtensionContext): SessionState {
+		let state = states.get(ctx.sessionManager);
+		if (!state) {
+			state = {
+				config: loadConfig(ctx.cwd),
+				override: "auto",
+			};
+			states.set(ctx.sessionManager, state);
+		}
+		return state;
+	}
+
+	pi.on("session_start", (_event, ctx) => {
+		const state: SessionState = {
+			config: loadConfig(ctx.cwd),
+			override: "auto",
+		};
+		states.set(ctx.sessionManager, state);
+		updateStatus(ctx, state);
+	});
+
+	pi.on("model_select", (_event, ctx) => {
+		updateStatus(ctx, getState(ctx));
+	});
+
+	pi.on("before_provider_request", (event, ctx) => {
+		const state = getState(ctx);
+		const nextPayload = injectFastServiceTier(event.payload, ctx, state);
+		updateStatus(ctx, state);
+		return nextPayload;
+	});
+
+	pi.registerCommand("fast", {
+		description: "Manage OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4/GPT-5.5",
+		getArgumentCompletions: (prefix) => {
+			const normalized = prefix.trim().toLowerCase();
+			const matches = COMMANDS.filter((command) => command.startsWith(normalized));
+			return matches.length > 0 ? matches.map((value) => ({ value, label: value })) : null;
+		},
+		handler: async (args, ctx) => {
+			const state = getState(ctx);
+			const action = args.trim().toLowerCase() || "status";
+
+			if (action === "on" || action === "enable") {
+				state.override = "on";
+				updateStatus(ctx, state);
+				ctx.ui.notify(getStatusMessage(ctx, state), "info");
+				return;
+			}
+
+			if (action === "off" || action === "disable") {
+				state.override = "off";
+				updateStatus(ctx, state);
+				ctx.ui.notify(getStatusMessage(ctx, state), "info");
+				return;
+			}
+
+			if (action === "auto" || action === "default") {
+				state.override = "auto";
+				state.config = loadConfig(ctx.cwd);
+				updateStatus(ctx, state);
+				ctx.ui.notify(getStatusMessage(ctx, state), "info");
+				return;
+			}
+
+			if (action === "toggle") {
+				state.override = isFastEnabled(state) ? "off" : "on";
+				updateStatus(ctx, state);
+				ctx.ui.notify(getStatusMessage(ctx, state), "info");
+				return;
+			}
+
+			if (action === "status") {
+				updateStatus(ctx, state);
+				ctx.ui.notify(getStatusMessage(ctx, state), "info");
+				return;
+			}
+
+			ctx.ui.notify("Usage: /fast on | off | auto | toggle | status", "warning");
+		},
+	});
+}

package/openai-fast.example.json

@@ -0,0 +1,4 @@
+{
+  "enabled": false,
+  "showStatus": true
+}

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "@diegopetrucci/pi-openai-fast",
+  "version": "0.1.0",
+  "description": "A pi extension that enables OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4 and GPT-5.5 by injecting the priority service tier.",
+  "keywords": ["pi-package", "pi", "openai", "codex", "fast"],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/diegopetrucci/pi-extensions.git",
+    "directory": "extensions/openai-fast"
+  },
+  "files": [
+    "index.ts",
+    "openai-fast.example.json",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "pi": {
+    "extensions": [
+      "index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  }
+}