@mastra/core 0.21.2-alpha.0 → 0.22.0

package/dist/ai-tracing/chunk-tracing.d.ts

@@ -0,0 +1,76 @@
+/**
+ * Model Model Span Tracing
+ *
+ * Provides span tracking for Model generations, including:
+ * - LLM_STEP spans (one per Model API call)
+ * - LLM_CHUNK spans (individual streaming chunks within a step)
+ *
+ * Hierarchy: LLM_GENERATION -> LLM_STEP -> LLM_CHUNK
+ */
+import type { ToolSet } from 'ai-v5';
+import type { OutputSchema } from '../stream/base/schema.js';
+import type { StepStartPayload, StepFinishPayload } from '../stream/types.js';
+import { AISpanType } from './types.js';
+import type { AISpan } from './types.js';
+/**
+ * Manages LLM_STEP and LLM_CHUNK span tracking for streaming Model responses.
+ *
+ * Should be instantiated once per LLM_GENERATION span and shared across
+ * all streaming steps (including after tool calls).
+ */
+export declare class ModelSpanTracker {
+    private modelSpan?;
+    private currentStepSpan?;
+    private currentChunkSpan?;
+    private accumulator;
+    private stepIndex;
+    private chunkSequence;
+    constructor(modelSpan?: AISpan<AISpanType.LLM_GENERATION>);
+    /**
+     * Start a new Model execution step
+     */
+    startStepSpan(payload?: StepStartPayload): void;
+    /**
+     * End the current Model execution step with token usage, finish reason, output, and metadata
+     */
+    endStepSpan<OUTPUT extends OutputSchema>(payload: StepFinishPayload<ToolSet, OUTPUT>): void;
+    /**
+     * Create a new chunk span (for multi-part chunks like text-start/delta/end)
+     */
+    startChunkSpan(chunkType: string, initialData?: Record<string, any>): void;
+    /**
+     * Append string content to a specific field in the accumulator
+     */
+    appendToAccumulator(field: string, text: string): void;
+    /**
+     * End the current chunk span.
+     * Safe to call multiple times - will no-op if span already ended.
+     */
+    endChunkSpan(output?: any): void;
+    /**
+     * Create an event span (for single chunks like tool-call)
+     */
+    createEventSpan(chunkType: string, output: any): void;
+    /**
+     * Check if there is currently an active chunk span
+     */
+    hasActiveChunkSpan(): boolean;
+    /**
+     * Get the current accumulator value
+     */
+    getAccumulator(): Record<string, any>;
+    /**
+     * Get the current step span (for making tool calls children of steps)
+     */
+    getCurrentStepSpan(): AISpan<AISpanType.LLM_STEP> | undefined;
+    /**
+     * Wraps a stream with model tracing transform to track LLM_STEP and LLM_CHUNK spans.
+     *
+     * This should be added to the stream pipeline to automatically
+     * create LLM_STEP and LLM_CHUNK spans for each semantic unit in the stream.
+     */
+    wrapStream<T extends {
+        pipeThrough: Function;
+    }>(stream: T): T;
+}
+//# sourceMappingURL=chunk-tracing.d.ts.map

package/dist/ai-tracing/chunk-tracing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunk-tracing.d.ts","sourceRoot":"","sources":["../../src/ai-tracing/chunk-tracing.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAGH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACrC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,KAAK,EAAa,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACtF,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEtC;;;;;GAKG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,SAAS,CAAC,CAAoC;IACtD,OAAO,CAAC,eAAe,CAAC,CAA8B;IACtD,OAAO,CAAC,gBAAgB,CAAC,CAA+B;IACxD,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,SAAS,CAAa;IAC9B,OAAO,CAAC,aAAa,CAAa;gBAEtB,SAAS,CAAC,EAAE,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC;IAIzD;;OAEG;IACH,aAAa,CAAC,OAAO,CAAC,EAAE,gBAAgB;IAexC;;OAEG;IACH,WAAW,CAAC,MAAM,SAAS,YAAY,EAAE,OAAO,EAAE,iBAAiB,CAAC,OAAO,EAAE,MAAM,CAAC;IA+BpF;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAiBnE;;OAEG;IACH,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;IAQ/C;;;OAGG;IACH,YAAY,CAAC,MAAM,CAAC,EAAE,GAAG;IAWzB;;OAEG;IACH,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG;IAqB9C;;OAEG;IACH,kBAAkB,IAAI,OAAO;IAI7B;;OAEG;IACH,cAAc,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAIrC;;OAEG;IACH,kBAAkB,IAAI,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,SAAS;IAI7D;;;;;OAKG;IACH,UAAU,CAAC,CAAC,SAAS;QAAE,WAAW,EAAE,QAAQ,CAAA;KAAE,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC;CA2E9D"}

package/dist/ai-tracing/exporters/base.d.ts

@@ -0,0 +1,111 @@
+/**
+ * Base Exporter for AI Tracing
+ *
+ * Provides common functionality shared by all AI tracing exporters:
+ * - Logger initialization with proper Mastra logger support
+ * - Disabled state management
+ * - Graceful shutdown lifecycle
+ */
+import { LogLevel } from '../../logger/index.js';
+import type { IMastraLogger } from '../../logger/index.js';
+import type { AITracingEvent, AITracingExporter } from '../types.js';
+/**
+ * Base configuration that all exporters should support
+ */
+export interface BaseExporterConfig {
+    /** Optional Mastra logger instance */
+    logger?: IMastraLogger;
+    /** Log level for the exporter (defaults to INFO) - accepts both enum and string */
+    logLevel?: LogLevel | 'debug' | 'info' | 'warn' | 'error';
+}
+/**
+ * Abstract base class for AI tracing exporters
+ *
+ * Handles common concerns:
+ * - Logger setup with proper Mastra logger
+ * - Disabled state management
+ * - Basic lifecycle methods
+ *
+ * @example
+ * ```typescript
+ * class MyExporter extends BaseExporter {
+ *   name = 'my-exporter';
+ *
+ *   constructor(config: MyExporterConfig) {
+ *     super(config);
+ *
+ *     if (!config.apiKey) {
+ *       this.setDisabled('Missing API key');
+ *       return;
+ *     }
+ *
+ *     // Initialize exporter-specific logic
+ *   }
+ *
+ *   async _exportEvent(event: AITracingEvent): Promise<void> {
+ *     // Export logic
+ *   }
+ * }
+ * ```
+ */
+export declare abstract class BaseExporter implements AITracingExporter {
+    /** Exporter name - must be implemented by subclasses */
+    abstract name: string;
+    /** Mastra logger instance */
+    protected logger: IMastraLogger;
+    /** Whether this exporter is disabled */
+    protected isDisabled: boolean;
+    /**
+     * Initialize the base exporter with logger
+     */
+    constructor(config?: BaseExporterConfig);
+    /**
+     * Set the logger for the exporter (called by Mastra/AITracing during initialization)
+     */
+    __setLogger(logger: IMastraLogger): void;
+    /**
+     * Convert string log level to LogLevel enum
+     */
+    private resolveLogLevel;
+    /**
+     * Mark the exporter as disabled and log a message
+     *
+     * @param reason - Reason why the exporter is disabled
+     */
+    protected setDisabled(reason: string): void;
+    /**
+     * Export a tracing event
+     *
+     * This method checks if the exporter is disabled before calling _exportEvent.
+     * Subclasses should implement _exportEvent instead of overriding this method.
+     */
+    exportEvent(event: AITracingEvent): Promise<void>;
+    /**
+     * Export a tracing event - must be implemented by subclasses
+     *
+     * This method is called by exportEvent after checking if the exporter is disabled.
+     */
+    protected abstract _exportEvent(event: AITracingEvent): Promise<void>;
+    /**
+     * Optional initialization hook called after Mastra is fully configured
+     */
+    init?(_config?: any): void;
+    /**
+     * Optional method to add scores to traces
+     */
+    addScoreToTrace?(_args: {
+        traceId: string;
+        spanId?: string;
+        score: number;
+        reason?: string;
+        scorerName: string;
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    /**
+     * Shutdown the exporter and clean up resources
+     *
+     * Default implementation just logs. Override to add custom cleanup.
+     */
+    shutdown(): Promise<void>;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/ai-tracing/exporters/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/base.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAiB,QAAQ,EAAE,MAAM,cAAc,CAAC;AACvD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAElE;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,sCAAsC;IACtC,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,mFAAmF;IACnF,QAAQ,CAAC,EAAE,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;CAC3D;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,8BAAsB,YAAa,YAAW,iBAAiB;IAC7D,wDAAwD;IACxD,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,6BAA6B;IAC7B,SAAS,CAAC,MAAM,EAAE,aAAa,CAAC;IAEhC,wCAAwC;IACxC,SAAS,CAAC,UAAU,EAAE,OAAO,CAAS;IAEtC;;OAEG;gBACS,MAAM,GAAE,kBAAuB;IAO3C;;OAEG;IACH,WAAW,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI;IAMxC;;OAEG;IACH,OAAO,CAAC,eAAe;IAqBvB;;;;OAIG;IACH,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAK3C;;;;;OAKG;IACG,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAOvD;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAErE;;OAEG;IACH,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,GAAG,IAAI;IAE1B;;OAEG;IACH,eAAe,CAAC,CAAC,KAAK,EAAE;QACtB,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,MAAM,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,GAAG,OAAO,CAAC,IAAI,CAAC;IAEjB;;;;OAIG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAGhC"}

package/dist/ai-tracing/exporters/cloud.d.ts

@@ -1,22 +1,20 @@
-import type { IMastraLogger } from '../../logger/index.js';
-import type { AITracingEvent, AITracingExporter } from '../types.js';
-export interface CloudExporterConfig {
+import type { AITracingEvent } from '../types.js';
+import { BaseExporter } from './base.js';
+import type { BaseExporterConfig } from './base.js';
+export interface CloudExporterConfig extends BaseExporterConfig {
     maxBatchSize?: number;
     maxBatchWaitMs?: number;
     maxRetries?: number;
     accessToken?: string;
     endpoint?: string;
-    logger?: IMastraLogger;
 }
-export declare class CloudExporter implements AITracingExporter {
+export declare class CloudExporter extends BaseExporter {
     name: string;
     private config;
     private buffer;
     private flushTimer;
-    private logger;
-    private isDisabled;
     constructor(config?: CloudExporterConfig);
-    exportEvent(event: AITracingEvent): Promise<void>;
+    protected _exportEvent(event: AITracingEvent): Promise<void>;
     private addToBuffer;
     private formatSpan;
     private shouldFlush;

package/dist/ai-tracing/exporters/cloud.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"cloud.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/cloud.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAGlD,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAqB,MAAM,UAAU,CAAC;AAErF,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,aAAa,CAAC;CACxB;AA0BD,qBAAa,aAAc,YAAW,iBAAiB;IACrD,IAAI,SAAsC;IAE1C,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,UAAU,CAA+B;IACjD,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,UAAU,CAAkB;gBAExB,MAAM,GAAE,mBAAwB;IA8BtC,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBvD,OAAO,CAAC,WAAW;IAYnB,OAAO,CAAC,UAAU;IAsBlB,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,aAAa;YAoBP,KAAK;IA8CnB;;OAEG;YACW,WAAW;IAiBzB,OAAO,CAAC,WAAW;IAMb,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAuChC"}
+{"version":3,"file":"cloud.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/cloud.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAqB,MAAM,UAAU,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,QAAQ,CAAC;AAEjD,MAAM,WAAW,mBAAoB,SAAQ,kBAAkB;IAC7D,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IAGpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AA0BD,qBAAa,aAAc,SAAQ,YAAY;IAC7C,IAAI,SAAsC;IAE1C,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,UAAU,CAA+B;gBAErC,MAAM,GAAE,mBAAwB;cA8B5B,YAAY,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBlE,OAAO,CAAC,WAAW;IAWnB,OAAO,CAAC,UAAU;IAsBlB,OAAO,CAAC,WAAW;IAiBnB,OAAO,CAAC,aAAa;YAoBP,KAAK;IA8CnB;;OAEG;YACW,WAAW;IAezB,OAAO,CAAC,WAAW;IAMb,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAuChC"}

package/dist/ai-tracing/exporters/console.d.ts

@@ -1,10 +1,10 @@
-import type { IMastraLogger } from '../../logger/index.js';
-import type { AITracingEvent, AITracingExporter } from '../types.js';
-export declare class ConsoleExporter implements AITracingExporter {
+import type { AITracingEvent } from '../types.js';
+import { BaseExporter } from './base.js';
+import type { BaseExporterConfig } from './base.js';
+export declare class ConsoleExporter extends BaseExporter {
     name: string;
-    private logger;
-    constructor(logger?: IMastraLogger);
-    exportEvent(event: AITracingEvent): Promise<void>;
+    constructor(config?: BaseExporterConfig);
+    protected _exportEvent(event: AITracingEvent): Promise<void>;
     shutdown(): Promise<void>;
 }
 //# sourceMappingURL=console.d.ts.map

package/dist/ai-tracing/exporters/console.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"console.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/console.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAElD,OAAO,KAAK,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAElE,qBAAa,eAAgB,YAAW,iBAAiB;IACvD,IAAI,SAA8B;IAClC,OAAO,CAAC,MAAM,CAAgB;gBAElB,MAAM,CAAC,EAAE,aAAa;IAS5B,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IA+EjD,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAGhC"}
+{"version":3,"file":"console.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/console.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,QAAQ,CAAC;AAEjD,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,IAAI,SAA8B;gBAEtB,MAAM,GAAE,kBAAuB;cAI3B,YAAY,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IA+E5D,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAGhC"}

package/dist/ai-tracing/exporters/index.d.ts

@@ -1,6 +1,7 @@
 /**
  * Mastra AI Tracing Exporters
  */
+export * from './base.js';
 export { CloudExporter } from './cloud.js';
 export type { CloudExporterConfig } from './cloud.js';
 export * from './console.js';

package/dist/ai-tracing/exporters/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACxC,YAAY,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AACnD,cAAc,WAAW,CAAC;AAC1B,cAAc,WAAW,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/exporters/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,cAAc,QAAQ,CAAC;AAGvB,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACxC,YAAY,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AACnD,cAAc,WAAW,CAAC;AAC1B,cAAc,WAAW,CAAC"}

package/dist/ai-tracing/index.cjs

@@ -1,132 +1,148 @@
 'use strict';
 
-var chunkURQI7RJ3_cjs = require('../chunk-URQI7RJ3.cjs');
+var chunk5GIAZ3CH_cjs = require('../chunk-5GIAZ3CH.cjs');
 
 
 
 Object.defineProperty(exports, "AISpanType", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.AISpanType; }
+  get: function () { return chunk5GIAZ3CH_cjs.AISpanType; }
 });
 Object.defineProperty(exports, "AITracingEventType", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.AITracingEventType; }
+  get: function () { return chunk5GIAZ3CH_cjs.AITracingEventType; }
 });
 Object.defineProperty(exports, "BaseAISpan", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.BaseAISpan; }
+  get: function () { return chunk5GIAZ3CH_cjs.BaseAISpan; }
 });
 Object.defineProperty(exports, "BaseAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.BaseAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.BaseAITracing; }
+});
+Object.defineProperty(exports, "BaseExporter", {
+  enumerable: true,
+  get: function () { return chunk5GIAZ3CH_cjs.BaseExporter; }
 });
 Object.defineProperty(exports, "CloudExporter", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.CloudExporter; }
+  get: function () { return chunk5GIAZ3CH_cjs.CloudExporter; }
 });
 Object.defineProperty(exports, "ConsoleExporter", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.ConsoleExporter; }
+  get: function () { return chunk5GIAZ3CH_cjs.ConsoleExporter; }
 });
 Object.defineProperty(exports, "DefaultAISpan", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.DefaultAISpan; }
+  get: function () { return chunk5GIAZ3CH_cjs.DefaultAISpan; }
 });
 Object.defineProperty(exports, "DefaultAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.DefaultAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.DefaultAITracing; }
 });
 Object.defineProperty(exports, "DefaultExporter", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.DefaultExporter; }
+  get: function () { return chunk5GIAZ3CH_cjs.DefaultExporter; }
 });
 Object.defineProperty(exports, "InternalSpans", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.InternalSpans; }
+  get: function () { return chunk5GIAZ3CH_cjs.InternalSpans; }
+});
+Object.defineProperty(exports, "ModelSpanTracker", {
+  enumerable: true,
+  get: function () { return chunk5GIAZ3CH_cjs.ModelSpanTracker; }
 });
 Object.defineProperty(exports, "NoOpAISpan", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.NoOpAISpan; }
+  get: function () { return chunk5GIAZ3CH_cjs.NoOpAISpan; }
 });
 Object.defineProperty(exports, "SamplingStrategyType", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.SamplingStrategyType; }
+  get: function () { return chunk5GIAZ3CH_cjs.SamplingStrategyType; }
 });
 Object.defineProperty(exports, "SensitiveDataFilter", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.SensitiveDataFilter; }
+  get: function () { return chunk5GIAZ3CH_cjs.SensitiveDataFilter; }
 });
 Object.defineProperty(exports, "clearAITracingRegistry", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.clearAITracingRegistry; }
+  get: function () { return chunk5GIAZ3CH_cjs.clearAITracingRegistry; }
 });
 Object.defineProperty(exports, "deepClean", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.deepClean; }
+  get: function () { return chunk5GIAZ3CH_cjs.deepClean; }
 });
 Object.defineProperty(exports, "getAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.getAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.getAITracing; }
 });
 Object.defineProperty(exports, "getAllAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.getAllAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.getAllAITracing; }
 });
 Object.defineProperty(exports, "getDefaultAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.getDefaultAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.getDefaultAITracing; }
+});
+Object.defineProperty(exports, "getNestedValue", {
+  enumerable: true,
+  get: function () { return chunk5GIAZ3CH_cjs.getNestedValue; }
 });
 Object.defineProperty(exports, "getOrCreateSpan", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.getOrCreateSpan; }
+  get: function () { return chunk5GIAZ3CH_cjs.getOrCreateSpan; }
 });
 Object.defineProperty(exports, "getSelectedAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.getSelectedAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.getSelectedAITracing; }
 });
 Object.defineProperty(exports, "getValidTraceId", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.getValidTraceId; }
+  get: function () { return chunk5GIAZ3CH_cjs.getValidTraceId; }
 });
 Object.defineProperty(exports, "hasAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.hasAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.hasAITracing; }
 });
 Object.defineProperty(exports, "isMastra", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.isMastra; }
+  get: function () { return chunk5GIAZ3CH_cjs.isMastra; }
 });
 Object.defineProperty(exports, "omitKeys", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.omitKeys; }
+  get: function () { return chunk5GIAZ3CH_cjs.omitKeys; }
 });
 Object.defineProperty(exports, "registerAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.registerAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.registerAITracing; }
 });
 Object.defineProperty(exports, "selectFields", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.selectFields; }
+  get: function () { return chunk5GIAZ3CH_cjs.selectFields; }
+});
+Object.defineProperty(exports, "setNestedValue", {
+  enumerable: true,
+  get: function () { return chunk5GIAZ3CH_cjs.setNestedValue; }
 });
 Object.defineProperty(exports, "setSelector", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.setSelector; }
+  get: function () { return chunk5GIAZ3CH_cjs.setSelector; }
 });
 Object.defineProperty(exports, "setupAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.setupAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.setupAITracing; }
 });
 Object.defineProperty(exports, "shutdownAITracingRegistry", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.shutdownAITracingRegistry; }
+  get: function () { return chunk5GIAZ3CH_cjs.shutdownAITracingRegistry; }
 });
 Object.defineProperty(exports, "unregisterAITracing", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.unregisterAITracing; }
+  get: function () { return chunk5GIAZ3CH_cjs.unregisterAITracing; }
 });
 Object.defineProperty(exports, "wrapMastra", {
   enumerable: true,
-  get: function () { return chunkURQI7RJ3_cjs.wrapMastra; }
+  get: function () { return chunk5GIAZ3CH_cjs.wrapMastra; }
 });
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/ai-tracing/index.d.ts

@@ -9,4 +9,5 @@ export * from './utils.js';
 export * from './context.js';
 export * from './exporters/index.js';
 export * from './span_processors/index.js';
+export * from './chunk-tracing.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/ai-tracing/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ai-tracing/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,cAAc,SAAS,CAAC;AACxB,cAAc,WAAW,CAAC;AAC1B,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,SAAS,CAAC;AACxB,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,mBAAmB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ai-tracing/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,cAAc,SAAS,CAAC;AACxB,cAAc,WAAW,CAAC;AAC1B,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,SAAS,CAAC;AACxB,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,mBAAmB,CAAC;AAClC,cAAc,iBAAiB,CAAC"}

package/dist/ai-tracing/index.js

@@ -1,3 +1,3 @@
-export { AISpanType, AITracingEventType, BaseAISpan, BaseAITracing, CloudExporter, ConsoleExporter, DefaultAISpan, DefaultAITracing, DefaultExporter, InternalSpans, NoOpAISpan, SamplingStrategyType, SensitiveDataFilter, clearAITracingRegistry, deepClean, getAITracing, getAllAITracing, getDefaultAITracing, getOrCreateSpan, getSelectedAITracing, getValidTraceId, hasAITracing, isMastra, omitKeys, registerAITracing, selectFields, setSelector, setupAITracing, shutdownAITracingRegistry, unregisterAITracing, wrapMastra } from '../chunk-BLVB2IJK.js';
+export { AISpanType, AITracingEventType, BaseAISpan, BaseAITracing, BaseExporter, CloudExporter, ConsoleExporter, DefaultAISpan, DefaultAITracing, DefaultExporter, InternalSpans, ModelSpanTracker, NoOpAISpan, SamplingStrategyType, SensitiveDataFilter, clearAITracingRegistry, deepClean, getAITracing, getAllAITracing, getDefaultAITracing, getNestedValue, getOrCreateSpan, getSelectedAITracing, getValidTraceId, hasAITracing, isMastra, omitKeys, registerAITracing, selectFields, setNestedValue, setSelector, setupAITracing, shutdownAITracingRegistry, unregisterAITracing, wrapMastra } from '../chunk-3CVE7NQC.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/ai-tracing/spans/base.d.ts

@@ -1,4 +1,4 @@
-import type { AISpan, AISpanTypeMap, AnyAISpan, ChildSpanOptions, ChildEventOptions, EndSpanOptions, ErrorSpanOptions, UpdateSpanOptions, CreateSpanOptions, AITracing, ExportedAISpan } from '../types.js';
+import type { AISpan, AISpanTypeMap, AnyAISpan, ChildSpanOptions, ChildEventOptions, EndSpanOptions, ErrorSpanOptions, UpdateSpanOptions, CreateSpanOptions, AITracing, ExportedAISpan, TraceState } from '../types.js';
 import { AISpanType } from '../types.js';
 export declare abstract class BaseAISpan<TType extends AISpanType = any> implements AISpan<TType> {
     abstract id: string;
@@ -22,6 +22,9 @@ export declare abstract class BaseAISpan<TType extends AISpanType = any> impleme
         details?: Record<string, any>;
     };
     metadata?: Record<string, any>;
+    traceState?: TraceState;
+    /** Parent span ID (for root spans that are children of external spans) */
+    protected parentSpanId?: string;
     constructor(options: CreateSpanOptions<TType>, aiTracing: AITracing);
     /** End the span */
     abstract end(options?: EndSpanOptions<TType>): void;

package/dist/ai-tracing/spans/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/spans/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EACN,aAAa,EACb,SAAS,EACT,gBAAgB,EAChB,iBAAiB,EACjB,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,iBAAiB,EACjB,SAAS,EACT,cAAc,EACf,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,UAAU,EAAiB,MAAM,UAAU,CAAC;AA2CrD,8BAAsB,UAAU,CAAC,KAAK,SAAS,UAAU,GAAG,GAAG,CAAE,YAAW,MAAM,CAAC,KAAK,CAAC;IACvF,SAAgB,EAAE,EAAE,MAAM,CAAC;IAC3B,SAAgB,OAAO,EAAE,MAAM,CAAC;IAEzB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,KAAK,CAAC;IACZ,UAAU,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,SAAS,EAAE,IAAI,CAAC;IAChB,OAAO,CAAC,EAAE,IAAI,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,SAAS,CAAC;IACrB,KAAK,CAAC,EAAE,GAAG,CAAC;IACZ,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,SAAS,CAAC,EAAE;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC/B,CAAC;IACK,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;gBAE1B,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,SAAS;IAqBnE,mBAAmB;IACnB,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,KAAK,CAAC,GAAG,IAAI;IAEnD,oEAAoE;IACpE,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI;IAEtD,6BAA6B;IAC7B,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,IAAI;IAExD,eAAe,CAAC,UAAU,SAAS,UAAU,EAAE,OAAO,EAAE,gBAAgB,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC;IAIzG,eAAe,CAAC,UAAU,SAAS,UAAU,EAAE,OAAO,EAAE,iBAAiB,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC;IAI1G,6DAA6D;IAC7D,IAAI,UAAU,IAAI,OAAO,CAExB;IAED,oEAAoE;IACpE,QAAQ,KAAK,OAAO,IAAI,OAAO,CAAC;IAEhC,gEAAgE;IACzD,eAAe,CAAC,oBAAoB,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS;IAQ1E,qFAAqF;IAC9E,UAAU,CAAC,CAAC,SAAS,UAAU,EAAE,QAAQ,EAAE,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,SAAS;IAa3E,kDAAkD;IAC3C,UAAU,CAAC,oBAAoB,CAAC,EAAE,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC;CAkBzE;AASD,MAAM,WAAW,gBAAgB;IAC/B,WAAW,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,SAAS,CACvB,KAAK,EAAE,GAAG,EACV,OAAO,GAAE,gBAAqB,EAC9B,KAAK,GAAE,OAAO,CAAC,GAAG,CAAiB,EACnC,MAAM,GAAE,MAAU,GACjB,GAAG,CAwCL"}
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/spans/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EACN,aAAa,EACb,SAAS,EACT,gBAAgB,EAChB,iBAAiB,EACjB,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,iBAAiB,EACjB,SAAS,EACT,cAAc,EACd,UAAU,EACX,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,UAAU,EAAiB,MAAM,UAAU,CAAC;AA4CrD,8BAAsB,UAAU,CAAC,KAAK,SAAS,UAAU,GAAG,GAAG,CAAE,YAAW,MAAM,CAAC,KAAK,CAAC;IACvF,SAAgB,EAAE,EAAE,MAAM,CAAC;IAC3B,SAAgB,OAAO,EAAE,MAAM,CAAC;IAEzB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,KAAK,CAAC;IACZ,UAAU,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,SAAS,EAAE,IAAI,CAAC;IAChB,OAAO,CAAC,EAAE,IAAI,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,SAAS,CAAC;IACrB,KAAK,CAAC,EAAE,GAAG,CAAC;IACZ,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,SAAS,CAAC,EAAE;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC/B,CAAC;IACK,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,UAAU,CAAC,EAAE,UAAU,CAAC;IAC/B,0EAA0E;IAC1E,SAAS,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;gBAEpB,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,SAAS;IAsBnE,mBAAmB;IACnB,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,KAAK,CAAC,GAAG,IAAI;IAEnD,oEAAoE;IACpE,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI;IAEtD,6BAA6B;IAC7B,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,IAAI;IAExD,eAAe,CAAC,UAAU,SAAS,UAAU,EAAE,OAAO,EAAE,gBAAgB,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC;IAIzG,eAAe,CAAC,UAAU,SAAS,UAAU,EAAE,OAAO,EAAE,iBAAiB,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC;IAI1G,6DAA6D;IAC7D,IAAI,UAAU,IAAI,OAAO,CAExB;IAED,oEAAoE;IACpE,QAAQ,KAAK,OAAO,IAAI,OAAO,CAAC;IAEhC,gEAAgE;IACzD,eAAe,CAAC,oBAAoB,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS;IAW1E,qFAAqF;IAC9E,UAAU,CAAC,CAAC,SAAS,UAAU,EAAE,QAAQ,EAAE,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,SAAS;IAa3E,kDAAkD;IAC3C,UAAU,CAAC,oBAAoB,CAAC,EAAE,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC;CAkBzE;AASD,MAAM,WAAW,gBAAgB;IAC/B,WAAW,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,SAAS,CACvB,KAAK,EAAE,GAAG,EACV,OAAO,GAAE,gBAAqB,EAC9B,KAAK,GAAE,OAAO,CAAC,GAAG,CAAiB,EACnC,MAAM,GAAE,MAAU,GACjB,GAAG,CAwCL"}

package/dist/ai-tracing/spans/default.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"default.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/spans/default.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,UAAU,EACV,SAAS,EACT,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AAE/C,qBAAa,aAAa,CAAC,KAAK,SAAS,UAAU,CAAE,SAAQ,UAAU,CAAC,KAAK,CAAC;IACrE,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;gBAEX,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,SAAS;IAcnE,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,KAAK,CAAC,GAAG,IAAI;IAiB1C,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI;IAoC7C,MAAM,CAAC,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,IAAI;IAoB/C,IAAI,OAAO,IAAI,OAAO,CAErB;IAEK,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC;CAUhC"}
+{"version":3,"file":"default.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/spans/default.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,UAAU,EACV,SAAS,EACT,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,UAAU,EAAa,MAAM,QAAQ,CAAC;AAE/C,qBAAa,aAAa,CAAC,KAAK,SAAS,UAAU,CAAE,SAAQ,UAAU,CAAC,KAAK,CAAC;IACrE,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;gBAEX,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,SAAS;IAmCnE,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,KAAK,CAAC,GAAG,IAAI;IAiB1C,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI;IAoC7C,MAAM,CAAC,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,IAAI;IAoB/C,IAAI,OAAO,IAAI,OAAO,CAErB;IAEK,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC;CAUhC"}

package/dist/ai-tracing/tracers/base.d.ts

@@ -3,7 +3,8 @@
  */
 import { MastraBase } from '../../base.js';
 import type { IMastraLogger } from '../../logger/index.js';
-import type { TracingConfig, AISpan, AISpanType, AITracingExporter, AISpanProcessor, AITracingEvent, AnyAISpan, StartSpanOptions, CreateSpanOptions, AITracing, CustomSamplerOptions, AnyExportedAISpan } from '../types.js';
+import type { RuntimeContext } from '../../runtime-context/index.js';
+import type { TracingConfig, AISpan, AISpanType, AITracingExporter, AISpanProcessor, AITracingEvent, AnyAISpan, StartSpanOptions, CreateSpanOptions, AITracing, CustomSamplerOptions, AnyExportedAISpan, TraceState, TracingOptions } from '../types.js';
 /**
  * Abstract base class for all AI Tracing implementations in Mastra.
  */
@@ -12,6 +13,7 @@ export declare abstract class BaseAITracing extends MastraBase implements AITrac
     constructor(config: TracingConfig);
     /**
      * Override setLogger to add AI tracing specific initialization log
+     * and propagate logger to exporters
      */
     __setLogger(logger: IMastraLogger): void;
     protected get exporters(): AITracingExporter[];
@@ -58,6 +60,18 @@ export declare abstract class BaseAITracing extends MastraBase implements AITrac
      * Check if an AI trace should be sampled
      */
     protected shouldSample(options?: CustomSamplerOptions): boolean;
+    /**
+     * Compute TraceState for a new trace based on configured and per-request keys
+     */
+    protected computeTraceState(tracingOptions?: TracingOptions): TraceState | undefined;
+    /**
+     * Extract metadata from RuntimeContext using TraceState
+     */
+    protected extractMetadataFromRuntimeContext(runtimeContext: RuntimeContext | undefined, explicitMetadata: Record<string, any> | undefined, traceState: TraceState | undefined): Record<string, any> | undefined;
+    /**
+     * Extract specific keys from RuntimeContext
+     */
+    protected extractKeys(runtimeContext: RuntimeContext, keys: string[]): Record<string, any>;
     /**
      * Process a span through all processors
      */

package/dist/ai-tracing/tracers/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/tracers/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAGlD,OAAO,KAAK,EACV,aAAa,EACb,MAAM,EACN,UAAU,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,EACd,SAAS,EAGT,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,oBAAoB,EACpB,iBAAiB,EAClB,MAAM,UAAU,CAAC;AAOlB;;GAEG;AACH,8BAAsB,aAAc,SAAQ,UAAW,YAAW,SAAS;IACzE,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;gBAE9B,MAAM,EAAE,aAAa;IAcjC;;OAEG;IACH,WAAW,CAAC,MAAM,EAAE,aAAa;IAYjC,SAAS,KAAK,SAAS,IAAI,iBAAiB,EAAE,CAE7C;IAED,SAAS,KAAK,UAAU,IAAI,eAAe,EAAE,CAE5C;IAMD;;OAEG;IACH,SAAS,CAAC,KAAK,SAAS,UAAU,EAAE,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IA0BpF;;;;;;;;;;;OAWG;IACH,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,SAAS,UAAU,EAAE,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IAMzG;;OAEG;IACH,SAAS,IAAI,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAQ9C;;OAEG;IACH,YAAY,IAAI,SAAS,iBAAiB,EAAE;IAI5C;;OAEG;IACH,aAAa,IAAI,SAAS,eAAe,EAAE;IAI3C;;OAEG;IACH,SAAS;IAQT;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAkCzB;;OAEG;IACH,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO;IAwB/D;;OAEG;IACH,OAAO,CAAC,WAAW;IAqBnB,gBAAgB,CAAC,IAAI,EAAE,SAAS,GAAG,iBAAiB,GAAG,SAAS;IAQhE;;OAEG;IACH,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAShD;;OAEG;IACH,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAS9C;;OAEG;IACH,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAShD;;OAEG;cACa,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBjE;;OAEG;IACH,IAAI,IAAI,IAAI;IASZ;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAUhC"}
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/ai-tracing/tracers/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAElD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE5D,OAAO,KAAK,EACV,aAAa,EACb,MAAM,EACN,UAAU,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,EACd,SAAS,EAGT,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,oBAAoB,EACpB,iBAAiB,EACjB,UAAU,EACV,cAAc,EACf,MAAM,UAAU,CAAC;AAQlB;;GAEG;AACH,8BAAsB,aAAc,SAAQ,UAAW,YAAW,SAAS;IACzE,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;gBAE9B,MAAM,EAAE,aAAa;IAejC;;;OAGG;IACH,WAAW,CAAC,MAAM,EAAE,aAAa;IAoBjC,SAAS,KAAK,SAAS,IAAI,iBAAiB,EAAE,CAE7C;IAED,SAAS,KAAK,UAAU,IAAI,eAAe,EAAE,CAE5C;IAMD;;OAEG;IACH,SAAS,CAAC,KAAK,SAAS,UAAU,EAAE,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IA4CpF;;;;;;;;;;;OAWG;IACH,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,SAAS,UAAU,EAAE,OAAO,EAAE,iBAAiB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IAMzG;;OAEG;IACH,SAAS,IAAI,QAAQ,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAQ9C;;OAEG;IACH,YAAY,IAAI,SAAS,iBAAiB,EAAE;IAI5C;;OAEG;IACH,aAAa,IAAI,SAAS,eAAe,EAAE;IAI3C;;OAEG;IACH,SAAS;IAQT;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAkCzB;;OAEG;IACH,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO;IAwB/D;;OAEG;IACH,SAAS,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,cAAc,GAAG,UAAU,GAAG,SAAS;IAgBpF;;OAEG;IACH,SAAS,CAAC,iCAAiC,CACzC,cAAc,EAAE,cAAc,GAAG,SAAS,EAC1C,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS,EACjD,UAAU,EAAE,UAAU,GAAG,SAAS,GACjC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS;IAkBlC;;OAEG;IACH,SAAS,CAAC,WAAW,CAAC,cAAc,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IA2B1F;;OAEG;IACH,OAAO,CAAC,WAAW;IAqBnB,gBAAgB,CAAC,IAAI,EAAE,SAAS,GAAG,iBAAiB,GAAG,SAAS;IAQhE;;OAEG;IACH,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAShD;;OAEG;IACH,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAS9C;;OAEG;IACH,SAAS,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAShD;;OAEG;cACa,WAAW,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBjE;;OAEG;IACH,IAAI,IAAI,IAAI;IASZ;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAUhC"}