@mastra/observability 1.16.6-alpha.0 → 1.16.6-alpha.3

package/dist/index.js

@@ -1759,6 +1759,34 @@ function addUsageStats(a, b) {
 function supportsModelInference() {
 	return coreFeatures.has("model-inference-span");
 }
+function parseGatewayCost(providerMetadata) {
+	const rawCost = providerMetadata?.gateway?.cost;
+	const cost = typeof rawCost === "number" ? rawCost : typeof rawCost === "string" && rawCost.trim().length > 0 ? Number(rawCost) : void 0;
+	return typeof cost === "number" && Number.isFinite(cost) && cost >= 0 ? cost : void 0;
+}
+function getGatewayCostContext({ stepProviderMetadata }) {
+	if (!stepProviderMetadata) return;
+	if (!stepProviderMetadata.some((metadata) => metadata?.gateway !== void 0)) return;
+	const costs = [];
+	for (const metadata of stepProviderMetadata) {
+		const cost = parseGatewayCost(metadata);
+		if (cost === void 0) return;
+		costs.push(cost);
+	}
+	const estimatedCost = costs.reduce((total, cost) => total + cost, 0);
+	if (!Number.isFinite(estimatedCost)) return;
+	return {
+		estimatedCost,
+		costUnit: "USD",
+		costMetadata: {
+			source: "provider_reported",
+			sdkProvider: "vercel_ai_gateway",
+			sdkCostField: "gateway.cost",
+			scope: "query_total",
+			reportedStepCount: costs.length
+		}
+	};
+}
 function formatPreviewLabel(label, fallback) {
 	return typeof label === "string" && label.length > 0 ? label : fallback;
 }
@@ -1940,10 +1968,11 @@ var ModelSpanTracker = class {
 	* If usage is provided, it will be converted to UsageStats with cache token details.
 	*/
 	endGeneration(options) {
-		const { usage, providerMetadata, ...spanOptions } = options ?? {};
+		const { usage, providerMetadata, stepProviderMetadata, ...spanOptions } = options ?? {};
 		if (spanOptions.attributes) {
 			spanOptions.attributes.completionStartTime = this.#completionStartTime;
 			spanOptions.attributes.usage = extractUsageMetrics(usage, providerMetadata);
+			if (!spanOptions.attributes.costContext) spanOptions.attributes.costContext = getGatewayCostContext({ stepProviderMetadata });
 		}
 		this.#modelSpan?.end(spanOptions);
 	}
@@ -2027,13 +2056,15 @@ var ModelSpanTracker = class {
 		if (!this.#currentInferenceSpan) return;
 		const { usage: rawUsage, ...otherOutput } = payload.output;
 		const usage = extractUsageMetrics(rawUsage, payload.metadata?.providerMetadata);
+		const responseModel = typeof payload.metadata?.modelId === "string" ? payload.metadata.modelId : void 0;
 		this.#currentInferenceSpan.end({
 			output: otherOutput,
 			attributes: {
 				usage,
 				finishReason: payload.stepResult.reason,
 				warnings: payload.stepResult.warnings,
-				completionStartTime: this.#completionStartTime
+				completionStartTime: this.#completionStartTime,
+				...responseModel?.trim() ? { responseModel } : {}
 			}
 		});
 		this.#currentInferenceSpan = void 0;
@@ -2431,12 +2462,12 @@ function isSpanInternal(spanType, flags) {
 /**
 * Get the external parent span ID from CreateSpanOptions.
 *
-* If the parent is internal, walks up the parent chain to find
-* the closest external ancestor. If the parent is already external,
-* returns its ID directly.
+* If the parent is internal or excluded from export (`excludeSpanTypes`),
+* walks up the parent chain to find the closest exported ancestor.
+* If the parent is already exportable, returns its ID directly.
 *
 * This is useful when exporting spans to external observability systems
-* that shouldn't include internal framework spans.
+* that shouldn't include internal framework spans or excluded types.
 *
 * @param options - Span creation options
 * @returns The external parent span ID, or undefined if no external parent exists
@@ -2457,8 +2488,9 @@ function isSpanInternal(spanType, flags) {
 */
 function getExternalParentId(options) {
 	if (!options.parent) return;
-	if (options.parent.isInternal) return options.parent.getParentSpanId(false);
-	else return options.parent.id;
+	const parent = options.parent;
+	if (parent.isInternal || parent.isExcluded) return parent.getParentSpanId(false);
+	return parent.id;
 }
 var BaseSpan = class {
 	name;
@@ -2497,6 +2529,10 @@ var BaseSpan = class {
 	* when the span is internal and includeInternalSpans is false, or when
 	* the subclass is always excluded (e.g., NoOpSpan).
 	*
+	* Public so parent-chain walks (`getParentSpan`) can skip ancestors that
+	* exporters will never receive — otherwise descendants export a
+	* `parentSpanId` pointing at a dropped span and become orphans.
+	*
 	* Note: metadata is still attached and deepCleaned because it is read in
 	* process by getCorrelationContext() and by getLoggerContext() /
 	* getMetricsContext() (which structuredClone it).
@@ -2568,14 +2604,20 @@ var BaseSpan = class {
 	get isRootSpan() {
 		return !this.parent;
 	}
-	/** Get the closest parent span, optionally skipping internal spans */
+	/**
+	* Get the closest parent span.
+	* Always skips ancestors dropped by `excludeSpanTypes` (`isExcluded`), so
+	* exported `parentSpanId` values never point at spans exporters omit.
+	* When `includeInternalSpans` is false/undefined, also skips `isInternal`
+	* ancestors.
+	*/
 	getParentSpan(includeInternalSpans) {
 		if (!this.parent) return;
-		if (includeInternalSpans) return this.parent;
-		if (this.parent.isInternal) return this.parent.getParentSpan(includeInternalSpans);
+		const parent = this.parent;
+		if (parent.isExcluded || !includeInternalSpans && parent.isInternal) return parent.getParentSpan(includeInternalSpans);
 		return this.parent;
 	}
-	/** Get the closest parent spanId that isn't an internal span */
+	/** Get the closest parent spanId that will reach exporters (unless includeInternalSpans) */
 	getParentSpanId(includeInternalSpans) {
 		if (!this.parent) return this.parentSpanId;
 		const parentSpan = this.getParentSpan(includeInternalSpans);