@gotgenes/pi-anthropic-auth 0.3.0 → 0.4.0

package/dist/constants.d.ts

@@ -43,3 +43,28 @@ export declare const BILLING_HEADER_SALT = "59cf53e54c78";
 export declare const BILLING_HEADER_POSITIONS: readonly [4, 7, 20];
 /** Entrypoint identifier included in the billing header. */
 export declare const CLAUDE_CODE_ENTRYPOINT = "sdk-cli";
+/**
+ * Strings whose presence in a paragraph marks it as Pi-specific and droppable.
+ *
+ * Each entry is checked with `paragraph.includes(anchor)`.
+ */
+export declare const PARAGRAPH_REMOVAL_ANCHORS: readonly string[];
+/**
+ * Inline text replacements applied after paragraph removal.
+ *
+ * These handle known Anthropic classifier trigger phrases that may appear
+ * in paragraphs we want to keep.  Each rule is applied with `replaceAll`.
+ *
+ * The "Here is some useful information..." phrase was isolated by
+ * `opencode-anthropic-auth` via sliding-window bisection of a 10KB failing
+ * prompt.  When it reaches Anthropic combined with typical agent context,
+ * /v1/messages responds with a 400 disguised as "You're out of extra usage."
+ * Replacing the word "useful" is enough to unblock the request.
+ *
+ * We don't currently emit this phrase, but it's included as a documented
+ * future risk per Issue #10.
+ */
+export declare const TEXT_REPLACEMENTS: readonly {
+    match: string;
+    replacement: string;
+}[];

package/dist/constants.js

@@ -60,3 +60,49 @@ export const BILLING_HEADER_SALT = "59cf53e54c78";
 export const BILLING_HEADER_POSITIONS = [4, 7, 20];
 /** Entrypoint identifier included in the billing header. */
 export const CLAUDE_CODE_ENTRYPOINT = "sdk-cli";
+// ---------------------------------------------------------------------------
+// Anchor-driven sanitizer constants
+//
+// Used by the system prompt sanitizer to remove Pi-specific paragraphs
+// (identity, documentation references, filler) while preserving extension-
+// contributed content (tool snippets, guidelines, appended content).
+//
+// A paragraph is any text between blank lines.  If a paragraph contains any
+// anchor string, it is dropped entirely.  This is resilient to upstream
+// rewording — as long as the anchor still appears somewhere in the paragraph,
+// removal works regardless of surrounding text changes.
+// ---------------------------------------------------------------------------
+/**
+ * Strings whose presence in a paragraph marks it as Pi-specific and droppable.
+ *
+ * Each entry is checked with `paragraph.includes(anchor)`.
+ */
+export const PARAGRAPH_REMOVAL_ANCHORS = [
+    // Pi identity sentence
+    "operating inside pi, a coding agent harness",
+    // Pi-specific filler about custom tools
+    "In addition to the tools above",
+    // Pi documentation block — references Pi-specific docs/paths
+    "Pi documentation (read only when the user asks about pi itself",
+];
+/**
+ * Inline text replacements applied after paragraph removal.
+ *
+ * These handle known Anthropic classifier trigger phrases that may appear
+ * in paragraphs we want to keep.  Each rule is applied with `replaceAll`.
+ *
+ * The "Here is some useful information..." phrase was isolated by
+ * `opencode-anthropic-auth` via sliding-window bisection of a 10KB failing
+ * prompt.  When it reaches Anthropic combined with typical agent context,
+ * /v1/messages responds with a 400 disguised as "You're out of extra usage."
+ * Replacing the word "useful" is enough to unblock the request.
+ *
+ * We don't currently emit this phrase, but it's included as a documented
+ * future risk per Issue #10.
+ */
+export const TEXT_REPLACEMENTS = [
+    {
+        match: "Here is some useful information about the environment you are running in:",
+        replacement: "Environment context you are running in:",
+    },
+];

package/dist/request-shaping.js

@@ -1,7 +1,6 @@
 import { createHash } from "node:crypto";
 import { BILLING_HEADER_POSITIONS, BILLING_HEADER_SALT, CLAUDE_CODE_ENTRYPOINT, CLAUDE_CODE_IDENTITY_PREFIX, CLAUDE_CODE_VERSION, MINIMAL_ANTHROPIC_OAUTH_PROMPT_PREFIX, } from "./constants.js";
 import { shapeSystemBlocks } from "./system-prompt-shaping.js";
-const ANTHROPIC_OAUTH_BETAS = ["claude-code-20250219", "oauth-2025-04-20"];
 function isRecord(value) {
     return value !== null && typeof value === "object" && !Array.isArray(value);
 }
@@ -90,13 +89,6 @@ function prependBillingHeader(system, messages) {
     const billingBlock = { type: "text", text: billingHeader };
     return [billingBlock, ...systemBlocks];
 }
-function mergeAnthropicBetas(betaHeader) {
-    const existing = (betaHeader ?? "")
-        .split(",")
-        .map((value) => value.trim())
-        .filter(Boolean);
-    return [...new Set([...ANTHROPIC_OAUTH_BETAS, ...existing])].join(",");
-}
 /**
  * Splits assistant messages that interleave text and tool_use blocks.
  *
@@ -139,13 +131,9 @@ export function shapeAnthropicOAuthPayload(payload) {
     const shapedSystem = Array.isArray(payload.system)
         ? shapeSystemBlocks(payload.system)
         : payload.system;
-    const shapedPayload = {
+    return {
         ...payload,
         messages: normalizedMessages,
         system: prependBillingHeader(shapedSystem, normalizedMessages),
     };
-    shapedPayload["anthropic-beta"] = mergeAnthropicBetas(typeof payload["anthropic-beta"] === "string"
-        ? payload["anthropic-beta"]
-        : undefined);
-    return shapedPayload;
 }

package/dist/system-prompt-shaping.d.ts

@@ -1,9 +1,27 @@
+/**
+ * Sanitize system prompt text by removing paragraphs containing known
+ * Pi-specific anchor strings and applying inline text replacements for
+ * known Anthropic classifier trigger phrases.
+ *
+ * A paragraph is any text between blank lines (`\n\n`).
+ *
+ * This approach is resilient to upstream rewording — as long as the anchor
+ * string still appears somewhere in the paragraph, removal works regardless
+ * of how the surrounding text changes.
+ */
+export declare function sanitizeSystemText(text: string): string;
 /**
  * Shape a system prompt string for Anthropic OAuth compatibility.
  *
- * Replaces Pi's verbose default preamble with a minimal neutral prompt while
- * preserving any project context that follows.  Returns the original string
- * unchanged when Pi's default preamble is not detected.
+ * Uses an anchor-driven sanitizer to remove Pi-specific paragraphs
+ * (identity, documentation references, filler) while preserving
+ * extension-contributed content (tool snippets, guidelines, appended
+ * content, project context, skills, and date/cwd footer).
+ *
+ * Prepends a minimal neutral prompt to replace the removed Pi identity.
+ *
+ * If the Pi default prompt prefix is not present, the prompt is returned
+ * unchanged — this gates shaping so non-Pi prompts pass through.
  */
 export declare function shapeAnthropicOAuthSystemPrompt(systemPrompt: string): string;
 type TextBlock = {

package/dist/system-prompt-shaping.js

@@ -1,24 +1,52 @@
-import { MINIMAL_ANTHROPIC_OAUTH_PROMPT, PI_DEFAULT_PROMPT_PREFIX, } from "./constants.js";
-function findProjectContextStart(systemPrompt) {
-    const marker = "\n\n# Project Context\n\n";
-    return systemPrompt.indexOf(marker);
+import { MINIMAL_ANTHROPIC_OAUTH_PROMPT, PARAGRAPH_REMOVAL_ANCHORS, PI_DEFAULT_PROMPT_PREFIX, TEXT_REPLACEMENTS, } from "./constants.js";
+/**
+ * Sanitize system prompt text by removing paragraphs containing known
+ * Pi-specific anchor strings and applying inline text replacements for
+ * known Anthropic classifier trigger phrases.
+ *
+ * A paragraph is any text between blank lines (`\n\n`).
+ *
+ * This approach is resilient to upstream rewording — as long as the anchor
+ * string still appears somewhere in the paragraph, removal works regardless
+ * of how the surrounding text changes.
+ */
+export function sanitizeSystemText(text) {
+    const paragraphs = text.split(/\n\n+/);
+    const filtered = paragraphs.filter((paragraph) => {
+        for (const anchor of PARAGRAPH_REMOVAL_ANCHORS) {
+            if (paragraph.includes(anchor))
+                return false;
+        }
+        return true;
+    });
+    let result = filtered.join("\n\n");
+    for (const rule of TEXT_REPLACEMENTS) {
+        result = result.replaceAll(rule.match, rule.replacement);
+    }
+    return result.trim();
 }
 /**
  * Shape a system prompt string for Anthropic OAuth compatibility.
  *
- * Replaces Pi's verbose default preamble with a minimal neutral prompt while
- * preserving any project context that follows.  Returns the original string
- * unchanged when Pi's default preamble is not detected.
+ * Uses an anchor-driven sanitizer to remove Pi-specific paragraphs
+ * (identity, documentation references, filler) while preserving
+ * extension-contributed content (tool snippets, guidelines, appended
+ * content, project context, skills, and date/cwd footer).
+ *
+ * Prepends a minimal neutral prompt to replace the removed Pi identity.
+ *
+ * If the Pi default prompt prefix is not present, the prompt is returned
+ * unchanged — this gates shaping so non-Pi prompts pass through.
  */
 export function shapeAnthropicOAuthSystemPrompt(systemPrompt) {
     if (!systemPrompt.includes(PI_DEFAULT_PROMPT_PREFIX)) {
         return systemPrompt;
     }
-    const projectContextStart = findProjectContextStart(systemPrompt);
-    if (projectContextStart === -1) {
+    const sanitized = sanitizeSystemText(systemPrompt);
+    if (!sanitized) {
         return MINIMAL_ANTHROPIC_OAUTH_PROMPT;
     }
-    return `${MINIMAL_ANTHROPIC_OAUTH_PROMPT}${systemPrompt.slice(projectContextStart)}`;
+    return `${MINIMAL_ANTHROPIC_OAUTH_PROMPT}\n\n${sanitized}`;
 }
 /**
  * Apply system prompt shaping to an array of Anthropic system text blocks.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gotgenes/pi-anthropic-auth",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Pi extension package for Anthropic OAuth compatibility",
   "author": {
     "name": "Chris Lasher"
@@ -38,7 +38,8 @@
     "lint:md": "markdownlint-cli2 '*.md'",
     "lint:all": "pnpm run lint && pnpm run lint:md",
     "format": "biome format --write .",
-    "test": "tsx --test test/**/*.test.ts",
+    "test": "vitest run",
+    "test:watch": "vitest",
     "prepublishOnly": "pnpm run check && pnpm run test && pnpm run build"
   },
   "pi": {
@@ -56,7 +57,7 @@
     "@mariozechner/pi-ai": "^0.68.0",
     "@mariozechner/pi-coding-agent": "^0.68.0",
     "markdownlint-cli2": "^0.22.0",
-    "tsx": "^4.20.6",
-    "typescript": "^6.0.3"
+    "typescript": "^6.0.3",
+    "vitest": "^4.1.5"
   }
 }