@diegopetrucci/pi-extensions 0.1.80 → 0.1.81

package/extensions/dirty-repo-guard/index.ts

@@ -12,9 +12,12 @@ async function checkDirtyRepo(
   ctx: ExtensionContext,
   action: string,
 ): Promise<{ cancel: boolean } | undefined> {
-  const { stdout, code } = await pi.exec("git", ["status", "--porcelain"]);
+  const { stdout, code, killed } = await pi.exec("git", ["status", "--porcelain"], {
+    timeout: 15_000,
+    signal: ctx.signal,
+  });
 
-  if (code !== 0) {
+  if (killed || code !== 0) {
     return;
   }
 

package/extensions/dirty-repo-guard/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-dirty-repo-guard",
-  "version": "0.1.12",
+  "version": "0.1.13",
   "description": "A pi extension that prompts before session changes when the current git repo has uncommitted changes.",
   "keywords": [
     "pi-package",

package/extensions/fast/README.md

@@ -34,11 +34,7 @@ For eligible Anthropic Claude models it injects:
 }
 ```
 
-It also adds the required Anthropic beta header value:
-
-```text
-anthropic-beta: fast-mode-2026-02-01
-```
+It also appends the required beta `fast-mode-2026-02-01` to the request's `betas` list, which Pi sends as the `anthropic-beta` header. The extension does not set the `anthropic-beta` header itself, so Pi's own betas are kept.
 
 ## Eligibility
 

package/extensions/fast/index.ts

@@ -12,9 +12,6 @@ const ANTHROPIC_PROVIDER_ID = "anthropic";
 const ANTHROPIC_API_ID = "anthropic-messages";
 const ANTHROPIC_FAST_SPEED = "fast";
 const ANTHROPIC_FAST_BETA = "fast-mode-2026-02-01";
-const CLAUDE_CODE_OAUTH_BETAS = ["claude-code-20250219", "oauth-2025-04-20"];
-const FINE_GRAINED_TOOL_STREAMING_BETA = "fine-grained-tool-streaming-2025-05-14";
-const INTERLEAVED_THINKING_BETA = "interleaved-thinking-2025-05-14";
 const ANTHROPIC_SUPPORTED_MODELS = new Set(["claude-opus-4-8", "claude-opus-5", "claude-opus-5-5"]);
 
 const OPENAI_PROVIDER_ID = "openai-codex";
@@ -79,11 +76,6 @@ type RecursivePartial<T> = {
 
 type PayloadRecord = Record<string, unknown>;
 
-type AnthropicHeaderCompat = {
-  forceAdaptiveThinking?: boolean;
-  supportsEagerToolInputStreaming?: boolean;
-};
-
 type Eligibility = {
   eligible: boolean;
   modelKey: string;
@@ -368,67 +360,17 @@ function getStatusMessage(ctx: ExtensionContext, state: SessionState): string {
   return `Fast mode is ${describeMode(state)}. Current model: ${eligibility.modelKey}.${injected}`;
 }
 
-function splitBetaHeader(value: string | null | undefined): string[] {
-  return (value ?? "")
-    .split(",")
-    .map((part) => part.trim())
-    .filter(Boolean);
-}
-
-async function isAnthropicOAuth(ctx: ExtensionContext): Promise<boolean> {
-  try {
-    const resolution = await ctx.modelRegistry.getProviderAuth(ANTHROPIC_PROVIDER_ID);
-    const apiKey = resolution?.auth.apiKey;
-    if (typeof apiKey === "string") return apiKey.includes("sk-ant-oat");
-  } catch {
-    // Fall back to the synchronous registry snapshot below.
-  }
-
-  return ctx.model ? ctx.modelRegistry.isUsingOAuth(ctx.model) : false;
-}
-
-function getAnthropicProviderBetas(ctx: ExtensionContext, hasActiveTools: boolean): string[] {
-  const model = ctx.model;
-  if (!model) return [];
-  const compat = model.compat as AnthropicHeaderCompat | undefined;
-
-  const betas: string[] = [];
-  if (hasActiveTools && compat?.supportsEagerToolInputStreaming === false) {
-    betas.push(FINE_GRAINED_TOOL_STREAMING_BETA);
-  }
-  if (compat?.forceAdaptiveThinking !== true) {
-    betas.push(INTERLEAVED_THINKING_BETA);
-  }
-  return betas;
-}
-
-async function injectAnthropicFastHeader(
-  headers: Record<string, string | null>,
-  ctx: ExtensionContext,
-  state: SessionState,
-  hasActiveTools: boolean,
-): Promise<void> {
-  if (state.ultrafast || !isFastEnabled(state)) return;
-  const eligibility = getEligibility(ctx, false);
-  if (!eligibility.eligible || eligibility.provider !== "anthropic") return;
-
-  const matchingKeys = Object.keys(headers).filter(
-    (name) => name.toLowerCase() === "anthropic-beta",
-  );
-  const existing = matchingKeys.flatMap((name) => splitBetaHeader(headers[name]));
-  const matchingModelEntries = Object.entries(ctx.model?.headers ?? {}).filter(
-    ([name]) => name.toLowerCase() === "anthropic-beta",
-  );
-  const modelBetas = matchingModelEntries.flatMap(([, value]) => splitBetaHeader(value));
-  const providerBetas = getAnthropicProviderBetas(ctx, hasActiveTools);
-  const oauthBetas = (await isAnthropicOAuth(ctx)) ? CLAUDE_CODE_OAUTH_BETAS : [];
-  const next = Array.from(
-    new Set([...modelBetas, ...existing, ...oauthBetas, ...providerBetas, ANTHROPIC_FAST_BETA]),
-  );
-
-  for (const key of matchingKeys) delete headers[key];
-  const outputKey = matchingModelEntries.at(-1)?.[0] ?? "anthropic-beta";
-  headers[outputKey] = next.join(",");
+/**
+ * Adds the Fast beta to the request's `betas` list instead of rewriting the
+ * `anthropic-beta` header. Pi derives the final beta list from the model and
+ * request (mid-conversation effort, server-side fallback, inline tools, OAuth,
+ * interleaved thinking, ...) and treats an explicit `anthropic-beta` header as
+ * the complete list, so setting that header here would drop any beta this
+ * extension does not know about.
+ */
+function withAnthropicFastBeta(betas: unknown[] | undefined): unknown[] {
+  const existing = betas ?? [];
+  return existing.includes(ANTHROPIC_FAST_BETA) ? existing : [...existing, ANTHROPIC_FAST_BETA];
 }
 
 function injectUltrafastPayload(
@@ -461,9 +403,10 @@ function injectFastPayload(
 
   if (eligibility.provider === "anthropic") {
     if ("speed" in payload) return undefined;
+    if (payload.betas !== undefined && !Array.isArray(payload.betas)) return undefined;
     state.lastInjectedAt = Date.now();
     state.lastInjectedModel = eligibility.modelKey;
-    return { ...payload, speed: ANTHROPIC_FAST_SPEED };
+    return { ...payload, speed: ANTHROPIC_FAST_SPEED, betas: withAnthropicFastBeta(payload.betas) };
   }
 
   if ("service_tier" in payload) return undefined;
@@ -552,12 +495,6 @@ export default function fastExtension(pi: ExtensionAPI) {
     updateStatus(ctx, state);
   });
 
-  pi.on("before_provider_headers", async (event, ctx) => {
-    const hasActiveTools =
-      typeof pi.getActiveTools === "function" && pi.getActiveTools().length > 0;
-    await injectAnthropicFastHeader(event.headers, ctx, getState(ctx), hasActiveTools);
-  });
-
   pi.on("before_provider_request", (event, ctx) => {
     const state = getState(ctx);
     if (state.ultrafast) {

package/extensions/fast/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-fast",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "A pi extension that enables Fast mode for supported direct OpenAI API, OpenAI Codex, and Anthropic Claude models based on the currently selected model.",
   "keywords": [
     "pi-package",

package/extensions/inline-bash/index.ts

@@ -49,6 +49,7 @@ export default function (pi: ExtensionAPI) {
       try {
         const bashResult = await pi.exec("bash", ["-c", command], {
           timeout: TIMEOUT_MS,
+          signal: ctx.signal,
         });
 
         const output = bashResult.stdout || bashResult.stderr || "";

package/extensions/inline-bash/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-inline-bash",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "description": "A pi extension that expands inline bash commands in user prompts.",
   "keywords": [
     "pi-package",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-extensions",
-  "version": "0.1.80",
+  "version": "0.1.81",
   "description": "A collection of pi extensions and skills for annotation UIs, context management, workflow audits, contrarian review, review-comment triage, notifications, brrr push alerts, safety guards, GitHub research, repo-local knowledge, todos, tool rendering, model/provider helpers, and Xiaohei-style article illustrations.",
   "keywords": [
     "pi-package",