@diegopetrucci/pi-claude-fast 0.1.1 → 0.1.3

package/README.md

@@ -51,7 +51,7 @@ Optional project config:
 .pi/claude-fast.json
 ```
 
-Project config overrides global config.
+Project config overrides global config after Pi reports that the project is trusted.
 
 ```json
 {

package/index.ts

@@ -6,6 +6,8 @@ import {
 	type ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
 
+const CONFIG_DIR_NAME = ".pi";
+
 const EXTENSION_ID = "claude-fast";
 const PROVIDER_ID = "anthropic";
 const API_ID = "anthropic-messages";
@@ -35,6 +37,11 @@ type SessionState = {
 	lastInjectedModel?: string;
 };
 
+type ProjectConfigContext = {
+	cwd: string;
+	isProjectTrusted?: () => boolean;
+};
+
 type RecursivePartial<T> = {
 	[P in keyof T]?: T[P] extends object ? RecursivePartial<T[P]> : T[P];
 };
@@ -77,21 +84,25 @@ function mergeConfig(
 	};
 }
 
+function canReadProjectConfig(ctx: ProjectConfigContext): boolean {
+	return typeof ctx.isProjectTrusted === "function" && ctx.isProjectTrusted();
+}
+
 function findProjectConfigPath(cwd: string): string {
 	let current = cwd;
 	while (true) {
-		const candidate = join(current, ".pi", "claude-fast.json");
+		const candidate = join(current, CONFIG_DIR_NAME, "claude-fast.json");
 		if (existsSync(candidate)) return candidate;
 
 		const parent = dirname(current);
-		if (parent === current) return join(cwd, ".pi", "claude-fast.json");
+		if (parent === current) return join(cwd, CONFIG_DIR_NAME, "claude-fast.json");
 		current = parent;
 	}
 }
 
-function loadConfig(cwd: string): ClaudeFastConfig {
+function loadConfig(ctx: ProjectConfigContext): ClaudeFastConfig {
 	const globalConfig = readConfigFile(join(getAgentDir(), "extensions", "claude-fast.json"));
-	const projectConfig = readConfigFile(findProjectConfigPath(cwd));
+	const projectConfig = canReadProjectConfig(ctx) ? readConfigFile(findProjectConfigPath(ctx.cwd)) : {};
 	return mergeConfig(mergeConfig(DEFAULT_CONFIG, globalConfig), projectConfig);
 }
 
@@ -167,7 +178,7 @@ function syncModelBetaHeader(ctx: ExtensionContext, state: SessionState): void {
 	const requiredBase = ctx.modelRegistry.isUsingOAuth(model) ? CLAUDE_CODE_OAUTH_BETAS : [];
 	const next = shouldEnable
 		? Array.from(new Set([...existing, ...requiredBase, FAST_BETA]))
-		: existing.filter((beta) => beta !== FAST_BETA && !CLAUDE_CODE_OAUTH_BETAS.includes(beta));
+		: existing.filter((beta) => beta !== FAST_BETA);
 
 	delete headers["Anthropic-Beta"];
 	if (next.length > 0) headers["anthropic-beta"] = next.join(",");
@@ -235,7 +246,7 @@ export default function claudeFastExtension(pi: ExtensionAPI) {
 		let state = states.get(ctx.sessionManager);
 		if (!state) {
 			state = {
-				config: loadConfig(ctx.cwd),
+				config: loadConfig(ctx),
 				override: "auto",
 			};
 			states.set(ctx.sessionManager, state);
@@ -245,7 +256,7 @@ export default function claudeFastExtension(pi: ExtensionAPI) {
 
 	pi.on("session_start", (_event, ctx) => {
 		const state: SessionState = {
-			config: loadConfig(ctx.cwd),
+			config: loadConfig(ctx),
 			override: "auto",
 		};
 		states.set(ctx.sessionManager, state);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-claude-fast",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "A pi extension that enables Anthropic Claude Fast mode for supported Claude Opus models by injecting speed=fast.",
   "keywords": [
     "pi-package",