@mastra/code-sdk 0.0.0-aggregate-trace-20261015182851

package/dist/providers/claude-max.js

@@ -0,0 +1,226 @@
+import { AuthStorage } from "../auth/storage.js";
+import { ANTHROPIC_THINKING_BUDGET_TOKENS, getAnthropicThinkingCapability } from "./anthropic-thinking.js";
+import { runCatalogThinkingLevel } from "../thinking-catalog.js";
+import { ProviderAuthRequiredError } from "../auth/provider-auth-error.js";
+import "./anthropic-prompt-cache.js";
+import { wrapLanguageModel } from "ai";
+import { createAnthropic } from "@ai-sdk/anthropic";
+//#region src/providers/claude-max.ts
+/**
+* Claude Max OAuth Provider
+*
+* Uses OAuth tokens from AuthStorage to authenticate with Claude Max plan.
+* The OAuth endpoint requires a specific system message to be present.
+*/
+const claudeCodeIdentity = "You are Claude Code, Anthropic's official CLI for Claude.";
+const OAUTH_REQUIRED_BETAS = [
+	"oauth-2025-04-20",
+	"claude-code-20250219",
+	"interleaved-thinking-2025-05-14",
+	"fine-grained-tool-streaming-2025-05-14"
+];
+let authStorageInstance = null;
+/**
+* Get or create the shared AuthStorage instance
+*/
+function getAuthStorage() {
+	if (!authStorageInstance) authStorageInstance = new AuthStorage();
+	return authStorageInstance;
+}
+/**
+* Set a custom AuthStorage instance (useful for TUI integration)
+*/
+function setAuthStorage(storage) {
+	authStorageInstance = storage ?? null;
+}
+/**
+* Middleware that injects the Claude Code identity system message
+* Required for Claude Max OAuth authentication
+*/
+const claudeCodeMiddleware = {
+	specificationVersion: "v3",
+	transformParams: async ({ params }) => {
+		const systemMessage = {
+			role: "system",
+			content: claudeCodeIdentity
+		};
+		if (params.temperature) delete params.topP;
+		return {
+			...params,
+			prompt: [systemMessage, ...params.prompt]
+		};
+	}
+};
+/**
+* Prompt caching middleware for Anthropic
+*
+* Adds cache breakpoints at strategic locations:
+* 1. Last system message (end of static instructions + dynamic memory)
+* 2. Most recent user/assistant message (conversation context), `conversation` scope only
+*
+* This allows Anthropic to cache:
+* - System prompts and instructions (rarely change)
+* - Conversation history up to the last message
+*/
+const createPromptCacheMiddleware = (scope) => ({
+	specificationVersion: "v3",
+	transformParams: async ({ params }) => {
+		const prompt = [...params.prompt];
+		const cacheControl = {
+			type: "ephemeral",
+			ttl: scope === "conversation" ? "1h" : "5m"
+		};
+		const addCacheToMessage = (msg) => {
+			if (typeof msg.content === "string") return {
+				...msg,
+				providerOptions: {
+					...msg.providerOptions,
+					anthropic: {
+						...msg.providerOptions?.anthropic,
+						cacheControl
+					}
+				}
+			};
+			if (Array.isArray(msg.content) && msg.content.length > 0) {
+				const content = [...msg.content];
+				const lastPart = content[content.length - 1];
+				content[content.length - 1] = {
+					...lastPart,
+					providerOptions: {
+						...lastPart.providerOptions,
+						anthropic: {
+							...lastPart.providerOptions?.anthropic,
+							cacheControl
+						}
+					}
+				};
+				return {
+					...msg,
+					content
+				};
+			}
+			return msg;
+		};
+		let lastSystemIdx = -1;
+		for (let i = prompt.length - 1; i >= 0; i--) if (prompt[i].role === "system") {
+			lastSystemIdx = i;
+			break;
+		}
+		if (lastSystemIdx >= 0) prompt[lastSystemIdx] = addCacheToMessage(prompt[lastSystemIdx]);
+		const lastIdx = prompt.length - 1;
+		if (scope === "conversation" && lastIdx >= 0 && lastIdx !== lastSystemIdx) prompt[lastIdx] = addCacheToMessage(prompt[lastIdx]);
+		return {
+			...params,
+			prompt
+		};
+	}
+});
+function runAnthropicRequestLevel(modelId, level) {
+	return runCatalogThinkingLevel(`anthropic/${modelId}`, level);
+}
+/**
+* Middleware that maps the session thinking level onto Anthropic extended
+* thinking. Returns `undefined` (no middleware) when the level is `off` or the
+* model doesn't support thinking, preserving the previous request shape.
+*
+* - Adaptive-era models (Sonnet 4.6+/Opus 4.6+): `thinking: adaptive` plus
+*   `output_config.effort`, the closest effort the model accepts.
+* - Budget-era models (Claude 3.7 – Opus 4.5): `thinking: enabled` with a
+*   `budget_tokens` value derived from the level.
+*/
+function createAnthropicThinkingMiddleware(modelId, thinkingLevel) {
+	if (!thinkingLevel || thinkingLevel === "off") return void 0;
+	const capability = getAnthropicThinkingCapability(modelId);
+	const level = runAnthropicRequestLevel(modelId, thinkingLevel);
+	if (level === "off") return void 0;
+	return {
+		specificationVersion: "v3",
+		transformParams: async ({ params }) => {
+			const anthropicOptions = params.providerOptions?.anthropic ?? {};
+			if (anthropicOptions.thinking !== void 0 || anthropicOptions.effort !== void 0) return params;
+			delete params.temperature;
+			delete params.topP;
+			delete params.topK;
+			params.providerOptions = {
+				...params.providerOptions,
+				anthropic: {
+					...anthropicOptions,
+					...capability === "adaptive" ? {
+						thinking: {
+							type: "adaptive",
+							display: "summarized"
+						},
+						effort: level
+					} : { thinking: {
+						type: "enabled",
+						budgetTokens: ANTHROPIC_THINKING_BUDGET_TOKENS[level]
+					} }
+				}
+			};
+			return params;
+		}
+	};
+}
+/**
+* Build a fetch function that handles Anthropic OAuth.
+* Preserves non-auth headers from init (critical for gateway auth header to survive
+* when used with the gateway). Strips `authorization` and `x-api-key`.
+*/
+function buildAnthropicOAuthFetch(opts = {}) {
+	return (async (url, init) => {
+		const storage = opts.authStorage ?? getAuthStorage();
+		storage.reload();
+		if (storage.get("anthropic")?.type === "api_key") throw new Error("Anthropic API key credential is configured, but OAuth is required.");
+		const accessToken = await storage.getApiKey("anthropic");
+		if (!accessToken) throw new ProviderAuthRequiredError("Not logged in to Anthropic.");
+		const headers = new Headers(url instanceof Request ? url.headers : void 0);
+		if (init?.headers) new Headers(init.headers).forEach((value, key) => headers.set(key, value));
+		headers.delete("authorization");
+		headers.delete("x-api-key");
+		headers.set("Authorization", `Bearer ${accessToken}`);
+		const requestBetas = (headers.get("anthropic-beta") ?? "").split(",").map((beta) => beta.trim()).filter(Boolean);
+		headers.set("anthropic-beta", Array.from(/* @__PURE__ */ new Set([...OAUTH_REQUIRED_BETAS, ...requestBetas])).join(","));
+		headers.set("anthropic-version", "2023-06-01");
+		try {
+			return await fetch(url, {
+				...init,
+				headers
+			});
+		} catch (error) {
+			if (error && typeof error === "object") Object.assign(error, { requestUrl: url instanceof URL ? url.toString() : typeof url === "string" ? url : url.url });
+			throw error;
+		}
+	});
+}
+/**
+* Creates an Anthropic model using Claude Max OAuth authentication
+* Uses OAuth tokens from AuthStorage (auto-refreshes when needed)
+*/
+function opencodeClaudeMaxProvider(modelId = "claude-sonnet-4-20250514", options) {
+	const headers = options?.headers;
+	const thinkingMiddleware = createAnthropicThinkingMiddleware(modelId, options?.thinkingLevel);
+	const middleware = [
+		claudeCodeMiddleware,
+		createPromptCacheMiddleware(options?.promptCacheScope ?? "conversation"),
+		...thinkingMiddleware ? [thinkingMiddleware] : []
+	];
+	if (process.env.NODE_ENV === "test" || process.env.VITEST) return wrapLanguageModel({
+		model: createAnthropic({
+			apiKey: "test-api-key",
+			headers
+		})(modelId),
+		middleware
+	});
+	return wrapLanguageModel({
+		model: createAnthropic({
+			apiKey: "oauth-placeholder",
+			headers,
+			fetch: buildAnthropicOAuthFetch({ authStorage: options?.authStorage })
+		})(modelId),
+		middleware
+	});
+}
+//#endregion
+export { buildAnthropicOAuthFetch, claudeCodeMiddleware, createAnthropicThinkingMiddleware, createPromptCacheMiddleware, getAuthStorage, opencodeClaudeMaxProvider, setAuthStorage };
+
+//# sourceMappingURL=claude-max.js.map

package/dist/providers/claude-max.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude-max.js","names":[],"sources":["../../src/providers/claude-max.ts"],"sourcesContent":["/**\n * Claude Max OAuth Provider\n *\n * Uses OAuth tokens from AuthStorage to authenticate with Claude Max plan.\n * The OAuth endpoint requires a specific system message to be present.\n */\n\nimport { createAnthropic } from '@ai-sdk/anthropic';\nimport type { MastraModelConfig } from '@mastra/core/llm';\nimport { wrapLanguageModel } from 'ai';\nimport type { LanguageModelMiddleware } from 'ai';\nimport { ProviderAuthRequiredError } from '../auth/provider-auth-error.js';\nimport { AuthStorage } from '../auth/storage.js';\nimport type { CredentialStore } from '../auth/types.js';\nimport { runCatalogThinkingLevel } from '../thinking-catalog.js';\nimport { ANTHROPIC_PROMPT_CACHE_TTL } from './anthropic-prompt-cache.js';\nimport type { AnthropicPromptCacheScope } from './anthropic-prompt-cache.js';\nimport { ANTHROPIC_THINKING_BUDGET_TOKENS, getAnthropicThinkingCapability } from './anthropic-thinking.js';\nimport type { ThinkingLevel } from './openai-codex.js';\n\n// Required for Claude Max plan OAuth - the endpoint checks for this system message\nconst claudeCodeIdentity = \"You are Claude Code, Anthropic's official CLI for Claude.\";\n\n// Betas required for Claude Max plan OAuth. Merged with (not replacing) any\n// betas the AI SDK already set on the request — e.g. the SDK adds\n// `server-side-fallback-2026-06-01` when `providerOptions.anthropic.fallbacks`\n// is configured; dropping it makes the API reject the `fallbacks` body field\n// with \"Extra inputs are not permitted\".\nconst OAUTH_REQUIRED_BETAS = [\n  'oauth-2025-04-20',\n  'claude-code-20250219',\n  'interleaved-thinking-2025-05-14',\n  'fine-grained-tool-streaming-2025-05-14',\n];\n\n// Singleton auth storage instance\nlet authStorageInstance: AuthStorage | null = null;\n\n/**\n * Get or create the shared AuthStorage instance\n */\nexport function getAuthStorage(): AuthStorage {\n  if (!authStorageInstance) {\n    authStorageInstance = new AuthStorage();\n  }\n  return authStorageInstance;\n}\n\n/**\n * Set a custom AuthStorage instance (useful for TUI integration)\n */\nexport function setAuthStorage(storage: AuthStorage | undefined): void {\n  authStorageInstance = storage ?? null;\n}\n\n/**\n * Middleware that injects the Claude Code identity system message\n * Required for Claude Max OAuth authentication\n */\nexport const claudeCodeMiddleware: LanguageModelMiddleware = {\n  specificationVersion: 'v3',\n  transformParams: async ({ params }) => {\n    // Prepend the Claude Code identity as the first system message\n    const systemMessage = {\n      role: 'system' as const,\n      content: claudeCodeIdentity,\n    };\n\n    if (params.temperature) {\n      delete params.topP;\n    }\n\n    return {\n      ...params,\n      prompt: [systemMessage, ...params.prompt],\n    };\n  },\n};\n\n/**\n * Prompt caching middleware for Anthropic\n *\n * Adds cache breakpoints at strategic locations:\n * 1. Last system message (end of static instructions + dynamic memory)\n * 2. Most recent user/assistant message (conversation context), `conversation` scope only\n *\n * This allows Anthropic to cache:\n * - System prompts and instructions (rarely change)\n * - Conversation history up to the last message\n */\nexport const createPromptCacheMiddleware = (scope: AnthropicPromptCacheScope): LanguageModelMiddleware => ({\n  specificationVersion: 'v3',\n  transformParams: async ({ params }) => {\n    const prompt = [...params.prompt];\n\n    const cacheControl = {\n      type: 'ephemeral' as const,\n      ttl: scope === 'conversation' ? ANTHROPIC_PROMPT_CACHE_TTL : ('5m' as const),\n    };\n\n    // Helper to add cache control to a message's last content part\n    const addCacheToMessage = (msg: any) => {\n      // For system messages with string content\n      if (typeof msg.content === 'string') {\n        return {\n          ...msg,\n          providerOptions: {\n            ...msg.providerOptions,\n            anthropic: { ...msg.providerOptions?.anthropic, cacheControl },\n          },\n        };\n      }\n\n      // For messages with array content, add to last part\n      if (Array.isArray(msg.content) && msg.content.length > 0) {\n        const content = [...msg.content];\n        const lastPart = content[content.length - 1];\n        content[content.length - 1] = {\n          ...lastPart,\n          providerOptions: {\n            ...lastPart.providerOptions,\n            anthropic: { ...lastPart.providerOptions?.anthropic, cacheControl },\n          },\n        };\n        return { ...msg, content };\n      }\n\n      return msg;\n    };\n\n    // Find the last system message index\n    let lastSystemIdx = -1;\n    for (let i = prompt.length - 1; i >= 0; i--) {\n      if ((prompt[i] as any).role === 'system') {\n        lastSystemIdx = i;\n        break;\n      }\n    }\n\n    // Add cache breakpoint to last system message\n    if (lastSystemIdx >= 0) {\n      prompt[lastSystemIdx] = addCacheToMessage(prompt[lastSystemIdx]);\n    }\n\n    // Add cache breakpoint to the most recent message (last in array)\n    const lastIdx = prompt.length - 1;\n    if (scope === 'conversation' && lastIdx >= 0 && lastIdx !== lastSystemIdx) {\n      prompt[lastIdx] = addCacheToMessage(prompt[lastIdx]);\n    }\n\n    return { ...params, prompt };\n  },\n});\n\nfunction runAnthropicRequestLevel(modelId: string, level: ThinkingLevel): ThinkingLevel {\n  return runCatalogThinkingLevel(`anthropic/${modelId}`, level);\n}\n\n/**\n * Middleware that maps the session thinking level onto Anthropic extended\n * thinking. Returns `undefined` (no middleware) when the level is `off` or the\n * model doesn't support thinking, preserving the previous request shape.\n *\n * - Adaptive-era models (Sonnet 4.6+/Opus 4.6+): `thinking: adaptive` plus\n *   `output_config.effort`, the closest effort the model accepts.\n * - Budget-era models (Claude 3.7 – Opus 4.5): `thinking: enabled` with a\n *   `budget_tokens` value derived from the level.\n */\nexport function createAnthropicThinkingMiddleware(\n  modelId: string,\n  thinkingLevel?: ThinkingLevel,\n): LanguageModelMiddleware | undefined {\n  if (!thinkingLevel || thinkingLevel === 'off') return undefined;\n  const capability = getAnthropicThinkingCapability(modelId);\n  const level = runAnthropicRequestLevel(modelId, thinkingLevel);\n  if (level === 'off') return undefined;\n\n  return {\n    specificationVersion: 'v3',\n    transformParams: async ({ params }) => {\n      const anthropicOptions = (params.providerOptions?.anthropic ?? {}) as Record<string, unknown>;\n      // Don't override explicit per-request thinking configuration.\n      if (anthropicOptions.thinking !== undefined || anthropicOptions.effort !== undefined) {\n        return params;\n      }\n\n      // Anthropic rejects sampling parameters when thinking is enabled.\n      delete params.temperature;\n      delete params.topP;\n      delete params.topK;\n\n      params.providerOptions = {\n        ...params.providerOptions,\n        anthropic: {\n          ...anthropicOptions,\n          ...(capability === 'adaptive'\n            ? { thinking: { type: 'adaptive', display: 'summarized' }, effort: level }\n            : { thinking: { type: 'enabled', budgetTokens: ANTHROPIC_THINKING_BUDGET_TOKENS[level] } }),\n        },\n      } as typeof params.providerOptions;\n\n      return params;\n    },\n  };\n}\n\n/**\n * Build a fetch function that handles Anthropic OAuth.\n * Preserves non-auth headers from init (critical for gateway auth header to survive\n * when used with the gateway). Strips `authorization` and `x-api-key`.\n */\nexport function buildAnthropicOAuthFetch(opts: { authStorage?: CredentialStore } = {}): typeof fetch {\n  return (async (url: string | URL | Request, init?: Parameters<typeof fetch>[1]) => {\n    const storage = opts.authStorage ?? getAuthStorage();\n    storage.reload();\n\n    const storedCred = storage.get('anthropic');\n    if (storedCred?.type === 'api_key') {\n      throw new Error('Anthropic API key credential is configured, but OAuth is required.');\n    }\n\n    const accessToken = await storage.getApiKey('anthropic');\n    if (!accessToken) {\n      throw new ProviderAuthRequiredError('Not logged in to Anthropic.');\n    }\n\n    // Preserve Request headers and let explicit init headers override them.\n    const headers = new Headers(url instanceof Request ? url.headers : undefined);\n    if (init?.headers) new Headers(init.headers).forEach((value, key) => headers.set(key, value));\n    headers.delete('authorization');\n    headers.delete('x-api-key');\n    headers.set('Authorization', `Bearer ${accessToken}`);\n    const requestBetas = (headers.get('anthropic-beta') ?? '')\n      .split(',')\n      .map(beta => beta.trim())\n      .filter(Boolean);\n    headers.set('anthropic-beta', Array.from(new Set([...OAUTH_REQUIRED_BETAS, ...requestBetas])).join(','));\n    headers.set('anthropic-version', '2023-06-01');\n\n    try {\n      return await fetch(url, { ...init, headers });\n    } catch (error) {\n      if (error && typeof error === 'object') {\n        Object.assign(error as Record<string, unknown>, {\n          requestUrl: url instanceof URL ? url.toString() : typeof url === 'string' ? url : url.url,\n        });\n      }\n      throw error;\n    }\n  }) as typeof fetch;\n}\n\n/**\n * Creates an Anthropic model using Claude Max OAuth authentication\n * Uses OAuth tokens from AuthStorage (auto-refreshes when needed)\n */\nexport function opencodeClaudeMaxProvider(\n  modelId: string = 'claude-sonnet-4-20250514',\n  options?: {\n    headers?: Record<string, string>;\n    authStorage?: CredentialStore;\n    thinkingLevel?: ThinkingLevel;\n    promptCacheScope?: AnthropicPromptCacheScope;\n  },\n): MastraModelConfig {\n  const headers = options?.headers;\n  const thinkingMiddleware = createAnthropicThinkingMiddleware(modelId, options?.thinkingLevel);\n  const middleware = [\n    claudeCodeMiddleware,\n    createPromptCacheMiddleware(options?.promptCacheScope ?? 'conversation'),\n    ...(thinkingMiddleware ? [thinkingMiddleware] : []),\n  ];\n\n  // Test environment: use API key\n  if (process.env.NODE_ENV === 'test' || process.env.VITEST) {\n    const anthropic = createAnthropic({\n      apiKey: 'test-api-key',\n      headers,\n    });\n    return wrapLanguageModel({\n      model: anthropic(modelId),\n      middleware,\n    });\n  }\n\n  const anthropic = createAnthropic({\n    apiKey: 'oauth-placeholder',\n    headers,\n    fetch: buildAnthropicOAuthFetch({ authStorage: options?.authStorage }) as any,\n  });\n\n  // Wrap with middleware to inject Claude Code identity and enable prompt caching\n  return wrapLanguageModel({\n    model: anthropic(modelId),\n    middleware,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;AAqBA,MAAM,qBAAqB;AAO3B,MAAM,uBAAuB;CAC3B;CACA;CACA;CACA;AACF;AAGA,IAAI,sBAA0C;;;;AAK9C,SAAgB,iBAA8B;CAC5C,IAAI,CAAC,qBACH,sBAAsB,IAAI,YAAY;CAExC,OAAO;AACT;;;;AAKA,SAAgB,eAAe,SAAwC;CACrE,sBAAsB,WAAW;AACnC;;;;;AAMA,MAAa,uBAAgD;CAC3D,sBAAsB;CACtB,iBAAiB,OAAO,EAAE,aAAa;EAErC,MAAM,gBAAgB;GACpB,MAAM;GACN,SAAS;EACX;EAEA,IAAI,OAAO,aACT,OAAO,OAAO;EAGhB,OAAO;GACL,GAAG;GACH,QAAQ,CAAC,eAAe,GAAG,OAAO,MAAM;EAC1C;CACF;AACF;;;;;;;;;;;;AAaA,MAAa,+BAA+B,WAA+D;CACzG,sBAAsB;CACtB,iBAAiB,OAAO,EAAE,aAAa;EACrC,MAAM,SAAS,CAAC,GAAG,OAAO,MAAM;EAEhC,MAAM,eAAe;GACnB,MAAM;GACN,KAAK,UAAU,iBAAA,OAA+C;EAChE;EAGA,MAAM,qBAAqB,QAAa;GAEtC,IAAI,OAAO,IAAI,YAAY,UACzB,OAAO;IACL,GAAG;IACH,iBAAiB;KACf,GAAG,IAAI;KACP,WAAW;MAAE,GAAG,IAAI,iBAAiB;MAAW;KAAa;IAC/D;GACF;GAIF,IAAI,MAAM,QAAQ,IAAI,OAAO,KAAK,IAAI,QAAQ,SAAS,GAAG;IACxD,MAAM,UAAU,CAAC,GAAG,IAAI,OAAO;IAC/B,MAAM,WAAW,QAAQ,QAAQ,SAAS;IAC1C,QAAQ,QAAQ,SAAS,KAAK;KAC5B,GAAG;KACH,iBAAiB;MACf,GAAG,SAAS;MACZ,WAAW;OAAE,GAAG,SAAS,iBAAiB;OAAW;MAAa;KACpE;IACF;IACA,OAAO;KAAE,GAAG;KAAK;IAAQ;GAC3B;GAEA,OAAO;EACT;EAGA,IAAI,gBAAgB;EACpB,KAAK,IAAI,IAAI,OAAO,SAAS,GAAG,KAAK,GAAG,KACtC,IAAK,OAAO,EAAE,CAAS,SAAS,UAAU;GACxC,gBAAgB;GAChB;EACF;EAIF,IAAI,iBAAiB,GACnB,OAAO,iBAAiB,kBAAkB,OAAO,cAAc;EAIjE,MAAM,UAAU,OAAO,SAAS;EAChC,IAAI,UAAU,kBAAkB,WAAW,KAAK,YAAY,eAC1D,OAAO,WAAW,kBAAkB,OAAO,QAAQ;EAGrD,OAAO;GAAE,GAAG;GAAQ;EAAO;CAC7B;AACF;AAEA,SAAS,yBAAyB,SAAiB,OAAqC;CACtF,OAAO,wBAAwB,aAAa,WAAW,KAAK;AAC9D;;;;;;;;;;;AAYA,SAAgB,kCACd,SACA,eACqC;CACrC,IAAI,CAAC,iBAAiB,kBAAkB,OAAO,OAAO,KAAA;CACtD,MAAM,aAAa,+BAA+B,OAAO;CACzD,MAAM,QAAQ,yBAAyB,SAAS,aAAa;CAC7D,IAAI,UAAU,OAAO,OAAO,KAAA;CAE5B,OAAO;EACL,sBAAsB;EACtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,MAAM,mBAAoB,OAAO,iBAAiB,aAAa,CAAC;GAEhE,IAAI,iBAAiB,aAAa,KAAA,KAAa,iBAAiB,WAAW,KAAA,GACzE,OAAO;GAIT,OAAO,OAAO;GACd,OAAO,OAAO;GACd,OAAO,OAAO;GAEd,OAAO,kBAAkB;IACvB,GAAG,OAAO;IACV,WAAW;KACT,GAAG;KACH,GAAI,eAAe,aACf;MAAE,UAAU;OAAE,MAAM;OAAY,SAAS;MAAa;MAAG,QAAQ;KAAM,IACvE,EAAE,UAAU;MAAE,MAAM;MAAW,cAAc,iCAAiC;KAAO,EAAE;IAC7F;GACF;GAEA,OAAO;EACT;CACF;AACF;;;;;;AAOA,SAAgB,yBAAyB,OAA0C,CAAC,GAAiB;CACnG,QAAQ,OAAO,KAA6B,SAAuC;EACjF,MAAM,UAAU,KAAK,eAAe,eAAe;EACnD,QAAQ,OAAO;EAGf,IADmB,QAAQ,IAAI,WAClB,CAAC,EAAE,SAAS,WACvB,MAAM,IAAI,MAAM,oEAAoE;EAGtF,MAAM,cAAc,MAAM,QAAQ,UAAU,WAAW;EACvD,IAAI,CAAC,aACH,MAAM,IAAI,0BAA0B,6BAA6B;EAInE,MAAM,UAAU,IAAI,QAAQ,eAAe,UAAU,IAAI,UAAU,KAAA,CAAS;EAC5E,IAAI,MAAM,SAAS,IAAI,QAAQ,KAAK,OAAO,CAAC,CAAC,SAAS,OAAO,QAAQ,QAAQ,IAAI,KAAK,KAAK,CAAC;EAC5F,QAAQ,OAAO,eAAe;EAC9B,QAAQ,OAAO,WAAW;EAC1B,QAAQ,IAAI,iBAAiB,UAAU,aAAa;EACpD,MAAM,gBAAgB,QAAQ,IAAI,gBAAgB,KAAK,GAAA,CACpD,MAAM,GAAG,CAAC,CACV,KAAI,SAAQ,KAAK,KAAK,CAAC,CAAC,CACxB,OAAO,OAAO;EACjB,QAAQ,IAAI,kBAAkB,MAAM,qBAAK,IAAI,IAAI,CAAC,GAAG,sBAAsB,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC;EACvG,QAAQ,IAAI,qBAAqB,YAAY;EAE7C,IAAI;GACF,OAAO,MAAM,MAAM,KAAK;IAAE,GAAG;IAAM;GAAQ,CAAC;EAC9C,SAAS,OAAO;GACd,IAAI,SAAS,OAAO,UAAU,UAC5B,OAAO,OAAO,OAAkC,EAC9C,YAAY,eAAe,MAAM,IAAI,SAAS,IAAI,OAAO,QAAQ,WAAW,MAAM,IAAI,IACxF,CAAC;GAEH,MAAM;EACR;CACF;AACF;;;;;AAMA,SAAgB,0BACd,UAAkB,4BAClB,SAMmB;CACnB,MAAM,UAAU,SAAS;CACzB,MAAM,qBAAqB,kCAAkC,SAAS,SAAS,aAAa;CAC5F,MAAM,aAAa;EACjB;EACA,4BAA4B,SAAS,oBAAoB,cAAc;EACvE,GAAI,qBAAqB,CAAC,kBAAkB,IAAI,CAAC;CACnD;CAGA,IAAI,QAAQ,IAAI,aAAa,UAAU,QAAQ,IAAI,QAKjD,OAAO,kBAAkB;EACvB,OALgB,gBAAgB;GAChC,QAAQ;GACR;EACF,CAEiB,CAAC,CAAC,OAAO;EACxB;CACF,CAAC;CAUH,OAAO,kBAAkB;EACvB,OARgB,gBAAgB;GAChC,QAAQ;GACR;GACA,OAAO,yBAAyB,EAAE,aAAa,SAAS,YAAY,CAAC;EACvE,CAIiB,CAAC,CAAC,OAAO;EACxB;CACF,CAAC;AACH"}

package/dist/providers/github-copilot-gateway.d.ts

@@ -0,0 +1,35 @@
+import { MastraModelGateway } from '@mastra/core/llm';
+import type { GatewayAuthRequest, GatewayAuthResult, GatewayLanguageModel, ProviderConfig } from '@mastra/core/llm';
+import type { CredentialStore } from '../auth/types.js';
+export declare const GITHUB_COPILOT_GATEWAY_ID = "github-copilot";
+/**
+ * Standalone GitHub Copilot gateway.
+ *
+ * models.dev excludes Copilot from the shared registry, so Mastra Code supplies
+ * the provider itself. It is its own gateway so catalog ids are the canonical
+ * `github-copilot/<model>` rather than `mastracode/github-copilot/<model>`.
+ */
+export declare class GitHubCopilotGateway extends MastraModelGateway {
+    #private;
+    readonly id = "github-copilot";
+    readonly name = "GitHub Copilot";
+    constructor(options?: {
+        credentialStore?: CredentialStore;
+    });
+    fetchProviders(): Promise<Record<string, ProviderConfig>>;
+    buildUrl(_modelId: string): string | undefined;
+    getApiKey(_modelId: string): Promise<string>;
+    resolveAuth(_request: GatewayAuthRequest): GatewayAuthResult | undefined;
+    resolveLanguageModel(args: {
+        modelId: string;
+        providerId: string;
+        apiKey: string;
+        headers?: Record<string, string>;
+        transport?: any;
+        responsesWebSocket?: any;
+    }): GatewayLanguageModel;
+}
+export declare function createGitHubCopilotGateway(options?: {
+    credentialStore?: CredentialStore;
+}): GitHubCopilotGateway;
+//# sourceMappingURL=github-copilot-gateway.d.ts.map

package/dist/providers/github-copilot-gateway.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"github-copilot-gateway.d.ts","sourceRoot":"","sources":["../../src/providers/github-copilot-gateway.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,KAAK,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACpH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAGxD,eAAO,MAAM,yBAAyB,mBAAmB,CAAC;AAE1D;;;;;;GAMG;AACH,qBAAa,oBAAqB,SAAQ,kBAAkB;;IAC1D,QAAQ,CAAC,EAAE,oBAA6B;IACxC,QAAQ,CAAC,IAAI,oBAAoB;gBAGrB,OAAO,GAAE;QAAE,eAAe,CAAC,EAAE,eAAe,CAAA;KAAO;IASzD,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IAkB/D,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAIxC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAIlD,WAAW,CAAC,QAAQ,EAAE,kBAAkB,GAAG,iBAAiB,GAAG,SAAS;IAOxE,oBAAoB,CAAC,IAAI,EAAE;QACzB,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,SAAS,CAAC,EAAE,GAAG,CAAC;QAChB,kBAAkB,CAAC,EAAE,GAAG,CAAC;KAC1B,GAAG,oBAAoB;CAMzB;AAED,wBAAgB,0BAA0B,CAAC,OAAO,CAAC,EAAE;IAAE,eAAe,CAAC,EAAE,eAAe,CAAA;CAAE,GAAG,oBAAoB,CAEhH"}

package/dist/providers/github-copilot-gateway.js

@@ -0,0 +1,61 @@
+import { getAuthStorage, getCopilotModelCatalog, githubCopilotProvider } from "./github-copilot.js";
+import { MastraModelGateway } from "@mastra/core/llm";
+//#region src/providers/github-copilot-gateway.ts
+const GITHUB_COPILOT_GATEWAY_ID = "github-copilot";
+/**
+* Standalone GitHub Copilot gateway.
+*
+* models.dev excludes Copilot from the shared registry, so Mastra Code supplies
+* the provider itself. It is its own gateway so catalog ids are the canonical
+* `github-copilot/<model>` rather than `mastracode/github-copilot/<model>`.
+*/
+var GitHubCopilotGateway = class extends MastraModelGateway {
+	id = GITHUB_COPILOT_GATEWAY_ID;
+	name = "GitHub Copilot";
+	#credentials;
+	constructor(options = {}) {
+		super();
+		this.#credentials = options.credentialStore;
+	}
+	get #store() {
+		return this.#credentials ?? getAuthStorage();
+	}
+	async fetchProviders() {
+		try {
+			const models = await getCopilotModelCatalog({ authStorage: this.#store });
+			return { [GITHUB_COPILOT_GATEWAY_ID]: {
+				name: "GitHub Copilot",
+				apiKeyEnvVar: "",
+				apiKeyHeader: "Authorization",
+				gateway: this.id,
+				models: models.map((model) => model.id)
+			} };
+		} catch (error) {
+			console.warn("Failed to load GitHub Copilot model catalog:", error);
+			return {};
+		}
+	}
+	buildUrl(_modelId) {}
+	async getApiKey(_modelId) {
+		return "";
+	}
+	resolveAuth(_request) {
+		return this.#store.get("github-copilot")?.type === "oauth" ? {
+			bearerToken: "oauth",
+			source: "gateway"
+		} : void 0;
+	}
+	resolveLanguageModel(args) {
+		return githubCopilotProvider(args.modelId, {
+			headers: args.headers,
+			authStorage: this.#store
+		});
+	}
+};
+function createGitHubCopilotGateway(options) {
+	return new GitHubCopilotGateway(options);
+}
+//#endregion
+export { GITHUB_COPILOT_GATEWAY_ID, GitHubCopilotGateway, createGitHubCopilotGateway };
+
+//# sourceMappingURL=github-copilot-gateway.js.map

package/dist/providers/github-copilot-gateway.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"github-copilot-gateway.js","names":["#credentials","#store"],"sources":["../../src/providers/github-copilot-gateway.ts"],"sourcesContent":["import { MastraModelGateway } from '@mastra/core/llm';\nimport type { GatewayAuthRequest, GatewayAuthResult, GatewayLanguageModel, ProviderConfig } from '@mastra/core/llm';\nimport type { CredentialStore } from '../auth/types.js';\nimport { getAuthStorage, getCopilotModelCatalog, githubCopilotProvider } from './github-copilot.js';\n\nexport const GITHUB_COPILOT_GATEWAY_ID = 'github-copilot';\n\n/**\n * Standalone GitHub Copilot gateway.\n *\n * models.dev excludes Copilot from the shared registry, so Mastra Code supplies\n * the provider itself. It is its own gateway so catalog ids are the canonical\n * `github-copilot/<model>` rather than `mastracode/github-copilot/<model>`.\n */\nexport class GitHubCopilotGateway extends MastraModelGateway {\n  readonly id = GITHUB_COPILOT_GATEWAY_ID;\n  readonly name = 'GitHub Copilot';\n  readonly #credentials?: CredentialStore;\n\n  constructor(options: { credentialStore?: CredentialStore } = {}) {\n    super();\n    this.#credentials = options.credentialStore;\n  }\n\n  get #store(): CredentialStore {\n    return this.#credentials ?? getAuthStorage();\n  }\n\n  async fetchProviders(): Promise<Record<string, ProviderConfig>> {\n    try {\n      const models = await getCopilotModelCatalog({ authStorage: this.#store });\n      return {\n        [GITHUB_COPILOT_GATEWAY_ID]: {\n          name: 'GitHub Copilot',\n          apiKeyEnvVar: '',\n          apiKeyHeader: 'Authorization',\n          gateway: this.id,\n          models: models.map(model => model.id),\n        },\n      };\n    } catch (error) {\n      console.warn('Failed to load GitHub Copilot model catalog:', error);\n      return {};\n    }\n  }\n\n  buildUrl(_modelId: string): string | undefined {\n    return undefined;\n  }\n\n  async getApiKey(_modelId: string): Promise<string> {\n    return '';\n  }\n\n  resolveAuth(_request: GatewayAuthRequest): GatewayAuthResult | undefined {\n    // The OAuth token is attached per request by the Copilot fetch wrapper.\n    return this.#store.get(GITHUB_COPILOT_GATEWAY_ID)?.type === 'oauth'\n      ? { bearerToken: 'oauth', source: 'gateway' }\n      : undefined;\n  }\n\n  resolveLanguageModel(args: {\n    modelId: string;\n    providerId: string;\n    apiKey: string;\n    headers?: Record<string, string>;\n    transport?: any;\n    responsesWebSocket?: any;\n  }): GatewayLanguageModel {\n    return githubCopilotProvider(args.modelId, {\n      headers: args.headers,\n      authStorage: this.#store,\n    }) as unknown as GatewayLanguageModel;\n  }\n}\n\nexport function createGitHubCopilotGateway(options?: { credentialStore?: CredentialStore }): GitHubCopilotGateway {\n  return new GitHubCopilotGateway(options);\n}\n"],"mappings":";;;AAKA,MAAa,4BAA4B;;;;;;;;AASzC,IAAa,uBAAb,cAA0C,mBAAmB;CAC3D,KAAc;CACd,OAAgB;CAChB;CAEA,YAAY,UAAiD,CAAC,GAAG;EAC/D,MAAM;EACN,KAAKA,eAAe,QAAQ;CAC9B;CAEA,IAAIC,SAA0B;EAC5B,OAAO,KAAKD,gBAAgB,eAAe;CAC7C;CAEA,MAAM,iBAA0D;EAC9D,IAAI;GACF,MAAM,SAAS,MAAM,uBAAuB,EAAE,aAAa,KAAKC,OAAO,CAAC;GACxE,OAAO,GACJ,4BAA4B;IAC3B,MAAM;IACN,cAAc;IACd,cAAc;IACd,SAAS,KAAK;IACd,QAAQ,OAAO,KAAI,UAAS,MAAM,EAAE;GACtC,EACF;EACF,SAAS,OAAO;GACd,QAAQ,KAAK,gDAAgD,KAAK;GAClE,OAAO,CAAC;EACV;CACF;CAEA,SAAS,UAAsC,CAE/C;CAEA,MAAM,UAAU,UAAmC;EACjD,OAAO;CACT;CAEA,YAAY,UAA6D;EAEvE,OAAO,KAAKA,OAAO,IAAA,gBAA6B,CAAC,EAAE,SAAS,UACxD;GAAE,aAAa;GAAS,QAAQ;EAAU,IAC1C,KAAA;CACN;CAEA,qBAAqB,MAOI;EACvB,OAAO,sBAAsB,KAAK,SAAS;GACzC,SAAS,KAAK;GACd,aAAa,KAAKA;EACpB,CAAC;CACH;AACF;AAEA,SAAgB,2BAA2B,SAAuE;CAChH,OAAO,IAAI,qBAAqB,OAAO;AACzC"}

package/dist/providers/github-copilot.d.ts

@@ -0,0 +1,60 @@
+/**
+ * GitHub Copilot OAuth Provider
+ *
+ * Uses OAuth tokens from AuthStorage to authenticate with GitHub Copilot's chat API.
+ * The Copilot API speaks an OpenAI-compatible chat format, so we plug
+ * `@ai-sdk/openai-compatible` into Copilot's API URL and use a custom fetch to inject
+ * the bearer token and Copilot-specific headers.
+ *
+ * Inspired by:
+ *   - opencode: https://github.com/anomalyco/opencode/blob/dev/packages/opencode/src/plugin/github-copilot/copilot.ts
+ *   - pi-mono:  https://github.com/badlogic/pi-mono/blob/main/packages/ai/src/utils/oauth/github-copilot.ts
+ */
+import type { MastraModelConfig } from '@mastra/core/llm';
+import type { CopilotModelEntry } from '../auth/providers/github-copilot.js';
+import { AuthStorage } from '../auth/storage.js';
+import type { CredentialStore } from '../auth/types.js';
+/** Get or create the shared AuthStorage instance. */
+export declare function getAuthStorage(): AuthStorage;
+/** Set a custom AuthStorage instance (useful for tests / TUI integration). */
+export declare function setAuthStorage(storage: AuthStorage | undefined): void;
+/**
+ * Build a fetch wrapper that authenticates with GitHub Copilot OAuth.
+ *
+ * - Injects the short-lived Copilot bearer token (auto-refreshed by AuthStorage).
+ * - Adds the VS Code-like Copilot headers required by the API.
+ * - Rewrites the request URL onto the per-token API base when `rewriteUrl` is true.
+ */
+export declare function buildGitHubCopilotOAuthFetch(opts?: {
+    authStorage?: CredentialStore;
+    rewriteUrl?: boolean;
+}): typeof fetch;
+/**
+ * Creates a model that talks to GitHub Copilot using OAuth credentials.
+ *
+ * Copilot's `/chat/completions` endpoint is OpenAI-compatible, but GitHub Copilot
+ * is not OpenAI. Use the generic OpenAI-compatible adapter with Copilot's base URL
+ * instead of the OpenAI provider plus URL rewriting.
+ */
+export declare function githubCopilotProvider(modelId?: string, options?: {
+    headers?: Record<string, string>;
+    authStorage?: CredentialStore;
+}): MastraModelConfig;
+/** Reset the in-process Copilot catalog cache (test seam, also useful after logout). */
+export declare function clearCopilotCatalogCache(): void;
+/**
+ * Return the user's currently-available Copilot models.
+ *
+ * - Returns `[]` when the user is not logged in to GitHub Copilot.
+ * - Returns the cached list when a recent fetch succeeded.
+ * - On the cache-miss / expired path, fetches `/models` with a 5s timeout, filters
+ *   to picker-enabled and non-policy-disabled models, then caches for 10 minutes.
+ * - On fetch failure, returns a small hard-coded fallback (so packs still work
+ *   offline) and caches that for 1 minute to avoid hammering the API.
+ *
+ * Concurrent calls during a fetch share the inflight promise.
+ */
+export declare function getCopilotModelCatalog(opts?: {
+    authStorage?: CredentialStore;
+}): Promise<CopilotModelEntry[]>;
+//# sourceMappingURL=github-copilot.d.ts.map

package/dist/providers/github-copilot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"github-copilot.d.ts","sourceRoot":"","sources":["../../src/providers/github-copilot.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAGH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAO1D,OAAO,KAAK,EAAE,iBAAiB,EAA4B,MAAM,qCAAqC,CAAC;AACvG,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,KAAK,EAAE,eAAe,EAAmB,MAAM,kBAAkB,CAAC;AAOzE,qDAAqD;AACrD,wBAAgB,cAAc,IAAI,WAAW,CAK5C;AAED,8EAA8E;AAC9E,wBAAgB,cAAc,CAAC,OAAO,EAAE,WAAW,GAAG,SAAS,GAAG,IAAI,CAErE;AAqED;;;;;;GAMG;AACH,wBAAgB,4BAA4B,CAC1C,IAAI,GAAE;IAAE,WAAW,CAAC,EAAE,eAAe,CAAC;IAAC,UAAU,CAAC,EAAE,OAAO,CAAA;CAAO,GACjE,OAAO,KAAK,CA+Ed;AAwED;;;;;;GAMG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,GAAE,MAAkB,EAC3B,OAAO,CAAC,EAAE;IAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAAC,WAAW,CAAC,EAAE,eAAe,CAAA;CAAE,GAC5E,iBAAiB,CAiBnB;AAsDD,wFAAwF;AACxF,wBAAgB,wBAAwB,IAAI,IAAI,CAG/C;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,sBAAsB,CAC1C,IAAI,GAAE;IAAE,WAAW,CAAC,EAAE,eAAe,CAAA;CAAO,GAC3C,OAAO,CAAC,iBAAiB,EAAE,CAAC,CA2F9B"}