pi-anthropic-oauth 0.1.7 → 0.1.9

package/README.md

@@ -1,12 +1,11 @@
 # Pi Anthropic OAuth
 
-[![npm](https://img.shields.io/npm/v/pi-anthropic-oauth?style=flat-square&logo=npm&logoColor=white&label=npm&color=7c3aed)](https://www.npmjs.com/package/pi-anthropic-oauth) [![downloads](https://img.shields.io/npm/dm/pi-anthropic-oauth?style=flat-square&logo=npm&logoColor=white&label=downloads&color=7c3aed)](https://www.npmjs.com/package/pi-anthropic-oauth) [![node](https://img.shields.io/badge/node-%3E%3D18-7c3aed?style=flat-square&logo=nodedotjs&logoColor=white)](https://nodejs.org)
+[![npm](https://img.shields.io/npm/v/pi-anthropic-oauth?style=flat-square&logo=npm&logoColor=white&label=npm&color=7c3aed)](https://www.npmjs.com/package/pi-anthropic-oauth) [![node](https://img.shields.io/badge/node-%3E%3D18-7c3aed?style=flat-square&logo=nodedotjs&logoColor=white)](https://nodejs.org)
 
 Claude Pro/Max OAuth extension for Pi.
 
 > [!WARNING]
 > Use this at your own risk. This may go against Anthropic's terms.
-> I do not recommend using this with gsd2 or any extensions that will cause heavy token usage.
 
 ## Install
 
@@ -28,10 +27,36 @@ Claude Pro/Max
 
 ## Notes
 
+- uses Pi's built-in Anthropic model list
 - sends Claude Code-like OAuth headers
 - rewrites Pi system identity where needed
 - auto-creates `~/.Claude Code` → `~/.pi` symlink
 
+## Custom models
+
+To add another Anthropic model, create `~/.pi/agent/models.json`:
+
+```json
+{
+  "providers": {
+    "anthropic": {
+      "baseUrl": "https://api.anthropic.com",
+      "apiKey": "REQUIRED_BY_PI_FOR_CUSTOM_MODELS",
+      "api": "anthropic-messages",
+      "models": [
+        {
+          "id": "your-model-id",
+          "name": "Your Model Name"
+        }
+      ]
+    }
+  }
+}
+```
+
+> [!NOTE]
+> Pi requires `baseUrl`, `apiKey`, and `api` when defining custom models in `models.json`. With this extension, requests normally authenticate through Claude Pro/Max OAuth after `/login anthropic`, so `apiKey` is present to satisfy Pi's config requirements and does not need to be a valid Anthropic API key.
+
 ## License
 
 MIT

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-anthropic-oauth",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "description": "Claude Pro/Max OAuth extension for Pi",
   "type": "module",
   "license": "MIT",

package/src/auth.ts

@@ -152,7 +152,6 @@ export async function refreshAnthropicToken(
           grant_type: "refresh_token",
           client_id: CLIENT_ID,
           refresh_token: credentials.refresh,
-          scope: SCOPES,
         }),
       },
       "Token refresh",

package/src/convert.ts

@@ -98,8 +98,6 @@ export function convertPiMessagesToAnthropic(
       for (const block of message.content) {
         if (block.type === "text" && block.text.trim()) {
           blocks.push({ type: "text", text: sanitizeSurrogates(block.text) });
-        } else if (block.type === "thinking" && block.thinking.trim()) {
-          blocks.push({ type: "text", text: sanitizeSurrogates(block.thinking) });
         } else if (block.type === "toolCall") {
           blocks.push({
             type: "tool_use",

package/src/index.ts

@@ -1,40 +1,27 @@
 import { existsSync, symlinkSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
-import type { ExtensionAPI, ProviderConfig } from "@mariozechner/pi-coding-agent";
+import {
+  AuthStorage,
+  ModelRegistry,
+  type ExtensionAPI,
+  type ProviderConfig,
+} from "@mariozechner/pi-coding-agent";
 import type { OAuthCredentials } from "@mariozechner/pi-ai";
 import { loginAnthropic, refreshAnthropicToken } from "./auth.js";
 import { streamAnthropicOAuth } from "./stream.js";
 
-const MODELS = [
-  {
-    id: "claude-opus-4-6",
-    name: "Claude Opus 4.6",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-    contextWindow: 1000000,
-    maxTokens: 128000,
-  },
-  {
-    id: "claude-sonnet-4-6",
-    name: "Claude Sonnet 4.6",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-    contextWindow: 1000000,
-    maxTokens: 64000,
-  },
-  {
-    id: "claude-haiku-4-5",
-    name: "Claude Haiku 4.5",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-    contextWindow: 200000,
-    maxTokens: 64000,
-  },
-];
+const DEFAULT_OPUS_4_7: NonNullable<ProviderConfig["models"]>[number] = {
+  id: "claude-opus-4-7",
+  name: "Claude Opus 4.7",
+  api: "anthropic-messages",
+  reasoning: true,
+  input: ["text", "image"],
+  cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  contextWindow: 1000000,
+  maxTokens: 128000,
+  compat: undefined,
+};
 
 function ensureClaudeCodeSymlink() {
   const target = join(homedir(), ".pi");
@@ -42,17 +29,42 @@ function ensureClaudeCodeSymlink() {
   if (existsSync(target) && !existsSync(link)) {
     try {
       symlinkSync(target, link);
-    } catch {}
+    } catch { }
   }
 }
 
+function getAnthropicModels(): NonNullable<ProviderConfig["models"]> {
+  const modelRegistry = ModelRegistry.create(AuthStorage.inMemory());
+  const models: NonNullable<ProviderConfig["models"]> = modelRegistry
+    .getAll()
+    .filter((model) => model.provider === "anthropic")
+    .map((model) => ({
+      id: model.id,
+      name: model.name,
+      api: model.api ?? "anthropic-messages",
+      reasoning: model.reasoning,
+      input: model.input,
+      cost: model.cost,
+      contextWindow: model.contextWindow,
+      maxTokens: model.maxTokens,
+      compat: model.compat,
+    }));
+
+  if (!models.some((model) => model.id === DEFAULT_OPUS_4_7.id)) {
+    models.push(DEFAULT_OPUS_4_7);
+  }
+
+  return models;
+}
+
 export default function (pi: ExtensionAPI) {
   ensureClaudeCodeSymlink();
+  const models = getAnthropicModels();
+
   pi.registerProvider("anthropic", {
     baseUrl: "https://api.anthropic.com",
-    apiKey: "ANTHROPIC_MAX_API_KEY",
-    api: "anthropic-max-api",
-    models: [...MODELS],
+    api: "anthropic-messages",
+    models,
     oauth: {
       name: "Claude Pro/Max",
       usesCallbackServer: true,

package/src/stream.ts

@@ -11,7 +11,7 @@ import {
   type SimpleStreamOptions,
   type StopReason,
 } from "@mariozechner/pi-ai";
-import { USER_AGENT } from "./auth.js";
+import { isClaudeOAuthAccessToken, USER_AGENT } from "./auth.js";
 import {
   convertPiMessagesToAnthropic,
   convertPiToolsToAnthropic,
@@ -63,8 +63,16 @@ function makeDefaultHeaders(
   }
 
   if (options?.headers) {
-    for (const [key, value] of Object.entries(options.headers))
+    for (const [key, value] of Object.entries(options.headers)) {
+      const normalizedKey = key.toLowerCase();
+      if (
+        isOAuth &&
+        (normalizedKey === "x-api-key" || normalizedKey === "authorization")
+      ) {
+        continue;
+      }
       headers[key] = value;
+    }
   }
 
   return headers;
@@ -97,8 +105,12 @@ export function streamAnthropicOAuth(
     };
 
     try {
-      const apiKey = options?.apiKey ?? "";
-      const isOAuth = true;
+      const apiKey = options?.apiKey;
+      if (!apiKey) {
+        throw new Error("No Anthropic auth available. Run /login anthropic.");
+      }
+
+      const isOAuth = isClaudeOAuthAccessToken(apiKey);
       const defaultHeaders = makeDefaultHeaders(isOAuth, options);
 
       if (isOAuth) defaultHeaders.authorization = `Bearer ${apiKey}`;