@mastra/core 1.21.0-alpha.2 → 1.22.0-alpha.0

package/dist/{chunk-ZRP4BAK4.js → chunk-B7PS7GR7.js}

@@ -1,9 +1,102 @@
+import { ModelsDevGateway, createOpenRouter } from './chunk-RZISMMDM.js';
+import { NetlifyGateway } from './chunk-46DASRFB.js';
+import { MastraModelGateway, createAnthropic, MASTRA_USER_AGENT, GATEWAY_AUTH_HEADER } from './chunk-EFXVHFAI.js';
+import { MastraError } from './chunk-FJEVLHJT.js';
 import fs from 'fs';
 import { createRequire } from 'module';
 import os from 'os';
 import path from 'path';
 
-// src/llm/model/provider-registry.ts
+// src/llm/model/gateways/mastra.ts
+var MastraGateway = class extends 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 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 createAnthropic({
+        apiKey: "oauth-gateway-placeholder",
+        baseURL,
+        headers: {
+          "User-Agent": MASTRA_USER_AGENT,
+          [GATEWAY_AUTH_HEADER]: `Bearer ${apiKey}`,
+          ...headers
+        },
+        fetch: this.config.customFetch
+      })(modelId);
+    }
+    if (this.config?.customFetch) {
+      return createOpenRouter({
+        apiKey: "oauth-gateway-placeholder",
+        baseURL,
+        headers: {
+          "User-Agent": MASTRA_USER_AGENT,
+          [GATEWAY_AUTH_HEADER]: `Bearer ${apiKey}`,
+          ...headers
+        },
+        fetch: this.config.customFetch
+      }).chat(fullModelId);
+    }
+    return createOpenRouter({
+      apiKey,
+      baseURL,
+      headers: {
+        "User-Agent": MASTRA_USER_AGENT,
+        ...headers
+      }
+    }).chat(fullModelId);
+  }
+};
 
 // src/llm/model/provider-registry.json
 var provider_registry_default = {
@@ -8130,6 +8223,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.join(os.homedir(), ".cache", "mastra");
 var CACHE_FILE = () => path.join(CACHE_DIR(), "gateway-refresh-time");
@@ -8243,9 +8357,15 @@ function getPackageRoot() {
     return process.cwd();
   }
 }
-function loadRegistry(useDynamicLoading) {
+function loadRegistry(useDynamicLoading, customGateways = []) {
+  const enabledGatewayIds = getEnabledGatewayIds([
+    new ModelsDevGateway({}),
+    new NetlifyGateway(),
+    new MastraGateway(),
+    ...customGateways
+  ]);
   if (!useDynamicLoading) {
-    return provider_registry_default;
+    return sanitizeRegistryDataForRuntime(provider_registry_default, enabledGatewayIds);
   }
   syncGlobalCacheToLocal();
   if (registryData) {
@@ -8265,7 +8385,8 @@ function loadRegistry(useDynamicLoading) {
   for (const jsonPath of possiblePaths) {
     try {
       const content = fs.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);
@@ -8280,7 +8401,7 @@ function loadRegistry(useDynamicLoading) {
           fs.unlinkSync(jsonPath);
         } catch {
         }
-        registryData = provider_registry_default;
+        registryData = sanitizeRegistryDataForRuntime(provider_registry_default, enabledGatewayIds);
         return registryData;
       }
       continue;
@@ -8291,7 +8412,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({}, {
@@ -8401,6 +8522,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;
   }
@@ -8432,10 +8557,15 @@ var GatewayRegistry = class _GatewayRegistry {
     }
     this.isRefreshing = true;
     try {
-      const { ModelsDevGateway } = await import('./models-dev-PGMVTTO7.js');
-      const { NetlifyGateway } = await import('./netlify-3BL6W3LY.js');
-      const { fetchProvidersFromGateways, writeRegistryFiles } = await import('./registry-generator-BRPCL2LG.js');
-      const defaultGateways = [new ModelsDevGateway({}), new NetlifyGateway()];
+      const { ModelsDevGateway: ModelsDevGateway2 } = await import('./models-dev-DA5OEOGQ.js');
+      const { NetlifyGateway: NetlifyGateway2 } = await import('./netlify-EENUHCAW.js');
+      const { MastraGateway: MastraGateway2 } = await import('./mastra-2EQZYM7C.js');
+      const { fetchProvidersFromGateways, writeRegistryFiles } = await import('./registry-generator-QMLHG25G.js');
+      const defaultGateways = [
+        new ModelsDevGateway2({}),
+        new NetlifyGateway2(),
+        ...writeToSrc ? [] : [new MastraGateway2()]
+      ];
       const gateways = [...defaultGateways, ...this.customGateways];
       const { providers, models } = await fetchProvidersFromGateways(gateways);
       const packageRoot = getPackageRoot();
@@ -8520,28 +8650,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";
@@ -8550,6 +8680,6 @@ if (autoRefreshEnabled) {
   GatewayRegistry.getInstance({ useDynamicLoading: isDev }).startAutoRefresh();
 }
 
-export { GatewayRegistry, PROVIDER_MODELS, PROVIDER_REGISTRY, getProviderConfig, getRegisteredProviders, isProviderRegistered, isValidModelId, parseModelString };
-//# sourceMappingURL=chunk-ZRP4BAK4.js.map
-//# sourceMappingURL=chunk-ZRP4BAK4.js.map
+export { GatewayRegistry, MastraGateway, PROVIDER_MODELS, PROVIDER_REGISTRY, getProviderConfig, getRegisteredProviders, isProviderRegistered, isValidModelId, parseModelString };
+//# sourceMappingURL=chunk-B7PS7GR7.js.map
+//# sourceMappingURL=chunk-B7PS7GR7.js.map