@mastra/observability 0.0.0-fix-local-pkg-cwd-20251226155239 → 0.0.0-fix-11640-persist-workflow-events-to-memory-20260106220027

package/dist/index.cjs

@@ -3839,7 +3839,7 @@ ZodNaN.create = (params) => {
     ...processCreateParams(params)
   });
 };
-var BRAND = Symbol("zod_brand");
+var BRAND = /* @__PURE__ */ Symbol("zod_brand");
 var ZodBranded = class extends ZodType {
   _parse(input) {
     const { ctx } = this._processInputParams(input);
@@ -4076,6 +4076,12 @@ var samplingStrategySchema = external_exports.discriminatedUnion("type", [
     sampler: external_exports.function().args(external_exports.any().optional()).returns(external_exports.boolean())
   })
 ]);
+var serializationOptionsSchema = external_exports.object({
+  maxStringLength: external_exports.number().int().positive().optional(),
+  maxDepth: external_exports.number().int().positive().optional(),
+  maxArrayLength: external_exports.number().int().positive().optional(),
+  maxObjectKeys: external_exports.number().int().positive().optional()
+}).optional();
 var observabilityInstanceConfigSchema = external_exports.object({
   name: external_exports.string().min(1, "Name is required"),
   serviceName: external_exports.string().min(1, "Service name is required"),
@@ -4084,7 +4090,8 @@ var observabilityInstanceConfigSchema = external_exports.object({
   bridge: external_exports.any().optional(),
   spanOutputProcessors: external_exports.array(external_exports.any()).optional(),
   includeInternalSpans: external_exports.boolean().optional(),
-  requestContextKeys: external_exports.array(external_exports.string()).optional()
+  requestContextKeys: external_exports.array(external_exports.string()).optional(),
+  serializationOptions: serializationOptionsSchema
 }).refine(
   (data) => {
     const hasExporters = data.exporters && data.exporters.length > 0;
@@ -4102,7 +4109,8 @@ var observabilityConfigValueSchema = external_exports.object({
   bridge: external_exports.any().optional(),
   spanOutputProcessors: external_exports.array(external_exports.any()).optional(),
   includeInternalSpans: external_exports.boolean().optional(),
-  requestContextKeys: external_exports.array(external_exports.string()).optional()
+  requestContextKeys: external_exports.array(external_exports.string()).optional(),
+  serializationOptions: serializationOptionsSchema
 }).refine(
   (data) => {
     const hasExporters = data.exporters && data.exporters.length > 0;
@@ -5136,9 +5144,14 @@ var ModelSpanTracker = class {
     this.#modelSpan?.update(options);
   }
   /**
-   * Start a new Model execution step
+   * Start a new Model execution step.
+   * This should be called at the beginning of LLM execution to capture accurate startTime.
+   * The step-start chunk payload can be passed later via updateStep() if needed.
    */
-  #startStepSpan(payload) {
+  startStep(payload) {
+    if (this.#currentStepSpan) {
+      return;
+    }
     this.#currentStepSpan = this.#modelSpan?.createChildSpan({
       name: `step: ${this.#stepIndex}`,
       type: observability.SpanType.MODEL_STEP,
@@ -5151,6 +5164,22 @@ var ModelSpanTracker = class {
     });
     this.#chunkSequence = 0;
   }
+  /**
+   * Update the current step span with additional payload data.
+   * Called when step-start chunk arrives with request/warnings info.
+   */
+  updateStep(payload) {
+    if (!this.#currentStepSpan || !payload) {
+      return;
+    }
+    this.#currentStepSpan.update({
+      input: payload.request,
+      attributes: {
+        ...payload.messageId ? { messageId: payload.messageId } : {},
+        ...payload.warnings?.length ? { warnings: payload.warnings } : {}
+      }
+    });
+  }
   /**
    * End the current Model execution step with token usage, finish reason, output, and metadata
    */
@@ -5185,7 +5214,7 @@ var ModelSpanTracker = class {
    */
   #startChunkSpan(chunkType, initialData) {
     if (!this.#currentStepSpan) {
-      this.#startStepSpan();
+      this.startStep();
     }
     this.#currentChunkSpan = this.#currentStepSpan?.createChildSpan({
       name: `chunk: '${chunkType}'`,
@@ -5225,7 +5254,7 @@ var ModelSpanTracker = class {
    */
   #createEventSpan(chunkType, output) {
     if (!this.#currentStepSpan) {
-      this.#startStepSpan();
+      this.startStep();
     }
     const span = this.#currentStepSpan?.createEventSpan({
       name: `chunk: '${chunkType}'`,
@@ -5344,7 +5373,7 @@ var ModelSpanTracker = class {
     let acc = this.#toolOutputAccumulators.get(toolCallId);
     if (!acc) {
       if (!this.#currentStepSpan) {
-        this.#startStepSpan();
+        this.startStep();
       }
       acc = {
         toolName: toolName || "unknown",
@@ -5444,7 +5473,11 @@ var ModelSpanTracker = class {
               this.#handleObjectChunk(chunk);
               break;
             case "step-start":
-              this.#startStepSpan(chunk.payload);
+              if (this.#currentStepSpan) {
+                this.updateStep(chunk.payload);
+              } else {
+                this.startStep(chunk.payload);
+              }
               break;
             case "step-finish":
               this.#endStepSpan(chunk.payload);
@@ -5503,6 +5536,18 @@ var DEFAULT_DEEP_CLEAN_OPTIONS = Object.freeze({
   maxArrayLength: 50,
   maxObjectKeys: 50
 });
+function mergeSerializationOptions(userOptions) {
+  if (!userOptions) {
+    return DEFAULT_DEEP_CLEAN_OPTIONS;
+  }
+  return {
+    keysToStrip: DEFAULT_KEYS_TO_STRIP,
+    maxDepth: userOptions.maxDepth ?? DEFAULT_DEEP_CLEAN_OPTIONS.maxDepth,
+    maxStringLength: userOptions.maxStringLength ?? DEFAULT_DEEP_CLEAN_OPTIONS.maxStringLength,
+    maxArrayLength: userOptions.maxArrayLength ?? DEFAULT_DEEP_CLEAN_OPTIONS.maxArrayLength,
+    maxObjectKeys: userOptions.maxObjectKeys ?? DEFAULT_DEEP_CLEAN_OPTIONS.maxObjectKeys
+  };
+}
 function truncateString(s, maxChars) {
   if (s.length <= maxChars) {
     return s;
@@ -5659,11 +5704,15 @@ var BaseSpan = class {
   entityName;
   /** Parent span ID (for root spans that are children of external spans) */
   parentSpanId;
+  /** Deep clean options for serialization */
+  deepCleanOptions;
   constructor(options, observabilityInstance) {
+    const serializationOptions = observabilityInstance.getConfig().serializationOptions;
+    this.deepCleanOptions = mergeSerializationOptions(serializationOptions);
     this.name = options.name;
     this.type = options.type;
-    this.attributes = deepClean(options.attributes) || {};
-    this.metadata = deepClean(options.metadata);
+    this.attributes = deepClean(options.attributes, this.deepCleanOptions) || {};
+    this.metadata = deepClean(options.metadata, this.deepCleanOptions);
     this.parent = options.parent;
     this.startTime = /* @__PURE__ */ new Date();
     this.observabilityInstance = observabilityInstance;
@@ -5675,9 +5724,9 @@ var BaseSpan = class {
     this.entityId = options.entityId;
     this.entityName = options.entityName;
     if (this.isEvent) {
-      this.output = deepClean(options.output);
+      this.output = deepClean(options.output, this.deepCleanOptions);
     } else {
-      this.input = deepClean(options.input);
+      this.input = deepClean(options.input, this.deepCleanOptions);
     }
   }
   createChildSpan(options) {
@@ -5809,13 +5858,13 @@ var DefaultSpan = class extends BaseSpan {
     }
     this.endTime = /* @__PURE__ */ new Date();
     if (options?.output !== void 0) {
-      this.output = deepClean(options.output);
+      this.output = deepClean(options.output, this.deepCleanOptions);
     }
     if (options?.attributes) {
-      this.attributes = { ...this.attributes, ...deepClean(options.attributes) };
+      this.attributes = { ...this.attributes, ...deepClean(options.attributes, this.deepCleanOptions) };
     }
     if (options?.metadata) {
-      this.metadata = { ...this.metadata, ...deepClean(options.metadata) };
+      this.metadata = { ...this.metadata, ...deepClean(options.metadata, this.deepCleanOptions) };
     }
   }
   error(options) {
@@ -5833,10 +5882,10 @@ var DefaultSpan = class extends BaseSpan {
       message: error$1.message
     };
     if (attributes) {
-      this.attributes = { ...this.attributes, ...deepClean(attributes) };
+      this.attributes = { ...this.attributes, ...deepClean(attributes, this.deepCleanOptions) };
     }
     if (metadata) {
-      this.metadata = { ...this.metadata, ...deepClean(metadata) };
+      this.metadata = { ...this.metadata, ...deepClean(metadata, this.deepCleanOptions) };
     }
     if (endSpan) {
       this.end();
@@ -5849,16 +5898,16 @@ var DefaultSpan = class extends BaseSpan {
       return;
     }
     if (options.input !== void 0) {
-      this.input = deepClean(options.input);
+      this.input = deepClean(options.input, this.deepCleanOptions);
     }
     if (options.output !== void 0) {
-      this.output = deepClean(options.output);
+      this.output = deepClean(options.output, this.deepCleanOptions);
     }
     if (options.attributes) {
-      this.attributes = { ...this.attributes, ...deepClean(options.attributes) };
+      this.attributes = { ...this.attributes, ...deepClean(options.attributes, this.deepCleanOptions) };
     }
     if (options.metadata) {
-      this.metadata = { ...this.metadata, ...deepClean(options.metadata) };
+      this.metadata = { ...this.metadata, ...deepClean(options.metadata, this.deepCleanOptions) };
     }
   }
   get isValid() {
@@ -5949,7 +5998,8 @@ var BaseObservabilityInstance = class extends base.MastraBase {
       spanOutputProcessors: config.spanOutputProcessors ?? [],
       bridge: config.bridge ?? void 0,
       includeInternalSpans: config.includeInternalSpans ?? false,
-      requestContextKeys: config.requestContextKeys ?? []
+      requestContextKeys: config.requestContextKeys ?? [],
+      serializationOptions: config.serializationOptions
     };
     if (this.config.bridge?.init) {
       this.config.bridge.init({ config: this.config });
@@ -6428,6 +6478,9 @@ var SensitiveDataFilter = class {
       return "[Circular Reference]";
     }
     seen.add(obj);
+    if (obj instanceof Date) {
+      return obj;
+    }
     if (Array.isArray(obj)) {
       return obj.map((item) => this.deepFilter(item, seen));
     }
@@ -6666,10 +6719,12 @@ exports.TestExporter = TestExporter;
 exports.buildTracingOptions = buildTracingOptions;
 exports.deepClean = deepClean;
 exports.getExternalParentId = getExternalParentId;
+exports.mergeSerializationOptions = mergeSerializationOptions;
 exports.observabilityConfigValueSchema = observabilityConfigValueSchema;
 exports.observabilityInstanceConfigSchema = observabilityInstanceConfigSchema;
 exports.observabilityRegistryConfigSchema = observabilityRegistryConfigSchema;
 exports.samplingStrategySchema = samplingStrategySchema;
+exports.serializationOptionsSchema = serializationOptionsSchema;
 exports.truncateString = truncateString;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map