@mastra/core 1.21.0 → 1.22.0-alpha.1

package/dist/{chunk-5WJWEAD2.cjs → chunk-XL636GZV.cjs}

@@ -1,5 +1,9 @@
 'use strict';
 
+var chunkSH3TVIAI_cjs = require('./chunk-SH3TVIAI.cjs');
+var chunk73LGQD54_cjs = require('./chunk-73LGQD54.cjs');
+var chunkWID3XI3A_cjs = require('./chunk-WID3XI3A.cjs');
+var chunk4U7ZLI36_cjs = require('./chunk-4U7ZLI36.cjs');
 var fs = require('fs');
 var module$1 = require('module');
 var os = require('os');
@@ -12,7 +16,96 @@ var fs__default = /*#__PURE__*/_interopDefault(fs);
 var os__default = /*#__PURE__*/_interopDefault(os);
 var path__default = /*#__PURE__*/_interopDefault(path);
 
-// src/llm/model/provider-registry.ts
+// src/llm/model/gateways/mastra.ts
+var MastraGateway = class extends chunkWID3XI3A_cjs.MastraModelGateway {
+  constructor(config) {
+    super();
+    this.config = config;
+  }
+  id = "mastra";
+  name = "Memory Gateway";
+  getBaseUrl() {
+    const raw = this.config?.baseUrl ?? process.env["MASTRA_GATEWAY_URL"] ?? "https://gateway-api.mastra.ai";
+    return raw.replace(/\/+$/, "").replace(/\/v1$/, "");
+  }
+  shouldEnable() {
+    return !!(this.config?.apiKey ?? process.env["MASTRA_GATEWAY_API_KEY"]);
+  }
+  async fetchProviders() {
+    if (!this.shouldEnable()) {
+      return {};
+    }
+    const openrouterConfig = PROVIDER_REGISTRY["openrouter"];
+    const models = openrouterConfig?.models ?? [];
+    const providers = {
+      mastra: {
+        apiKeyEnvVar: "MASTRA_GATEWAY_API_KEY",
+        apiKeyHeader: "Authorization",
+        name: "Memory Gateway",
+        gateway: "mastra",
+        models: [...models],
+        docUrl: "https://mastra.ai/docs/gateway"
+      }
+    };
+    return providers;
+  }
+  async buildUrl(_modelId) {
+    return `${this.getBaseUrl()}/v1`;
+  }
+  async getApiKey() {
+    const apiKey = this.config?.apiKey ?? process.env["MASTRA_GATEWAY_API_KEY"];
+    if (!apiKey) {
+      throw new chunk4U7ZLI36_cjs.MastraError({
+        id: "MASTRA_GATEWAY_NO_API_KEY",
+        domain: "LLM",
+        category: "UNKNOWN",
+        text: "Missing MASTRA_GATEWAY_API_KEY environment variable"
+      });
+    }
+    return apiKey;
+  }
+  resolveLanguageModel({
+    modelId,
+    providerId,
+    apiKey,
+    headers
+  }) {
+    const baseURL = `${this.getBaseUrl()}/v1`;
+    const fullModelId = `${providerId}/${modelId}`;
+    if (this.config?.customFetch && providerId === "anthropic") {
+      return chunkWID3XI3A_cjs.createAnthropic({
+        apiKey: "oauth-gateway-placeholder",
+        baseURL,
+        headers: {
+          "User-Agent": chunkWID3XI3A_cjs.MASTRA_USER_AGENT,
+          [chunkWID3XI3A_cjs.GATEWAY_AUTH_HEADER]: `Bearer ${apiKey}`,
+          ...headers
+        },
+        fetch: this.config.customFetch
+      })(modelId);
+    }
+    if (this.config?.customFetch) {
+      return chunkSH3TVIAI_cjs.createOpenRouter({
+        apiKey: "oauth-gateway-placeholder",
+        baseURL,
+        headers: {
+          "User-Agent": chunkWID3XI3A_cjs.MASTRA_USER_AGENT,
+          [chunkWID3XI3A_cjs.GATEWAY_AUTH_HEADER]: `Bearer ${apiKey}`,
+          ...headers
+        },
+        fetch: this.config.customFetch
+      }).chat(fullModelId);
+    }
+    return chunkSH3TVIAI_cjs.createOpenRouter({
+      apiKey,
+      baseURL,
+      headers: {
+        "User-Agent": chunkWID3XI3A_cjs.MASTRA_USER_AGENT,
+        ...headers
+      }
+    }).chat(fullModelId);
+  }
+};
 
 // src/llm/model/provider-registry.json
 var provider_registry_default = {
@@ -8139,6 +8232,27 @@ var provider_registry_default = {
 };
 
 // src/llm/model/provider-registry.ts
+function getEnabledGatewayIds(gateways) {
+  const enabledGatewayIds = /* @__PURE__ */ new Set();
+  for (const gateway of gateways) {
+    const enabled = gateway.shouldEnable();
+    if (enabled) {
+      enabledGatewayIds.add(gateway.id);
+    }
+  }
+  return enabledGatewayIds;
+}
+function sanitizeRegistryDataForRuntime(data, enabledGatewayIds) {
+  const providers = Object.fromEntries(
+    Object.entries(data.providers).filter(([, config]) => enabledGatewayIds.has(config.gateway))
+  );
+  const models = Object.fromEntries(Object.entries(data.models).filter(([providerId]) => providerId in providers));
+  return {
+    ...data,
+    providers,
+    models
+  };
+}
 var registryData = null;
 var CACHE_DIR = () => path__default.default.join(os__default.default.homedir(), ".cache", "mastra");
 var CACHE_FILE = () => path__default.default.join(CACHE_DIR(), "gateway-refresh-time");
@@ -8245,16 +8359,22 @@ function saveLastRefreshTimeToDisk(date) {
 }
 function getPackageRoot() {
   try {
-    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-5WJWEAD2.cjs', document.baseURI).href)) || "file://");
+    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-XL636GZV.cjs', document.baseURI).href)) || "file://");
     const packageJsonPath = require2.resolve("@mastra/core/package.json");
     return path__default.default.dirname(packageJsonPath);
   } catch {
     return process.cwd();
   }
 }
-function loadRegistry(useDynamicLoading) {
+function loadRegistry(useDynamicLoading, customGateways = []) {
+  const enabledGatewayIds = getEnabledGatewayIds([
+    new chunkSH3TVIAI_cjs.ModelsDevGateway({}),
+    new chunk73LGQD54_cjs.NetlifyGateway(),
+    new MastraGateway(),
+    ...customGateways
+  ]);
   if (!useDynamicLoading) {
-    return provider_registry_default;
+    return sanitizeRegistryDataForRuntime(provider_registry_default, enabledGatewayIds);
   }
   syncGlobalCacheToLocal();
   if (registryData) {
@@ -8274,7 +8394,8 @@ function loadRegistry(useDynamicLoading) {
   for (const jsonPath of possiblePaths) {
     try {
       const content = fs__default.default.readFileSync(jsonPath, "utf-8");
-      registryData = JSON.parse(content);
+      const parsed = JSON.parse(content);
+      registryData = sanitizeRegistryDataForRuntime(parsed, enabledGatewayIds);
       return registryData;
     } catch (err) {
       const errorMessage = err instanceof Error ? err.message : String(err);
@@ -8289,7 +8410,7 @@ function loadRegistry(useDynamicLoading) {
           fs__default.default.unlinkSync(jsonPath);
         } catch {
         }
-        registryData = provider_registry_default;
+        registryData = sanitizeRegistryDataForRuntime(provider_registry_default, enabledGatewayIds);
         return registryData;
       }
       continue;
@@ -8300,7 +8421,7 @@ function loadRegistry(useDynamicLoading) {
 Tried paths:
 ${errors.join("\n")}`
   );
-  registryData = provider_registry_default;
+  registryData = sanitizeRegistryDataForRuntime(provider_registry_default, enabledGatewayIds);
   return registryData;
 }
 var PROVIDER_REGISTRY = new Proxy({}, {
@@ -8410,6 +8531,10 @@ var GatewayRegistry = class _GatewayRegistry {
   static getInstance(options) {
     if (!_GatewayRegistry.instance) {
       _GatewayRegistry.instance = new _GatewayRegistry(options);
+      return _GatewayRegistry.instance;
+    }
+    if (options?.useDynamicLoading === true) {
+      _GatewayRegistry.instance.useDynamicLoading = true;
     }
     return _GatewayRegistry.instance;
   }
@@ -8441,10 +8566,15 @@ var GatewayRegistry = class _GatewayRegistry {
     }
     this.isRefreshing = true;
     try {
-      const { ModelsDevGateway } = await import('./models-dev-ARAO7S6W.cjs');
-      const { NetlifyGateway } = await import('./netlify-XPEVGT6A.cjs');
-      const { fetchProvidersFromGateways, writeRegistryFiles } = await import('./registry-generator-PUWKZKVW.cjs');
-      const defaultGateways = [new ModelsDevGateway({}), new NetlifyGateway()];
+      const { ModelsDevGateway: ModelsDevGateway2 } = await import('./models-dev-TVW7HD3S.cjs');
+      const { NetlifyGateway: NetlifyGateway2 } = await import('./netlify-AEHADB4C.cjs');
+      const { MastraGateway: MastraGateway2 } = await import('./mastra-UBBD2C4R.cjs');
+      const { fetchProvidersFromGateways, writeRegistryFiles } = await import('./registry-generator-VNYH6GDH.cjs');
+      const defaultGateways = [
+        new ModelsDevGateway2({}),
+        new NetlifyGateway2(),
+        ...writeToSrc ? [] : [new MastraGateway2()]
+      ];
       const gateways = [...defaultGateways, ...this.customGateways];
       const { providers, models } = await fetchProvidersFromGateways(gateways);
       const packageRoot = getPackageRoot();
@@ -8529,28 +8659,28 @@ var GatewayRegistry = class _GatewayRegistry {
    * Get provider configuration by ID
    */
   getProviderConfig(providerId) {
-    const data = loadRegistry(this.useDynamicLoading);
+    const data = loadRegistry(this.useDynamicLoading, this.customGateways);
     return data.providers[providerId];
   }
   /**
    * Check if a provider is registered
    */
   isProviderRegistered(providerId) {
-    const data = loadRegistry(this.useDynamicLoading);
+    const data = loadRegistry(this.useDynamicLoading, this.customGateways);
     return providerId in data.providers;
   }
   /**
    * Get all registered providers
    */
   getProviders() {
-    const data = loadRegistry(this.useDynamicLoading);
+    const data = loadRegistry(this.useDynamicLoading, this.customGateways);
     return data.providers;
   }
   /**
    * Get all models
    */
   getModels() {
-    return loadRegistry(this.useDynamicLoading).models;
+    return loadRegistry(this.useDynamicLoading, this.customGateways).models;
   }
 };
 var isDev = process.env.MASTRA_DEV === "true" || process.env.MASTRA_DEV === "1";
@@ -8560,6 +8690,7 @@ if (autoRefreshEnabled) {
 }
 
 exports.GatewayRegistry = GatewayRegistry;
+exports.MastraGateway = MastraGateway;
 exports.PROVIDER_MODELS = PROVIDER_MODELS;
 exports.PROVIDER_REGISTRY = PROVIDER_REGISTRY;
 exports.getProviderConfig = getProviderConfig;
@@ -8567,5 +8698,5 @@ exports.getRegisteredProviders = getRegisteredProviders;
 exports.isProviderRegistered = isProviderRegistered;
 exports.isValidModelId = isValidModelId;
 exports.parseModelString = parseModelString;
-//# sourceMappingURL=chunk-5WJWEAD2.cjs.map
-//# sourceMappingURL=chunk-5WJWEAD2.cjs.map
+//# sourceMappingURL=chunk-XL636GZV.cjs.map
+//# sourceMappingURL=chunk-XL636GZV.cjs.map