@mastra/observability 1.16.1 → 1.16.2-alpha.0

package/dist/index.js

@@ -999,6 +999,13 @@ var PricingModel = class {
 // src/metrics/pricing-registry.ts
 var DATA_FILE_NAME = "pricing-data.jsonl";
 var BEDROCK_GEOGRAPHY_PREFIXES = /* @__PURE__ */ new Set(["global", "us", "eu", "apac", "jp", "au"]);
+var AI_SDK_VERCEL_GATEWAY_PROVIDER_ID = "gateway";
+var VERCEL_PRICING_PROVIDER_ID = "vercel";
+var AI_SDK_PROVIDER_NAMESPACE_ALIASES = /* @__PURE__ */ new Map([
+  ["google.vertex", "google-vertex"],
+  ["vertex.anthropic", "google-vertex-anthropic"],
+  ["vertex.maas", "google-vertex"]
+]);
 var MINIFIED_METER_TO_CANONICAL = {
   it: "input_tokens",
   ot: "output_tokens",
@@ -1033,11 +1040,13 @@ var PricingRegistry = class _PricingRegistry {
     return _PricingRegistry.globalRegistry;
   }
   get(args) {
-    const variants = getModelVariants(args.model, normalizeProvider(args.provider));
-    for (const variant of variants) {
-      const key = makePricingKey({ provider: args.provider, model: variant });
-      const match = this.pricingModels.get(key);
-      if (match) return match;
+    for (const provider of getPricingProviderCandidates(args.provider, args.model)) {
+      const variants = getModelVariants(args.model, provider);
+      for (const variant of variants) {
+        const key = makePricingKey({ provider, model: variant });
+        const match = this.pricingModels.get(key);
+        if (match) return match;
+      }
     }
     return null;
   }
@@ -1117,15 +1126,41 @@ function getPackageRoot() {
   }
 }
 function makePricingKey(args) {
-  return `${normalizeProvider(args.provider)}::${normalizeKeyPart(args.model)}`;
+  return `${normalizeKeyPart(args.provider)}::${normalizeKeyPart(args.model)}`;
 }
 function normalizeKeyPart(value) {
   return value.trim().toLowerCase();
 }
-function normalizeProvider(provider) {
-  const normalized = provider.trim().toLowerCase();
-  const dotIndex = normalized.indexOf(".");
-  return dotIndex !== -1 ? normalized.substring(0, dotIndex) : normalized;
+function getPricingProviderCandidates(provider, model) {
+  const normalizedProvider = normalizeKeyPart(provider);
+  const providerCandidates = getNamespacedProviderCandidates(normalizedProvider) ?? getBaseProviderCandidates(normalizedProvider, model);
+  return [.../* @__PURE__ */ new Set([normalizedProvider, ...providerCandidates])];
+}
+function getNamespacedProviderCandidates(provider) {
+  for (const [providerNamespace, pricingProvider] of AI_SDK_PROVIDER_NAMESPACE_ALIASES) {
+    if (matchesProviderNamespace(provider, providerNamespace)) {
+      return [providerNamespace, pricingProvider];
+    }
+  }
+  return null;
+}
+function matchesProviderNamespace(provider, providerNamespace) {
+  return provider === providerNamespace || provider.startsWith(`${providerNamespace}.`);
+}
+function getBaseProviderCandidates(provider, model) {
+  const baseProvider = getBaseProvider(provider);
+  return isVercelGatewayModel(baseProvider, model) ? [baseProvider, VERCEL_PRICING_PROVIDER_ID] : [baseProvider];
+}
+function getBaseProvider(provider) {
+  const capabilitySeparator = provider.indexOf(".");
+  return capabilitySeparator === -1 ? provider : provider.substring(0, capabilitySeparator);
+}
+function isVercelGatewayModel(provider, model) {
+  return provider === AI_SDK_VERCEL_GATEWAY_PROVIDER_ID && hasCreatorModelIdShape(model);
+}
+function hasCreatorModelIdShape(model) {
+  const segments = model.split("/");
+  return segments.length === 2 && segments.every((segment) => segment.trim().length > 0);
 }
 function getModelVariants(model, provider) {
   const variants = /* @__PURE__ */ new Set();
@@ -1161,7 +1196,9 @@ function getModelVariants(model, provider) {
   return [...variants];
 }
 function stripDateSuffix(model) {
-  let stripped = model.replace(/-20\d{2}-\d{2}-\d{2}$/, "");
+  let stripped = model.replace(/@20\d{6}$/, "");
+  if (stripped !== model) return stripped;
+  stripped = model.replace(/-20\d{2}-\d{2}-\d{2}$/, "");
   if (stripped !== model) return stripped;
   stripped = model.replace(/-20\d{6}(-[a-z]+)?$/, "$1");
   if (stripped !== model) return stripped;