@mastra/observability 1.17.2-alpha.1 → 1.17.2

package/dist/index.js

@@ -2693,7 +2693,7 @@ function getExternalParentId(options) {
 	if (parent.isInternal || parent.isExcluded) return parent.getParentSpanId(false);
 	return parent.id;
 }
-var BaseSpan = class {
+var BaseSpan = class BaseSpan {
 	name;
 	type;
 	attributes;
@@ -2743,6 +2743,8 @@ var BaseSpan = class {
 	isExcluded;
 	/** Cached canonical correlation context for this live span */
 	correlationContext;
+	/** Child spans that have started but not yet ended */
+	#openChildren;
 	/**
 	* Subclasses can override to unconditionally mark the span as excluded.
 	* NoOpSpan uses this because it is never exported regardless of config.
@@ -2764,6 +2766,8 @@ var BaseSpan = class {
 		this.isEvent = options.isEvent ?? false;
 		this.tracingPolicy = options.tracingPolicy;
 		this.traceState = options.traceState;
+		const parent = options.parent;
+		if (!this.isEvent && parent instanceof BaseSpan && parent.isValid) (parent.#openChildren ??= /* @__PURE__ */ new Set()).add(this);
 		this.tags = !options.parent && options.tags?.length ? options.tags : void 0;
 		const entityParent = this.getParentSpan(false);
 		this.entityType = options.entityType ?? entityParent?.entityType;
@@ -2801,6 +2805,16 @@ var BaseSpan = class {
 			return value;
 		}
 	}
+	/** End the span and any descendant spans that are still open, applying `options` to each */
+	endTree(options) {
+		if (this.#openChildren) for (const child of [...this.#openChildren]) child.endTree(options);
+		this.end(options);
+	}
+	/** Release this span from its parent's open-child set once it has ended */
+	detachFromParent() {
+		const parent = this.parent;
+		if (parent instanceof BaseSpan) parent.#openChildren?.delete(this);
+	}
 	createChildSpan(options) {
 		return this.observabilityInstance.startSpan({
 			...options,
@@ -3004,7 +3018,9 @@ var DefaultSpan = class extends BaseSpan {
 	}
 	end(options) {
 		if (this.isEvent) return;
+		if (this.endTime) return;
 		this.endTime = /* @__PURE__ */ new Date();
+		this.detachFromParent();
 		if (options?.metadata) this.metadata = {
 			...this.metadata,
 			...deepClean(this.prepareSpanMetadata(options.metadata), this.deepCleanOptions)
@@ -3460,6 +3476,7 @@ var BaseObservabilityInstance = class extends MastraBase {
 				this.logger.warn(`End event is not available on event spans`);
 				return;
 			}
+			if (span.endTime) return;
 			const rollupTarget = this.captureModelUsageRollup(span, options);
 			const excludedModelUsage = this.captureExcludedModelUsage(span, options);
 			originalEnd(options);