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

package/dist/index.js

@@ -3837,7 +3837,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);
@@ -4074,6 +4074,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"),
@@ -4082,7 +4088,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;
@@ -4100,7 +4107,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;
@@ -5134,9 +5142,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: SpanType.MODEL_STEP,
@@ -5149,6 +5162,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
    */
@@ -5183,7 +5212,7 @@ var ModelSpanTracker = class {
    */
   #startChunkSpan(chunkType, initialData) {
     if (!this.#currentStepSpan) {
-      this.#startStepSpan();
+      this.startStep();
     }
     this.#currentChunkSpan = this.#currentStepSpan?.createChildSpan({
       name: `chunk: '${chunkType}'`,
@@ -5223,7 +5252,7 @@ var ModelSpanTracker = class {
    */
   #createEventSpan(chunkType, output) {
     if (!this.#currentStepSpan) {
-      this.#startStepSpan();
+      this.startStep();
     }
     const span = this.#currentStepSpan?.createEventSpan({
       name: `chunk: '${chunkType}'`,
@@ -5342,7 +5371,7 @@ var ModelSpanTracker = class {
     let acc = this.#toolOutputAccumulators.get(toolCallId);
     if (!acc) {
       if (!this.#currentStepSpan) {
-        this.#startStepSpan();
+        this.startStep();
       }
       acc = {
         toolName: toolName || "unknown",
@@ -5442,7 +5471,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);
@@ -5501,6 +5534,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;
@@ -5657,11 +5702,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;
@@ -5673,9 +5722,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) {
@@ -5807,13 +5856,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) {
@@ -5831,10 +5880,10 @@ var DefaultSpan = class extends BaseSpan {
       message: error.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();
@@ -5847,16 +5896,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() {
@@ -5947,7 +5996,8 @@ var BaseObservabilityInstance = class extends 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 });
@@ -6426,6 +6476,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));
     }
@@ -6645,6 +6698,6 @@ function buildTracingOptions(...updaters) {
   return updaters.reduce((opts, updater) => updater(opts), {});
 }
 
-export { BaseExporter, BaseObservabilityInstance, BaseSpan, CloudExporter, ConsoleExporter, DEFAULT_DEEP_CLEAN_OPTIONS, DEFAULT_KEYS_TO_STRIP, DefaultExporter, DefaultObservabilityInstance, DefaultSpan, ModelSpanTracker, NoOpSpan, Observability, SamplingStrategyType, SensitiveDataFilter, TestExporter, buildTracingOptions, deepClean, getExternalParentId, observabilityConfigValueSchema, observabilityInstanceConfigSchema, observabilityRegistryConfigSchema, samplingStrategySchema, truncateString };
+export { BaseExporter, BaseObservabilityInstance, BaseSpan, CloudExporter, ConsoleExporter, DEFAULT_DEEP_CLEAN_OPTIONS, DEFAULT_KEYS_TO_STRIP, DefaultExporter, DefaultObservabilityInstance, DefaultSpan, ModelSpanTracker, NoOpSpan, Observability, SamplingStrategyType, SensitiveDataFilter, TestExporter, buildTracingOptions, deepClean, getExternalParentId, mergeSerializationOptions, observabilityConfigValueSchema, observabilityInstanceConfigSchema, observabilityRegistryConfigSchema, samplingStrategySchema, serializationOptionsSchema, truncateString };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map