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

package/dist/providers/github-copilot.js

@@ -0,0 +1,318 @@
+import { COPILOT_HEADERS, fetchCopilotModels, getGitHubCopilotBaseUrl } from "../auth/providers/github-copilot.js";
+import { AuthStorage } from "../auth/storage.js";
+import { ProviderAuthRequiredError } from "../auth/provider-auth-error.js";
+import { wrapLanguageModel } from "ai";
+import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
+import { GoogleSchemaCompatLayer, applyCompatLayer } from "@mastra/schema-compat";
+//#region src/providers/github-copilot.ts
+/**
+* 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
+*/
+const COPILOT_PROVIDER_ID = "github-copilot";
+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 tests / TUI integration). */
+function setAuthStorage(storage) {
+	authStorageInstance = storage ?? null;
+}
+/**
+* Heuristic: did this request come from the agent (e.g. tool result follow-ups) rather
+* than a fresh user turn? Mirrors opencode's `isAgent` logic — Copilot bills these
+* differently via the `x-initiator` header.
+*/
+function detectIsAgent(body) {
+	if (!body || typeof body !== "object") return false;
+	const obj = body;
+	const messages = obj.messages;
+	if (Array.isArray(messages) && messages.length > 0) {
+		const last = messages[messages.length - 1];
+		if (last?.role && last.role !== "user") return true;
+		if (Array.isArray(last?.content)) {
+			if (last.content.some((part) => part && typeof part === "object" && part.type === "tool_result")) return true;
+		}
+	}
+	const input = obj.input;
+	if (Array.isArray(input) && input.length > 0) {
+		const last = input[input.length - 1];
+		if (last?.role && last.role !== "user") return true;
+	}
+	return false;
+}
+/** Detect image/vision content in a request body. */
+function detectIsVision(body) {
+	if (!body || typeof body !== "object") return false;
+	const obj = body;
+	const matchPart = (part) => {
+		if (!part || typeof part !== "object") return false;
+		const t = part.type;
+		return t === "image" || t === "image_url" || t === "input_image";
+	};
+	const messages = obj.messages;
+	if (Array.isArray(messages)) return messages.some((msg) => msg && typeof msg === "object" && Array.isArray(msg.content) && msg.content.some(matchPart));
+	const input = obj.input;
+	if (Array.isArray(input)) return input.some((item) => item && typeof item === "object" && Array.isArray(item.content) && item.content.some(matchPart));
+	return false;
+}
+/**
+* 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.
+*/
+function buildGitHubCopilotOAuthFetch(opts = {}) {
+	return (async (url, init) => {
+		const { headers: initHeaders, ...requestInit } = init ?? {};
+		const request = new Request(url, requestInit);
+		const storage = opts.authStorage ?? getAuthStorage();
+		storage.reload();
+		const cred = storage.get(COPILOT_PROVIDER_ID);
+		if (!cred || cred.type !== "oauth") throw new ProviderAuthRequiredError("Not logged in to GitHub Copilot.");
+		let accessToken;
+		let activeCred;
+		if (storage.getOAuthCredential) {
+			activeCred = await storage.getOAuthCredential(COPILOT_PROVIDER_ID);
+			accessToken = activeCred?.access;
+		} else {
+			accessToken = await storage.getApiKey(COPILOT_PROVIDER_ID);
+			storage.reload();
+			const reloaded = storage.get(COPILOT_PROVIDER_ID);
+			activeCred = reloaded?.type === "oauth" ? {
+				...reloaded,
+				access: accessToken ?? reloaded.access
+			} : void 0;
+		}
+		if (!accessToken || !activeCred) throw new ProviderAuthRequiredError("Failed to refresh the GitHub Copilot token.");
+		const enterpriseUrl = activeCred.enterpriseUrl;
+		let parsedBody;
+		try {
+			const body = await request.clone().text();
+			parsedBody = body ? JSON.parse(body) : void 0;
+		} catch {
+			parsedBody = void 0;
+		}
+		const isAgent = detectIsAgent(parsedBody);
+		const isVision = detectIsVision(parsedBody);
+		const headers = new Headers(url instanceof Request ? url.headers : void 0);
+		if (initHeaders) new Headers(initHeaders).forEach((value, key) => headers.set(key, value));
+		headers.delete("authorization");
+		headers.delete("x-api-key");
+		headers.set("Authorization", `Bearer ${accessToken}`);
+		headers.set("x-initiator", isAgent ? "agent" : "user");
+		headers.set("Openai-Intent", "conversation-edits");
+		if (isVision) headers.set("Copilot-Vision-Request", "true");
+		for (const [key, value] of Object.entries(COPILOT_HEADERS)) if (!headers.has(key)) headers.set(key, value);
+		const finalUrl = opts.rewriteUrl !== false ? rewriteToCopilotBase(request, accessToken, enterpriseUrl) : new URL(request.url);
+		const body = request.method === "GET" || request.method === "HEAD" ? void 0 : request.body;
+		const finalRequest = new Request(finalUrl, {
+			method: request.method,
+			headers,
+			body,
+			signal: request.signal,
+			redirect: request.redirect,
+			integrity: request.integrity,
+			...body ? { duplex: "half" } : {}
+		});
+		try {
+			return await fetch(finalRequest);
+		} catch (error) {
+			if (error && typeof error === "object") Object.assign(error, { requestUrl: finalUrl.toString() });
+			throw error;
+		}
+	});
+}
+function rewriteToCopilotBase(url, token, enterpriseDomain) {
+	const original = url instanceof URL ? url : new URL(typeof url === "string" ? url : url.url);
+	const base = new URL(getGitHubCopilotBaseUrl(token, enterpriseDomain));
+	const pathname = original.pathname.replace(/^\/v1(\/|$)/, "/");
+	return new URL(`${pathname}${original.search}`, base);
+}
+function isGeminiModel(modelId) {
+	return modelId.startsWith("gemini-");
+}
+function applyGeminiSchemaCompatToTools(modelId, tools) {
+	if (!Array.isArray(tools)) return tools;
+	const compatLayer = new GoogleSchemaCompatLayer({
+		provider: COPILOT_PROVIDER_ID,
+		modelId,
+		supportsStructuredOutputs: false
+	});
+	return tools.map((tool) => {
+		if (!tool || typeof tool !== "object" || tool.type !== "function") return tool;
+		const functionTool = tool;
+		if (!functionTool.inputSchema) return tool;
+		return {
+			...functionTool,
+			inputSchema: applyCompatLayer({
+				schema: functionTool.inputSchema,
+				compatLayers: [compatLayer],
+				mode: "aiSdkSchema"
+			}).jsonSchema
+		};
+	});
+}
+/** Middleware that prevents sending parameters Copilot's endpoint rejects. */
+function createCopilotMiddleware(modelId) {
+	return {
+		specificationVersion: "v3",
+		transformParams: async ({ params }) => {
+			if (params.temperature !== void 0 && params.temperature !== null) delete params.topP;
+			if (isGeminiModel(modelId)) params.tools = applyGeminiSchemaCompatToTools(modelId, params.tools);
+			return params;
+		}
+	};
+}
+/**
+* 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.
+*/
+function githubCopilotProvider(modelId = "gpt-4.1", options) {
+	const headers = options?.headers;
+	return wrapLanguageModel({
+		model: createOpenAICompatible({
+			name: COPILOT_PROVIDER_ID,
+			baseURL: "https://api.githubcopilot.com",
+			apiKey: process.env.NODE_ENV === "test" || process.env.VITEST ? "test-api-key" : "oauth-placeholder",
+			headers,
+			fetch: process.env.NODE_ENV === "test" || process.env.VITEST ? void 0 : buildGitHubCopilotOAuthFetch({
+				rewriteUrl: false,
+				authStorage: options?.authStorage
+			})
+		}).chatModel(modelId),
+		middleware: [createCopilotMiddleware(modelId)]
+	});
+}
+/**
+* Hard-coded fallback advertised when the live `/models` request fails (network
+* down, expired token, etc.). Keep this conservative because the live catalog is
+* the source of truth for the user's currently-enabled Copilot models.
+*
+* Available across all paid Copilot tiers and free of premium-request charges.
+*/
+const COPILOT_FALLBACK_MODELS = [{
+	id: "gpt-4.1",
+	name: "GPT-4.1",
+	vendor: "OpenAI",
+	supportedEndpoints: ["/chat/completions"],
+	isAnthropicShaped: false,
+	supportsVision: true,
+	supportsToolCalls: true
+}];
+const CATALOG_TTL_MS = 600 * 1e3;
+const CATALOG_FAILURE_TTL_MS = 60 * 1e3;
+const CATALOG_FETCH_TIMEOUT_MS = 5e3;
+const catalogCache = /* @__PURE__ */ new Map();
+const inflightFetches = /* @__PURE__ */ new Map();
+/**
+* Resolve the account identity for a credential read through `getApiKey`, which
+* may await a refresh with a rotation landing inside it. The registry is
+* consulted on both sides of that await, and only an id that survived it names
+* the account that produced the token: an id that moved describes whichever
+* account is active *now*, so trusting it would file this fetch's models under
+* the wrong account and serve them to it. Unidentified is the safe answer — the
+* caller then skips the TTL cache, exactly as it does when the store exposes no
+* registry at all. (`AuthStorage` never needs this: its snapshots name their
+* account.)
+*/
+function readStableAccountId(storage, idBeforeToken) {
+	const idAfterToken = storage.getActiveAccount?.(COPILOT_PROVIDER_ID)?.id;
+	return idBeforeToken !== void 0 && idBeforeToken === idAfterToken ? idAfterToken : void 0;
+}
+/** Reset the in-process Copilot catalog cache (test seam, also useful after logout). */
+function clearCopilotCatalogCache() {
+	catalogCache.clear();
+	inflightFetches.clear();
+}
+/**
+* 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.
+*/
+async function getCopilotModelCatalog(opts = {}) {
+	const storage = opts.authStorage ?? getAuthStorage();
+	let accessToken;
+	let accountInstanceId;
+	let enterpriseUrl;
+	if (storage.getOAuthCredential) {
+		const credential = await storage.getOAuthCredential(COPILOT_PROVIDER_ID);
+		if (!credential || credential.type !== "oauth") return [];
+		accessToken = credential.access;
+		accountInstanceId = credential.accountInstanceId;
+		enterpriseUrl = credential.enterpriseUrl;
+	} else {
+		const idBeforeToken = storage.getActiveAccount?.(COPILOT_PROVIDER_ID)?.id;
+		accessToken = await storage.getApiKey(COPILOT_PROVIDER_ID);
+		if (!accessToken) return [];
+		accountInstanceId = readStableAccountId(storage, idBeforeToken);
+		const stored = storage.get(COPILOT_PROVIDER_ID);
+		if (stored?.type === "oauth") enterpriseUrl = stored.enterpriseUrl;
+	}
+	const baseUrl = getGitHubCopilotBaseUrl(accessToken, enterpriseUrl);
+	const credentialKey = accountInstanceId === void 0 ? void 0 : `${accountInstanceId}\0${baseUrl}`;
+	const dedupeKey = credentialKey;
+	const now = Date.now();
+	const cached = credentialKey ? catalogCache.get(credentialKey) : void 0;
+	if (cached && now - cached.fetchedAt < cached.ttl) return cached.models;
+	const existingFetch = dedupeKey === void 0 ? void 0 : inflightFetches.get(dedupeKey);
+	if (existingFetch) return existingFetch;
+	const fetchPromise = (async () => {
+		try {
+			const controller = new AbortController();
+			const timer = setTimeout(() => controller.abort(), CATALOG_FETCH_TIMEOUT_MS);
+			try {
+				const models = await fetchCopilotModels({
+					baseUrl,
+					bearerToken: accessToken,
+					signal: controller.signal
+				});
+				if (credentialKey) catalogCache.set(credentialKey, {
+					fetchedAt: Date.now(),
+					ttl: CATALOG_TTL_MS,
+					models
+				});
+				return models;
+			} finally {
+				clearTimeout(timer);
+			}
+		} catch (error) {
+			if (credentialKey) catalogCache.set(credentialKey, {
+				fetchedAt: Date.now(),
+				ttl: CATALOG_FAILURE_TTL_MS,
+				models: COPILOT_FALLBACK_MODELS
+			});
+			console.warn("Failed to fetch live GitHub Copilot models, using fallback list:", error instanceof Error ? error.message : error);
+			return COPILOT_FALLBACK_MODELS;
+		} finally {
+			if (dedupeKey !== void 0) inflightFetches.delete(dedupeKey);
+		}
+	})();
+	if (dedupeKey !== void 0) inflightFetches.set(dedupeKey, fetchPromise);
+	return fetchPromise;
+}
+//#endregion
+export { buildGitHubCopilotOAuthFetch, clearCopilotCatalogCache, getAuthStorage, getCopilotModelCatalog, githubCopilotProvider, setAuthStorage };
+
+//# sourceMappingURL=github-copilot.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"github-copilot.js","names":[],"sources":["../../src/providers/github-copilot.ts"],"sourcesContent":["/**\n * GitHub Copilot OAuth Provider\n *\n * Uses OAuth tokens from AuthStorage to authenticate with GitHub Copilot's chat API.\n * The Copilot API speaks an OpenAI-compatible chat format, so we plug\n * `@ai-sdk/openai-compatible` into Copilot's API URL and use a custom fetch to inject\n * the bearer token and Copilot-specific headers.\n *\n * Inspired by:\n *   - opencode: https://github.com/anomalyco/opencode/blob/dev/packages/opencode/src/plugin/github-copilot/copilot.ts\n *   - pi-mono:  https://github.com/badlogic/pi-mono/blob/main/packages/ai/src/utils/oauth/github-copilot.ts\n */\n\nimport { createOpenAICompatible } from '@ai-sdk/openai-compatible';\nimport type { MastraModelConfig } from '@mastra/core/llm';\nimport type { JSONSchema7 } from '@mastra/schema-compat';\nimport { applyCompatLayer, GoogleSchemaCompatLayer } from '@mastra/schema-compat';\nimport { wrapLanguageModel } from 'ai';\nimport type { LanguageModelMiddleware } from 'ai';\nimport { ProviderAuthRequiredError } from '../auth/provider-auth-error.js';\nimport { COPILOT_HEADERS, fetchCopilotModels, getGitHubCopilotBaseUrl } from '../auth/providers/github-copilot.js';\nimport type { CopilotModelEntry, GitHubCopilotCredentials } from '../auth/providers/github-copilot.js';\nimport { AuthStorage } from '../auth/storage.js';\nimport type { CredentialStore, OAuthCredential } from '../auth/types.js';\n\nconst COPILOT_PROVIDER_ID = 'github-copilot';\n\n// Singleton auth storage instance (shared with claude-max.ts / openai-codex.ts when not overridden).\nlet authStorageInstance: AuthStorage | null = null;\n\n/** Get or create the shared AuthStorage instance. */\nexport function getAuthStorage(): AuthStorage {\n  if (!authStorageInstance) {\n    authStorageInstance = new AuthStorage();\n  }\n  return authStorageInstance;\n}\n\n/** Set a custom AuthStorage instance (useful for tests / TUI integration). */\nexport function setAuthStorage(storage: AuthStorage | undefined): void {\n  authStorageInstance = storage ?? null;\n}\n\n/**\n * Heuristic: did this request come from the agent (e.g. tool result follow-ups) rather\n * than a fresh user turn? Mirrors opencode's `isAgent` logic — Copilot bills these\n * differently via the `x-initiator` header.\n */\nfunction detectIsAgent(body: unknown): boolean {\n  if (!body || typeof body !== 'object') return false;\n  const obj = body as Record<string, unknown>;\n\n  const messages = obj.messages;\n  if (Array.isArray(messages) && messages.length > 0) {\n    const last = messages[messages.length - 1] as { role?: string; content?: unknown };\n    if (last?.role && last.role !== 'user') return true;\n    if (Array.isArray(last?.content)) {\n      // If the last user turn carries any tool_result parts, treat it as an agent turn.\n      const hasToolResult = last.content.some(\n        (part: unknown) => part && typeof part === 'object' && (part as { type?: string }).type === 'tool_result',\n      );\n      if (hasToolResult) return true;\n    }\n  }\n\n  const input = obj.input;\n  if (Array.isArray(input) && input.length > 0) {\n    const last = input[input.length - 1] as { role?: string };\n    if (last?.role && last.role !== 'user') return true;\n  }\n\n  return false;\n}\n\n/** Detect image/vision content in a request body. */\nfunction detectIsVision(body: unknown): boolean {\n  if (!body || typeof body !== 'object') return false;\n  const obj = body as Record<string, unknown>;\n\n  const matchPart = (part: unknown): boolean => {\n    if (!part || typeof part !== 'object') return false;\n    const t = (part as { type?: string }).type;\n    return t === 'image' || t === 'image_url' || t === 'input_image';\n  };\n\n  const messages = obj.messages;\n  if (Array.isArray(messages)) {\n    return messages.some(\n      (msg: unknown) =>\n        msg &&\n        typeof msg === 'object' &&\n        Array.isArray((msg as { content?: unknown }).content) &&\n        ((msg as { content: unknown[] }).content as unknown[]).some(matchPart),\n    );\n  }\n\n  const input = obj.input;\n  if (Array.isArray(input)) {\n    return input.some(\n      (item: unknown) =>\n        item &&\n        typeof item === 'object' &&\n        Array.isArray((item as { content?: unknown }).content) &&\n        ((item as { content: unknown[] }).content as unknown[]).some(matchPart),\n    );\n  }\n\n  return false;\n}\n\n/**\n * Build a fetch wrapper that authenticates with GitHub Copilot OAuth.\n *\n * - Injects the short-lived Copilot bearer token (auto-refreshed by AuthStorage).\n * - Adds the VS Code-like Copilot headers required by the API.\n * - Rewrites the request URL onto the per-token API base when `rewriteUrl` is true.\n */\nexport function buildGitHubCopilotOAuthFetch(\n  opts: { authStorage?: CredentialStore; rewriteUrl?: boolean } = {},\n): typeof fetch {\n  return (async (url: string | URL | Request, init?: Parameters<typeof fetch>[1]) => {\n    const { headers: initHeaders, ...requestInit } = init ?? {};\n    const request = new Request(url, requestInit);\n    const storage = opts.authStorage ?? getAuthStorage();\n    storage.reload();\n\n    const cred = storage.get(COPILOT_PROVIDER_ID);\n    if (!cred || cred.type !== 'oauth') {\n      throw new ProviderAuthRequiredError('Not logged in to GitHub Copilot.');\n    }\n\n    let accessToken: string | undefined;\n    let activeCred: OAuthCredential | undefined;\n    if (storage.getOAuthCredential) {\n      activeCred = await storage.getOAuthCredential(COPILOT_PROVIDER_ID);\n      accessToken = activeCred?.access;\n    } else {\n      accessToken = await storage.getApiKey(COPILOT_PROVIDER_ID);\n      storage.reload();\n      const reloaded = storage.get(COPILOT_PROVIDER_ID);\n      activeCred = reloaded?.type === 'oauth' ? { ...reloaded, access: accessToken ?? reloaded.access } : undefined;\n    }\n    if (!accessToken || !activeCred) {\n      throw new ProviderAuthRequiredError('Failed to refresh the GitHub Copilot token.');\n    }\n    const enterpriseUrl = (activeCred as GitHubCopilotCredentials).enterpriseUrl;\n\n    let parsedBody: unknown;\n    try {\n      const body = await request.clone().text();\n      parsedBody = body ? JSON.parse(body) : undefined;\n    } catch {\n      parsedBody = undefined;\n    }\n    const isAgent = detectIsAgent(parsedBody);\n    const isVision = detectIsVision(parsedBody);\n\n    const headers = new Headers(url instanceof Request ? url.headers : undefined);\n    if (initHeaders) new Headers(initHeaders).forEach((value, key) => headers.set(key, value));\n    headers.delete('authorization');\n    headers.delete('x-api-key');\n    headers.set('Authorization', `Bearer ${accessToken}`);\n    headers.set('x-initiator', isAgent ? 'agent' : 'user');\n    headers.set('Openai-Intent', 'conversation-edits');\n    if (isVision) {\n      headers.set('Copilot-Vision-Request', 'true');\n    }\n    for (const [key, value] of Object.entries(COPILOT_HEADERS)) {\n      // Only set if caller didn't already provide it (allow overrides for tests).\n      if (!headers.has(key)) {\n        headers.set(key, value);\n      }\n    }\n\n    const finalUrl =\n      opts.rewriteUrl !== false ? rewriteToCopilotBase(request, accessToken, enterpriseUrl) : new URL(request.url);\n\n    const body = request.method === 'GET' || request.method === 'HEAD' ? undefined : request.body;\n    const finalRequest = new Request(finalUrl, {\n      method: request.method,\n      headers,\n      body,\n      signal: request.signal,\n      redirect: request.redirect,\n      integrity: request.integrity,\n      ...(body ? ({ duplex: 'half' } as RequestInit) : {}),\n    });\n    try {\n      return await fetch(finalRequest);\n    } catch (error) {\n      if (error && typeof error === 'object') {\n        Object.assign(error as Record<string, unknown>, {\n          requestUrl: finalUrl.toString(),\n        });\n      }\n      throw error;\n    }\n  }) as typeof fetch;\n}\n\nfunction rewriteToCopilotBase(url: string | URL | Request, token: string, enterpriseDomain?: string): URL {\n  const original = url instanceof URL ? url : new URL(typeof url === 'string' ? url : (url as Request).url);\n  const base = new URL(getGitHubCopilotBaseUrl(token, enterpriseDomain));\n  // Copilot's OpenAI-compatible API serves endpoints at the root of the base host\n  // (`/chat/completions`, `/responses`, `/models`, ...) — not under a `/v1/` prefix\n  // like api.openai.com does. The @ai-sdk/openai default baseURL is\n  // `https://api.openai.com/v1`, so the SDK builds requests like\n  // `https://api.openai.com/v1/chat/completions`. Strip the leading `/v1` segment\n  // when rewriting onto the Copilot base or Copilot will return 404 Not Found.\n  const pathname = original.pathname.replace(/^\\/v1(\\/|$)/, '/');\n  return new URL(`${pathname}${original.search}`, base);\n}\n\nfunction isGeminiModel(modelId: string): boolean {\n  return modelId.startsWith('gemini-');\n}\n\nfunction applyGeminiSchemaCompatToTools(modelId: string, tools: unknown): unknown {\n  if (!Array.isArray(tools)) {\n    return tools;\n  }\n\n  const compatLayer = new GoogleSchemaCompatLayer({\n    provider: COPILOT_PROVIDER_ID,\n    modelId,\n    supportsStructuredOutputs: false,\n  });\n\n  return tools.map(tool => {\n    if (!tool || typeof tool !== 'object' || (tool as { type?: unknown }).type !== 'function') {\n      return tool;\n    }\n\n    const functionTool = tool as { inputSchema?: JSONSchema7 };\n    if (!functionTool.inputSchema) {\n      return tool;\n    }\n\n    return {\n      ...functionTool,\n      inputSchema: applyCompatLayer({\n        schema: functionTool.inputSchema,\n        compatLayers: [compatLayer],\n        mode: 'aiSdkSchema',\n      }).jsonSchema as JSONSchema7,\n    };\n  });\n}\n\n/** Middleware that prevents sending parameters Copilot's endpoint rejects. */\nfunction createCopilotMiddleware(modelId: string): LanguageModelMiddleware {\n  return {\n    specificationVersion: 'v3',\n    transformParams: async ({ params }) => {\n      if (params.temperature !== undefined && params.temperature !== null) {\n        delete params.topP;\n      }\n\n      if (isGeminiModel(modelId)) {\n        (params as { tools?: unknown }).tools = applyGeminiSchemaCompatToTools(\n          modelId,\n          (params as { tools?: unknown }).tools,\n        );\n      }\n\n      return params;\n    },\n  };\n}\n\n/**\n * Creates a model that talks to GitHub Copilot using OAuth credentials.\n *\n * Copilot's `/chat/completions` endpoint is OpenAI-compatible, but GitHub Copilot\n * is not OpenAI. Use the generic OpenAI-compatible adapter with Copilot's base URL\n * instead of the OpenAI provider plus URL rewriting.\n */\nexport function githubCopilotProvider(\n  modelId: string = 'gpt-4.1',\n  options?: { headers?: Record<string, string>; authStorage?: CredentialStore },\n): MastraModelConfig {\n  const headers = options?.headers;\n  const copilot = createOpenAICompatible({\n    name: COPILOT_PROVIDER_ID,\n    baseURL: 'https://api.githubcopilot.com',\n    apiKey: process.env.NODE_ENV === 'test' || process.env.VITEST ? 'test-api-key' : 'oauth-placeholder',\n    headers,\n    fetch:\n      process.env.NODE_ENV === 'test' || process.env.VITEST\n        ? undefined\n        : (buildGitHubCopilotOAuthFetch({ rewriteUrl: false, authStorage: options?.authStorage }) as any),\n  });\n\n  return wrapLanguageModel({\n    model: copilot.chatModel(modelId),\n    middleware: [createCopilotMiddleware(modelId)],\n  });\n}\n\n// ---------------------------------------------------------------------------\n// Live model catalog\n// ---------------------------------------------------------------------------\n\n/**\n * Hard-coded fallback advertised when the live `/models` request fails (network\n * down, expired token, etc.). Keep this conservative because the live catalog is\n * the source of truth for the user's currently-enabled Copilot models.\n *\n * Available across all paid Copilot tiers and free of premium-request charges.\n */\nconst COPILOT_FALLBACK_MODELS: CopilotModelEntry[] = [\n  {\n    id: 'gpt-4.1',\n    name: 'GPT-4.1',\n    vendor: 'OpenAI',\n    supportedEndpoints: ['/chat/completions'],\n    isAnthropicShaped: false,\n    supportsVision: true,\n    supportsToolCalls: true,\n  },\n];\n\nconst CATALOG_TTL_MS = 10 * 60 * 1000;\nconst CATALOG_FAILURE_TTL_MS = 60 * 1000;\nconst CATALOG_FETCH_TIMEOUT_MS = 5_000;\n\ninterface CatalogCacheEntry {\n  fetchedAt: number;\n  ttl: number;\n  models: CopilotModelEntry[];\n}\n\nconst catalogCache = new Map<string, CatalogCacheEntry>();\nconst inflightFetches = new Map<string, Promise<CopilotModelEntry[]>>();\n\n/**\n * Resolve the account identity for a credential read through `getApiKey`, which\n * may await a refresh with a rotation landing inside it. The registry is\n * consulted on both sides of that await, and only an id that survived it names\n * the account that produced the token: an id that moved describes whichever\n * account is active *now*, so trusting it would file this fetch's models under\n * the wrong account and serve them to it. Unidentified is the safe answer — the\n * caller then skips the TTL cache, exactly as it does when the store exposes no\n * registry at all. (`AuthStorage` never needs this: its snapshots name their\n * account.)\n */\nfunction readStableAccountId(storage: CredentialStore, idBeforeToken: string | undefined): string | undefined {\n  const idAfterToken = storage.getActiveAccount?.(COPILOT_PROVIDER_ID)?.id;\n  return idBeforeToken !== undefined && idBeforeToken === idAfterToken ? idAfterToken : undefined;\n}\n\n/** Reset the in-process Copilot catalog cache (test seam, also useful after logout). */\nexport function clearCopilotCatalogCache(): void {\n  catalogCache.clear();\n  inflightFetches.clear();\n}\n\n/**\n * Return the user's currently-available Copilot models.\n *\n * - Returns `[]` when the user is not logged in to GitHub Copilot.\n * - Returns the cached list when a recent fetch succeeded.\n * - On the cache-miss / expired path, fetches `/models` with a 5s timeout, filters\n *   to picker-enabled and non-policy-disabled models, then caches for 10 minutes.\n * - On fetch failure, returns a small hard-coded fallback (so packs still work\n *   offline) and caches that for 1 minute to avoid hammering the API.\n *\n * Concurrent calls during a fetch share the inflight promise.\n */\nexport async function getCopilotModelCatalog(\n  opts: { authStorage?: CredentialStore } = {},\n): Promise<CopilotModelEntry[]> {\n  const storage = opts.authStorage ?? getAuthStorage();\n\n  // Resolve one coherent credential snapshot before consulting the cache so a\n  // token can never be paired with another account's enterprise endpoint.\n  // `getOAuthCredential` is optional (deployed stores have no local registry),\n  // so fall back to the always-present `getApiKey`, taking any enterprise\n  // endpoint from the same slot via `get()`.\n  let accessToken: string | undefined;\n  let accountInstanceId: string | undefined;\n  let enterpriseUrl: string | undefined;\n  if (storage.getOAuthCredential) {\n    const credential = await storage.getOAuthCredential(COPILOT_PROVIDER_ID);\n    if (!credential || credential.type !== 'oauth') return [];\n    accessToken = credential.access;\n    // A snapshot that names its account is coherent by construction — the store\n    // read the identity and the token in the same breath. One that does not\n    // cannot be attributed to a registry entry we then read afterwards, because\n    // a rotation may have landed in between; leave it unnamed so the caller\n    // skips the TTL cache instead of guessing. (`AuthStorage` always stamps.)\n    accountInstanceId = credential.accountInstanceId;\n    enterpriseUrl = (credential as GitHubCopilotCredentials).enterpriseUrl;\n  } else {\n    // This store republishes its registry but not the OAuth snapshot API, so the\n    // identity has to come from the registry. `getApiKey` may await a refresh and\n    // a rotation can land inside it, so bracket the fetch: only an id that held\n    // still names the account that produced this token.\n    const idBeforeToken = storage.getActiveAccount?.(COPILOT_PROVIDER_ID)?.id;\n    accessToken = await storage.getApiKey(COPILOT_PROVIDER_ID);\n    if (!accessToken) return [];\n    accountInstanceId = readStableAccountId(storage, idBeforeToken);\n    const stored = storage.get(COPILOT_PROVIDER_ID);\n    if (stored?.type === 'oauth') {\n      enterpriseUrl = (stored as GitHubCopilotCredentials).enterpriseUrl;\n    }\n    // The id is non-secret, and keying per account is what keeps entitlements\n    // (per account, per `github-copilot-catalog.test.ts`) from leaking across a\n    // rotation. Deriving the key from the token is what `cba11389e7` removed.\n  }\n  const baseUrl = getGitHubCopilotBaseUrl(accessToken, enterpriseUrl);\n  // No identity means the store cannot name the account it just served, so the\n  // TTL cache cannot be keyed per account — one entry would outlive the request\n  // and serve those models to every other account of the store. The in-flight\n  // map is no safer: two callers that resolved *different* accounts of an\n  // unnamed store would otherwise collide on one key and share whichever\n  // catalog was fetched first. So both caches are identified-only, and an\n  // unnamed store pays one `/models` request per call.\n  const credentialKey = accountInstanceId === undefined ? undefined : `${accountInstanceId}\\0${baseUrl}`;\n  const dedupeKey = credentialKey;\n\n  const now = Date.now();\n  const cached = credentialKey ? catalogCache.get(credentialKey) : undefined;\n  if (cached && now - cached.fetchedAt < cached.ttl) return cached.models;\n\n  const existingFetch = dedupeKey === undefined ? undefined : inflightFetches.get(dedupeKey);\n  if (existingFetch) return existingFetch;\n\n  const fetchPromise = (async (): Promise<CopilotModelEntry[]> => {\n    try {\n      const controller = new AbortController();\n      const timer = setTimeout(() => controller.abort(), CATALOG_FETCH_TIMEOUT_MS);\n      try {\n        const models = await fetchCopilotModels({\n          baseUrl,\n          bearerToken: accessToken,\n          signal: controller.signal,\n        });\n        if (credentialKey) catalogCache.set(credentialKey, { fetchedAt: Date.now(), ttl: CATALOG_TTL_MS, models });\n        return models;\n      } finally {\n        clearTimeout(timer);\n      }\n    } catch (error) {\n      if (credentialKey) {\n        catalogCache.set(credentialKey, {\n          fetchedAt: Date.now(),\n          ttl: CATALOG_FAILURE_TTL_MS,\n          models: COPILOT_FALLBACK_MODELS,\n        });\n      }\n      console.warn(\n        'Failed to fetch live GitHub Copilot models, using fallback list:',\n        error instanceof Error ? error.message : error,\n      );\n      return COPILOT_FALLBACK_MODELS;\n    } finally {\n      if (dedupeKey !== undefined) inflightFetches.delete(dedupeKey);\n    }\n  })();\n  if (dedupeKey !== undefined) inflightFetches.set(dedupeKey, fetchPromise);\n  return fetchPromise;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAyBA,MAAM,sBAAsB;AAG5B,IAAI,sBAA0C;;AAG9C,SAAgB,iBAA8B;CAC5C,IAAI,CAAC,qBACH,sBAAsB,IAAI,YAAY;CAExC,OAAO;AACT;;AAGA,SAAgB,eAAe,SAAwC;CACrE,sBAAsB,WAAW;AACnC;;;;;;AAOA,SAAS,cAAc,MAAwB;CAC7C,IAAI,CAAC,QAAQ,OAAO,SAAS,UAAU,OAAO;CAC9C,MAAM,MAAM;CAEZ,MAAM,WAAW,IAAI;CACrB,IAAI,MAAM,QAAQ,QAAQ,KAAK,SAAS,SAAS,GAAG;EAClD,MAAM,OAAO,SAAS,SAAS,SAAS;EACxC,IAAI,MAAM,QAAQ,KAAK,SAAS,QAAQ,OAAO;EAC/C,IAAI,MAAM,QAAQ,MAAM,OAAO,GAEP;OAAA,KAAK,QAAQ,MAChC,SAAkB,QAAQ,OAAO,SAAS,YAAa,KAA2B,SAAS,aAE9E,GAAG,OAAO;EAAA;CAE9B;CAEA,MAAM,QAAQ,IAAI;CAClB,IAAI,MAAM,QAAQ,KAAK,KAAK,MAAM,SAAS,GAAG;EAC5C,MAAM,OAAO,MAAM,MAAM,SAAS;EAClC,IAAI,MAAM,QAAQ,KAAK,SAAS,QAAQ,OAAO;CACjD;CAEA,OAAO;AACT;;AAGA,SAAS,eAAe,MAAwB;CAC9C,IAAI,CAAC,QAAQ,OAAO,SAAS,UAAU,OAAO;CAC9C,MAAM,MAAM;CAEZ,MAAM,aAAa,SAA2B;EAC5C,IAAI,CAAC,QAAQ,OAAO,SAAS,UAAU,OAAO;EAC9C,MAAM,IAAK,KAA2B;EACtC,OAAO,MAAM,WAAW,MAAM,eAAe,MAAM;CACrD;CAEA,MAAM,WAAW,IAAI;CACrB,IAAI,MAAM,QAAQ,QAAQ,GACxB,OAAO,SAAS,MACb,QACC,OACA,OAAO,QAAQ,YACf,MAAM,QAAS,IAA8B,OAAO,KAClD,IAA+B,QAAsB,KAAK,SAAS,CACzE;CAGF,MAAM,QAAQ,IAAI;CAClB,IAAI,MAAM,QAAQ,KAAK,GACrB,OAAO,MAAM,MACV,SACC,QACA,OAAO,SAAS,YAChB,MAAM,QAAS,KAA+B,OAAO,KACnD,KAAgC,QAAsB,KAAK,SAAS,CAC1E;CAGF,OAAO;AACT;;;;;;;;AASA,SAAgB,6BACd,OAAgE,CAAC,GACnD;CACd,QAAQ,OAAO,KAA6B,SAAuC;EACjF,MAAM,EAAE,SAAS,aAAa,GAAG,gBAAgB,QAAQ,CAAC;EAC1D,MAAM,UAAU,IAAI,QAAQ,KAAK,WAAW;EAC5C,MAAM,UAAU,KAAK,eAAe,eAAe;EACnD,QAAQ,OAAO;EAEf,MAAM,OAAO,QAAQ,IAAI,mBAAmB;EAC5C,IAAI,CAAC,QAAQ,KAAK,SAAS,SACzB,MAAM,IAAI,0BAA0B,kCAAkC;EAGxE,IAAI;EACJ,IAAI;EACJ,IAAI,QAAQ,oBAAoB;GAC9B,aAAa,MAAM,QAAQ,mBAAmB,mBAAmB;GACjE,cAAc,YAAY;EAC5B,OAAO;GACL,cAAc,MAAM,QAAQ,UAAU,mBAAmB;GACzD,QAAQ,OAAO;GACf,MAAM,WAAW,QAAQ,IAAI,mBAAmB;GAChD,aAAa,UAAU,SAAS,UAAU;IAAE,GAAG;IAAU,QAAQ,eAAe,SAAS;GAAO,IAAI,KAAA;EACtG;EACA,IAAI,CAAC,eAAe,CAAC,YACnB,MAAM,IAAI,0BAA0B,6CAA6C;EAEnF,MAAM,gBAAiB,WAAwC;EAE/D,IAAI;EACJ,IAAI;GACF,MAAM,OAAO,MAAM,QAAQ,MAAM,CAAC,CAAC,KAAK;GACxC,aAAa,OAAO,KAAK,MAAM,IAAI,IAAI,KAAA;EACzC,QAAQ;GACN,aAAa,KAAA;EACf;EACA,MAAM,UAAU,cAAc,UAAU;EACxC,MAAM,WAAW,eAAe,UAAU;EAE1C,MAAM,UAAU,IAAI,QAAQ,eAAe,UAAU,IAAI,UAAU,KAAA,CAAS;EAC5E,IAAI,aAAa,IAAI,QAAQ,WAAW,CAAC,CAAC,SAAS,OAAO,QAAQ,QAAQ,IAAI,KAAK,KAAK,CAAC;EACzF,QAAQ,OAAO,eAAe;EAC9B,QAAQ,OAAO,WAAW;EAC1B,QAAQ,IAAI,iBAAiB,UAAU,aAAa;EACpD,QAAQ,IAAI,eAAe,UAAU,UAAU,MAAM;EACrD,QAAQ,IAAI,iBAAiB,oBAAoB;EACjD,IAAI,UACF,QAAQ,IAAI,0BAA0B,MAAM;EAE9C,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,eAAe,GAEvD,IAAI,CAAC,QAAQ,IAAI,GAAG,GAClB,QAAQ,IAAI,KAAK,KAAK;EAI1B,MAAM,WACJ,KAAK,eAAe,QAAQ,qBAAqB,SAAS,aAAa,aAAa,IAAI,IAAI,IAAI,QAAQ,GAAG;EAE7G,MAAM,OAAO,QAAQ,WAAW,SAAS,QAAQ,WAAW,SAAS,KAAA,IAAY,QAAQ;EACzF,MAAM,eAAe,IAAI,QAAQ,UAAU;GACzC,QAAQ,QAAQ;GAChB;GACA;GACA,QAAQ,QAAQ;GAChB,UAAU,QAAQ;GAClB,WAAW,QAAQ;GACnB,GAAI,OAAQ,EAAE,QAAQ,OAAO,IAAoB,CAAC;EACpD,CAAC;EACD,IAAI;GACF,OAAO,MAAM,MAAM,YAAY;EACjC,SAAS,OAAO;GACd,IAAI,SAAS,OAAO,UAAU,UAC5B,OAAO,OAAO,OAAkC,EAC9C,YAAY,SAAS,SAAS,EAChC,CAAC;GAEH,MAAM;EACR;CACF;AACF;AAEA,SAAS,qBAAqB,KAA6B,OAAe,kBAAgC;CACxG,MAAM,WAAW,eAAe,MAAM,MAAM,IAAI,IAAI,OAAO,QAAQ,WAAW,MAAO,IAAgB,GAAG;CACxG,MAAM,OAAO,IAAI,IAAI,wBAAwB,OAAO,gBAAgB,CAAC;CAOrE,MAAM,WAAW,SAAS,SAAS,QAAQ,eAAe,GAAG;CAC7D,OAAO,IAAI,IAAI,GAAG,WAAW,SAAS,UAAU,IAAI;AACtD;AAEA,SAAS,cAAc,SAA0B;CAC/C,OAAO,QAAQ,WAAW,SAAS;AACrC;AAEA,SAAS,+BAA+B,SAAiB,OAAyB;CAChF,IAAI,CAAC,MAAM,QAAQ,KAAK,GACtB,OAAO;CAGT,MAAM,cAAc,IAAI,wBAAwB;EAC9C,UAAU;EACV;EACA,2BAA2B;CAC7B,CAAC;CAED,OAAO,MAAM,KAAI,SAAQ;EACvB,IAAI,CAAC,QAAQ,OAAO,SAAS,YAAa,KAA4B,SAAS,YAC7E,OAAO;EAGT,MAAM,eAAe;EACrB,IAAI,CAAC,aAAa,aAChB,OAAO;EAGT,OAAO;GACL,GAAG;GACH,aAAa,iBAAiB;IAC5B,QAAQ,aAAa;IACrB,cAAc,CAAC,WAAW;IAC1B,MAAM;GACR,CAAC,CAAC,CAAC;EACL;CACF,CAAC;AACH;;AAGA,SAAS,wBAAwB,SAA0C;CACzE,OAAO;EACL,sBAAsB;EACtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,IAAI,OAAO,gBAAgB,KAAA,KAAa,OAAO,gBAAgB,MAC7D,OAAO,OAAO;GAGhB,IAAI,cAAc,OAAO,GACvB,OAAgC,QAAQ,+BACtC,SACC,OAA+B,KAClC;GAGF,OAAO;EACT;CACF;AACF;;;;;;;;AASA,SAAgB,sBACd,UAAkB,WAClB,SACmB;CACnB,MAAM,UAAU,SAAS;CAYzB,OAAO,kBAAkB;EACvB,OAZc,uBAAuB;GACrC,MAAM;GACN,SAAS;GACT,QAAQ,QAAQ,IAAI,aAAa,UAAU,QAAQ,IAAI,SAAS,iBAAiB;GACjF;GACA,OACE,QAAQ,IAAI,aAAa,UAAU,QAAQ,IAAI,SAC3C,KAAA,IACC,6BAA6B;IAAE,YAAY;IAAO,aAAa,SAAS;GAAY,CAAC;EAC9F,CAGe,CAAC,CAAC,UAAU,OAAO;EAChC,YAAY,CAAC,wBAAwB,OAAO,CAAC;CAC/C,CAAC;AACH;;;;;;;;AAaA,MAAM,0BAA+C,CACnD;CACE,IAAI;CACJ,MAAM;CACN,QAAQ;CACR,oBAAoB,CAAC,mBAAmB;CACxC,mBAAmB;CACnB,gBAAgB;CAChB,mBAAmB;AACrB,CACF;AAEA,MAAM,iBAAiB,MAAU;AACjC,MAAM,yBAAyB,KAAK;AACpC,MAAM,2BAA2B;AAQjC,MAAM,+BAAe,IAAI,IAA+B;AACxD,MAAM,kCAAkB,IAAI,IAA0C;;;;;;;;;;;;AAatE,SAAS,oBAAoB,SAA0B,eAAuD;CAC5G,MAAM,eAAe,QAAQ,mBAAmB,mBAAmB,CAAC,EAAE;CACtE,OAAO,kBAAkB,KAAA,KAAa,kBAAkB,eAAe,eAAe,KAAA;AACxF;;AAGA,SAAgB,2BAAiC;CAC/C,aAAa,MAAM;CACnB,gBAAgB,MAAM;AACxB;;;;;;;;;;;;;AAcA,eAAsB,uBACpB,OAA0C,CAAC,GACb;CAC9B,MAAM,UAAU,KAAK,eAAe,eAAe;CAOnD,IAAI;CACJ,IAAI;CACJ,IAAI;CACJ,IAAI,QAAQ,oBAAoB;EAC9B,MAAM,aAAa,MAAM,QAAQ,mBAAmB,mBAAmB;EACvE,IAAI,CAAC,cAAc,WAAW,SAAS,SAAS,OAAO,CAAC;EACxD,cAAc,WAAW;EAMzB,oBAAoB,WAAW;EAC/B,gBAAiB,WAAwC;CAC3D,OAAO;EAKL,MAAM,gBAAgB,QAAQ,mBAAmB,mBAAmB,CAAC,EAAE;EACvE,cAAc,MAAM,QAAQ,UAAU,mBAAmB;EACzD,IAAI,CAAC,aAAa,OAAO,CAAC;EAC1B,oBAAoB,oBAAoB,SAAS,aAAa;EAC9D,MAAM,SAAS,QAAQ,IAAI,mBAAmB;EAC9C,IAAI,QAAQ,SAAS,SACnB,gBAAiB,OAAoC;CAKzD;CACA,MAAM,UAAU,wBAAwB,aAAa,aAAa;CAQlE,MAAM,gBAAgB,sBAAsB,KAAA,IAAY,KAAA,IAAY,GAAG,kBAAkB,IAAI;CAC7F,MAAM,YAAY;CAElB,MAAM,MAAM,KAAK,IAAI;CACrB,MAAM,SAAS,gBAAgB,aAAa,IAAI,aAAa,IAAI,KAAA;CACjE,IAAI,UAAU,MAAM,OAAO,YAAY,OAAO,KAAK,OAAO,OAAO;CAEjE,MAAM,gBAAgB,cAAc,KAAA,IAAY,KAAA,IAAY,gBAAgB,IAAI,SAAS;CACzF,IAAI,eAAe,OAAO;CAE1B,MAAM,gBAAgB,YAA0C;EAC9D,IAAI;GACF,MAAM,aAAa,IAAI,gBAAgB;GACvC,MAAM,QAAQ,iBAAiB,WAAW,MAAM,GAAG,wBAAwB;GAC3E,IAAI;IACF,MAAM,SAAS,MAAM,mBAAmB;KACtC;KACA,aAAa;KACb,QAAQ,WAAW;IACrB,CAAC;IACD,IAAI,eAAe,aAAa,IAAI,eAAe;KAAE,WAAW,KAAK,IAAI;KAAG,KAAK;KAAgB;IAAO,CAAC;IACzG,OAAO;GACT,UAAU;IACR,aAAa,KAAK;GACpB;EACF,SAAS,OAAO;GACd,IAAI,eACF,aAAa,IAAI,eAAe;IAC9B,WAAW,KAAK,IAAI;IACpB,KAAK;IACL,QAAQ;GACV,CAAC;GAEH,QAAQ,KACN,oEACA,iBAAiB,QAAQ,MAAM,UAAU,KAC3C;GACA,OAAO;EACT,UAAU;GACR,IAAI,cAAc,KAAA,GAAW,gBAAgB,OAAO,SAAS;EAC/D;CACF,EAAA,CAAG;CACH,IAAI,cAAc,KAAA,GAAW,gBAAgB,IAAI,WAAW,YAAY;CACxE,OAAO;AACT"}

package/dist/providers/google-thinking.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Google (Gemini) thinking-level middleware.
+ *
+ * Maps the Mastra Code session thinking level (`off|low|medium|high|xhigh|max`)
+ * onto the thinking control each Gemini family accepts:
+ * - Gemini 3+: `thinkingConfig.thinkingLevel` (Gemini 3 Pro only accepts `low|high`).
+ * - Gemini 2.5: `thinkingConfig.thinkingBudget` (token budget).
+ * Unknown model families get no thinking config, so requests are unchanged.
+ */
+import type { LanguageModelMiddleware } from 'ai';
+import type { ActiveThinkingLevel, ThinkingLevelSetting } from '../thinking.js';
+type GeminiLevel = 'low' | 'medium' | 'high';
+type GoogleThinkingConfig = {
+    thinkingLevel: 'minimal' | GeminiLevel;
+} | {
+    thinkingBudget: number;
+};
+export declare function runGeminiThinkingLevel(modelId: string, level: ActiveThinkingLevel): ThinkingLevelSetting;
+/**
+ * Resolve the Google thinking config for a model and session level.
+ * Returns `undefined` for `off`, unset levels, and unrecognized model families.
+ */
+export declare function resolveGoogleThinkingConfig(modelId: string, level: ThinkingLevelSetting | undefined): GoogleThinkingConfig | undefined;
+/**
+ * Create middleware that injects the resolved config into
+ * `providerOptions.google.thinkingConfig`. Returns `undefined` when there is
+ * nothing to inject, so callers wrap nothing.
+ */
+export declare function createGoogleThinkingMiddleware(modelId: string, level: ThinkingLevelSetting | undefined): LanguageModelMiddleware | undefined;
+export {};
+//# sourceMappingURL=google-thinking.d.ts.map

package/dist/providers/google-thinking.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-thinking.d.ts","sourceRoot":"","sources":["../../src/providers/google-thinking.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,IAAI,CAAC;AAClD,OAAO,KAAK,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAEhF,KAAK,WAAW,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAE7C,KAAK,oBAAoB,GAAG;IAAE,aAAa,EAAE,SAAS,GAAG,WAAW,CAAA;CAAE,GAAG;IAAE,cAAc,EAAE,MAAM,CAAA;CAAE,CAAC;AAuBpG,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,mBAAmB,GAAG,oBAAoB,CAGxG;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,oBAAoB,GAAG,SAAS,GACtC,oBAAoB,GAAG,SAAS,CASlC;AAED;;;;GAIG;AACH,wBAAgB,8BAA8B,CAC5C,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,oBAAoB,GAAG,SAAS,GACtC,uBAAuB,GAAG,SAAS,CAkBrC"}

package/dist/providers/google-thinking.js

@@ -0,0 +1,67 @@
+//#region src/providers/google-thinking.ts
+const GEMINI_25_BUDGETS = {
+	low: 1024,
+	medium: 8192,
+	high: 24576
+};
+function getGeminiThinkingFamily(modelId) {
+	const id = modelId.toLowerCase();
+	if (id.startsWith("gemini-2.5")) return "budget";
+	if (id.startsWith("gemini-3-pro")) return "low-high";
+	if (id.startsWith("gemini-3.1-flash-image") || id.startsWith("gemini-3.1-flash-lite-image")) return "minimal-high";
+	if (/^gemini-\d/.test(id) && !id.startsWith("gemini-1") && !id.startsWith("gemini-2")) return "levels";
+}
+function geminiLevelForFamily(family, level) {
+	const cappedAtHigh = level === "xhigh" || level === "max" ? "high" : level;
+	if ((family === "low-high" || family === "minimal-high") && cappedAtHigh === "medium") return "high";
+	return cappedAtHigh;
+}
+function runGeminiThinkingLevel(modelId, level) {
+	const family = getGeminiThinkingFamily(modelId);
+	return family ? geminiLevelForFamily(family, level) : "off";
+}
+/**
+* Resolve the Google thinking config for a model and session level.
+* Returns `undefined` for `off`, unset levels, and unrecognized model families.
+*/
+function resolveGoogleThinkingConfig(modelId, level) {
+	if (!level || level === "off") return void 0;
+	const family = getGeminiThinkingFamily(modelId);
+	if (!family) return void 0;
+	const geminiLevel = geminiLevelForFamily(family, level);
+	if (family === "budget") return { thinkingBudget: GEMINI_25_BUDGETS[geminiLevel] };
+	if (family === "minimal-high") return { thinkingLevel: geminiLevel === "low" ? "minimal" : "high" };
+	return { thinkingLevel: geminiLevel };
+}
+/**
+* Create middleware that injects the resolved config into
+* `providerOptions.google.thinkingConfig`. Returns `undefined` when there is
+* nothing to inject, so callers wrap nothing.
+*/
+function createGoogleThinkingMiddleware(modelId, level) {
+	const config = resolveGoogleThinkingConfig(modelId, level);
+	if (!config) return void 0;
+	return {
+		specificationVersion: "v3",
+		transformParams: async ({ params }) => {
+			const google = params.providerOptions?.google ?? {};
+			const thinkingConfig = google.thinkingConfig ?? {};
+			if (thinkingConfig.thinkingBudget !== void 0 || thinkingConfig.thinkingLevel !== void 0) return params;
+			params.providerOptions = {
+				...params.providerOptions,
+				google: {
+					...google,
+					thinkingConfig: {
+						...thinkingConfig,
+						...config
+					}
+				}
+			};
+			return params;
+		}
+	};
+}
+//#endregion
+export { createGoogleThinkingMiddleware, resolveGoogleThinkingConfig, runGeminiThinkingLevel };
+
+//# sourceMappingURL=google-thinking.js.map

package/dist/providers/google-thinking.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-thinking.js","names":[],"sources":["../../src/providers/google-thinking.ts"],"sourcesContent":["/**\n * Google (Gemini) thinking-level middleware.\n *\n * Maps the Mastra Code session thinking level (`off|low|medium|high|xhigh|max`)\n * onto the thinking control each Gemini family accepts:\n * - Gemini 3+: `thinkingConfig.thinkingLevel` (Gemini 3 Pro only accepts `low|high`).\n * - Gemini 2.5: `thinkingConfig.thinkingBudget` (token budget).\n * Unknown model families get no thinking config, so requests are unchanged.\n */\n\nimport type { LanguageModelMiddleware } from 'ai';\nimport type { ActiveThinkingLevel, ThinkingLevelSetting } from '../thinking.js';\n\ntype GeminiLevel = 'low' | 'medium' | 'high';\ntype GeminiThinkingFamily = 'budget' | 'low-high' | 'minimal-high' | 'levels';\ntype GoogleThinkingConfig = { thinkingLevel: 'minimal' | GeminiLevel } | { thinkingBudget: number };\n\n// Budgets stay within the smallest Gemini 2.5 maximum (24576 for Flash / Flash-Lite).\nconst GEMINI_25_BUDGETS: Record<GeminiLevel, number> = { low: 1024, medium: 8192, high: 24576 };\n\nfunction getGeminiThinkingFamily(modelId: string): GeminiThinkingFamily | undefined {\n  const id = modelId.toLowerCase();\n  if (id.startsWith('gemini-2.5')) return 'budget';\n  // Only Gemini 3 Pro lacks `medium`; Gemini 3.1 Pro accepts it.\n  if (id.startsWith('gemini-3-pro')) return 'low-high';\n  // Gemini 3.1 Flash Image models only accept `minimal|high`.\n  if (id.startsWith('gemini-3.1-flash-image') || id.startsWith('gemini-3.1-flash-lite-image')) return 'minimal-high';\n  if (/^gemini-\\d/.test(id) && !id.startsWith('gemini-1') && !id.startsWith('gemini-2')) return 'levels';\n  return undefined;\n}\n\nfunction geminiLevelForFamily(family: GeminiThinkingFamily, level: ActiveThinkingLevel): GeminiLevel {\n  const cappedAtHigh = level === 'xhigh' || level === 'max' ? 'high' : level;\n  const hasNoMedium = family === 'low-high' || family === 'minimal-high';\n  if (hasNoMedium && cappedAtHigh === 'medium') return 'high';\n  return cappedAtHigh;\n}\n\nexport function runGeminiThinkingLevel(modelId: string, level: ActiveThinkingLevel): ThinkingLevelSetting {\n  const family = getGeminiThinkingFamily(modelId);\n  return family ? geminiLevelForFamily(family, level) : 'off';\n}\n\n/**\n * Resolve the Google thinking config for a model and session level.\n * Returns `undefined` for `off`, unset levels, and unrecognized model families.\n */\nexport function resolveGoogleThinkingConfig(\n  modelId: string,\n  level: ThinkingLevelSetting | undefined,\n): GoogleThinkingConfig | undefined {\n  if (!level || level === 'off') return undefined;\n  const family = getGeminiThinkingFamily(modelId);\n  if (!family) return undefined;\n\n  const geminiLevel = geminiLevelForFamily(family, level);\n  if (family === 'budget') return { thinkingBudget: GEMINI_25_BUDGETS[geminiLevel] };\n  if (family === 'minimal-high') return { thinkingLevel: geminiLevel === 'low' ? 'minimal' : 'high' };\n  return { thinkingLevel: geminiLevel };\n}\n\n/**\n * Create middleware that injects the resolved config into\n * `providerOptions.google.thinkingConfig`. Returns `undefined` when there is\n * nothing to inject, so callers wrap nothing.\n */\nexport function createGoogleThinkingMiddleware(\n  modelId: string,\n  level: ThinkingLevelSetting | undefined,\n): LanguageModelMiddleware | undefined {\n  const config = resolveGoogleThinkingConfig(modelId, level);\n  if (!config) return undefined;\n\n  return {\n    specificationVersion: 'v3',\n    transformParams: async ({ params }) => {\n      const google = (params.providerOptions?.google ?? {}) as Record<string, unknown>;\n      const thinkingConfig = (google.thinkingConfig ?? {}) as Record<string, unknown>;\n      // Explicit caller settings win; Google rejects budget and level together.\n      if (thinkingConfig.thinkingBudget !== undefined || thinkingConfig.thinkingLevel !== undefined) return params;\n      params.providerOptions = {\n        ...params.providerOptions,\n        google: { ...google, thinkingConfig: { ...thinkingConfig, ...config } },\n      } as typeof params.providerOptions;\n      return params;\n    },\n  };\n}\n"],"mappings":";AAkBA,MAAM,oBAAiD;CAAE,KAAK;CAAM,QAAQ;CAAM,MAAM;AAAM;AAE9F,SAAS,wBAAwB,SAAmD;CAClF,MAAM,KAAK,QAAQ,YAAY;CAC/B,IAAI,GAAG,WAAW,YAAY,GAAG,OAAO;CAExC,IAAI,GAAG,WAAW,cAAc,GAAG,OAAO;CAE1C,IAAI,GAAG,WAAW,wBAAwB,KAAK,GAAG,WAAW,6BAA6B,GAAG,OAAO;CACpG,IAAI,aAAa,KAAK,EAAE,KAAK,CAAC,GAAG,WAAW,UAAU,KAAK,CAAC,GAAG,WAAW,UAAU,GAAG,OAAO;AAEhG;AAEA,SAAS,qBAAqB,QAA8B,OAAyC;CACnG,MAAM,eAAe,UAAU,WAAW,UAAU,QAAQ,SAAS;CAErE,KADoB,WAAW,cAAc,WAAW,mBACrC,iBAAiB,UAAU,OAAO;CACrD,OAAO;AACT;AAEA,SAAgB,uBAAuB,SAAiB,OAAkD;CACxG,MAAM,SAAS,wBAAwB,OAAO;CAC9C,OAAO,SAAS,qBAAqB,QAAQ,KAAK,IAAI;AACxD;;;;;AAMA,SAAgB,4BACd,SACA,OACkC;CAClC,IAAI,CAAC,SAAS,UAAU,OAAO,OAAO,KAAA;CACtC,MAAM,SAAS,wBAAwB,OAAO;CAC9C,IAAI,CAAC,QAAQ,OAAO,KAAA;CAEpB,MAAM,cAAc,qBAAqB,QAAQ,KAAK;CACtD,IAAI,WAAW,UAAU,OAAO,EAAE,gBAAgB,kBAAkB,aAAa;CACjF,IAAI,WAAW,gBAAgB,OAAO,EAAE,eAAe,gBAAgB,QAAQ,YAAY,OAAO;CAClG,OAAO,EAAE,eAAe,YAAY;AACtC;;;;;;AAOA,SAAgB,+BACd,SACA,OACqC;CACrC,MAAM,SAAS,4BAA4B,SAAS,KAAK;CACzD,IAAI,CAAC,QAAQ,OAAO,KAAA;CAEpB,OAAO;EACL,sBAAsB;EACtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,MAAM,SAAU,OAAO,iBAAiB,UAAU,CAAC;GACnD,MAAM,iBAAkB,OAAO,kBAAkB,CAAC;GAElD,IAAI,eAAe,mBAAmB,KAAA,KAAa,eAAe,kBAAkB,KAAA,GAAW,OAAO;GACtG,OAAO,kBAAkB;IACvB,GAAG,OAAO;IACV,QAAQ;KAAE,GAAG;KAAQ,gBAAgB;MAAE,GAAG;MAAgB,GAAG;KAAO;IAAE;GACxE;GACA,OAAO;EACT;CACF;AACF"}

package/dist/providers/kimi-coding.d.ts

@@ -0,0 +1,14 @@
+import type { MastraModelConfig } from '@mastra/core/llm';
+import { AuthStorage } from '../auth/storage.js';
+import type { CredentialStore } from '../auth/types.js';
+export declare function setAuthStorage(storage: AuthStorage | undefined): void;
+export declare function buildKimiCodingOAuthFetch(options?: {
+    credentialStore?: CredentialStore;
+}): typeof fetch;
+export declare function buildKimiCodingApiKeyFetch(apiKey: string): typeof fetch;
+export declare function kimiCodingProvider(modelId: string, options: {
+    apiKey: string;
+    headers?: Record<string, string>;
+    credentialStore?: CredentialStore;
+}): MastraModelConfig;
+//# sourceMappingURL=kimi-coding.d.ts.map

package/dist/providers/kimi-coding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"kimi-coding.d.ts","sourceRoot":"","sources":["../../src/providers/kimi-coding.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAG1D,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,KAAK,EAAE,eAAe,EAAmB,MAAM,kBAAkB,CAAC;AASzE,wBAAgB,cAAc,CAAC,OAAO,EAAE,WAAW,GAAG,SAAS,GAAG,IAAI,CAErE;AAOD,wBAAgB,yBAAyB,CAAC,OAAO,GAAE;IAAE,eAAe,CAAC,EAAE,eAAe,CAAA;CAAO,GAAG,OAAO,KAAK,CAoC3G;AAED,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,KAAK,CASvE;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,EACf,OAAO,EAAE;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAAC,eAAe,CAAC,EAAE,eAAe,CAAA;CAAE,GAC/F,iBAAiB,CAYnB"}

package/dist/providers/kimi-coding.js

@@ -0,0 +1,78 @@
+import { getKimiCodingDeviceHeaders, isKimiCodingDeviceId } from "../auth/providers/kimi-coding.js";
+import { AuthStorage } from "../auth/storage.js";
+import { ProviderAuthRequiredError } from "../auth/provider-auth-error.js";
+import { createAnthropic } from "@ai-sdk/anthropic";
+//#region src/providers/kimi-coding.ts
+const PROVIDER_ID = "kimi-for-coding";
+const BASE_URL = "https://api.kimi.com/coding/v1";
+let authStorageInstance = null;
+function setAuthStorage(storage) {
+	authStorageInstance = storage ?? null;
+}
+function getAuthStorage() {
+	if (!authStorageInstance) authStorageInstance = new AuthStorage();
+	return authStorageInstance;
+}
+function buildKimiCodingOAuthFetch(options = {}) {
+	return (async (input, init) => {
+		const store = options.credentialStore ?? getAuthStorage();
+		store.reload();
+		const credential = store.get(PROVIDER_ID);
+		if (!credential || credential.type !== "oauth") throw new ProviderAuthRequiredError("Not logged in to Kimi For Coding.");
+		if (!isKimiCodingDeviceId(credential.deviceId)) throw new ProviderAuthRequiredError("Kimi For Coding credentials are invalid. Please reconnect the account.");
+		let token;
+		let activeCredential = credential;
+		if (store.getOAuthCredential) {
+			activeCredential = await store.getOAuthCredential(PROVIDER_ID);
+			token = activeCredential?.access;
+		} else {
+			token = await store.getApiKey(PROVIDER_ID);
+			store.reload();
+			const reloaded = store.get(PROVIDER_ID);
+			activeCredential = reloaded?.type === "oauth" ? {
+				...reloaded,
+				access: token ?? reloaded.access
+			} : void 0;
+		}
+		if (!token) throw new ProviderAuthRequiredError("Failed to refresh the Kimi For Coding token.");
+		if (!activeCredential || !isKimiCodingDeviceId(activeCredential.deviceId)) throw new ProviderAuthRequiredError("Kimi For Coding credentials are invalid. Please reconnect the account.");
+		const deviceHeaders = getKimiCodingDeviceHeaders(activeCredential.deviceId);
+		const headers = new Headers(input instanceof Request ? input.headers : void 0);
+		if (init?.headers) new Headers(init.headers).forEach((value, key) => headers.set(key, value));
+		for (const [key, value] of Object.entries(deviceHeaders)) headers.set(key, value);
+		headers.delete("authorization");
+		headers.delete("x-api-key");
+		headers.set("Authorization", `Bearer ${token}`);
+		return fetch(input, {
+			...init,
+			headers
+		});
+	});
+}
+function buildKimiCodingApiKeyFetch(apiKey) {
+	return (async (input, init) => {
+		const headers = new Headers(input instanceof Request ? input.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 ${apiKey}`);
+		return fetch(input, {
+			...init,
+			headers
+		});
+	});
+}
+function kimiCodingProvider(modelId, options) {
+	const usesOAuth = options.credentialStore?.get(PROVIDER_ID)?.type === "oauth";
+	return createAnthropic({
+		name: PROVIDER_ID,
+		apiKey: "auth-placeholder",
+		baseURL: BASE_URL,
+		headers: options.headers,
+		fetch: usesOAuth ? buildKimiCodingOAuthFetch({ credentialStore: options.credentialStore }) : buildKimiCodingApiKeyFetch(options.apiKey)
+	})(modelId);
+}
+//#endregion
+export { buildKimiCodingApiKeyFetch, buildKimiCodingOAuthFetch, kimiCodingProvider, setAuthStorage };
+
+//# sourceMappingURL=kimi-coding.js.map

package/dist/providers/kimi-coding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kimi-coding.js","names":[],"sources":["../../src/providers/kimi-coding.ts"],"sourcesContent":["import { createAnthropic } from '@ai-sdk/anthropic';\nimport type { MastraModelConfig } from '@mastra/core/llm';\nimport { ProviderAuthRequiredError } from '../auth/provider-auth-error.js';\nimport { getKimiCodingDeviceHeaders, isKimiCodingDeviceId } from '../auth/providers/kimi-coding.js';\nimport { AuthStorage } from '../auth/storage.js';\nimport type { CredentialStore, OAuthCredential } from '../auth/types.js';\n\nconst PROVIDER_ID = 'kimi-for-coding';\n// Pi's provider root is https://api.kimi.com/coding. The AI SDK expects the\n// versioned Anthropic base URL and appends /messages itself.\nconst BASE_URL = 'https://api.kimi.com/coding/v1';\n\nlet authStorageInstance: AuthStorage | null = null;\n\nexport function setAuthStorage(storage: AuthStorage | undefined): void {\n  authStorageInstance = storage ?? null;\n}\n\nfunction getAuthStorage(): AuthStorage {\n  if (!authStorageInstance) authStorageInstance = new AuthStorage();\n  return authStorageInstance;\n}\n\nexport function buildKimiCodingOAuthFetch(options: { credentialStore?: CredentialStore } = {}): typeof fetch {\n  return (async (input: string | URL | Request, init?: Parameters<typeof fetch>[1]) => {\n    const store = options.credentialStore ?? getAuthStorage();\n    store.reload();\n    const credential = store.get(PROVIDER_ID);\n    if (!credential || credential.type !== 'oauth') {\n      throw new ProviderAuthRequiredError('Not logged in to Kimi For Coding.');\n    }\n    if (!isKimiCodingDeviceId(credential.deviceId)) {\n      throw new ProviderAuthRequiredError('Kimi For Coding credentials are invalid. Please reconnect the account.');\n    }\n    let token: string | undefined;\n    let activeCredential: OAuthCredential | undefined = credential;\n    if (store.getOAuthCredential) {\n      activeCredential = await store.getOAuthCredential(PROVIDER_ID);\n      token = activeCredential?.access;\n    } else {\n      token = await store.getApiKey(PROVIDER_ID);\n      store.reload();\n      const reloaded = store.get(PROVIDER_ID);\n      activeCredential = reloaded?.type === 'oauth' ? { ...reloaded, access: token ?? reloaded.access } : undefined;\n    }\n    if (!token) throw new ProviderAuthRequiredError('Failed to refresh the Kimi For Coding token.');\n    if (!activeCredential || !isKimiCodingDeviceId(activeCredential.deviceId)) {\n      throw new ProviderAuthRequiredError('Kimi For Coding credentials are invalid. Please reconnect the account.');\n    }\n    const deviceHeaders = getKimiCodingDeviceHeaders(activeCredential.deviceId);\n\n    const headers = new Headers(input instanceof Request ? input.headers : undefined);\n    if (init?.headers) new Headers(init.headers).forEach((value, key) => headers.set(key, value));\n    for (const [key, value] of Object.entries(deviceHeaders)) headers.set(key, value);\n    headers.delete('authorization');\n    headers.delete('x-api-key');\n    headers.set('Authorization', `Bearer ${token}`);\n    return fetch(input, { ...init, headers });\n  }) as typeof fetch;\n}\n\nexport function buildKimiCodingApiKeyFetch(apiKey: string): typeof fetch {\n  return (async (input: string | URL | Request, init?: Parameters<typeof fetch>[1]) => {\n    const headers = new Headers(input instanceof Request ? input.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 ${apiKey}`);\n    return fetch(input, { ...init, headers });\n  }) as typeof fetch;\n}\n\nexport function kimiCodingProvider(\n  modelId: string,\n  options: { apiKey: string; headers?: Record<string, string>; credentialStore?: CredentialStore },\n): MastraModelConfig {\n  const usesOAuth = options.credentialStore?.get(PROVIDER_ID)?.type === 'oauth';\n  const provider = createAnthropic({\n    name: PROVIDER_ID,\n    apiKey: 'auth-placeholder',\n    baseURL: BASE_URL,\n    headers: options.headers,\n    fetch: usesOAuth\n      ? (buildKimiCodingOAuthFetch({ credentialStore: options.credentialStore }) as any)\n      : (buildKimiCodingApiKeyFetch(options.apiKey) as any),\n  });\n  return provider(modelId) as unknown as MastraModelConfig;\n}\n"],"mappings":";;;;;AAOA,MAAM,cAAc;AAGpB,MAAM,WAAW;AAEjB,IAAI,sBAA0C;AAE9C,SAAgB,eAAe,SAAwC;CACrE,sBAAsB,WAAW;AACnC;AAEA,SAAS,iBAA8B;CACrC,IAAI,CAAC,qBAAqB,sBAAsB,IAAI,YAAY;CAChE,OAAO;AACT;AAEA,SAAgB,0BAA0B,UAAiD,CAAC,GAAiB;CAC3G,QAAQ,OAAO,OAA+B,SAAuC;EACnF,MAAM,QAAQ,QAAQ,mBAAmB,eAAe;EACxD,MAAM,OAAO;EACb,MAAM,aAAa,MAAM,IAAI,WAAW;EACxC,IAAI,CAAC,cAAc,WAAW,SAAS,SACrC,MAAM,IAAI,0BAA0B,mCAAmC;EAEzE,IAAI,CAAC,qBAAqB,WAAW,QAAQ,GAC3C,MAAM,IAAI,0BAA0B,wEAAwE;EAE9G,IAAI;EACJ,IAAI,mBAAgD;EACpD,IAAI,MAAM,oBAAoB;GAC5B,mBAAmB,MAAM,MAAM,mBAAmB,WAAW;GAC7D,QAAQ,kBAAkB;EAC5B,OAAO;GACL,QAAQ,MAAM,MAAM,UAAU,WAAW;GACzC,MAAM,OAAO;GACb,MAAM,WAAW,MAAM,IAAI,WAAW;GACtC,mBAAmB,UAAU,SAAS,UAAU;IAAE,GAAG;IAAU,QAAQ,SAAS,SAAS;GAAO,IAAI,KAAA;EACtG;EACA,IAAI,CAAC,OAAO,MAAM,IAAI,0BAA0B,8CAA8C;EAC9F,IAAI,CAAC,oBAAoB,CAAC,qBAAqB,iBAAiB,QAAQ,GACtE,MAAM,IAAI,0BAA0B,wEAAwE;EAE9G,MAAM,gBAAgB,2BAA2B,iBAAiB,QAAQ;EAE1E,MAAM,UAAU,IAAI,QAAQ,iBAAiB,UAAU,MAAM,UAAU,KAAA,CAAS;EAChF,IAAI,MAAM,SAAS,IAAI,QAAQ,KAAK,OAAO,CAAC,CAAC,SAAS,OAAO,QAAQ,QAAQ,IAAI,KAAK,KAAK,CAAC;EAC5F,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,aAAa,GAAG,QAAQ,IAAI,KAAK,KAAK;EAChF,QAAQ,OAAO,eAAe;EAC9B,QAAQ,OAAO,WAAW;EAC1B,QAAQ,IAAI,iBAAiB,UAAU,OAAO;EAC9C,OAAO,MAAM,OAAO;GAAE,GAAG;GAAM;EAAQ,CAAC;CAC1C;AACF;AAEA,SAAgB,2BAA2B,QAA8B;CACvE,QAAQ,OAAO,OAA+B,SAAuC;EACnF,MAAM,UAAU,IAAI,QAAQ,iBAAiB,UAAU,MAAM,UAAU,KAAA,CAAS;EAChF,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,QAAQ;EAC/C,OAAO,MAAM,OAAO;GAAE,GAAG;GAAM;EAAQ,CAAC;CAC1C;AACF;AAEA,SAAgB,mBACd,SACA,SACmB;CACnB,MAAM,YAAY,QAAQ,iBAAiB,IAAI,WAAW,CAAC,EAAE,SAAS;CAUtE,OATiB,gBAAgB;EAC/B,MAAM;EACN,QAAQ;EACR,SAAS;EACT,SAAS,QAAQ;EACjB,OAAO,YACF,0BAA0B,EAAE,iBAAiB,QAAQ,gBAAgB,CAAC,IACtE,2BAA2B,QAAQ,MAAM;CAChD,CACc,CAAC,CAAC,OAAO;AACzB"}

package/dist/providers/model-ids.d.ts

@@ -0,0 +1,30 @@
+/**
+ * Model-id helpers shared by the model gateway and the browser settings
+ * module. Kept dependency-free so neither side has to import the other.
+ */
+export declare const OPENAI_PREFIX = "openai/";
+export declare const ANTHROPIC_PREFIX = "anthropic/";
+export declare const MASTRA_GATEWAY_PREFIX = "mastra/";
+/** Anthropic's API only accepts dashed ids (`claude-opus-4-6`), but catalogs may list dotted ones. */
+export declare function normalizeAnthropicModelId(modelId: string): string;
+export declare function stripMastraGatewayPrefix(modelId: string): string;
+/**
+ * Kimi For Coding subscriptions are served from `api.kimi.com`, which models.dev
+ * lists as `kimi-code-plan-cn`. Mastra Code augments that provider with the
+ * Kimi OAuth/API-key credential instead of defining its own.
+ */
+export declare const KIMI_CODING_PROVIDER_ID = "kimi-code-plan-cn";
+/** Credential id the Kimi login stores accounts under. */
+export declare const KIMI_CODING_CREDENTIAL_ID = "kimi-for-coding";
+/**
+ * Map model ids saved by older Mastra Code versions (packs, OM settings,
+ * thread metadata) to the current catalog id:
+ * `mastracode/kimi-for-coding/k3` and `kimi-for-coding/k3` → `kimi-code-plan-cn/k3`,
+ * `mastracode/github-copilot/<m>` → `github-copilot/<m>`,
+ * `mastracode/amazon-bedrock/<m>` → `amazon-bedrock/<m>`.
+ */
+export declare function normalizeLegacyModelId(modelId: string): string;
+/** Credential-store id that authenticates a catalog provider id. */
+export declare function getCredentialProviderId(providerId: string): string;
+export declare function remapOpenAIModelForCodexOAuth(modelId: string): string;
+//# sourceMappingURL=model-ids.d.ts.map

package/dist/providers/model-ids.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-ids.d.ts","sourceRoot":"","sources":["../../src/providers/model-ids.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,eAAO,MAAM,aAAa,YAAY,CAAC;AACvC,eAAO,MAAM,gBAAgB,eAAe,CAAC;AAC7C,eAAO,MAAM,qBAAqB,YAAY,CAAC;AAE/C,sGAAsG;AACtG,wBAAgB,yBAAyB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAEjE;AAeD,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAEhE;AAED;;;;GAIG;AACH,eAAO,MAAM,uBAAuB,sBAAsB,CAAC;AAC3D,0DAA0D;AAC1D,eAAO,MAAM,yBAAyB,oBAAoB,CAAC;AAS3D;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAU9D;AAED,oEAAoE;AACpE,wBAAgB,uBAAuB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAIlE;AAED,wBAAgB,6BAA6B,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAoBrE"}