@mastra/code-sdk 1.0.3-alpha.1 → 1.1.0-alpha.2

package/dist/agents/mastracode-gateway.js

@@ -1,473 +1,400 @@
-import { createAnthropic } from "@ai-sdk/anthropic";
-import { createOpenAI } from "@ai-sdk/openai";
-import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
-import {
-  GATEWAY_AUTH_HEADER,
-  MastraGateway,
-  MastraModelGateway,
-  ModelRouterLanguageModel,
-  PROVIDER_REGISTRY
-} from "@mastra/core/llm";
-import { wrapLanguageModel } from "ai";
 import { AuthStorage } from "../auth/storage.js";
-import { getCustomProviderId, loadSettings, MASTRA_GATEWAY_PROVIDER } from "../onboarding/settings.js";
-import {
-  buildAnthropicOAuthFetch,
-  claudeCodeMiddleware,
-  opencodeClaudeMaxProvider,
-  promptCacheMiddleware
-} from "../providers/claude-max.js";
+import { THINKING_LEVEL_TO_REASONING_EFFORT, buildOpenAICodexOAuthFetch, createCodexMiddleware, getEffectiveThinkingLevel, openaiCodexProvider } from "../providers/openai-codex.js";
+import { getCustomProviderId, loadSettings } from "../onboarding/settings.js";
+import { resolveCustomProviders } from "./custom-provider-source.js";
+import { buildAnthropicOAuthFetch, claudeCodeMiddleware, opencodeClaudeMaxProvider, promptCacheMiddleware } from "../providers/claude-max.js";
 import { getCopilotModelCatalog, githubCopilotProvider } from "../providers/github-copilot.js";
-import {
-  buildOpenAICodexOAuthFetch,
-  createCodexMiddleware,
-  getEffectiveThinkingLevel,
-  openaiCodexProvider,
-  THINKING_LEVEL_TO_REASONING_EFFORT
-} from "../providers/openai-codex.js";
 import { xaiProvider } from "../providers/xai.js";
-import { resolveCustomProviders } from "./custom-provider-source.js";
+import { createOpenAI } from "@ai-sdk/openai";
+import { wrapLanguageModel } from "ai";
+import { GATEWAY_AUTH_HEADER, MastraGateway, MastraModelGateway, ModelRouterLanguageModel, PROVIDER_REGISTRY } from "@mastra/core/llm";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
+//#region src/agents/mastracode-gateway.ts
 const OPENAI_PREFIX = "openai/";
 const MASTRA_GATEWAY_PREFIX = "mastra/";
 const MASTRACODE_GATEWAY_ID = "mastracode";
 const CODEX_OPENAI_MODEL_REMAPS = {
-  "gpt-5.3": "gpt-5.3-codex",
-  "gpt-5.2": "gpt-5.2-codex",
-  "gpt-5.1": "gpt-5.1-codex",
-  "gpt-5.1-mini": "gpt-5.1-codex-mini",
-  "gpt-5": "gpt-5-codex"
+	"gpt-5.3": "gpt-5.3-codex",
+	"gpt-5.2": "gpt-5.2-codex",
+	"gpt-5.1": "gpt-5.1-codex",
+	"gpt-5.1-mini": "gpt-5.1-codex-mini",
+	"gpt-5": "gpt-5-codex"
 };
 const authStorage = new AuthStorage();
 function reloadAuthStorage() {
-  authStorage.reload();
+	authStorage.reload();
 }
 function stripMastraGatewayPrefix(modelId) {
-  return modelId.startsWith(MASTRA_GATEWAY_PREFIX) ? modelId.substring(MASTRA_GATEWAY_PREFIX.length) : modelId;
+	return modelId.startsWith("mastra/") ? modelId.substring(7) : modelId;
 }
 function normalizeAnthropicModelId(modelId) {
-  return modelId.replace(/\.(?=\d)/g, "-");
+	return modelId.replace(/\.(?=\d)/g, "-");
 }
 function remapOpenAIModelForCodexOAuth(modelId) {
-  const normalizedModelId = stripMastraGatewayPrefix(modelId);
-  if (!normalizedModelId.startsWith(OPENAI_PREFIX)) {
-    return modelId;
-  }
-  const openaiModelId = normalizedModelId.substring(OPENAI_PREFIX.length);
-  if (openaiModelId.includes("-codex")) {
-    return modelId;
-  }
-  const codexModelId = CODEX_OPENAI_MODEL_REMAPS[openaiModelId];
-  if (!codexModelId) {
-    return modelId;
-  }
-  const remappedModelId = `${OPENAI_PREFIX}${codexModelId}`;
-  return modelId.startsWith(MASTRA_GATEWAY_PREFIX) ? `${MASTRA_GATEWAY_PREFIX}${remappedModelId}` : remappedModelId;
+	const normalizedModelId = stripMastraGatewayPrefix(modelId);
+	if (!normalizedModelId.startsWith("openai/")) return modelId;
+	const openaiModelId = normalizedModelId.substring(7);
+	if (openaiModelId.includes("-codex")) return modelId;
+	const codexModelId = CODEX_OPENAI_MODEL_REMAPS[openaiModelId];
+	if (!codexModelId) return modelId;
+	const remappedModelId = `${OPENAI_PREFIX}${codexModelId}`;
+	return modelId.startsWith("mastra/") ? `${MASTRA_GATEWAY_PREFIX}${remappedModelId}` : remappedModelId;
 }
+/**
+* Resolve the Anthropic API key.
+* Main slot → dedicated apikey: slot → env var.
+*/
 function getAnthropicApiKey(credentials = authStorage) {
-  const storedCred = credentials.get("anthropic");
-  if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) {
-    return storedCred.key.trim();
-  }
-  const dedicatedKey = credentials.getStoredApiKey("anthropic")?.trim();
-  if (dedicatedKey) return dedicatedKey;
-  return credentials.allowEnvironmentFallback === false ? void 0 : process.env.ANTHROPIC_API_KEY?.trim() || void 0;
+	const storedCred = credentials.get("anthropic");
+	if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return storedCred.key.trim();
+	const dedicatedKey = credentials.getStoredApiKey("anthropic")?.trim();
+	if (dedicatedKey) return dedicatedKey;
+	return credentials.allowEnvironmentFallback === false ? void 0 : process.env.ANTHROPIC_API_KEY?.trim() || void 0;
 }
+/**
+* Resolve the OpenAI API key.
+* Main slot → dedicated apikey: slot → env var.
+*/
 function getOpenAIApiKey(credentials = authStorage) {
-  const storedCred = credentials.get("openai-codex");
-  if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) {
-    return storedCred.key.trim();
-  }
-  const dedicatedKey = credentials.getStoredApiKey("openai-codex")?.trim();
-  if (dedicatedKey) return dedicatedKey;
-  return credentials.allowEnvironmentFallback === false ? void 0 : process.env.OPENAI_API_KEY?.trim() || void 0;
+	const storedCred = credentials.get("openai-codex");
+	if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return storedCred.key.trim();
+	const dedicatedKey = credentials.getStoredApiKey("openai-codex")?.trim();
+	if (dedicatedKey) return dedicatedKey;
+	return credentials.allowEnvironmentFallback === false ? void 0 : process.env.OPENAI_API_KEY?.trim() || void 0;
 }
 function anthropicApiKeyProvider(modelId, apiKey, headers) {
-  const anthropic = createAnthropic({ apiKey, headers });
-  return wrapLanguageModel({
-    model: anthropic(modelId),
-    middleware: [promptCacheMiddleware]
-  });
+	return wrapLanguageModel({
+		model: createAnthropic({
+			apiKey,
+			headers
+		})(modelId),
+		middleware: [promptCacheMiddleware]
+	});
 }
 function openaiApiKeyProvider(modelId, apiKey, headers) {
-  const openai = createOpenAI({ apiKey, baseURL: process.env.OPENAI_BASE_URL, headers });
-  return wrapLanguageModel({
-    model: openai.responses(modelId),
-    middleware: []
-  });
+	return wrapLanguageModel({
+		model: createOpenAI({
+			apiKey,
+			baseURL: process.env.OPENAI_BASE_URL,
+			headers
+		}).responses(modelId),
+		middleware: []
+	});
 }
 function getAuthProviderId(providerId) {
-  return providerId === "openai" ? "openai-codex" : providerId;
+	return providerId === "openai" ? "openai-codex" : providerId;
 }
 function getProviderAuthKey(providerId, credentials = authStorage) {
-  const authProviderId = getAuthProviderId(providerId);
-  const storedCred = credentials.get(authProviderId);
-  if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) {
-    return storedCred.key.trim();
-  }
-  return credentials.getStoredApiKey(authProviderId)?.trim() || void 0;
+	const authProviderId = getAuthProviderId(providerId);
+	const storedCred = credentials.get(authProviderId);
+	if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return storedCred.key.trim();
+	return credentials.getStoredApiKey(authProviderId)?.trim() || void 0;
 }
 function resolveAuth(request, mastraGatewayApiKey) {
-  return MastraCodeGateway.resolveProviderAuth(request, mastraGatewayApiKey);
+	return MastraCodeGateway.resolveProviderAuth(request, mastraGatewayApiKey);
 }
 function getGatewayProviderKey(gatewayId, providerId) {
-  if (gatewayId === "models.dev") return providerId;
-  return providerId === gatewayId ? gatewayId : `${gatewayId}/${providerId}`;
+	if (gatewayId === "models.dev") return providerId;
+	return providerId === gatewayId ? gatewayId : `${gatewayId}/${providerId}`;
 }
 function parseGatewayRouterId(routerId, gateway) {
-  const [firstPart = "", secondPart = "", ...restParts] = routerId.split("/");
-  const gatewayId = gateway.id;
-  if (firstPart === gatewayId && secondPart) {
-    return {
-      gatewayId,
-      providerId: secondPart,
-      modelId: restParts.join("/")
-    };
-  }
-  return {
-    gatewayId,
-    providerId: firstPart,
-    modelId: [secondPart, ...restParts].filter(Boolean).join("/")
-  };
+	const [firstPart = "", secondPart = "", ...restParts] = routerId.split("/");
+	const gatewayId = gateway.id;
+	if (firstPart === gatewayId && secondPart) return {
+		gatewayId,
+		providerId: secondPart,
+		modelId: restParts.join("/")
+	};
+	return {
+		gatewayId,
+		providerId: firstPart,
+		modelId: [secondPart, ...restParts].filter(Boolean).join("/")
+	};
 }
 function hasResolvedAuth(auth) {
-  if (!auth) return false;
-  if (auth.apiKey || auth.bearerToken) return true;
-  return auth.headers ? Object.keys(auth.headers).length > 0 : false;
+	if (!auth) return false;
+	if (auth.apiKey || auth.bearerToken) return true;
+	return auth.headers ? Object.keys(auth.headers).length > 0 : false;
 }
 async function resolveGatewayProviderAuth(gateway, routerId) {
-  if (!gateway.resolveAuth) return void 0;
-  const parsed = parseGatewayRouterId(routerId, gateway);
-  try {
-    const result = await gateway.resolveAuth({
-      gatewayId: parsed.gatewayId,
-      providerId: parsed.providerId,
-      modelId: parsed.modelId,
-      routerId
-    });
-    return hasResolvedAuth(result) ? result : void 0;
-  } catch {
-    return void 0;
-  }
+	if (!gateway.resolveAuth) return void 0;
+	const parsed = parseGatewayRouterId(routerId, gateway);
+	try {
+		const result = await gateway.resolveAuth({
+			gatewayId: parsed.gatewayId,
+			providerId: parsed.providerId,
+			modelId: parsed.modelId,
+			routerId
+		});
+		return hasResolvedAuth(result) ? result : void 0;
+	} catch {
+		return;
+	}
 }
 async function getMastraCodeProviderConfigs(gateway) {
-  const providers = { ...PROVIDER_REGISTRY };
-  try {
-    const gatewayProviders = await gateway.fetchProviders();
-    for (const [providerId, config] of Object.entries(gatewayProviders)) {
-      providers[getGatewayProviderKey(gateway.id, providerId)] = {
-        ...config,
-        gateway: gateway.id
-      };
-    }
-  } catch (error) {
-    console.warn(`Failed to load providers from gateway ${gateway.id}:`, error);
-  }
-  return providers;
+	const providers = { ...PROVIDER_REGISTRY };
+	try {
+		const gatewayProviders = await gateway.fetchProviders();
+		for (const [providerId, config] of Object.entries(gatewayProviders)) providers[getGatewayProviderKey(gateway.id, providerId)] = {
+			...config,
+			gateway: gateway.id
+		};
+	} catch (error) {
+		console.warn(`Failed to load providers from gateway ${gateway.id}:`, error);
+	}
+	return providers;
 }
 function getApiKeyEnvVar(providerConfig) {
-  const envVars = providerConfig?.apiKeyEnvVar;
-  return Array.isArray(envVars) ? envVars[0] : envVars;
-}
-class MastraCodeGateway extends MastraModelGateway {
-  id = MASTRACODE_GATEWAY_ID;
-  name = "MastraCode Gateway";
-  #mastraGateway;
-  #mastraGatewayBaseUrl;
-  #mastraGatewayApiKey;
-  #routeThroughMastraGateway;
-  #thinkingLevel;
-  #customProviders;
-  #settingsPath;
-  #credentials;
-  constructor({
-    mastraGatewayBaseUrl,
-    mastraGatewayApiKey,
-    routeThroughMastraGateway,
-    thinkingLevel,
-    customProviders,
-    settingsPath,
-    credentialStore
-  }) {
-    super();
-    this.#mastraGateway = new MastraGateway({ baseUrl: mastraGatewayBaseUrl });
-    this.#mastraGatewayBaseUrl = mastraGatewayBaseUrl;
-    this.#mastraGatewayApiKey = mastraGatewayApiKey;
-    this.#routeThroughMastraGateway = routeThroughMastraGateway;
-    this.#thinkingLevel = thinkingLevel;
-    this.#customProviders = customProviders;
-    this.#settingsPath = settingsPath;
-    this.#credentials = credentialStore ?? authStorage;
-  }
-  static getMastraGatewayApiKey() {
-    return authStorage.getStoredApiKey(MASTRA_GATEWAY_PROVIDER) ?? process.env["MASTRA_GATEWAY_API_KEY"];
-  }
-  /** @deprecated Renamed to {@link MastraCodeGateway.getMastraGatewayApiKey}. */
-  static getMemoryGatewayApiKey() {
-    return MastraCodeGateway.getMastraGatewayApiKey();
-  }
-  /**
-   * Claim an unprefixed model id (e.g. a bare `anthropic/...`) when this gateway
-   * can authenticate it via OAuth or a stored/env credential. This lets the
-   * shared gateway router select MastraCode for auth resolution, so the status
-   * bar and `/api-keys` reflect OAuth logins instead of only checking env vars.
-   */
-  handlesModel(modelId) {
-    const parsed = parseGatewayRouterId(modelId, this);
-    if (!parsed.providerId || !parsed.modelId) return false;
-    if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) return true;
-    const customProvider = this.#getCustomProviders().find(
-      (provider) => parsed.providerId === getCustomProviderId(provider.name)
-    );
-    if (customProvider?.apiKey) return true;
-    return hasResolvedAuth(
-      MastraCodeGateway.resolveProviderAuth(
-        {
-          gatewayId: this.id,
-          providerId: parsed.providerId,
-          modelId: parsed.modelId,
-          routerId: modelId
-        },
-        void 0,
-        this.#credentials
-      )
-    );
-  }
-  static resolveProviderAuth(request, mastraGatewayApiKey, credentials = authStorage) {
-    if (request.gatewayId === "mastra" && mastraGatewayApiKey) {
-      return { apiKey: mastraGatewayApiKey, source: "gateway" };
-    }
-    const storedCred = credentials.get(getAuthProviderId(request.providerId));
-    if (storedCred?.type === "oauth") {
-      return { bearerToken: "oauth", source: "gateway" };
-    }
-    const apiKey = getProviderAuthKey(request.providerId, credentials);
-    return apiKey ? { apiKey, source: "gateway" } : void 0;
-  }
-  static createModelCatalogProvider(gateway) {
-    return async () => {
-      const registry = await getMastraCodeProviderConfigs(gateway);
-      const models = [];
-      for (const [provider, providerConfig] of Object.entries(registry)) {
-        const apiKeyEnvVar = getApiKeyEnvVar(providerConfig);
-        const hasEnvKey = apiKeyEnvVar ? Boolean(process.env[apiKeyEnvVar]) : false;
-        const modelNames = providerConfig.models;
-        if (!Array.isArray(modelNames)) continue;
-        const gatewayAuth = modelNames[0] ? await resolveGatewayProviderAuth(gateway, `${provider}/${modelNames[0]}`) : void 0;
-        for (const modelName of modelNames) {
-          const id = `${provider}/${modelName}`;
-          models.push({
-            id,
-            provider,
-            modelName,
-            hasApiKey: hasEnvKey || Boolean(gatewayAuth),
-            apiKeyEnvVar: apiKeyEnvVar || void 0
-          });
-        }
-      }
-      return models;
-    };
-  }
-  createModelCatalogProvider() {
-    return MastraCodeGateway.createModelCatalogProvider(this);
-  }
-  #getCustomProviders() {
-    return this.#customProviders ?? resolveCustomProviders() ?? loadSettings(this.#settingsPath).customProviders;
-  }
-  async fetchProviders() {
-    const providers = {};
-    for (const provider of this.#getCustomProviders()) {
-      const models = provider.models ?? [];
-      if (!models.length) continue;
-      providers[getCustomProviderId(provider.name)] = {
-        name: provider.name,
-        url: provider.url,
-        apiKeyEnvVar: "",
-        apiKeyHeader: "Authorization",
-        gateway: this.id,
-        models
-      };
-    }
-    try {
-      const copilotModels = await getCopilotModelCatalog({ authStorage });
-      providers["github-copilot"] = {
-        name: "GitHub Copilot",
-        apiKeyEnvVar: "",
-        apiKeyHeader: "Authorization",
-        gateway: this.id,
-        models: copilotModels.map((model) => model.id)
-      };
-    } catch (error) {
-      console.warn("Failed to load GitHub Copilot model catalog:", error);
-    }
-    return providers;
-  }
-  buildUrl(modelId) {
-    return this.#routeThroughMastraGateway ? this.#mastraGateway.buildUrl(modelId) : modelId;
-  }
-  async getApiKey(modelId) {
-    const providerId = stripMastraGatewayPrefix(modelId).split("/", 1)[0];
-    if (this.#routeThroughMastraGateway) return this.#mastraGatewayApiKey ?? "";
-    return providerId ? getProviderAuthKey(providerId, this.#credentials) ?? "" : "";
-  }
-  resolveAuth(request) {
-    if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) {
-      return { apiKey: this.#mastraGatewayApiKey, source: "gateway" };
-    }
-    const customProvider = this.#getCustomProviders().find(
-      (provider) => request.providerId === getCustomProviderId(provider.name)
-    );
-    if (customProvider?.apiKey) {
-      return { apiKey: customProvider.apiKey, source: "gateway" };
-    }
-    return MastraCodeGateway.resolveProviderAuth(request, void 0, this.#credentials);
-  }
-  resolveLanguageModel(args) {
-    const customProvider = this.#getCustomProviders().find(
-      (provider) => args.providerId === getCustomProviderId(provider.name)
-    );
-    if (customProvider) {
-      const provider = createOpenAICompatible({
-        name: args.providerId,
-        baseURL: customProvider.url,
-        apiKey: args.apiKey,
-        headers: args.headers
-      });
-      return provider.chatModel(args.modelId);
-    }
-    if (args.providerId === "github-copilot") {
-      return githubCopilotProvider(args.modelId, {
-        headers: args.headers,
-        authStorage: this.#credentials
-      });
-    }
-    if (args.providerId === "moonshotai") {
-      const apiKey = args.apiKey?.trim() || process.env.MOONSHOT_API_KEY?.trim() || process.env.MOONSHOT_AI_API_KEY?.trim();
-      if (!apiKey) {
-        throw new Error("Need MOONSHOT_API_KEY");
-      }
-      return createAnthropic({
-        apiKey,
-        baseURL: "https://api.moonshot.ai/anthropic/v1",
-        name: "moonshotai.anthropicv1",
-        headers: args.headers
-      })(args.modelId);
-    }
-    if (args.providerId === "anthropic") {
-      return this.#resolveAnthropicModel(args);
-    }
-    if (args.providerId === "openai") {
-      const openaiModel = this.#resolveOpenAIModel(args);
-      if (openaiModel) return openaiModel;
-    }
-    if (args.providerId === "xai" && this.#credentials.get("xai")?.type === "oauth") {
-      return xaiProvider(args.modelId, {
-        headers: args.headers,
-        authStorage: this.#credentials
-      });
-    }
-    if (this.#routeThroughMastraGateway) {
-      return this.#mastraGateway.resolveLanguageModel(args);
-    }
-    return new ModelRouterLanguageModel({
-      id: `${args.providerId}/${args.modelId}`,
-      apiKey: args.apiKey,
-      headers: args.headers
-    });
-  }
-  #resolveAnthropicModel(args) {
-    const bareModelId = normalizeAnthropicModelId(args.modelId);
-    const storedCred = this.#credentials.get("anthropic");
-    if (this.#routeThroughMastraGateway) {
-      if (storedCred?.type === "oauth") {
-        const anthropic = createAnthropic({
-          apiKey: "oauth-gateway-placeholder",
-          baseURL: `${this.#mastraGatewayBaseUrl}/v1`,
-          headers: {
-            [GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,
-            ...args.headers
-          },
-          fetch: buildAnthropicOAuthFetch({ authStorage: this.#credentials })
-        });
-        return wrapLanguageModel({
-          model: anthropic(bareModelId),
-          middleware: [claudeCodeMiddleware, promptCacheMiddleware]
-        });
-      }
-      return this.#mastraGateway.resolveLanguageModel({ ...args, modelId: bareModelId });
-    }
-    if (storedCred?.type === "oauth") {
-      return opencodeClaudeMaxProvider(bareModelId, {
-        headers: args.headers,
-        authStorage: this.#credentials
-      });
-    }
-    if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) {
-      return anthropicApiKeyProvider(
-        bareModelId,
-        storedCred.key.trim(),
-        args.headers
-      );
-    }
-    const apiKey = getAnthropicApiKey(this.#credentials);
-    if (apiKey) {
-      return anthropicApiKeyProvider(bareModelId, apiKey, args.headers);
-    }
-    return opencodeClaudeMaxProvider(bareModelId, {
-      headers: args.headers,
-      authStorage: this.#credentials
-    });
-  }
-  #resolveOpenAIModel(args) {
-    const storedCred = this.#credentials.get("openai-codex");
-    if (this.#routeThroughMastraGateway) {
-      if (storedCred?.type === "oauth") {
-        const resolvedModelId = remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`);
-        const resolvedBareModelId = resolvedModelId.substring(OPENAI_PREFIX.length);
-        const requestedLevel = this.#thinkingLevel ?? "medium";
-        const effectiveLevel = getEffectiveThinkingLevel(resolvedBareModelId, requestedLevel);
-        const reasoningEffort = THINKING_LEVEL_TO_REASONING_EFFORT[effectiveLevel];
-        const middleware = createCodexMiddleware(reasoningEffort);
-        const openai = createOpenAI({
-          apiKey: "oauth-gateway-placeholder",
-          baseURL: `${this.#mastraGatewayBaseUrl}/v1`,
-          headers: {
-            [GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,
-            ...args.headers
-          },
-          fetch: buildOpenAICodexOAuthFetch({ authStorage: this.#credentials, rewriteUrl: false })
-        });
-        return wrapLanguageModel({
-          model: openai.responses(resolvedBareModelId),
-          middleware: [middleware]
-        });
-      }
-      return this.#mastraGateway.resolveLanguageModel(args);
-    }
-    if (storedCred?.type === "oauth") {
-      const resolvedModelId = remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`);
-      return openaiCodexProvider(resolvedModelId.substring(OPENAI_PREFIX.length), {
-        thinkingLevel: this.#thinkingLevel,
-        headers: args.headers,
-        authStorage: this.#credentials
-      });
-    }
-    const apiKey = getOpenAIApiKey(this.#credentials);
-    if (apiKey) {
-      return openaiApiKeyProvider(args.modelId, apiKey, args.headers);
-    }
-    return void 0;
-  }
+	const envVars = providerConfig?.apiKeyEnvVar;
+	return Array.isArray(envVars) ? envVars[0] : envVars;
 }
-export {
-  MASTRACODE_GATEWAY_ID,
-  MASTRA_GATEWAY_PREFIX,
-  MastraCodeGateway,
-  OPENAI_PREFIX,
-  getAnthropicApiKey,
-  getOpenAIApiKey,
-  reloadAuthStorage,
-  remapOpenAIModelForCodexOAuth,
-  resolveAuth,
-  stripMastraGatewayPrefix
+var MastraCodeGateway = class MastraCodeGateway extends MastraModelGateway {
+	id = MASTRACODE_GATEWAY_ID;
+	name = "MastraCode Gateway";
+	#mastraGateway;
+	#mastraGatewayBaseUrl;
+	#mastraGatewayApiKey;
+	#routeThroughMastraGateway;
+	#thinkingLevel;
+	#customProviders;
+	#settingsPath;
+	#credentials;
+	constructor({ mastraGatewayBaseUrl, mastraGatewayApiKey, routeThroughMastraGateway, thinkingLevel, customProviders, settingsPath, credentialStore }) {
+		super();
+		this.#mastraGateway = new MastraGateway({ baseUrl: mastraGatewayBaseUrl });
+		this.#mastraGatewayBaseUrl = mastraGatewayBaseUrl;
+		this.#mastraGatewayApiKey = mastraGatewayApiKey;
+		this.#routeThroughMastraGateway = routeThroughMastraGateway;
+		this.#thinkingLevel = thinkingLevel;
+		this.#customProviders = customProviders;
+		this.#settingsPath = settingsPath;
+		this.#credentials = credentialStore ?? authStorage;
+	}
+	static getMastraGatewayApiKey() {
+		return authStorage.getStoredApiKey("mastra-gateway") ?? process.env["MASTRA_GATEWAY_API_KEY"];
+	}
+	/** @deprecated Renamed to {@link MastraCodeGateway.getMastraGatewayApiKey}. */
+	static getMemoryGatewayApiKey() {
+		return MastraCodeGateway.getMastraGatewayApiKey();
+	}
+	/**
+	* Claim an unprefixed model id (e.g. a bare `anthropic/...`) when this gateway
+	* can authenticate it via OAuth or a stored/env credential. This lets the
+	* shared gateway router select MastraCode for auth resolution, so the status
+	* bar and `/api-keys` reflect OAuth logins instead of only checking env vars.
+	*/
+	handlesModel(modelId) {
+		const parsed = parseGatewayRouterId(modelId, this);
+		if (!parsed.providerId || !parsed.modelId) return false;
+		if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) return true;
+		if (this.#getCustomProviders().find((provider) => parsed.providerId === getCustomProviderId(provider.name))?.apiKey) return true;
+		return hasResolvedAuth(MastraCodeGateway.resolveProviderAuth({
+			gatewayId: this.id,
+			providerId: parsed.providerId,
+			modelId: parsed.modelId,
+			routerId: modelId
+		}, void 0, this.#credentials));
+	}
+	static resolveProviderAuth(request, mastraGatewayApiKey, credentials = authStorage) {
+		if (request.gatewayId === "mastra" && mastraGatewayApiKey) return {
+			apiKey: mastraGatewayApiKey,
+			source: "gateway"
+		};
+		if (credentials.get(getAuthProviderId(request.providerId))?.type === "oauth") return {
+			bearerToken: "oauth",
+			source: "gateway"
+		};
+		const apiKey = getProviderAuthKey(request.providerId, credentials);
+		return apiKey ? {
+			apiKey,
+			source: "gateway"
+		} : void 0;
+	}
+	static createModelCatalogProvider(gateway) {
+		return async () => {
+			const registry = await getMastraCodeProviderConfigs(gateway);
+			const models = [];
+			for (const [provider, providerConfig] of Object.entries(registry)) {
+				const apiKeyEnvVar = getApiKeyEnvVar(providerConfig);
+				const hasEnvKey = apiKeyEnvVar ? Boolean(process.env[apiKeyEnvVar]) : false;
+				const modelNames = providerConfig.models;
+				if (!Array.isArray(modelNames)) continue;
+				const gatewayAuth = modelNames[0] ? await resolveGatewayProviderAuth(gateway, `${provider}/${modelNames[0]}`) : void 0;
+				for (const modelName of modelNames) {
+					const id = `${provider}/${modelName}`;
+					models.push({
+						id,
+						provider,
+						modelName,
+						hasApiKey: hasEnvKey || Boolean(gatewayAuth),
+						apiKeyEnvVar: apiKeyEnvVar || void 0
+					});
+				}
+			}
+			return models;
+		};
+	}
+	createModelCatalogProvider() {
+		return MastraCodeGateway.createModelCatalogProvider(this);
+	}
+	#getCustomProviders() {
+		return this.#customProviders ?? resolveCustomProviders() ?? loadSettings(this.#settingsPath).customProviders;
+	}
+	async fetchProviders() {
+		const providers = {};
+		for (const provider of this.#getCustomProviders()) {
+			const models = provider.models ?? [];
+			if (!models.length) continue;
+			providers[getCustomProviderId(provider.name)] = {
+				name: provider.name,
+				url: provider.url,
+				apiKeyEnvVar: "",
+				apiKeyHeader: "Authorization",
+				gateway: this.id,
+				models
+			};
+		}
+		try {
+			const copilotModels = await getCopilotModelCatalog({ authStorage });
+			providers["github-copilot"] = {
+				name: "GitHub Copilot",
+				apiKeyEnvVar: "",
+				apiKeyHeader: "Authorization",
+				gateway: this.id,
+				models: copilotModels.map((model) => model.id)
+			};
+		} catch (error) {
+			console.warn("Failed to load GitHub Copilot model catalog:", error);
+		}
+		return providers;
+	}
+	buildUrl(modelId) {
+		return this.#routeThroughMastraGateway ? this.#mastraGateway.buildUrl(modelId) : modelId;
+	}
+	async getApiKey(modelId) {
+		const providerId = stripMastraGatewayPrefix(modelId).split("/", 1)[0];
+		if (this.#routeThroughMastraGateway) return this.#mastraGatewayApiKey ?? "";
+		return providerId ? getProviderAuthKey(providerId, this.#credentials) ?? "" : "";
+	}
+	resolveAuth(request) {
+		if (this.#routeThroughMastraGateway && this.#mastraGatewayApiKey) return {
+			apiKey: this.#mastraGatewayApiKey,
+			source: "gateway"
+		};
+		const customProvider = this.#getCustomProviders().find((provider) => request.providerId === getCustomProviderId(provider.name));
+		if (customProvider?.apiKey) return {
+			apiKey: customProvider.apiKey,
+			source: "gateway"
+		};
+		return MastraCodeGateway.resolveProviderAuth(request, void 0, this.#credentials);
+	}
+	resolveLanguageModel(args) {
+		const customProvider = this.#getCustomProviders().find((provider) => args.providerId === getCustomProviderId(provider.name));
+		if (customProvider) return createOpenAICompatible({
+			name: args.providerId,
+			baseURL: customProvider.url,
+			apiKey: args.apiKey,
+			headers: args.headers
+		}).chatModel(args.modelId);
+		if (args.providerId === "github-copilot") return githubCopilotProvider(args.modelId, {
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+		if (args.providerId === "moonshotai") {
+			const apiKey = args.apiKey?.trim() || process.env.MOONSHOT_API_KEY?.trim() || process.env.MOONSHOT_AI_API_KEY?.trim();
+			if (!apiKey) throw new Error("Need MOONSHOT_API_KEY");
+			return createAnthropic({
+				apiKey,
+				baseURL: "https://api.moonshot.ai/anthropic/v1",
+				name: "moonshotai.anthropicv1",
+				headers: args.headers
+			})(args.modelId);
+		}
+		if (args.providerId === "anthropic") return this.#resolveAnthropicModel(args);
+		if (args.providerId === "openai") {
+			const openaiModel = this.#resolveOpenAIModel(args);
+			if (openaiModel) return openaiModel;
+		}
+		if (args.providerId === "xai" && this.#credentials.get("xai")?.type === "oauth") return xaiProvider(args.modelId, {
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+		if (this.#routeThroughMastraGateway) return this.#mastraGateway.resolveLanguageModel(args);
+		return new ModelRouterLanguageModel({
+			id: `${args.providerId}/${args.modelId}`,
+			apiKey: args.apiKey,
+			headers: args.headers
+		});
+	}
+	#resolveAnthropicModel(args) {
+		const bareModelId = normalizeAnthropicModelId(args.modelId);
+		const storedCred = this.#credentials.get("anthropic");
+		if (this.#routeThroughMastraGateway) {
+			if (storedCred?.type === "oauth") return wrapLanguageModel({
+				model: createAnthropic({
+					apiKey: "oauth-gateway-placeholder",
+					baseURL: `${this.#mastraGatewayBaseUrl}/v1`,
+					headers: {
+						[GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,
+						...args.headers
+					},
+					fetch: buildAnthropicOAuthFetch({ authStorage: this.#credentials })
+				})(bareModelId),
+				middleware: [claudeCodeMiddleware, promptCacheMiddleware]
+			});
+			return this.#mastraGateway.resolveLanguageModel({
+				...args,
+				modelId: bareModelId
+			});
+		}
+		if (storedCred?.type === "oauth") return opencodeClaudeMaxProvider(bareModelId, {
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+		if (storedCred?.type === "api_key" && storedCred.key.trim().length > 0) return anthropicApiKeyProvider(bareModelId, storedCred.key.trim(), args.headers);
+		const apiKey = getAnthropicApiKey(this.#credentials);
+		if (apiKey) return anthropicApiKeyProvider(bareModelId, apiKey, args.headers);
+		return opencodeClaudeMaxProvider(bareModelId, {
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+	}
+	#resolveOpenAIModel(args) {
+		const storedCred = this.#credentials.get("openai-codex");
+		if (this.#routeThroughMastraGateway) {
+			if (storedCred?.type === "oauth") {
+				const resolvedBareModelId = remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`).substring(7);
+				const reasoningEffort = THINKING_LEVEL_TO_REASONING_EFFORT[getEffectiveThinkingLevel(resolvedBareModelId, this.#thinkingLevel ?? "medium")];
+				const middleware = createCodexMiddleware(reasoningEffort);
+				return wrapLanguageModel({
+					model: createOpenAI({
+						apiKey: "oauth-gateway-placeholder",
+						baseURL: `${this.#mastraGatewayBaseUrl}/v1`,
+						headers: {
+							[GATEWAY_AUTH_HEADER]: `Bearer ${args.apiKey}`,
+							...args.headers
+						},
+						fetch: buildOpenAICodexOAuthFetch({
+							authStorage: this.#credentials,
+							rewriteUrl: false
+						})
+					}).responses(resolvedBareModelId),
+					middleware: [middleware]
+				});
+			}
+			return this.#mastraGateway.resolveLanguageModel(args);
+		}
+		if (storedCred?.type === "oauth") return openaiCodexProvider(remapOpenAIModelForCodexOAuth(`openai/${args.modelId}`).substring(7), {
+			thinkingLevel: this.#thinkingLevel,
+			headers: args.headers,
+			authStorage: this.#credentials
+		});
+		const apiKey = getOpenAIApiKey(this.#credentials);
+		if (apiKey) return openaiApiKeyProvider(args.modelId, apiKey, args.headers);
+	}
 };
+//#endregion
+export { MASTRACODE_GATEWAY_ID, MASTRA_GATEWAY_PREFIX, MastraCodeGateway, OPENAI_PREFIX, getAnthropicApiKey, getOpenAIApiKey, reloadAuthStorage, remapOpenAIModelForCodexOAuth, resolveAuth, stripMastraGatewayPrefix };
+
 //# sourceMappingURL=mastracode-gateway.js.map