@mastra/core 1.18.0-alpha.0 → 1.18.0-alpha.1

package/dist/{chunk-QVQ3PGG2.cjs → chunk-7IDACSBM.cjs}

@@ -106,6 +106,207 @@ var listScoresResponseSchema = v4.z.object({
   pagination: chunkG5HKDGNT_cjs.paginationInfoSchema,
   scores: v4.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]));
@@ -951,6 +1152,9 @@ exports.buildStorageSchema = buildStorageSchema;
 exports.computeTraceStatus = computeTraceStatus;
 exports.createSpanArgsSchema = createSpanArgsSchema;
 exports.createSpanRecordSchema = createSpanRecordSchema;
+exports.extractTrajectory = extractTrajectory;
+exports.extractTrajectoryFromTrace = extractTrajectoryFromTrace;
+exports.extractWorkflowTrajectory = extractWorkflowTrajectory;
 exports.getRootSpanArgsSchema = getRootSpanArgsSchema;
 exports.getRootSpanResponseSchema = getRootSpanResponseSchema;
 exports.getSpanArgsSchema = getSpanArgsSchema;
@@ -987,5 +1191,5 @@ exports.tracesOrderByFieldSchema = tracesOrderByFieldSchema;
 exports.tracesOrderBySchema = tracesOrderBySchema;
 exports.updateSpanArgsSchema = updateSpanArgsSchema;
 exports.updateSpanRecordSchema = updateSpanRecordSchema;
-//# sourceMappingURL=chunk-QVQ3PGG2.cjs.map
-//# sourceMappingURL=chunk-QVQ3PGG2.cjs.map
+//# sourceMappingURL=chunk-7IDACSBM.cjs.map
+//# sourceMappingURL=chunk-7IDACSBM.cjs.map