@mastra/core 1.20.0 → 1.21.0-alpha.1

package/dist/{chunk-57OYWNNV.cjs → chunk-EX2LVHTL.cjs}

@@ -1,18 +1,13 @@
 'use strict';
 
 var v4 = require('zod/v4');
+var async_hooks = require('async_hooks');
 
-// src/observability/types/core.ts
-var SamplingStrategyType = /* @__PURE__ */ ((SamplingStrategyType2) => {
-  SamplingStrategyType2["ALWAYS"] = "always";
-  SamplingStrategyType2["NEVER"] = "never";
-  SamplingStrategyType2["RATIO"] = "ratio";
-  SamplingStrategyType2["CUSTOM"] = "custom";
-  return SamplingStrategyType2;
-})(SamplingStrategyType || {});
+// ../_internal-core/dist/index.js
 var EntityType = /* @__PURE__ */ ((EntityType2) => {
   EntityType2["AGENT"] = "agent";
-  EntityType2["EVAL"] = "eval";
+  EntityType2["SCORER"] = "scorer";
+  EntityType2["TRAJECTORY"] = "trajectory";
   EntityType2["INPUT_PROCESSOR"] = "input_processor";
   EntityType2["INPUT_STEP_PROCESSOR"] = "input_step_processor";
   EntityType2["OUTPUT_PROCESSOR"] = "output_processor";
@@ -200,8 +195,9 @@ var listLogsResponseSchema = v4.z.object({
   logs: v4.z.array(logRecordSchema)
 });
 var scorerIdField = v4.z.string().describe("Identifier of the scorer (e.g., relevance, accuracy)");
+var scorerNameField = v4.z.string().describe("Display name of the scorer");
 var scorerVersionField = v4.z.string().describe("Version of the scorer");
-var scoreSourceField = v4.z.string().describe("Source of the score (e.g., manual, automated, experiment)");
+var scoreSourceField = v4.z.string().describe("How the score was produced (e.g., manual, automated, experiment)");
 var scoreValueField = v4.z.number().describe("Score value (range defined by scorer)");
 var scoreReasonField = v4.z.string().describe("Explanation for the score");
 var scoreRecordSchema = v4.z.object({
@@ -211,6 +207,7 @@ var scoreRecordSchema = v4.z.object({
   spanId: spanIdField.nullish().describe("Span ID this score applies to"),
   // Score data
   scorerId: scorerIdField,
+  scorerName: scorerNameField.nullish(),
   scorerVersion: scorerVersionField.nullish(),
   scoreSource: scoreSourceField.nullish(),
   /**
@@ -228,6 +225,7 @@ var scoreRecordSchema = v4.z.object({
 }).describe("Score record as stored in the database");
 var scoreInputSchema = v4.z.object({
   scorerId: scorerIdField,
+  scorerName: scorerNameField.optional(),
   scorerVersion: scorerVersionField.optional(),
   scoreSource: scoreSourceField.optional(),
   /**
@@ -238,7 +236,8 @@ var scoreInputSchema = v4.z.object({
   reason: scoreReasonField.optional(),
   metadata: v4.z.record(v4.z.string(), v4.z.unknown()).optional().describe("Additional scorer-specific metadata"),
   experimentId: experimentIdField.optional(),
-  scoreTraceId: v4.z.string().optional().describe("Trace ID of the scoring run for debugging score generation")
+  scoreTraceId: v4.z.string().optional().describe("Trace ID of the scoring run for debugging score generation"),
+  targetEntityType: entityTypeField.optional().describe("Entity type the scorer evaluated when known")
 }).describe("User-provided score input");
 var createScoreRecordSchema = scoreRecordSchema;
 var createScoreArgsSchema = v4.z.object({
@@ -255,11 +254,11 @@ var scoresFilterSchema = v4.z.object({
   ...commonFilterFields,
   // Score-specific filters
   scorerId: v4.z.union([v4.z.string(), v4.z.array(v4.z.string())]).optional().describe("Filter by scorer ID(s)"),
-  scoreSource: scoreSourceField.optional().describe("Filter by score source"),
+  scoreSource: scoreSourceField.optional().describe("Filter by how the score was produced"),
   /**
    * @deprecated Use `scoreSource` instead.
    */
-  source: scoreSourceField.optional().describe("Filter by score source")
+  source: scoreSourceField.optional().describe("Filter by how the score was produced")
 }).describe("Filters for querying scores");
 var scoresOrderByFieldSchema = v4.z.enum(["timestamp", "score"]).describe("Field to order by: 'timestamp' | 'score'");
 var scoresOrderBySchema = v4.z.object({
@@ -696,6 +695,8 @@ var getTagsResponseSchema = v4.z.object({
 // src/observability/types/tracing.ts
 var SpanType = /* @__PURE__ */ ((SpanType2) => {
   SpanType2["AGENT_RUN"] = "agent_run";
+  SpanType2["SCORER_RUN"] = "scorer_run";
+  SpanType2["SCORER_STEP"] = "scorer_step";
   SpanType2["GENERIC"] = "generic";
   SpanType2["MODEL_GENERATION"] = "model_generation";
   SpanType2["MODEL_STEP"] = "model_step";
@@ -728,290 +729,98 @@ var TracingEventType = /* @__PURE__ */ ((TracingEventType2) => {
   TracingEventType2["SPAN_ENDED"] = "span_ended";
   return TracingEventType2;
 })(TracingEventType || {});
-
-// src/observability/types/metrics.ts
-var DEFAULT_BLOCKED_LABELS = [
-  "trace_id",
-  "span_id",
-  "run_id",
-  "request_id",
-  "user_id",
-  "resource_id",
-  "session_id",
-  "thread_id"
-];
-
-// src/observability/no-op.ts
-var noOpCounter = {
-  add() {
-  }
-};
-var noOpGauge = {
-  set() {
-  }
-};
-var noOpHistogram = {
-  record() {
-  }
-};
-var noOpTracingContext = {
-  currentSpan: void 0
-};
-var noOpLoggerContext = {
-  debug() {
-  },
-  info() {
-  },
-  warn() {
-  },
-  error() {
-  },
-  fatal() {
-  }
-};
-var noOpMetricsContext = {
-  emit() {
-  },
-  counter() {
-    return noOpCounter;
-  },
-  gauge() {
-    return noOpGauge;
-  },
-  histogram() {
-    return noOpHistogram;
-  }
-};
-var NoOpObservability = class {
-  setMastraContext(_options) {
-    return;
-  }
-  setLogger(_options) {
-    return;
-  }
-  getSelectedInstance(_options) {
-    return;
-  }
-  async getRecordedTrace(_args) {
-    return null;
-  }
-  async addScore(_args) {
-    return;
-  }
-  async addFeedback(_args) {
-    return;
-  }
-  registerInstance(_name, _instance, _isDefault = false) {
-    return;
-  }
-  getInstance(_name) {
-    return;
-  }
-  getDefaultInstance() {
-    return;
-  }
-  listInstances() {
-    return /* @__PURE__ */ new Map();
-  }
-  unregisterInstance(_name) {
-    return false;
-  }
-  hasInstance(_name) {
-    return false;
-  }
-  setConfigSelector(_selector) {
-    return;
-  }
-  clear() {
-    return;
-  }
-  async shutdown() {
-    return;
-  }
-};
-
-// src/observability/context-factory.ts
-function deriveLoggerContext(tracing) {
-  const span = tracing.currentSpan;
-  return span?.observabilityInstance?.getLoggerContext?.(span) ?? noOpLoggerContext;
-}
-function deriveMetricsContext(tracing) {
-  const span = tracing.currentSpan;
-  return span?.observabilityInstance?.getMetricsContext?.(span) ?? noOpMetricsContext;
-}
-function createObservabilityContext(tracingContext) {
-  const tracing = tracingContext ?? noOpTracingContext;
-  return {
-    tracing,
-    loggerVNext: deriveLoggerContext(tracing),
-    metrics: deriveMetricsContext(tracing),
-    tracingContext: tracing
-    // alias — preferred at forwarding sites
-  };
+var entityTypeValues = new Set(Object.values(EntityType));
+var spanContextStorage = new async_hooks.AsyncLocalStorage();
+function getCurrentSpan() {
+  return spanContextStorage.getStore();
 }
-function resolveObservabilityContext(partial) {
-  const tracing = partial.tracing ?? partial.tracingContext ?? noOpTracingContext;
-  return {
-    tracing,
-    loggerVNext: partial.loggerVNext ?? deriveLoggerContext(tracing),
-    metrics: partial.metrics ?? deriveMetricsContext(tracing),
-    tracingContext: tracing
-    // alias — preferred at forwarding sites
+function getOrCreateSpan(options) {
+  const { type, attributes, tracingContext, requestContext, tracingOptions, ...rest } = options;
+  const metadata = {
+    ...rest.metadata ?? {},
+    ...tracingOptions?.metadata ?? {}
   };
-}
-
-// src/observability/context.ts
-var AGENT_GETTERS = ["getAgent", "getAgentById"];
-var AGENT_METHODS_TO_WRAP = ["generate", "stream", "generateLegacy", "streamLegacy"];
-var WORKFLOW_GETTERS = ["getWorkflow", "getWorkflowById"];
-var WORKFLOW_METHODS_TO_WRAP = ["execute", "createRun", "createRun"];
-function isNoOpSpan(span) {
-  return span.constructor.name === "NoOpSpan" || span.__isNoOp === true;
-}
-function isMastra(mastra) {
-  const hasAgentGetters = AGENT_GETTERS.every((method) => typeof mastra?.[method] === "function");
-  const hasWorkflowGetters = WORKFLOW_GETTERS.every((method) => typeof mastra?.[method] === "function");
-  return hasAgentGetters && hasWorkflowGetters;
-}
-function wrapMastra(mastra, tracingContext) {
-  if (!tracingContext.currentSpan || isNoOpSpan(tracingContext.currentSpan)) {
-    return mastra;
-  }
-  if (!isMastra(mastra)) {
-    return mastra;
-  }
-  try {
-    return new Proxy(mastra, {
-      get(target, prop) {
-        try {
-          if (AGENT_GETTERS.includes(prop)) {
-            return (...args) => {
-              const agent = target[prop](...args);
-              return wrapAgent(agent, tracingContext);
-            };
-          }
-          if (WORKFLOW_GETTERS.includes(prop)) {
-            return (...args) => {
-              const workflow = target[prop](...args);
-              return wrapWorkflow(workflow, tracingContext);
-            };
-          }
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        } catch (error) {
-          console.warn("Tracing: Failed to wrap method, falling back to original", error);
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        }
-      }
+  if (tracingContext?.currentSpan) {
+    return tracingContext.currentSpan.createChildSpan({
+      type,
+      attributes,
+      ...rest,
+      metadata,
+      requestContext
     });
-  } catch (error) {
-    console.warn("Tracing: Failed to create proxy, using original Mastra instance", error);
-    return mastra;
   }
+  const instance = options.mastra?.observability?.getSelectedInstance({ requestContext });
+  return instance?.startSpan({
+    type,
+    attributes,
+    ...rest,
+    metadata,
+    requestContext,
+    tracingOptions,
+    traceId: tracingOptions?.traceId,
+    parentSpanId: tracingOptions?.parentSpanId,
+    customSamplerOptions: {
+      requestContext,
+      metadata
+    }
+  });
 }
-function wrapAgent(agent, tracingContext) {
-  if (!tracingContext.currentSpan || isNoOpSpan(tracingContext.currentSpan)) {
-    return agent;
+async function executeWithContext(params) {
+  const { span, fn } = params;
+  const wrappedFn = span ? () => spanContextStorage.run(span, fn) : fn;
+  if (span?.executeInContext) {
+    return span.executeInContext(wrappedFn);
   }
-  try {
-    return new Proxy(agent, {
-      get(target, prop) {
-        try {
-          if (AGENT_METHODS_TO_WRAP.includes(prop)) {
-            return (input, options = {}) => {
-              return target[prop](input, {
-                ...options,
-                ...createObservabilityContext(tracingContext)
-              });
-            };
-          }
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        } catch (error) {
-          console.warn("Tracing: Failed to wrap agent method, falling back to original", error);
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        }
-      }
-    });
-  } catch (error) {
-    console.warn("Tracing: Failed to create agent proxy, using original instance", error);
-    return agent;
+  return wrappedFn();
+}
+function executeWithContextSync(params) {
+  const { span, fn } = params;
+  const wrappedFn = span ? () => spanContextStorage.run(span, fn) : fn;
+  if (span?.executeInContextSync) {
+    return span.executeInContextSync(wrappedFn);
   }
+  return wrappedFn();
 }
-function wrapWorkflow(workflow, tracingContext) {
-  if (!tracingContext.currentSpan || isNoOpSpan(tracingContext.currentSpan)) {
-    return workflow;
+function getRootExportSpan(span) {
+  if (!span?.isValid) {
+    return void 0;
   }
-  try {
-    return new Proxy(workflow, {
-      get(target, prop) {
-        try {
-          if (WORKFLOW_METHODS_TO_WRAP.includes(prop)) {
-            if (prop === "createRun" || prop === "createRun") {
-              return async (options = {}) => {
-                const run = await target[prop](options);
-                return run ? wrapRun(run, tracingContext) : run;
-              };
-            }
-            return (input, options = {}) => {
-              return target[prop](input, {
-                ...options,
-                ...createObservabilityContext(tracingContext)
-              });
-            };
-          }
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        } catch (error) {
-          console.warn("Tracing: Failed to wrap workflow method, falling back to original", error);
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        }
-      }
-    });
-  } catch (error) {
-    console.warn("Tracing: Failed to create workflow proxy, using original instance", error);
-    return workflow;
+  let current = span;
+  let rootExportSpan = span.isInternal ? void 0 : span;
+  while (current?.parent) {
+    current = current.parent;
+    if (!current.isInternal) {
+      rootExportSpan = current;
+    }
   }
+  return rootExportSpan;
 }
-function wrapRun(run, tracingContext) {
-  if (!tracingContext.currentSpan || isNoOpSpan(tracingContext.currentSpan)) {
-    return run;
+function getEntityTypeForSpan(span) {
+  if (span.entityType && entityTypeValues.has(span.entityType)) {
+    return span.entityType;
   }
-  try {
-    return new Proxy(run, {
-      get(target, prop) {
-        try {
-          if (prop === "start") {
-            return (startOptions = {}) => {
-              return target.start({
-                ...startOptions,
-                ...createObservabilityContext(startOptions.tracingContext ?? tracingContext)
-              });
-            };
-          }
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        } catch (error) {
-          console.warn("Tracing: Failed to wrap run method, falling back to original", error);
-          const value = target[prop];
-          return typeof value === "function" ? value.bind(target) : value;
-        }
-      }
-    });
-  } catch (error) {
-    console.warn("Tracing: Failed to create run proxy, using original instance", error);
-    return run;
+  switch (span.spanType) {
+    case "agent_run" /* AGENT_RUN */:
+      return EntityType.AGENT;
+    case "scorer_run" /* SCORER_RUN */:
+    case "scorer_step" /* SCORER_STEP */:
+      return EntityType.SCORER;
+    case "workflow_run" /* WORKFLOW_RUN */:
+      return EntityType.WORKFLOW_RUN;
+    case "workflow_step" /* WORKFLOW_STEP */:
+      return EntityType.WORKFLOW_STEP;
+    case "tool_call" /* TOOL_CALL */:
+    case "mcp_tool_call" /* MCP_TOOL_CALL */:
+      return EntityType.TOOL;
+    case "processor_run" /* PROCESSOR_RUN */:
+      return EntityType.OUTPUT_PROCESSOR;
+    default:
+      return void 0;
   }
 }
 
-exports.DEFAULT_BLOCKED_LABELS = DEFAULT_BLOCKED_LABELS;
 exports.EntityType = EntityType;
 exports.InternalSpans = InternalSpans;
-exports.NoOpObservability = NoOpObservability;
-exports.SamplingStrategyType = SamplingStrategyType;
 exports.SpanType = SpanType;
 exports.TracingEventType = TracingEventType;
 exports.aggregationIntervalSchema = aggregationIntervalSchema;
@@ -1029,7 +838,6 @@ exports.createFeedbackRecordSchema = createFeedbackRecordSchema;
 exports.createFeedbackResponseSchema = createFeedbackResponseSchema;
 exports.createLogRecordSchema = createLogRecordSchema;
 exports.createMetricRecordSchema = createMetricRecordSchema;
-exports.createObservabilityContext = createObservabilityContext;
 exports.createScoreArgsSchema = createScoreArgsSchema;
 exports.createScoreBodySchema = createScoreBodySchema;
 exports.createScoreRecordSchema = createScoreRecordSchema;
@@ -1041,6 +849,8 @@ exports.entityIdField = entityIdField;
 exports.entityNameField = entityNameField;
 exports.entityTypeField = entityTypeField;
 exports.environmentField = environmentField;
+exports.executeWithContext = executeWithContext;
+exports.executeWithContextSync = executeWithContextSync;
 exports.executionSourceField = executionSourceField;
 exports.experimentIdField = experimentIdField;
 exports.feedbackFilterSchema = feedbackFilterSchema;
@@ -1048,8 +858,10 @@ exports.feedbackInputSchema = feedbackInputSchema;
 exports.feedbackOrderByFieldSchema = feedbackOrderByFieldSchema;
 exports.feedbackOrderBySchema = feedbackOrderBySchema;
 exports.feedbackRecordSchema = feedbackRecordSchema;
+exports.getCurrentSpan = getCurrentSpan;
 exports.getEntityNamesArgsSchema = getEntityNamesArgsSchema;
 exports.getEntityNamesResponseSchema = getEntityNamesResponseSchema;
+exports.getEntityTypeForSpan = getEntityTypeForSpan;
 exports.getEntityTypesArgsSchema = getEntityTypesArgsSchema;
 exports.getEntityTypesResponseSchema = getEntityTypesResponseSchema;
 exports.getEnvironmentsArgsSchema = getEnvironmentsArgsSchema;
@@ -1076,6 +888,8 @@ exports.getMetricPercentilesArgsSchema = getMetricPercentilesArgsSchema;
 exports.getMetricPercentilesResponseSchema = getMetricPercentilesResponseSchema;
 exports.getMetricTimeSeriesArgsSchema = getMetricTimeSeriesArgsSchema;
 exports.getMetricTimeSeriesResponseSchema = getMetricTimeSeriesResponseSchema;
+exports.getOrCreateSpan = getOrCreateSpan;
+exports.getRootExportSpan = getRootExportSpan;
 exports.getScoreAggregateArgsSchema = getScoreAggregateArgsSchema;
 exports.getScoreAggregateResponseSchema = getScoreAggregateResponseSchema;
 exports.getScoreBreakdownArgsSchema = getScoreBreakdownArgsSchema;
@@ -1110,9 +924,6 @@ exports.metricsAggregationSchema = metricsAggregationSchema;
 exports.metricsFilterSchema = metricsFilterSchema;
 exports.metricsOrderByFieldSchema = metricsOrderByFieldSchema;
 exports.metricsOrderBySchema = metricsOrderBySchema;
-exports.noOpLoggerContext = noOpLoggerContext;
-exports.noOpMetricsContext = noOpMetricsContext;
-exports.noOpTracingContext = noOpTracingContext;
 exports.organizationIdField = organizationIdField;
 exports.paginationArgsSchema = paginationArgsSchema;
 exports.paginationInfoSchema = paginationInfoSchema;
@@ -1120,7 +931,6 @@ exports.parentEntityIdField = parentEntityIdField;
 exports.parentEntityNameField = parentEntityNameField;
 exports.parentEntityTypeField = parentEntityTypeField;
 exports.requestIdField = requestIdField;
-exports.resolveObservabilityContext = resolveObservabilityContext;
 exports.resourceIdField = resourceIdField;
 exports.rootEntityIdField = rootEntityIdField;
 exports.rootEntityNameField = rootEntityNameField;
@@ -1143,6 +953,5 @@ exports.threadIdField = threadIdField;
 exports.traceIdField = traceIdField;
 exports.updatedAtField = updatedAtField;
 exports.userIdField = userIdField;
-exports.wrapMastra = wrapMastra;
-//# sourceMappingURL=chunk-57OYWNNV.cjs.map
-//# sourceMappingURL=chunk-57OYWNNV.cjs.map
+//# sourceMappingURL=chunk-EX2LVHTL.cjs.map
+//# sourceMappingURL=chunk-EX2LVHTL.cjs.map