pi-anthropic-oauth 0.1.8 → 0.1.9

package/README.md

@@ -27,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.8",
+  "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/index.ts

@@ -1,58 +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-7",
-    name: "Claude Opus 4.7",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-    contextWindow: 1000000,
-    maxTokens: 128000,
-  },
-  {
-    id: "claude-opus-4-6",
-    name: "Claude Opus 4.6",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
-    contextWindow: 1000000,
-    maxTokens: 128000,
-  },
-  {
-    id: "claude-sonnet-4-6",
-    name: "Claude Sonnet 4.6",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
-    contextWindow: 1000000,
-    maxTokens: 64000,
-  },
-  {
-    id: "claude-haiku-4-5",
-    name: "Claude Haiku 4.5",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
-    contextWindow: 200000,
-    maxTokens: 64000,
-  },
-  {
-    id: "claude-opus-4-5-20251101",
-    name: "Claude Opus 4.5",
-    reasoning: true,
-    input: ["text", "image"] as ("text" | "image")[],
-    cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
-    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");
@@ -64,13 +33,38 @@ function ensureClaudeCodeSymlink() {
   }
 }
 
+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,
@@ -105,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}`;