@benvargas/pi-openai-fast 1.0.1 → 1.0.4

package/README.md

@@ -4,7 +4,7 @@
 
 This extension does not change the model, thinking level, tools, or prompts. It only adds `service_tier=priority` to provider requests when fast mode is active and the current model matches the configured supported-model list.
 
-Requires pi `0.57.0` or newer.
+Requires pi `0.74.0` or newer.
 
 ## Install
 
@@ -26,6 +26,7 @@ pi -e npm:@benvargas/pi-openai-fast
 - `/fast status` reports the current fast-mode state.
 - `--fast` starts the session with fast mode enabled.
 - By default, fast mode persists across new pi sessions via a JSON config file.
+- Startup state comes from the selected config file, not from resumed session/thread history.
 
 Example:
 
@@ -50,7 +51,9 @@ Default config:
   "active": false,
   "supportedModels": [
     "openai/gpt-5.4",
-    "openai-codex/gpt-5.4"
+    "openai/gpt-5.5",
+    "openai-codex/gpt-5.4",
+    "openai-codex/gpt-5.5"
   ]
 }
 ```
@@ -58,15 +61,15 @@ Default config:
 Settings:
 
 - `persistState`: when `true`, `/fast` writes the current on/off state to config so it resumes in new pi sessions. Default: `true`.
-- `active`: persisted fast-mode state used when `persistState` is enabled.
+- `active`: persisted fast-mode state used on startup when `persistState` is enabled.
 - `supportedModels`: list of `provider/model-id` strings that should receive `service_tier=priority`.
 
-Project config overrides global config. If fast mode is enabled on a model that is not in `supportedModels`, the setting stays on but requests are left unchanged until you switch back to a configured model.
+Project config overrides global config. `/fast on` and `/fast off` write to the selected config file, so if a project config exists the remembered state is project-specific. If fast mode is enabled on a model that is not in `supportedModels`, the setting stays on but requests are left unchanged until you switch back to a configured model.
 
 ## Notes
 
-- Fast mode still stores session state in the current session branch.
 - When `persistState` is enabled, the last `/fast` setting also carries across brand-new pi sessions.
+- Resumed sessions do not override the config-backed startup state.
 - On configured models, fast mode maps to OpenAI `service_tier=priority`.
 
 ## Uninstall

package/extensions/index.ts

@@ -1,15 +1,32 @@
+/**
+ * OpenAI fast mode for pi.
+ *
+ * `/fast` and `--fast` toggle `service_tier=priority` for configured models.
+ * This extension does not change the selected model, thinking level, tools, or prompts.
+ *
+ * Startup state comes from `pi-openai-fast.json`, not resumed session history.
+ * Config precedence is project `.pi/extensions/pi-openai-fast.json` over
+ * global `~/.pi/agent/extensions/pi-openai-fast.json`.
+ *
+ * `supportedModels` controls which `provider/model-id` pairs receive the flag.
+ */
 import { existsSync, 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";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 const FAST_COMMAND = "fast";
 const FAST_FLAG = "fast";
-const FAST_STATE_ENTRY = "pi-openai-fast.state";
 const FAST_CONFIG_BASENAME = "pi-openai-fast.json";
 const FAST_COMMAND_ARGS = ["on", "off", "status"] as const;
 const FAST_SERVICE_TIER = "priority";
-const DEFAULT_SUPPORTED_MODEL_KEYS = ["openai/gpt-5.4", "openai-codex/gpt-5.4"] as const;
+const DEFAULT_SUPPORTED_MODEL_KEYS = [
+	"openai/gpt-5.4",
+	"openai/gpt-5.5",
+	"openai-codex/gpt-5.4",
+	"openai-codex/gpt-5.5",
+] as const;
+const LEGACY_DEFAULT_SUPPORTED_MODEL_KEYS = ["openai/gpt-5.4", "openai-codex/gpt-5.4"] as const;
 
 interface FastModeState {
 	active: boolean;
@@ -121,22 +138,18 @@ function parseSupportedModels(value: unknown): FastSupportedModel[] | undefined
 	return models;
 }
 
-function parseFastModeState(value: unknown): FastModeState | undefined {
-	if (!isRecord(value) || typeof value.active !== "boolean") {
-		return undefined;
+function sameModelKeys(left: readonly string[] | undefined, right: readonly string[]): boolean {
+	if (!left || left.length !== right.length) {
+		return false;
 	}
-	return { active: value.active };
+	return left.every((value, index) => value === right[index]);
 }
 
-function getSavedFastModeState(ctx: ExtensionContext): FastModeState | undefined {
-	const entries = ctx.sessionManager.getBranch();
-	for (let i = entries.length - 1; i >= 0; i--) {
-		const entry = entries[i];
-		if (entry.type === "custom" && entry.customType === FAST_STATE_ENTRY) {
-			return parseFastModeState(entry.data);
-		}
+function migrateSupportedModelKeys(value: string[] | undefined): string[] | undefined {
+	if (sameModelKeys(value, LEGACY_DEFAULT_SUPPORTED_MODEL_KEYS)) {
+		return [...DEFAULT_SUPPORTED_MODEL_KEYS];
 	}
-	return undefined;
+	return value;
 }
 
 function readConfigFile(filePath: string): FastConfigFile | null {
@@ -194,7 +207,8 @@ function resolveFastConfig(cwd: string, homeDir: string = homedir()): ResolvedFa
 	const selectedConfigPath = existsSync(projectConfigPath) ? projectConfigPath : globalConfigPath;
 	const merged = { ...globalConfig, ...projectConfig };
 	const supportedModels =
-		parseSupportedModels(merged.supportedModels) ?? parseSupportedModels(DEFAULT_SUPPORTED_MODEL_KEYS);
+		parseSupportedModels(migrateSupportedModelKeys(merged.supportedModels)) ??
+		parseSupportedModels(DEFAULT_SUPPORTED_MODEL_KEYS);
 
 	return {
 		configPath: selectedConfigPath,
@@ -251,9 +265,19 @@ function applyFastServiceTier(payload: unknown): unknown {
 
 export default function piOpenAIFast(pi: ExtensionAPI): void {
 	let state: FastModeState = { active: false };
+	let cachedConfig: ResolvedFastConfig | undefined;
+
+	function refreshConfig(ctx: ExtensionContext): ResolvedFastConfig {
+		cachedConfig = resolveFastConfig(getConfigCwd(ctx));
+		return cachedConfig;
+	}
+
+	function getConfig(ctx: ExtensionContext): ResolvedFastConfig {
+		return cachedConfig ?? refreshConfig(ctx);
+	}
 
 	function persistState(config: ResolvedFastConfig): void {
-		pi.appendEntry(FAST_STATE_ENTRY, state);
+		cachedConfig = { ...config, active: state.active };
 		if (!config.persistState) {
 			return;
 		}
@@ -262,7 +286,7 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 	}
 
 	async function enableFastMode(ctx: ExtensionContext, options?: { notify?: boolean }): Promise<void> {
-		const config = resolveFastConfig(getConfigCwd(ctx));
+		const config = refreshConfig(ctx);
 		if (state.active) {
 			if (options?.notify !== false) {
 				ctx.ui.notify("Fast mode is already on.", "info");
@@ -279,7 +303,7 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 	}
 
 	async function disableFastMode(ctx: ExtensionContext, options?: { notify?: boolean }): Promise<void> {
-		const config = resolveFastConfig(getConfigCwd(ctx));
+		const config = refreshConfig(ctx);
 		if (!state.active) {
 			if (options?.notify !== false) {
 				ctx.ui.notify("Fast mode is already off.", "info");
@@ -334,10 +358,7 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 					await disableFastMode(ctx);
 					return;
 				case "status":
-					ctx.ui.notify(
-						describeCurrentState(ctx, state.active, resolveFastConfig(getConfigCwd(ctx)).supportedModels),
-						"info",
-					);
+					ctx.ui.notify(describeCurrentState(ctx, state.active, refreshConfig(ctx).supportedModels), "info");
 					return;
 				default:
 					ctx.ui.notify("Usage: /fast [on|off|status]", "error");
@@ -346,7 +367,7 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 	});
 
 	pi.on("before_provider_request", (event, ctx) => {
-		const config = resolveFastConfig(getConfigCwd(ctx));
+		const config = getConfig(ctx);
 		if (!state.active || !isFastSupportedModel(ctx.model, config.supportedModels)) {
 			return;
 		}
@@ -354,11 +375,8 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 	});
 
 	pi.on("session_start", async (_event, ctx) => {
-		const config = resolveFastConfig(getConfigCwd(ctx));
-		const savedState = getSavedFastModeState(ctx);
-		const persistedState =
-			config.persistState && typeof config.active === "boolean" ? { active: config.active } : undefined;
-		state = savedState ?? persistedState ?? { active: false };
+		const config = refreshConfig(ctx);
+		state = config.persistState && typeof config.active === "boolean" ? { active: config.active } : { active: false };
 
 		if (pi.getFlag(FAST_FLAG) === true) {
 			if (!state.active) {
@@ -369,7 +387,7 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 			return;
 		}
 
-		if (!savedState && state.active) {
+		if (state.active) {
 			ctx.ui.notify(describeCurrentState(ctx, state.active, config.supportedModels), "info");
 		}
 	});
@@ -378,16 +396,16 @@ export default function piOpenAIFast(pi: ExtensionAPI): void {
 export const _test = {
 	FAST_COMMAND,
 	FAST_FLAG,
-	FAST_STATE_ENTRY,
 	FAST_CONFIG_BASENAME,
 	FAST_COMMAND_ARGS,
 	FAST_SERVICE_TIER,
 	DEFAULT_SUPPORTED_MODEL_KEYS,
+	LEGACY_DEFAULT_SUPPORTED_MODEL_KEYS,
 	DEFAULT_CONFIG_FILE,
 	getConfigPaths,
-	parseFastModeState,
 	parseSupportedModelKey,
 	parseSupportedModels,
+	migrateSupportedModelKeys,
 	readConfigFile,
 	resolveFastConfig,
 	isFastSupportedModel,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@benvargas/pi-openai-fast",
-  "version": "1.0.1",
-  "description": "OpenAI fast mode toggle for pi - Enables priority service tier on supported GPT-5.4 models",
+  "version": "1.0.4",
+  "description": "OpenAI fast mode toggle for pi - Enables priority service tier on supported GPT-5 models",
   "keywords": [
     "pi",
     "pi-package",
@@ -10,6 +10,7 @@
     "openai",
     "codex",
     "gpt-5.4",
+    "gpt-5.5",
     "fast",
     "priority",
     "service-tier"
@@ -26,7 +27,7 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": ">=0.57.0"
+    "@earendil-works/pi-coding-agent": ">=0.74.0"
   },
   "repository": {
     "type": "git",