agent-inspect 6.31.12 → 6.31.13

package/packages/core/dist/persisted.cjs

@@ -1535,18 +1535,16 @@ var TreeBuilder = class {
         inc(confidenceBreakdown, e.confidence);
         kinds[e.kind] = (kinds[e.kind] ?? 0) + 1;
       }
-      const startedAt = sorted.length > 0 ? sorted[0].timestamp : void 0;
-      const endedAt = sorted.length > 0 ? sorted[sorted.length - 1].timestamp : void 0;
-      const status = computeRunStatus(sorted);
-      const durationMs = startedAt !== void 0 && endedAt !== void 0 && Number.isFinite(startedAt) && Number.isFinite(endedAt) && endedAt >= startedAt && status !== "running" ? endedAt - startedAt : void 0;
       const name = sorted.find((e) => e.kind === "RUN")?.name;
+      const status = computeRunStatus(sorted);
+      const timing = resolveRunTiming(sorted, status);
       out.push({
         runId,
         name,
         status,
-        startedAt,
-        endedAt: status === "running" ? void 0 : endedAt,
-        durationMs,
+        startedAt: timing.startedAt,
+        endedAt: timing.endedAt,
+        durationMs: timing.durationMs,
         children: roots,
         metadata: {
           totalEvents: sorted.length,
@@ -1560,6 +1558,57 @@ var TreeBuilder = class {
     return out;
   }
 };
+function resolveRunTiming(sorted, status) {
+  let startedAtMs;
+  let endedAtMs;
+  let explicitDurationMs;
+  for (const event of sorted) {
+    if (event.kind !== "RUN") continue;
+    const attrs = event.attributes ?? {};
+    if (typeof attrs.startedAtMs === "number" && Number.isFinite(attrs.startedAtMs)) {
+      startedAtMs = startedAtMs === void 0 ? attrs.startedAtMs : Math.min(startedAtMs, attrs.startedAtMs);
+    }
+    if (typeof attrs.endedAtMs === "number" && Number.isFinite(attrs.endedAtMs)) {
+      endedAtMs = endedAtMs === void 0 ? attrs.endedAtMs : Math.max(endedAtMs, attrs.endedAtMs);
+    }
+    if ((event.status === "ok" || event.status === "error") && typeof event.durationMs === "number" && Number.isFinite(event.durationMs) && event.durationMs >= 0) {
+      explicitDurationMs = event.durationMs;
+      if (endedAtMs === void 0 && startedAtMs !== void 0) {
+        endedAtMs = startedAtMs + event.durationMs;
+      } else if (endedAtMs === void 0 && Number.isFinite(event.timestamp)) {
+        endedAtMs = event.timestamp + event.durationMs;
+        if (startedAtMs === void 0) {
+          startedAtMs = event.timestamp;
+        }
+      }
+    }
+    if (startedAtMs === void 0 && event.status === "running" && Number.isFinite(event.timestamp)) {
+      startedAtMs = event.timestamp;
+    }
+  }
+  if (startedAtMs !== void 0 && endedAtMs !== void 0 && Number.isFinite(startedAtMs) && Number.isFinite(endedAtMs) && endedAtMs >= startedAtMs && status !== "running") {
+    return {
+      startedAt: startedAtMs,
+      endedAt: endedAtMs,
+      durationMs: endedAtMs - startedAtMs
+    };
+  }
+  if (startedAtMs !== void 0 && explicitDurationMs !== void 0 && status !== "running") {
+    return {
+      startedAt: startedAtMs,
+      endedAt: startedAtMs + explicitDurationMs,
+      durationMs: explicitDurationMs
+    };
+  }
+  const startedAt = sorted.length > 0 ? sorted[0].timestamp : void 0;
+  const endedAt = sorted.length > 0 ? sorted[sorted.length - 1].timestamp : void 0;
+  const durationMs = startedAt !== void 0 && endedAt !== void 0 && Number.isFinite(startedAt) && Number.isFinite(endedAt) && endedAt >= startedAt && status !== "running" ? endedAt - startedAt : void 0;
+  return {
+    startedAt,
+    endedAt: status === "running" ? void 0 : endedAt,
+    durationMs
+  };
+}
 
 // packages/core/src/persisted/tree-bridge.ts
 function logicalToPersisted(event) {