agent-inspect 6.31.29 → 6.31.31

package/packages/core/dist/exporters.mjs

@@ -418,6 +418,159 @@ function redactRunTreeForExport(tree, options) {
   return clone;
 }
 
+// packages/core/src/exporters/presentation-lifecycle.ts
+function stepIdOf(event) {
+  const value = event.attributes?.stepId;
+  return typeof value === "string" && value.trim() !== "" ? value : void 0;
+}
+function attemptIdOf(event) {
+  const direct = event.attributes?.attemptId;
+  if (typeof direct === "string" && direct.trim() !== "") return direct;
+  const meta = event.attributes?.metadata;
+  if (meta && typeof meta === "object" && !Array.isArray(meta)) {
+    const nested = meta.attemptId;
+    if (typeof nested === "string" && nested.trim() !== "") return nested;
+  }
+  return void 0;
+}
+function lifecycleKey(event) {
+  if (event.kind === "RUN") return `run:${event.runId}`;
+  const stepId = stepIdOf(event);
+  if (stepId !== void 0) return `step:${stepId}`;
+  return `event:${event.eventId}`;
+}
+function isTerminal(status) {
+  return status === "ok" || status === "error";
+}
+function eventTime(event) {
+  return typeof event.timestamp === "number" ? event.timestamp : 0;
+}
+function cloneNode(node) {
+  return {
+    event: {
+      ...node.event,
+      ...node.event.attributes !== void 0 ? { attributes: { ...node.event.attributes } } : {}
+    },
+    children: node.children.map(cloneNode),
+    depth: node.depth
+  };
+}
+function joinOnePair(start, terminal) {
+  const childMap = /* @__PURE__ */ new Map();
+  if (start) {
+    for (const child of start.children) {
+      childMap.set(child.event.eventId, child);
+    }
+  }
+  for (const child of terminal.children) {
+    childMap.set(child.event.eventId, child);
+  }
+  return {
+    event: {
+      ...terminal.event,
+      name: start?.event.name ?? terminal.event.name,
+      kind: start?.event.kind ?? terminal.event.kind,
+      ...start && attemptIdOf(terminal.event) === void 0 && attemptIdOf(start.event) ? {
+        attributes: {
+          ...terminal.event.attributes,
+          attemptId: attemptIdOf(start.event)
+        }
+      } : {}
+    },
+    children: presentSiblingNodes([...childMap.values()]),
+    depth: terminal.depth
+  };
+}
+function joinSiblingGroup(group) {
+  if (group.length <= 1) return group;
+  const terminals = group.filter((n) => isTerminal(n.event.status)).sort((a, b) => eventTime(a.event) - eventTime(b.event));
+  const runnings = group.filter((n) => n.event.status === "running").sort((a, b) => eventTime(a.event) - eventTime(b.event));
+  const others = group.filter(
+    (n) => !isTerminal(n.event.status) && n.event.status !== "running"
+  );
+  const terminalStatuses = new Set(terminals.map((n) => n.event.status));
+  if (terminalStatuses.size > 1) {
+    return group;
+  }
+  if (terminals.length > 1 && terminals.every((t) => attemptIdOf(t.event) === void 0) && runnings.every((r) => attemptIdOf(r.event) === void 0)) {
+    return group;
+  }
+  const usedRunning = /* @__PURE__ */ new Set();
+  const out = [];
+  for (const terminal of terminals) {
+    const terminalAttempt = attemptIdOf(terminal.event);
+    const terminalTs = eventTime(terminal.event);
+    const candidate = runnings.find((r) => {
+      if (usedRunning.has(r.event.eventId)) return false;
+      if (eventTime(r.event) > terminalTs) return false;
+      const runningAttempt = attemptIdOf(r.event);
+      if (terminalAttempt !== void 0) {
+        return runningAttempt === terminalAttempt;
+      }
+      if (runningAttempt !== void 0) {
+        return true;
+      }
+      return true;
+    });
+    if (candidate) {
+      usedRunning.add(candidate.event.eventId);
+      out.push(joinOnePair(candidate, terminal));
+    } else {
+      out.push({
+        ...terminal,
+        children: presentSiblingNodes(terminal.children)
+      });
+    }
+  }
+  for (const running of runnings) {
+    if (usedRunning.has(running.event.eventId)) continue;
+    out.push({
+      ...running,
+      children: presentSiblingNodes(running.children)
+    });
+  }
+  for (const other of others) {
+    out.push({
+      ...other,
+      children: presentSiblingNodes(other.children)
+    });
+  }
+  return out;
+}
+function presentSiblingNodes(nodes) {
+  const prepared = nodes.map((n) => ({
+    ...cloneNode(n),
+    children: presentSiblingNodes(n.children)
+  }));
+  const order = [];
+  const groups = /* @__PURE__ */ new Map();
+  for (const node of prepared) {
+    const key = lifecycleKey(node.event);
+    if (!groups.has(key)) {
+      groups.set(key, []);
+      order.push(key);
+    }
+    groups.get(key).push(node);
+  }
+  const out = [];
+  for (const key of order) {
+    out.push(...joinSiblingGroup(groups.get(key)));
+  }
+  return out;
+}
+function presentLifecycleJoinedTree(tree) {
+  return {
+    ...tree,
+    children: presentSiblingNodes(tree.children),
+    metadata: {
+      ...tree.metadata,
+      confidenceBreakdown: { ...tree.metadata.confidenceBreakdown },
+      kinds: { ...tree.metadata.kinds },
+      ...tree.metadata.relationshipSummary !== void 0 ? { relationshipSummary: { ...tree.metadata.relationshipSummary } } : {}
+    }
+  };
+}
+
 // packages/core/src/exporters/html-exporter.ts
 function renderTreeHtml(nodes, ulClass = "tree") {
   if (nodes.length === 0) return "";
@@ -445,32 +598,33 @@ function exportHtml(tree, options) {
   const includeErrors = options?.includeErrors ?? true;
   const maxLen = options?.maxAttributeLength ?? 500;
   const redacted = options?.redacted ?? true;
-  const titleName = escapeHtml(tree.name ?? tree.runId);
+  const view = presentLifecycleJoinedTree(tree);
+  const titleName = escapeHtml(view.name ?? view.runId);
   const summaryRows = [];
   summaryRows.push(
-    `<tr><th scope="row">runId</th><td><code>${escapeHtml(tree.runId)}</code></td></tr>`
+    `<tr><th scope="row">runId</th><td><code>${escapeHtml(view.runId)}</code></td></tr>`
   );
-  if (tree.name !== void 0) {
-    summaryRows.push(`<tr><th scope="row">name</th><td>${escapeHtml(tree.name)}</td></tr>`);
+  if (view.name !== void 0) {
+    summaryRows.push(`<tr><th scope="row">name</th><td>${escapeHtml(view.name)}</td></tr>`);
   }
   summaryRows.push(
-    `<tr><th scope="row">status</th><td>${escapeHtml(String(tree.status ?? "unknown"))}</td></tr>`
+    `<tr><th scope="row">status</th><td>${escapeHtml(String(view.status ?? "unknown"))}</td></tr>`
   );
   summaryRows.push(
-    `<tr><th scope="row">durationMs</th><td>${tree.durationMs !== void 0 ? escapeHtml(String(tree.durationMs)) : "\u2014"}</td></tr>`
+    `<tr><th scope="row">durationMs</th><td>${view.durationMs !== void 0 ? escapeHtml(String(view.durationMs)) : "\u2014"}</td></tr>`
   );
   summaryRows.push(
-    `<tr><th scope="row">startedAt</th><td>${tree.startedAt !== void 0 ? escapeHtml(String(tree.startedAt)) : "\u2014"}</td></tr>`
+    `<tr><th scope="row">startedAt</th><td>${view.startedAt !== void 0 ? escapeHtml(String(view.startedAt)) : "\u2014"}</td></tr>`
   );
   summaryRows.push(
-    `<tr><th scope="row">endedAt</th><td>${tree.endedAt !== void 0 ? escapeHtml(String(tree.endedAt)) : "\u2014"}</td></tr>`
+    `<tr><th scope="row">endedAt</th><td>${view.endedAt !== void 0 ? escapeHtml(String(view.endedAt)) : "\u2014"}</td></tr>`
   );
   summaryRows.push(
-    `<tr><th scope="row">totalEvents</th><td>${escapeHtml(String(tree.metadata.totalEvents))}</td></tr>`
+    `<tr><th scope="row">totalEvents</th><td>${escapeHtml(String(view.metadata.totalEvents))}</td></tr>`
   );
   let confidenceHtml = "";
   if (includeMetadata) {
-    const cb = tree.metadata.confidenceBreakdown;
+    const cb = view.metadata.confidenceBreakdown;
     confidenceHtml += "<h3>Confidence breakdown</h3><table><thead><tr><th>bucket</th><th>count</th></tr></thead><tbody>";
     for (const k of Object.keys(cb).sort()) {
       const key = k;
@@ -478,14 +632,14 @@ function exportHtml(tree, options) {
     }
     confidenceHtml += "</tbody></table>";
     confidenceHtml += "<h3>Kind breakdown</h3><table><thead><tr><th>kind</th><th>count</th></tr></thead><tbody>";
-    for (const k of Object.keys(tree.metadata.kinds).sort()) {
+    for (const k of Object.keys(view.metadata.kinds).sort()) {
       const key = k;
-      const c = tree.metadata.kinds[key];
+      const c = view.metadata.kinds[key];
       if (c > 0) confidenceHtml += `<tr><td>${escapeHtml(key)}</td><td>${c}</td></tr>`;
     }
     confidenceHtml += "</tbody></table>";
   }
-  const flat = flattenTree(tree);
+  const flat = flattenTree(view);
   const errors = flat.filter((n) => n.event.status === "error");
   let errorsHtml = "";
   if (includeErrors && errors.length > 0) {
@@ -540,7 +694,7 @@ footer{margin-top:2rem;font-size:0.85rem;color:#555}
 <header><h1>AgentInspect Run: ${titleName}</h1></header>
 <p class="note">Generated locally by AgentInspect.</p>
 ${includeMetadata ? `<section class="summary"><h2>Summary</h2><table>${summaryRows.join("")}</table>${confidenceHtml}</section>` : ""}
-<section class="tree"><h2>Execution tree</h2>${tree.children.length > 0 ? renderTreeHtml(tree.children) : "<p>No steps recorded.</p>"}</section>
+<section class="tree"><h2>Execution tree</h2>${view.children.length > 0 ? renderTreeHtml(view.children) : "<p>No steps recorded.</p>"}</section>
 ${errorsHtml}
 ${attrsHtml}
 <footer>Generated locally by AgentInspect. Review for sensitive data before sharing.</footer>
@@ -581,7 +735,8 @@ function exportMarkdown(tree, options) {
   const includeErrors = options?.includeErrors ?? true;
   const maxLen = options?.maxAttributeLength ?? 500;
   const redacted = options?.redacted ?? true;
-  const titleName = tree.name ?? tree.runId;
+  const view = presentLifecycleJoinedTree(tree);
+  const titleName = view.name ?? view.runId;
   const lines = [];
   lines.push(`# AgentInspect Run: ${escapeMarkdown(titleName)}`);
   lines.push("");
@@ -590,28 +745,28 @@ function exportMarkdown(tree, options) {
   if (includeMetadata) {
     lines.push("## Summary");
     lines.push("");
-    lines.push(`- **runId**: ${escapeMarkdown(tree.runId)}`);
-    if (tree.name !== void 0) lines.push(`- **name**: ${escapeMarkdown(tree.name)}`);
-    lines.push(`- **status**: ${escapeMarkdown(String(tree.status ?? "unknown"))}`);
+    lines.push(`- **runId**: ${escapeMarkdown(view.runId)}`);
+    if (view.name !== void 0) lines.push(`- **name**: ${escapeMarkdown(view.name)}`);
+    lines.push(`- **status**: ${escapeMarkdown(String(view.status ?? "unknown"))}`);
     lines.push(
-      `- **durationMs**: ${tree.durationMs !== void 0 ? escapeMarkdown(String(tree.durationMs)) : "-"}`
+      `- **durationMs**: ${view.durationMs !== void 0 ? escapeMarkdown(String(view.durationMs)) : "-"}`
     );
     lines.push(
-      `- **startedAt**: ${tree.startedAt !== void 0 ? escapeMarkdown(String(tree.startedAt)) : "-"}`
+      `- **startedAt**: ${view.startedAt !== void 0 ? escapeMarkdown(String(view.startedAt)) : "-"}`
     );
     lines.push(
-      `- **endedAt**: ${tree.endedAt !== void 0 ? escapeMarkdown(String(tree.endedAt)) : "-"}`
+      `- **endedAt**: ${view.endedAt !== void 0 ? escapeMarkdown(String(view.endedAt)) : "-"}`
     );
-    lines.push(`- **totalEvents**: ${tree.metadata.totalEvents}`);
+    lines.push(`- **totalEvents**: ${view.metadata.totalEvents}`);
     lines.push("");
     lines.push("### Confidence breakdown");
     lines.push("");
     lines.push("| bucket | count |");
     lines.push("| --- | --- |");
-    for (const k of Object.keys(tree.metadata.confidenceBreakdown).sort()) {
+    for (const k of Object.keys(view.metadata.confidenceBreakdown).sort()) {
       const key = k;
       lines.push(
-        `| ${escapeMarkdown(key)} | ${tree.metadata.confidenceBreakdown[key]} |`
+        `| ${escapeMarkdown(key)} | ${view.metadata.confidenceBreakdown[key]} |`
       );
     }
     lines.push("");
@@ -619,9 +774,9 @@ function exportMarkdown(tree, options) {
     lines.push("");
     lines.push("| kind | count |");
     lines.push("| --- | --- |");
-    for (const k of Object.keys(tree.metadata.kinds).sort()) {
+    for (const k of Object.keys(view.metadata.kinds).sort()) {
       const key = k;
-      const c = tree.metadata.kinds[key];
+      const c = view.metadata.kinds[key];
       if (c > 0) lines.push(`| ${escapeMarkdown(key)} | ${c} |`);
     }
     lines.push("");
@@ -630,11 +785,11 @@ function exportMarkdown(tree, options) {
   lines.push("");
   lines.push("```text");
   lines.push(
-    tree.children.length > 0 ? renderTreeAscii(tree.children) : "(no steps)"
+    view.children.length > 0 ? renderTreeAscii(view.children) : "(no steps)"
   );
   lines.push("```");
   lines.push("");
-  const flat = flattenTree(tree);
+  const flat = flattenTree(view);
   const errors = flat.filter((n) => n.event.status === "error");
   if (includeErrors && errors.length > 0) {
     lines.push("## Errors");