@mastra/core 1.20.0 → 1.21.0-alpha.1

package/dist/{chunk-2NISBCML.js → chunk-WG7QIVL7.js}

@@ -1,16 +1,11 @@
 import { z } from 'zod/v4';
+import { AsyncLocalStorage } from '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";
@@ -198,8 +193,9 @@ var listLogsResponseSchema = z.object({
   logs: z.array(logRecordSchema)
 });
 var scorerIdField = z.string().describe("Identifier of the scorer (e.g., relevance, accuracy)");
+var scorerNameField = z.string().describe("Display name of the scorer");
 var scorerVersionField = z.string().describe("Version of the scorer");
-var scoreSourceField = z.string().describe("Source of the score (e.g., manual, automated, experiment)");
+var scoreSourceField = z.string().describe("How the score was produced (e.g., manual, automated, experiment)");
 var scoreValueField = z.number().describe("Score value (range defined by scorer)");
 var scoreReasonField = z.string().describe("Explanation for the score");
 var scoreRecordSchema = z.object({
@@ -209,6 +205,7 @@ var scoreRecordSchema = 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(),
   /**
@@ -226,6 +223,7 @@ var scoreRecordSchema = z.object({
 }).describe("Score record as stored in the database");
 var scoreInputSchema = z.object({
   scorerId: scorerIdField,
+  scorerName: scorerNameField.optional(),
   scorerVersion: scorerVersionField.optional(),
   scoreSource: scoreSourceField.optional(),
   /**
@@ -236,7 +234,8 @@ var scoreInputSchema = z.object({
   reason: scoreReasonField.optional(),
   metadata: z.record(z.string(), z.unknown()).optional().describe("Additional scorer-specific metadata"),
   experimentId: experimentIdField.optional(),
-  scoreTraceId: z.string().optional().describe("Trace ID of the scoring run for debugging score generation")
+  scoreTraceId: 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 = z.object({
@@ -253,11 +252,11 @@ var scoresFilterSchema = z.object({
   ...commonFilterFields,
   // Score-specific filters
   scorerId: z.union([z.string(), z.array(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 = z.enum(["timestamp", "score"]).describe("Field to order by: 'timestamp' | 'score'");
 var scoresOrderBySchema = z.object({
@@ -694,6 +693,8 @@ var getTagsResponseSchema = 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";
@@ -726,285 +727,96 @@ 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 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;
   }
 }
 
-export { DEFAULT_BLOCKED_LABELS, EntityType, InternalSpans, NoOpObservability, SamplingStrategyType, SpanType, TracingEventType, aggregationIntervalSchema, aggregationTypeSchema, batchCreateFeedbackArgsSchema, batchCreateLogsArgsSchema, batchCreateMetricsArgsSchema, batchCreateScoresArgsSchema, commonFilterFields, comparePeriodSchema, contextFields, createFeedbackArgsSchema, createFeedbackBodySchema, createFeedbackRecordSchema, createFeedbackResponseSchema, createLogRecordSchema, createMetricRecordSchema, createObservabilityContext, createScoreArgsSchema, createScoreBodySchema, createScoreRecordSchema, createScoreResponseSchema, createdAtField, dateRangeSchema, dbTimestamps, entityIdField, entityNameField, entityTypeField, environmentField, executionSourceField, experimentIdField, feedbackFilterSchema, feedbackInputSchema, feedbackOrderByFieldSchema, feedbackOrderBySchema, feedbackRecordSchema, getEntityNamesArgsSchema, getEntityNamesResponseSchema, getEntityTypesArgsSchema, getEntityTypesResponseSchema, getEnvironmentsArgsSchema, getEnvironmentsResponseSchema, getFeedbackAggregateArgsSchema, getFeedbackAggregateResponseSchema, getFeedbackBreakdownArgsSchema, getFeedbackBreakdownResponseSchema, getFeedbackPercentilesArgsSchema, getFeedbackPercentilesResponseSchema, getFeedbackTimeSeriesArgsSchema, getFeedbackTimeSeriesResponseSchema, getMetricAggregateArgsSchema, getMetricAggregateResponseSchema, getMetricBreakdownArgsSchema, getMetricBreakdownResponseSchema, getMetricLabelKeysArgsSchema, getMetricLabelKeysResponseSchema, getMetricLabelValuesArgsSchema, getMetricLabelValuesResponseSchema, getMetricNamesArgsSchema, getMetricNamesResponseSchema, getMetricPercentilesArgsSchema, getMetricPercentilesResponseSchema, getMetricTimeSeriesArgsSchema, getMetricTimeSeriesResponseSchema, getScoreAggregateArgsSchema, getScoreAggregateResponseSchema, getScoreBreakdownArgsSchema, getScoreBreakdownResponseSchema, getScorePercentilesArgsSchema, getScorePercentilesResponseSchema, getScoreTimeSeriesArgsSchema, getScoreTimeSeriesResponseSchema, getServiceNamesArgsSchema, getServiceNamesResponseSchema, getTagsArgsSchema, getTagsResponseSchema, listFeedbackArgsSchema, listFeedbackResponseSchema, listLogsArgsSchema, listLogsResponseSchema, listMetricsArgsSchema, listMetricsResponseSchema, listScoresArgsSchema, listScoresResponseSchema, logLevelSchema, logRecordInputSchema, logRecordSchema, logsFilterSchema, logsOrderByFieldSchema, logsOrderBySchema, metadataField, metricInputSchema, metricRecordSchema, metricTypeSchema, metricsAggregationSchema, metricsFilterSchema, metricsOrderByFieldSchema, metricsOrderBySchema, noOpLoggerContext, noOpMetricsContext, noOpTracingContext, organizationIdField, paginationArgsSchema, paginationInfoSchema, parentEntityIdField, parentEntityNameField, parentEntityTypeField, requestIdField, resolveObservabilityContext, resourceIdField, rootEntityIdField, rootEntityNameField, rootEntityTypeField, runIdField, scopeField, scoreInputSchema, scoreRecordSchema, scoresFilterSchema, scoresOrderByFieldSchema, scoresOrderBySchema, serviceNameField, sessionIdField, sortDirectionSchema, sourceField, spanContextFields, spanIdField, tagsField, threadIdField, traceIdField, updatedAtField, userIdField, wrapMastra };
-//# sourceMappingURL=chunk-2NISBCML.js.map
-//# sourceMappingURL=chunk-2NISBCML.js.map
+export { EntityType, InternalSpans, SpanType, TracingEventType, aggregationIntervalSchema, aggregationTypeSchema, batchCreateFeedbackArgsSchema, batchCreateLogsArgsSchema, batchCreateMetricsArgsSchema, batchCreateScoresArgsSchema, commonFilterFields, comparePeriodSchema, contextFields, createFeedbackArgsSchema, createFeedbackBodySchema, createFeedbackRecordSchema, createFeedbackResponseSchema, createLogRecordSchema, createMetricRecordSchema, createScoreArgsSchema, createScoreBodySchema, createScoreRecordSchema, createScoreResponseSchema, createdAtField, dateRangeSchema, dbTimestamps, entityIdField, entityNameField, entityTypeField, environmentField, executeWithContext, executeWithContextSync, executionSourceField, experimentIdField, feedbackFilterSchema, feedbackInputSchema, feedbackOrderByFieldSchema, feedbackOrderBySchema, feedbackRecordSchema, getCurrentSpan, getEntityNamesArgsSchema, getEntityNamesResponseSchema, getEntityTypeForSpan, getEntityTypesArgsSchema, getEntityTypesResponseSchema, getEnvironmentsArgsSchema, getEnvironmentsResponseSchema, getFeedbackAggregateArgsSchema, getFeedbackAggregateResponseSchema, getFeedbackBreakdownArgsSchema, getFeedbackBreakdownResponseSchema, getFeedbackPercentilesArgsSchema, getFeedbackPercentilesResponseSchema, getFeedbackTimeSeriesArgsSchema, getFeedbackTimeSeriesResponseSchema, getMetricAggregateArgsSchema, getMetricAggregateResponseSchema, getMetricBreakdownArgsSchema, getMetricBreakdownResponseSchema, getMetricLabelKeysArgsSchema, getMetricLabelKeysResponseSchema, getMetricLabelValuesArgsSchema, getMetricLabelValuesResponseSchema, getMetricNamesArgsSchema, getMetricNamesResponseSchema, getMetricPercentilesArgsSchema, getMetricPercentilesResponseSchema, getMetricTimeSeriesArgsSchema, getMetricTimeSeriesResponseSchema, getOrCreateSpan, getRootExportSpan, getScoreAggregateArgsSchema, getScoreAggregateResponseSchema, getScoreBreakdownArgsSchema, getScoreBreakdownResponseSchema, getScorePercentilesArgsSchema, getScorePercentilesResponseSchema, getScoreTimeSeriesArgsSchema, getScoreTimeSeriesResponseSchema, getServiceNamesArgsSchema, getServiceNamesResponseSchema, getTagsArgsSchema, getTagsResponseSchema, listFeedbackArgsSchema, listFeedbackResponseSchema, listLogsArgsSchema, listLogsResponseSchema, listMetricsArgsSchema, listMetricsResponseSchema, listScoresArgsSchema, listScoresResponseSchema, logLevelSchema, logRecordInputSchema, logRecordSchema, logsFilterSchema, logsOrderByFieldSchema, logsOrderBySchema, metadataField, metricInputSchema, metricRecordSchema, metricTypeSchema, metricsAggregationSchema, metricsFilterSchema, metricsOrderByFieldSchema, metricsOrderBySchema, organizationIdField, paginationArgsSchema, paginationInfoSchema, parentEntityIdField, parentEntityNameField, parentEntityTypeField, requestIdField, resourceIdField, rootEntityIdField, rootEntityNameField, rootEntityTypeField, runIdField, scopeField, scoreInputSchema, scoreRecordSchema, scoresFilterSchema, scoresOrderByFieldSchema, scoresOrderBySchema, serviceNameField, sessionIdField, sortDirectionSchema, sourceField, spanContextFields, spanIdField, tagsField, threadIdField, traceIdField, updatedAtField, userIdField };
+//# sourceMappingURL=chunk-WG7QIVL7.js.map
+//# sourceMappingURL=chunk-WG7QIVL7.js.map