@gotgenes/pi-anthropic-auth 0.4.6 → 0.5.0

package/dist/request-shaping.js

@@ -1,189 +0,0 @@
-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 { debugLog, isToolUseOnlyDebugEnabled } from "./debug.js";
-import { shapeSystemBlocks } from "./system-prompt-shaping.js";
-function isRecord(value) {
-    return value !== null && typeof value === "object" && !Array.isArray(value);
-}
-function isAnthropicMessagesPayload(payload) {
-    return (isRecord(payload) &&
-        typeof payload.model === "string" &&
-        Array.isArray(payload.messages) &&
-        typeof payload.stream === "boolean");
-}
-function isOAuthAnthropicPayload(payload) {
-    if (!Array.isArray(payload.system)) {
-        return false;
-    }
-    return payload.system.some(hasOAuthAnthropicSystemMarker);
-}
-function hasOAuthAnthropicSystemMarker(block) {
-    if (!isRecord(block) ||
-        block.type !== "text" ||
-        typeof block.text !== "string") {
-        return false;
-    }
-    return (block.text.includes(CLAUDE_CODE_IDENTITY_PREFIX) ||
-        block.text.includes("x-anthropic-billing-header:") ||
-        block.text.startsWith(MINIMAL_ANTHROPIC_OAUTH_PROMPT_PREFIX));
-}
-function getFirstUserText(messages) {
-    const firstUserMessage = messages.find((message) => message.role === "user");
-    if (!firstUserMessage)
-        return "";
-    if (typeof firstUserMessage.content === "string") {
-        return firstUserMessage.content;
-    }
-    if (!Array.isArray(firstUserMessage.content)) {
-        return "";
-    }
-    const firstTextBlock = firstUserMessage.content.find((block) => block?.type === "text" && typeof block.text === "string");
-    return typeof firstTextBlock?.text === "string" ? firstTextBlock.text : "";
-}
-function buildBillingHeaderValue(messages) {
-    const messageText = getFirstUserText(messages);
-    if (!messageText) {
-        return undefined;
-    }
-    const cch = createHash("sha256")
-        .update(messageText)
-        .digest("hex")
-        .slice(0, 5);
-    const sampledCharacters = BILLING_HEADER_POSITIONS.map((index) => messageText[index] || "0").join("");
-    const suffix = createHash("sha256")
-        .update(`${BILLING_HEADER_SALT}${sampledCharacters}${CLAUDE_CODE_VERSION}`)
-        .digest("hex")
-        .slice(0, 3);
-    return [
-        "x-anthropic-billing-header:",
-        `cc_version=${CLAUDE_CODE_VERSION}.${suffix};`,
-        `cc_entrypoint=${CLAUDE_CODE_ENTRYPOINT};`,
-        `cch=${cch};`,
-    ].join(" ");
-}
-function normalizeSystemBlock(block) {
-    if (typeof block === "string") {
-        return { type: "text", text: block };
-    }
-    if (isRecord(block) && typeof block.text === "string") {
-        return {
-            ...block,
-            type: "text",
-            text: block.text,
-        };
-    }
-    return { type: "text", text: String(block ?? "") };
-}
-function prependBillingHeader(system, messages) {
-    const billingHeader = buildBillingHeaderValue(messages);
-    if (!billingHeader) {
-        return system;
-    }
-    const systemBlocks = Array.isArray(system)
-        ? system.map(normalizeSystemBlock)
-        : system == null
-            ? []
-            : [normalizeSystemBlock(system)];
-    if (systemBlocks.some((block) => block.text.includes("x-anthropic-billing-header:"))) {
-        return systemBlocks;
-    }
-    const billingBlock = { type: "text", text: billingHeader };
-    return [billingBlock, ...systemBlocks];
-}
-/**
- * Splits assistant messages that interleave text and tool_use blocks.
- *
- * The Anthropic API rejects assistant turns where non-tool_use blocks follow
- * a tool_use block.  Pi's serializer can produce this ordering, so we split
- * the message into two consecutive assistant turns: one with text blocks and
- * one with tool_use blocks.  The reordering is safe because the text and
- * tool_use blocks are semantically independent within a single turn.
- */
-function splitAssistantToolUseTrailingContent(messages) {
-    return messages.flatMap((message) => {
-        if (message.role !== "assistant" || !Array.isArray(message.content)) {
-            return [message];
-        }
-        const firstToolUseIndex = message.content.findIndex((block) => block?.type === "tool_use");
-        if (firstToolUseIndex === -1) {
-            return [message];
-        }
-        const trailingBlocks = message.content.slice(firstToolUseIndex);
-        if (!trailingBlocks.some((block) => block?.type !== "tool_use")) {
-            return [message];
-        }
-        const nonToolUseBlocks = message.content.filter((block) => block?.type !== "tool_use");
-        const toolUseBlocks = message.content.filter((block) => block?.type === "tool_use");
-        return [
-            { ...message, content: nonToolUseBlocks },
-            { ...message, content: toolUseBlocks },
-        ];
-    });
-}
-function getToolDefinitionNames(payload) {
-    const tools = payload.tools;
-    if (!Array.isArray(tools)) {
-        return [];
-    }
-    return tools
-        .map((tool) => isRecord(tool) && typeof tool.name === "string" ? tool.name : undefined)
-        .filter((name) => typeof name === "string");
-}
-function getToolUseNames(messages) {
-    return messages.flatMap((message) => {
-        if (!Array.isArray(message.content)) {
-            return [];
-        }
-        return message.content
-            .map((block) => block?.type === "tool_use" && typeof block.name === "string"
-            ? block.name
-            : undefined)
-            .filter((name) => typeof name === "string");
-    });
-}
-function countAssistantMessages(messages) {
-    return messages.filter((message) => message.role === "assistant").length;
-}
-function shouldLogRequestDebug(messages) {
-    if (!isToolUseOnlyDebugEnabled()) {
-        return true;
-    }
-    return getToolUseNames(messages).length > 0;
-}
-export function shapeAnthropicOAuthPayload(payload) {
-    if (!isAnthropicMessagesPayload(payload)) {
-        return payload;
-    }
-    const messages = payload.messages;
-    if (!isOAuthAnthropicPayload(payload)) {
-        return payload;
-    }
-    const normalizedMessages = splitAssistantToolUseTrailingContent(messages);
-    const shapedSystem = Array.isArray(payload.system)
-        ? shapeSystemBlocks(payload.system)
-        : payload.system;
-    const finalSystem = prependBillingHeader(shapedSystem, normalizedMessages);
-    const toolUseNamesBefore = getToolUseNames(messages);
-    const toolUseNamesAfter = getToolUseNames(normalizedMessages);
-    if (shouldLogRequestDebug(messages)) {
-        debugLog("before-provider-request", {
-            model: payload.model,
-            systemBlockCountBefore: Array.isArray(payload.system)
-                ? payload.system.length
-                : 0,
-            systemBlockCountAfter: Array.isArray(finalSystem)
-                ? finalSystem.length
-                : 0,
-            assistantMessagesBefore: countAssistantMessages(messages),
-            assistantMessagesAfter: countAssistantMessages(normalizedMessages),
-            toolDefinitions: getToolDefinitionNames(payload),
-            toolUseNamesBefore,
-            toolUseNamesAfter,
-        });
-    }
-    return {
-        ...payload,
-        messages: normalizedMessages,
-        system: finalSystem,
-    };
-}

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

@@ -1,53 +0,0 @@
-/**
- * Reset the one-time terminator-missing warning latch. Exposed for tests.
- */
-export declare function _resetShapingWarnings(): void;
-export type SanitizedSystemTextReport = {
-    text: string;
-    removedParagraphs: Array<{
-        anchor: string;
-        preview: string;
-    }>;
-    replacementMatches: string[];
-};
-/**
- * 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 sanitizeSystemTextWithReport(text: string): SanitizedSystemTextReport;
-export declare function sanitizeSystemText(text: string): string;
-/**
- * Shape a system prompt string for Anthropic OAuth compatibility.
- *
- * For the normal upstream Pi prompt shape, sanitize only the known preamble
- * span and replace its identity paragraph with the minimal neutral prompt.
- * This preserves downstream configuration/extension points embedded in the
- * preamble (tool snippets and guideline bullets) while still stripping the
- * Pi-specific identity, filler, and documentation paragraphs.
- *
- * If Pi's known preamble terminator drifts upstream, we fall back to slicing
- * from `# Project Context`. If that section is also absent, we return the
- * minimal prompt only.
- */
-export declare function shapeAnthropicOAuthSystemPrompt(systemPrompt: string): string;
-type TextBlock = {
-    type: "text";
-    text: string;
-    [key: string]: unknown;
-};
-/**
- * Apply system prompt shaping to an array of Anthropic system text blocks.
- *
- * Finds the first block containing Pi's default prompt preamble and replaces
- * its text in-place (returning a new array).  Blocks without the preamble are
- * passed through unchanged.
- */
-export declare function shapeSystemBlocks(blocks: TextBlock[]): TextBlock[];
-export {};

package/dist/system-prompt-shaping.js

@@ -1,146 +0,0 @@
-import { MINIMAL_ANTHROPIC_OAUTH_PROMPT, PARAGRAPH_REMOVAL_ANCHORS, PI_DEFAULT_PROMPT_PREFIX, PI_DEFAULT_PROMPT_TERMINATOR, TEXT_REPLACEMENTS, } from "./constants.js";
-import { debugLog, isToolUseOnlyDebugEnabled } from "./debug.js";
-let warnedTerminatorMissing = false;
-function warnTerminatorMissingOnce() {
-    if (warnedTerminatorMissing) {
-        return;
-    }
-    warnedTerminatorMissing = true;
-    console.warn("[pi-anthropic-auth] Pi default preamble terminator not found; falling back to '# Project Context' anchor. " +
-        "Upstream Pi may have reworded its preamble — update PI_DEFAULT_PROMPT_TERMINATOR.");
-}
-/**
- * Reset the one-time terminator-missing warning latch. Exposed for tests.
- */
-export function _resetShapingWarnings() {
-    warnedTerminatorMissing = false;
-}
-function previewParagraph(paragraph) {
-    return paragraph.replace(/\s+/g, " ").trim().slice(0, 140);
-}
-function shouldLogPromptDebug(report) {
-    if (!isToolUseOnlyDebugEnabled()) {
-        return true;
-    }
-    return (report.removedParagraphs.length === 0 &&
-        report.replacementMatches.length > 0);
-}
-/**
- * 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 sanitizeSystemTextWithReport(text) {
-    const paragraphs = text.split(/\n\n+/);
-    const removedParagraphs = [];
-    const filtered = paragraphs.filter((paragraph) => {
-        for (const anchor of PARAGRAPH_REMOVAL_ANCHORS) {
-            if (!paragraph.includes(anchor)) {
-                continue;
-            }
-            removedParagraphs.push({
-                anchor,
-                preview: previewParagraph(paragraph),
-            });
-            return false;
-        }
-        return true;
-    });
-    let result = filtered.join("\n\n");
-    const replacementMatches = [];
-    for (const rule of TEXT_REPLACEMENTS) {
-        if (result.includes(rule.match)) {
-            replacementMatches.push(rule.match);
-        }
-        result = result.replaceAll(rule.match, rule.replacement);
-    }
-    return {
-        text: result.trim(),
-        removedParagraphs,
-        replacementMatches,
-    };
-}
-export function sanitizeSystemText(text) {
-    return sanitizeSystemTextWithReport(text).text;
-}
-function findProjectContextStart(systemPrompt) {
-    const marker = "\n\n# Project Context\n\n";
-    return systemPrompt.indexOf(marker);
-}
-/**
- * Shape a system prompt string for Anthropic OAuth compatibility.
- *
- * For the normal upstream Pi prompt shape, sanitize only the known preamble
- * span and replace its identity paragraph with the minimal neutral prompt.
- * This preserves downstream configuration/extension points embedded in the
- * preamble (tool snippets and guideline bullets) while still stripping the
- * Pi-specific identity, filler, and documentation paragraphs.
- *
- * If Pi's known preamble terminator drifts upstream, we fall back to slicing
- * from `# Project Context`. If that section is also absent, we return the
- * minimal prompt only.
- */
-export function shapeAnthropicOAuthSystemPrompt(systemPrompt) {
-    const prefixIdx = systemPrompt.indexOf(PI_DEFAULT_PROMPT_PREFIX);
-    if (prefixIdx === -1) {
-        return systemPrompt;
-    }
-    const terminatorIdx = systemPrompt.indexOf(PI_DEFAULT_PROMPT_TERMINATOR, prefixIdx);
-    if (terminatorIdx !== -1) {
-        const terminatorEnd = terminatorIdx + PI_DEFAULT_PROMPT_TERMINATOR.length;
-        const preamble = systemPrompt.slice(prefixIdx, terminatorEnd);
-        const report = sanitizeSystemTextWithReport(preamble);
-        const shapedPreamble = report.text
-            ? `${MINIMAL_ANTHROPIC_OAUTH_PROMPT}\n\n${report.text}`
-            : MINIMAL_ANTHROPIC_OAUTH_PROMPT;
-        if (shouldLogPromptDebug(report)) {
-            debugLog("system-prompt-shaping", {
-                mode: "terminator",
-                originalLength: systemPrompt.length,
-                preambleLength: preamble.length,
-                sanitizedPreambleLength: report.text.length,
-                removedParagraphCount: report.removedParagraphs.length,
-                removedAnchors: report.removedParagraphs.map((entry) => entry.anchor),
-                removedParagraphPreviews: report.removedParagraphs.map((entry) => entry.preview),
-                replacementMatches: report.replacementMatches,
-            });
-        }
-        return (systemPrompt.slice(0, prefixIdx) +
-            shapedPreamble +
-            systemPrompt.slice(terminatorEnd));
-    }
-    warnTerminatorMissingOnce();
-    if (!isToolUseOnlyDebugEnabled()) {
-        debugLog("system-prompt-shaping", {
-            mode: "project-context-fallback",
-            originalLength: systemPrompt.length,
-        });
-    }
-    const projectContextStart = findProjectContextStart(systemPrompt);
-    if (projectContextStart === -1) {
-        return MINIMAL_ANTHROPIC_OAUTH_PROMPT;
-    }
-    return `${MINIMAL_ANTHROPIC_OAUTH_PROMPT}${systemPrompt.slice(projectContextStart)}`;
-}
-/**
- * Apply system prompt shaping to an array of Anthropic system text blocks.
- *
- * Finds the first block containing Pi's default prompt preamble and replaces
- * its text in-place (returning a new array).  Blocks without the preamble are
- * passed through unchanged.
- */
-export function shapeSystemBlocks(blocks) {
-    return blocks.map((block) => {
-        if (block.type !== "text" ||
-            !block.text.includes(PI_DEFAULT_PROMPT_PREFIX)) {
-            return block;
-        }
-        return { ...block, text: shapeAnthropicOAuthSystemPrompt(block.text) };
-    });
-}