@mastra/core 1.17.0 → 1.18.0-alpha.1

package/dist/{chunk-AGEYVARR.js → chunk-VMKNS3YO.js}

@@ -104,6 +104,207 @@ var listScoresResponseSchema = z.object({
   pagination: paginationInfoSchema,
   scores: z.array(scoreRowDataSchema)
 });
+function extractTrajectory(output) {
+  const steps = [];
+  for (const message of output) {
+    const toolInvocations = message?.content?.toolInvocations;
+    if (!toolInvocations) continue;
+    for (const invocation of toolInvocations) {
+      if (invocation && invocation.toolName && (invocation.state === "result" || invocation.state === "call")) {
+        const toolArgs = invocation.args != null && typeof invocation.args === "object" && !Array.isArray(invocation.args) ? invocation.args : invocation.args != null ? { value: invocation.args } : void 0;
+        const rawResult = invocation.state === "result" ? invocation.result : void 0;
+        const toolResult = rawResult != null && typeof rawResult === "object" && !Array.isArray(rawResult) ? rawResult : rawResult != null ? { value: rawResult } : void 0;
+        steps.push({
+          stepType: "tool_call",
+          name: invocation.toolName,
+          toolArgs,
+          toolResult,
+          success: invocation.state === "result"
+        });
+      }
+    }
+  }
+  return { steps, rawOutput: output };
+}
+function extractWorkflowTrajectory(stepResults, stepExecutionPath) {
+  const steps = [];
+  const stepIds = stepExecutionPath ?? Object.keys(stepResults);
+  let totalStartedAt;
+  let totalEndedAt;
+  for (const stepId of stepIds) {
+    const result = stepResults[stepId];
+    if (!result) continue;
+    if (result.startedAt != null) {
+      if (totalStartedAt == null || result.startedAt < totalStartedAt) {
+        totalStartedAt = result.startedAt;
+      }
+    }
+    const endedAt = "endedAt" in result ? result.endedAt : void 0;
+    if (endedAt != null) {
+      if (totalEndedAt == null || endedAt > totalEndedAt) {
+        totalEndedAt = endedAt;
+      }
+    }
+    const durationMs = result.startedAt != null && endedAt != null ? endedAt - result.startedAt : void 0;
+    const output = "output" in result && result.output != null && typeof result.output === "object" && !Array.isArray(result.output) ? result.output : "output" in result && result.output != null ? { value: result.output } : void 0;
+    steps.push({
+      stepType: "workflow_step",
+      name: stepId,
+      stepId,
+      status: result.status,
+      output,
+      durationMs,
+      metadata: result.metadata
+    });
+  }
+  const totalDurationMs = totalStartedAt != null && totalEndedAt != null ? totalEndedAt - totalStartedAt : void 0;
+  return {
+    steps,
+    totalDurationMs,
+    rawWorkflowResult: { stepResults, stepExecutionPath }
+  };
+}
+var SKIPPED_SPAN_TYPES = /* @__PURE__ */ new Set([
+  "generic" /* GENERIC */,
+  "model_step" /* MODEL_STEP */,
+  "model_chunk" /* MODEL_CHUNK */,
+  "workflow_conditional_eval" /* WORKFLOW_CONDITIONAL_EVAL */
+]);
+function spanToTrajectorySteps(node) {
+  const { span, children: childNodes } = node;
+  if (SKIPPED_SPAN_TYPES.has(span.spanType)) {
+    return childNodes.flatMap(spanToTrajectorySteps);
+  }
+  const durationMs = span.endedAt != null && span.startedAt != null ? span.endedAt.getTime() - span.startedAt.getTime() : void 0;
+  const childSteps = childNodes.flatMap(spanToTrajectorySteps);
+  const base = {
+    name: span.name,
+    durationMs,
+    metadata: span.metadata,
+    ...childSteps.length > 0 ? { children: childSteps } : {}
+  };
+  const attrs = span.attributes ?? {};
+  switch (span.spanType) {
+    case "tool_call" /* TOOL_CALL */: {
+      const toolArgs = toRecordOrUndefined(span.input);
+      const toolResult = toRecordOrUndefined(span.output);
+      return [
+        {
+          ...base,
+          stepType: "tool_call",
+          toolArgs,
+          toolResult,
+          success: typeof attrs.success === "boolean" ? attrs.success : void 0
+        }
+      ];
+    }
+    case "mcp_tool_call" /* MCP_TOOL_CALL */: {
+      const toolArgs = toRecordOrUndefined(span.input);
+      const toolResult = toRecordOrUndefined(span.output);
+      return [
+        {
+          ...base,
+          stepType: "mcp_tool_call",
+          toolArgs,
+          toolResult,
+          mcpServer: typeof attrs.mcpServer === "string" ? attrs.mcpServer : void 0,
+          success: typeof attrs.success === "boolean" ? attrs.success : void 0
+        }
+      ];
+    }
+    case "model_generation" /* MODEL_GENERATION */: {
+      const usage = attrs.usage;
+      return [
+        {
+          ...base,
+          stepType: "model_generation",
+          modelId: typeof attrs.model === "string" ? attrs.model : void 0,
+          promptTokens: usage?.inputTokens,
+          completionTokens: usage?.outputTokens,
+          finishReason: typeof attrs.finishReason === "string" ? attrs.finishReason : void 0
+        }
+      ];
+    }
+    case "agent_run" /* AGENT_RUN */:
+      return [{ ...base, stepType: "agent_run", agentId: span.entityId ?? void 0 }];
+    case "workflow_run" /* WORKFLOW_RUN */:
+      return [{ ...base, stepType: "workflow_run", workflowId: span.entityId ?? void 0 }];
+    case "workflow_step" /* WORKFLOW_STEP */: {
+      const output = toRecordOrUndefined(span.output);
+      return [{ ...base, stepType: "workflow_step", stepId: span.name, output }];
+    }
+    case "workflow_conditional" /* WORKFLOW_CONDITIONAL */:
+      return [{ ...base, stepType: "workflow_conditional" }];
+    case "workflow_parallel" /* WORKFLOW_PARALLEL */:
+      return [{ ...base, stepType: "workflow_parallel" }];
+    case "workflow_loop" /* WORKFLOW_LOOP */:
+      return [{ ...base, stepType: "workflow_loop" }];
+    case "workflow_sleep" /* WORKFLOW_SLEEP */:
+      return [{ ...base, stepType: "workflow_sleep" }];
+    case "workflow_wait_event" /* WORKFLOW_WAIT_EVENT */:
+      return [{ ...base, stepType: "workflow_wait_event" }];
+    case "processor_run" /* PROCESSOR_RUN */:
+      return [{ ...base, stepType: "processor_run" }];
+    default:
+      return childSteps;
+  }
+}
+function toRecordOrUndefined(value) {
+  if (value == null) return void 0;
+  if (typeof value === "object" && !Array.isArray(value)) {
+    return value;
+  }
+  return { value };
+}
+function extractTrajectoryFromTrace(spans, rootSpanId) {
+  if (spans.length === 0) {
+    return { steps: [] };
+  }
+  const nodeMap = /* @__PURE__ */ new Map();
+  for (const span of spans) {
+    nodeMap.set(span.spanId, { span, children: [] });
+  }
+  const roots = [];
+  for (const span of spans) {
+    const node = nodeMap.get(span.spanId);
+    if (span.parentSpanId && nodeMap.has(span.parentSpanId)) {
+      nodeMap.get(span.parentSpanId).children.push(node);
+    } else {
+      roots.push(node);
+    }
+  }
+  for (const node of nodeMap.values()) {
+    node.children.sort((a, b) => a.span.startedAt.getTime() - b.span.startedAt.getTime());
+  }
+  let targetRoots;
+  if (rootSpanId) {
+    const rootNode = nodeMap.get(rootSpanId);
+    targetRoots = rootNode ? [rootNode] : roots;
+  } else {
+    targetRoots = roots;
+  }
+  let stepsToConvert;
+  if (targetRoots.length === 1) {
+    const root = targetRoots[0];
+    const containerTypes = /* @__PURE__ */ new Set(["workflow_run" /* WORKFLOW_RUN */, "agent_run" /* AGENT_RUN */]);
+    if (containerTypes.has(root.span.spanType)) {
+      stepsToConvert = root.children;
+    } else {
+      stepsToConvert = targetRoots;
+    }
+  } else {
+    stepsToConvert = targetRoots;
+  }
+  const steps = stepsToConvert.flatMap(spanToTrajectorySteps);
+  let totalDurationMs;
+  if (targetRoots.length === 1) {
+    const root = targetRoots[0].span;
+    if (root.endedAt && root.startedAt) {
+      totalDurationMs = root.endedAt.getTime() - root.startedAt.getTime();
+    }
+  }
+  return { steps, totalDurationMs };
+}
 
 // src/storage/domains/observability/tracing.ts
 var createOmitKeys = (shape) => Object.fromEntries(Object.keys(shape).map((k) => [k, true]));
@@ -886,6 +1087,6 @@ var OBSERVATIONAL_MEMORY_TABLE_SCHEMA = {
   [TABLE_OBSERVATIONAL_MEMORY]: OBSERVATIONAL_MEMORY_SCHEMA
 };
 
-export { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, DATASETS_SCHEMA, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCP_CLIENTS_SCHEMA, MCP_CLIENT_VERSIONS_SCHEMA, MCP_SERVERS_SCHEMA, MCP_SERVER_VERSIONS_SCHEMA, OBSERVATIONAL_MEMORY_SCHEMA, OBSERVATIONAL_MEMORY_TABLE_SCHEMA, OLD_SPAN_SCHEMA, PROMPT_BLOCKS_SCHEMA, PROMPT_BLOCK_VERSIONS_SCHEMA, SCORERS_SCHEMA, SCORER_DEFINITIONS_SCHEMA, SCORER_DEFINITION_VERSIONS_SCHEMA, SKILLS_SCHEMA, SKILL_BLOBS_SCHEMA, SKILL_VERSIONS_SCHEMA, SPAN_SCHEMA, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_OBSERVATIONAL_MEMORY, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WORKSPACES_SCHEMA, WORKSPACE_VERSIONS_SCHEMA, batchCreateSpansArgsSchema, batchDeleteTracesArgsSchema, batchUpdateSpansArgsSchema, buildStorageSchema, computeTraceStatus, createSpanArgsSchema, createSpanRecordSchema, getRootSpanArgsSchema, getRootSpanResponseSchema, getSpanArgsSchema, getSpanResponseSchema, getTraceArgsSchema, getTraceResponseSchema, listScoresBySpanResponseSchema, listScoresResponseSchema, listTracesArgsSchema, listTracesResponseSchema, saveScorePayloadSchema, scoreResultSchema, scoreRowDataSchema, scoreTracesRequestSchema, scoreTracesResponseSchema, scoringEntityTypeSchema, scoringExtractStepResultSchema, scoringHookInputSchema, scoringInputSchema, scoringInputWithExtractStepResultAndAnalyzeStepResultSchema, scoringInputWithExtractStepResultAndScoreAndReasonSchema, scoringInputWithExtractStepResultSchema, scoringPromptsSchema, scoringSourceSchema, scoringValueSchema, spanIds, spanIdsSchema, spanRecordSchema, toTraceSpan, toTraceSpans, traceSpanSchema, tracesFilterSchema, tracesOrderByFieldSchema, tracesOrderBySchema, updateSpanArgsSchema, updateSpanRecordSchema };
-//# sourceMappingURL=chunk-AGEYVARR.js.map
-//# sourceMappingURL=chunk-AGEYVARR.js.map
+export { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, DATASETS_SCHEMA, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCP_CLIENTS_SCHEMA, MCP_CLIENT_VERSIONS_SCHEMA, MCP_SERVERS_SCHEMA, MCP_SERVER_VERSIONS_SCHEMA, OBSERVATIONAL_MEMORY_SCHEMA, OBSERVATIONAL_MEMORY_TABLE_SCHEMA, OLD_SPAN_SCHEMA, PROMPT_BLOCKS_SCHEMA, PROMPT_BLOCK_VERSIONS_SCHEMA, SCORERS_SCHEMA, SCORER_DEFINITIONS_SCHEMA, SCORER_DEFINITION_VERSIONS_SCHEMA, SKILLS_SCHEMA, SKILL_BLOBS_SCHEMA, SKILL_VERSIONS_SCHEMA, SPAN_SCHEMA, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_OBSERVATIONAL_MEMORY, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, TraceStatus, WORKSPACES_SCHEMA, WORKSPACE_VERSIONS_SCHEMA, batchCreateSpansArgsSchema, batchDeleteTracesArgsSchema, batchUpdateSpansArgsSchema, buildStorageSchema, computeTraceStatus, createSpanArgsSchema, createSpanRecordSchema, extractTrajectory, extractTrajectoryFromTrace, extractWorkflowTrajectory, getRootSpanArgsSchema, getRootSpanResponseSchema, getSpanArgsSchema, getSpanResponseSchema, getTraceArgsSchema, getTraceResponseSchema, listScoresBySpanResponseSchema, listScoresResponseSchema, listTracesArgsSchema, listTracesResponseSchema, saveScorePayloadSchema, scoreResultSchema, scoreRowDataSchema, scoreTracesRequestSchema, scoreTracesResponseSchema, scoringEntityTypeSchema, scoringExtractStepResultSchema, scoringHookInputSchema, scoringInputSchema, scoringInputWithExtractStepResultAndAnalyzeStepResultSchema, scoringInputWithExtractStepResultAndScoreAndReasonSchema, scoringInputWithExtractStepResultSchema, scoringPromptsSchema, scoringSourceSchema, scoringValueSchema, spanIds, spanIdsSchema, spanRecordSchema, toTraceSpan, toTraceSpans, traceSpanSchema, tracesFilterSchema, tracesOrderByFieldSchema, tracesOrderBySchema, updateSpanArgsSchema, updateSpanRecordSchema };
+//# sourceMappingURL=chunk-VMKNS3YO.js.map
+//# sourceMappingURL=chunk-VMKNS3YO.js.map