@mastra/observability 1.11.0 → 1.11.1-alpha.0

package/dist/index.js

@@ -18733,7 +18733,11 @@ function summarizeRequestBody(body) {
   }
   return Object.keys(summary).length > 0 ? summary : "[request body]";
 }
-function extractStepInput(request) {
+function extractStepInput(payload) {
+  if (Array.isArray(payload?.inputMessages)) {
+    return normalizeMessages(payload.inputMessages);
+  }
+  const request = payload?.request;
   if (!request) return void 0;
   const { body } = request;
   if (body == null) return request;
@@ -18753,6 +18757,7 @@ var ModelSpanTracker = class {
   #stepIndex = 0;
   #chunkSequence = 0;
   #completionStartTime;
+  #currentStepInputIsFinal = false;
   /** When true, step-finish chunks don't auto-close the step span (for durable execution) */
   #deferStepClose = false;
   /** Stored step-finish payload when defer mode is enabled */
@@ -18846,6 +18851,7 @@ var ModelSpanTracker = class {
     if (this.#currentStepSpan) {
       return;
     }
+    const input = extractStepInput(payload);
     this.#currentStepSpan = this.#modelSpan?.createChildSpan({
       name: `step: ${this.#stepIndex}`,
       type: SpanType.MODEL_STEP,
@@ -18854,8 +18860,9 @@ var ModelSpanTracker = class {
         ...payload?.messageId ? { messageId: payload.messageId } : {},
         ...payload?.warnings?.length ? { warnings: payload.warnings } : {}
       },
-      input: extractStepInput(payload?.request)
+      input
     });
+    this.#currentStepInputIsFinal = Array.isArray(payload?.inputMessages);
     this.#chunkSequence = 0;
   }
   /**
@@ -18866,13 +18873,18 @@ var ModelSpanTracker = class {
     if (!this.#currentStepSpan || !payload) {
       return;
     }
+    const hasFinalInput = Array.isArray(payload.inputMessages);
+    const input = hasFinalInput || !this.#currentStepInputIsFinal ? extractStepInput(payload) : void 0;
     this.#currentStepSpan.update({
-      input: extractStepInput(payload.request),
+      ...input !== void 0 ? { input } : {},
       attributes: {
         ...payload.messageId ? { messageId: payload.messageId } : {},
         ...payload.warnings?.length ? { warnings: payload.warnings } : {}
       }
     });
+    if (hasFinalInput) {
+      this.#currentStepInputIsFinal = true;
+    }
   }
   /**
    * End the current Model execution step with token usage, finish reason, output, and metadata
@@ -18904,6 +18916,7 @@ var ModelSpanTracker = class {
       }
     });
     this.#currentStepSpan = void 0;
+    this.#currentStepInputIsFinal = false;
     this.#stepIndex++;
   }
   /**
@@ -19479,7 +19492,8 @@ var BaseSpan = class {
   async executeInContext(fn) {
     const bridge = this.observabilityInstance.getBridge();
     if (bridge?.executeInContext) {
-      return bridge.executeInContext(this.id, fn);
+      const bridgeContextSpan = this.isInternal ? this.getParentSpan(false) : this;
+      return bridge.executeInContext(bridgeContextSpan?.id ?? this.id, fn);
     }
     return fn();
   }
@@ -19490,7 +19504,8 @@ var BaseSpan = class {
   executeInContextSync(fn) {
     const bridge = this.observabilityInstance.getBridge();
     if (bridge?.executeInContextSync) {
-      return bridge.executeInContextSync(this.id, fn);
+      const bridgeContextSpan = this.isInternal ? this.getParentSpan(false) : this;
+      return bridge.executeInContextSync(bridgeContextSpan?.id ?? this.id, fn);
     }
     return fn();
   }