@mastra/core 0.17.0-alpha.3 → 0.17.0-alpha.4

package/dist/{chunk-CUL4XPO7.cjs → chunk-CP4TZ4HO.cjs}

@@ -15,7 +15,79 @@ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
 var jsonSchemaToZod__default = /*#__PURE__*/_interopDefault(jsonSchemaToZod);
 
+// src/ai-tracing/types.ts
+var AISpanType = /* @__PURE__ */ ((AISpanType2) => {
+  AISpanType2["AGENT_RUN"] = "agent_run";
+  AISpanType2["GENERIC"] = "generic";
+  AISpanType2["LLM_GENERATION"] = "llm_generation";
+  AISpanType2["LLM_CHUNK"] = "llm_chunk";
+  AISpanType2["MCP_TOOL_CALL"] = "mcp_tool_call";
+  AISpanType2["TOOL_CALL"] = "tool_call";
+  AISpanType2["WORKFLOW_RUN"] = "workflow_run";
+  AISpanType2["WORKFLOW_STEP"] = "workflow_step";
+  AISpanType2["WORKFLOW_CONDITIONAL"] = "workflow_conditional";
+  AISpanType2["WORKFLOW_CONDITIONAL_EVAL"] = "workflow_conditional_eval";
+  AISpanType2["WORKFLOW_PARALLEL"] = "workflow_parallel";
+  AISpanType2["WORKFLOW_LOOP"] = "workflow_loop";
+  AISpanType2["WORKFLOW_SLEEP"] = "workflow_sleep";
+  AISpanType2["WORKFLOW_WAIT_EVENT"] = "workflow_wait_event";
+  return AISpanType2;
+})(AISpanType || {});
+var InternalSpans = /* @__PURE__ */ ((InternalSpans2) => {
+  InternalSpans2[InternalSpans2["NONE"] = 0] = "NONE";
+  InternalSpans2[InternalSpans2["WORKFLOW"] = 1] = "WORKFLOW";
+  InternalSpans2[InternalSpans2["AGENT"] = 2] = "AGENT";
+  InternalSpans2[InternalSpans2["TOOL"] = 4] = "TOOL";
+  InternalSpans2[InternalSpans2["LLM"] = 8] = "LLM";
+  InternalSpans2[InternalSpans2["ALL"] = 15] = "ALL";
+  return InternalSpans2;
+})(InternalSpans || {});
+var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
+  SamplingStrategyType2["ALWAYS"] = "always";
+  SamplingStrategyType2["NEVER"] = "never";
+  SamplingStrategyType2["RATIO"] = "ratio";
+  SamplingStrategyType2["CUSTOM"] = "custom";
+  return SamplingStrategyType2;
+})(SamplingStrategyType || {});
+var AITracingEventType = /* @__PURE__ */ ((AITracingEventType2) => {
+  AITracingEventType2["SPAN_STARTED"] = "span_started";
+  AITracingEventType2["SPAN_UPDATED"] = "span_updated";
+  AITracingEventType2["SPAN_ENDED"] = "span_ended";
+  return AITracingEventType2;
+})(AITracingEventType || {});
+
 // src/ai-tracing/spans/base.ts
+function isSpanInternal(spanType, flags) {
+  if (flags === void 0 || flags === 0 /* NONE */) {
+    return false;
+  }
+  switch (spanType) {
+    // Workflow-related spans
+    case "workflow_run" /* WORKFLOW_RUN */:
+    case "workflow_step" /* WORKFLOW_STEP */:
+    case "workflow_conditional" /* WORKFLOW_CONDITIONAL */:
+    case "workflow_conditional_eval" /* WORKFLOW_CONDITIONAL_EVAL */:
+    case "workflow_parallel" /* WORKFLOW_PARALLEL */:
+    case "workflow_loop" /* WORKFLOW_LOOP */:
+    case "workflow_sleep" /* WORKFLOW_SLEEP */:
+    case "workflow_wait_event" /* WORKFLOW_WAIT_EVENT */:
+      return (flags & 1 /* WORKFLOW */) !== 0;
+    // Agent-related spans
+    case "agent_run" /* AGENT_RUN */:
+      return (flags & 2 /* AGENT */) !== 0;
+    // Tool-related spans
+    case "tool_call" /* TOOL_CALL */:
+    case "mcp_tool_call" /* MCP_TOOL_CALL */:
+      return (flags & 4 /* TOOL */) !== 0;
+    // LLM-related spans
+    case "llm_generation" /* LLM_GENERATION */:
+    case "llm_chunk" /* LLM_CHUNK */:
+      return (flags & 8 /* LLM */) !== 0;
+    // Default: never internal
+    default:
+      return false;
+  }
+}
 var BaseAISpan = class {
   name;
   type;
@@ -39,7 +111,7 @@ var BaseAISpan = class {
     this.startTime = /* @__PURE__ */ new Date();
     this.aiTracing = aiTracing;
     this.isEvent = options.isEvent ?? false;
-    this.isInternal = options.isInternal ?? false;
+    this.isInternal = isSpanInternal(this.type, options.tracingPolicy?.internal);
     if (this.isEvent) {
       this.output = deepClean(options.output);
     } else {
@@ -252,38 +324,6 @@ var NoOpAISpan = class extends BaseAISpan {
   }
 };
 
-// src/ai-tracing/types.ts
-var AISpanType = /* @__PURE__ */ ((AISpanType2) => {
-  AISpanType2["AGENT_RUN"] = "agent_run";
-  AISpanType2["GENERIC"] = "generic";
-  AISpanType2["LLM_GENERATION"] = "llm_generation";
-  AISpanType2["LLM_CHUNK"] = "llm_chunk";
-  AISpanType2["MCP_TOOL_CALL"] = "mcp_tool_call";
-  AISpanType2["TOOL_CALL"] = "tool_call";
-  AISpanType2["WORKFLOW_RUN"] = "workflow_run";
-  AISpanType2["WORKFLOW_STEP"] = "workflow_step";
-  AISpanType2["WORKFLOW_CONDITIONAL"] = "workflow_conditional";
-  AISpanType2["WORKFLOW_CONDITIONAL_EVAL"] = "workflow_conditional_eval";
-  AISpanType2["WORKFLOW_PARALLEL"] = "workflow_parallel";
-  AISpanType2["WORKFLOW_LOOP"] = "workflow_loop";
-  AISpanType2["WORKFLOW_SLEEP"] = "workflow_sleep";
-  AISpanType2["WORKFLOW_WAIT_EVENT"] = "workflow_wait_event";
-  return AISpanType2;
-})(AISpanType || {});
-var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
-  SamplingStrategyType2["ALWAYS"] = "always";
-  SamplingStrategyType2["NEVER"] = "never";
-  SamplingStrategyType2["RATIO"] = "ratio";
-  SamplingStrategyType2["CUSTOM"] = "custom";
-  return SamplingStrategyType2;
-})(SamplingStrategyType || {});
-var AITracingEventType = /* @__PURE__ */ ((AITracingEventType2) => {
-  AITracingEventType2["SPAN_STARTED"] = "span_started";
-  AITracingEventType2["SPAN_UPDATED"] = "span_updated";
-  AITracingEventType2["SPAN_ENDED"] = "span_ended";
-  return AITracingEventType2;
-})(AITracingEventType || {});
-
 // src/ai-tracing/tracers/base.ts
 var BaseAITracing = class extends chunkKXCUCBEI_cjs.MastraBase {
   config;
@@ -600,7 +640,7 @@ var CoreToolBuilder = class extends chunkKXCUCBEI_cjs.MastraBase {
           toolDescription: options.description,
           toolType: logType || "tool"
         },
-        isInternal: options.tracingContext?.isInternal
+        tracingPolicy: options.tracingPolicy
       });
       try {
         let result;
@@ -1351,6 +1391,8 @@ var DefaultExporter = class {
   resolvedStrategy;
   buffer;
   flushTimer = null;
+  // Track all spans that have been created, persists across flushes
+  allCreatedSpans = /* @__PURE__ */ new Set();
   constructor(config = {}, logger) {
     if (logger) {
       this.logger = logger;
@@ -1371,6 +1413,7 @@ var DefaultExporter = class {
       insertOnly: [],
       seenSpans: /* @__PURE__ */ new Set(),
       spanSequences: /* @__PURE__ */ new Map(),
+      completedSpans: /* @__PURE__ */ new Set(),
       outOfOrderCount: 0,
       totalSize: 0
     };
@@ -1434,6 +1477,7 @@ var DefaultExporter = class {
     this.logger.warn("Out-of-order span update detected - skipping event", {
       spanId: event.exportedSpan.id,
       traceId: event.exportedSpan.traceId,
+      spanName: event.exportedSpan.name,
       eventType: event.type
     });
   }
@@ -1457,11 +1501,12 @@ var DefaultExporter = class {
           };
           this.buffer.creates.push(createRecord);
           this.buffer.seenSpans.add(spanKey);
+          this.allCreatedSpans.add(spanKey);
         }
         break;
       case "span_updated" /* SPAN_UPDATED */:
         if (this.resolvedStrategy === "batch-with-updates") {
-          if (this.buffer.seenSpans.has(spanKey)) {
+          if (this.allCreatedSpans.has(spanKey)) {
             this.buffer.updates.push({
               traceId: event.exportedSpan.traceId,
               spanId: event.exportedSpan.id,
@@ -1479,7 +1524,7 @@ var DefaultExporter = class {
         break;
       case "span_ended" /* SPAN_ENDED */:
         if (this.resolvedStrategy === "batch-with-updates") {
-          if (this.buffer.seenSpans.has(spanKey)) {
+          if (this.allCreatedSpans.has(spanKey)) {
             this.buffer.updates.push({
               traceId: event.exportedSpan.traceId,
               spanId: event.exportedSpan.id,
@@ -1489,6 +1534,7 @@ var DefaultExporter = class {
               },
               sequenceNumber: this.getNextSequence(spanKey)
             });
+            this.buffer.completedSpans.add(spanKey);
           } else if (event.exportedSpan.isEvent) {
             const createRecord = {
               traceId: event.exportedSpan.traceId,
@@ -1499,6 +1545,8 @@ var DefaultExporter = class {
             };
             this.buffer.creates.push(createRecord);
             this.buffer.seenSpans.add(spanKey);
+            this.allCreatedSpans.add(spanKey);
+            this.buffer.completedSpans.add(spanKey);
           } else {
             this.handleOutOfOrderUpdate(event);
             this.buffer.outOfOrderCount++;
@@ -1512,6 +1560,8 @@ var DefaultExporter = class {
             updatedAt: null
           };
           this.buffer.insertOnly.push(createRecord);
+          this.buffer.completedSpans.add(spanKey);
+          this.allCreatedSpans.add(spanKey);
         }
         break;
     }
@@ -1538,15 +1588,19 @@ var DefaultExporter = class {
   /**
    * Resets the buffer after successful flush
    */
-  resetBuffer() {
+  resetBuffer(completedSpansToCleanup = /* @__PURE__ */ new Set()) {
     this.buffer.creates = [];
     this.buffer.updates = [];
     this.buffer.insertOnly = [];
     this.buffer.seenSpans.clear();
     this.buffer.spanSequences.clear();
+    this.buffer.completedSpans.clear();
     this.buffer.outOfOrderCount = 0;
     this.buffer.firstEventTime = void 0;
     this.buffer.totalSize = 0;
+    for (const spanKey of completedSpansToCleanup) {
+      this.allCreatedSpans.delete(spanKey);
+    }
   }
   /**
    * Schedules a flush using setTimeout
@@ -1627,6 +1681,7 @@ var DefaultExporter = class {
    */
   async handleRealtimeEvent(event, storage) {
     const span = event.exportedSpan;
+    const spanKey = this.buildSpanKey(span.traceId, span.id);
     if (span.isEvent) {
       if (event.type === "span_ended" /* SPAN_ENDED */) {
         await storage.createAISpan({
@@ -1649,8 +1704,18 @@ var DefaultExporter = class {
             createdAt: /* @__PURE__ */ new Date(),
             updatedAt: null
           });
+          this.allCreatedSpans.add(spanKey);
           break;
         case "span_updated" /* SPAN_UPDATED */:
+          await storage.updateAISpan({
+            traceId: span.traceId,
+            spanId: span.id,
+            updates: {
+              ...this.buildUpdateRecord(span),
+              updatedAt: /* @__PURE__ */ new Date()
+            }
+          });
+          break;
         case "span_ended" /* SPAN_ENDED */:
           await storage.updateAISpan({
             traceId: span.traceId,
@@ -1660,6 +1725,7 @@ var DefaultExporter = class {
               updatedAt: /* @__PURE__ */ new Date()
             }
           });
+          this.allCreatedSpans.delete(spanKey);
           break;
         default:
           this.logger.warn(`Tracing event type not implemented for span spans: ${event.type}`);
@@ -1732,6 +1798,7 @@ var DefaultExporter = class {
       insertOnly: [...this.buffer.insertOnly],
       seenSpans: new Set(this.buffer.seenSpans),
       spanSequences: new Map(this.buffer.spanSequences),
+      completedSpans: new Set(this.buffer.completedSpans),
       outOfOrderCount: this.buffer.outOfOrderCount,
       firstEventTime: this.buffer.firstEventTime,
       totalSize: this.buffer.totalSize
@@ -1771,6 +1838,9 @@ var DefaultExporter = class {
           await storage.batchCreateAISpans({ records: buffer.insertOnly });
         }
       }
+      for (const spanKey of buffer.completedSpans) {
+        this.allCreatedSpans.delete(spanKey);
+      }
     } catch (error) {
       if (attempt < this.config.maxRetries) {
         const retryDelay = this.calculateRetryDelay(attempt);
@@ -1789,6 +1859,9 @@ var DefaultExporter = class {
           droppedBatchSize: buffer.totalSize,
           error: error instanceof Error ? error.message : String(error)
         });
+        for (const spanKey of buffer.completedSpans) {
+          this.allCreatedSpans.delete(spanKey);
+        }
       }
     }
   }
@@ -2146,18 +2219,17 @@ function getValidTraceId(span) {
   return span?.isValid ? span.traceId : void 0;
 }
 function getOrCreateSpan(options) {
-  const { type, attributes, tracingContext, tracingOptions, runtimeContext, ...rest } = options;
+  const { type, attributes, tracingContext, runtimeContext, ...rest } = options;
   const metadata = {
     ...rest.metadata ?? {},
-    ...tracingOptions?.metadata ?? {}
+    ...rest.tracingOptions?.metadata ?? {}
   };
   if (tracingContext?.currentSpan) {
     return tracingContext.currentSpan.createChildSpan({
       type,
       attributes,
       ...rest,
-      metadata,
-      isInternal: tracingContext?.isInternal
+      metadata
     });
   }
   const aiTracing = getSelectedAITracing({
@@ -2166,12 +2238,12 @@ function getOrCreateSpan(options) {
   return aiTracing?.startSpan({
     type,
     attributes,
+    ...rest,
+    metadata,
     customSamplerOptions: {
       runtimeContext,
       metadata
-    },
-    ...rest,
-    metadata
+    }
   });
 }
 
@@ -2331,6 +2403,7 @@ exports.ConsoleExporter = ConsoleExporter;
 exports.DefaultAISpan = DefaultAISpan;
 exports.DefaultAITracing = DefaultAITracing;
 exports.DefaultExporter = DefaultExporter;
+exports.InternalSpans = InternalSpans;
 exports.NoOpAISpan = NoOpAISpan;
 exports.SamplingStrategyType = SamplingStrategyType;
 exports.SensitiveDataFilter = SensitiveDataFilter;
@@ -2368,5 +2441,5 @@ exports.setupAITracing = setupAITracing;
 exports.shutdownAITracingRegistry = shutdownAITracingRegistry;
 exports.unregisterAITracing = unregisterAITracing;
 exports.wrapMastra = wrapMastra;
-//# sourceMappingURL=chunk-CUL4XPO7.cjs.map
-//# sourceMappingURL=chunk-CUL4XPO7.cjs.map
+//# sourceMappingURL=chunk-CP4TZ4HO.cjs.map
+//# sourceMappingURL=chunk-CP4TZ4HO.cjs.map