@mastra/observability 1.19.0-alpha.3 → 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;