@mastra/observability 1.14.1-alpha.0 → 1.14.1-alpha.1

package/dist/index.js

@@ -1430,18 +1430,23 @@ function emitAutoExtractedMetrics(span, metrics) {
 }
 function emitUsageMetrics(attrs, usage, metrics) {
   let metricCosts = /* @__PURE__ */ new Map();
-  try {
-    const provider = attrs.provider;
-    const model = attrs.responseModel ?? attrs.model;
-    if (provider && model) {
-      metricCosts = estimateCosts({
-        provider,
-        model,
-        usage
-      });
+  const providedCostContext = getProvidedCostContext(attrs, usage);
+  if (providedCostContext) {
+    metricCosts = providedCostContext;
+  } else {
+    try {
+      const provider = attrs.provider;
+      const model = attrs.responseModel ?? attrs.model;
+      if (provider && model) {
+        metricCosts = estimateCosts({
+          provider,
+          model,
+          usage
+        });
+      }
+    } catch {
+      metricCosts = /* @__PURE__ */ new Map();
     }
-  } catch {
-    metricCosts = /* @__PURE__ */ new Map();
   }
   const emit = (name, value) => {
     const costContext = metricCosts.get(name);
@@ -1455,6 +1460,32 @@ function emitUsageMetrics(attrs, usage, metrics) {
     emit(sample.name, sample.value);
   }
 }
+function getProvidedCostContext(attrs, usage) {
+  const costContext = attrs.costContext;
+  if (typeof costContext?.estimatedCost !== "number") {
+    return void 0;
+  }
+  const carrierMetric = usage.inputTokens !== void 0 ? TokenMetrics.TOTAL_INPUT : TokenMetrics.TOTAL_OUTPUT;
+  const provider = costContext.provider ?? attrs.provider;
+  const model = costContext.model ?? attrs.responseModel ?? attrs.model;
+  const contexts = /* @__PURE__ */ new Map();
+  for (const sample of getTokenMetricSamples(usage)) {
+    contexts.set(sample.name, {
+      provider,
+      model
+    });
+  }
+  contexts.set(carrierMetric, {
+    ...costContext,
+    provider,
+    model,
+    costMetadata: {
+      ...costContext.costMetadata,
+      allocation: "query_total"
+    }
+  });
+  return contexts;
+}
 function getDurationMetricName(span) {
   switch (span.type) {
     case SpanType.AGENT_RUN: