@mastra/observability 1.19.0-alpha.2 → 1.19.0-alpha.4

package/dist/index.js

@@ -2240,6 +2240,7 @@ var ModelSpanTracker = class {
 	#stepIndex = 0;
 	#chunkSequence = 0;
 	#completionStartTime;
+	#inferenceCompletionStartTime;
 	#currentStepInputIsFinal = false;
 	/** When true, step-finish chunks don't auto-close the step span (for durable execution) */
 	#deferStepClose = false;
@@ -2261,8 +2262,9 @@ var ModelSpanTracker = class {
 	* Capture the completion start time (time to first token) when the first content chunk arrives.
 	*/
 	#captureCompletionStartTime() {
-		if (this.#completionStartTime) return;
-		this.#completionStartTime = /* @__PURE__ */ new Date();
+		const now = /* @__PURE__ */ new Date();
+		this.#completionStartTime ??= now;
+		this.#inferenceCompletionStartTime ??= now;
 	}
 	/**
 	* Get the tracing context for creating child spans.
@@ -2400,7 +2402,7 @@ var ModelSpanTracker = class {
 				usage,
 				finishReason: payload.stepResult.reason,
 				warnings: payload.stepResult.warnings,
-				completionStartTime: this.#completionStartTime,
+				completionStartTime: this.#inferenceCompletionStartTime,
 				...responseModel?.trim() ? { responseModel } : {}
 			}
 		});
@@ -2424,6 +2426,7 @@ var ModelSpanTracker = class {
 	startInference(payload) {
 		if (!supportsModelInference()) return;
 		if (!this.#currentStepSpan || this.#currentInferenceSpan) return;
+		this.#inferenceCompletionStartTime = void 0;
 		const input = extractStepInput(payload);
 		const generationAttrs = this.#modelSpan?.attributes;
 		const ctx = this.#inferenceContext;
@@ -2447,6 +2450,16 @@ var ModelSpanTracker = class {
 		});
 	}
 	/**
+	* End the open MODEL_INFERENCE span as interrupted while keeping MODEL_STEP open,
+	* so a replacement request for the same step gets its own inference span.
+	*/
+	interruptInference() {
+		this.#endChunkSpan();
+		if (!this.#currentInferenceSpan) return;
+		this.#currentInferenceSpan.end({ attributes: { finishReason: "interrupted" } });
+		this.#currentInferenceSpan = void 0;
+	}
+	/**
 	* Update the current step span with additional payload data.
 	* Called when step-start chunk arrives with request/warnings info.
 	*/