@pasko70/pibo 1.16.6 → 2.0.0

package/dist/shared/trace-event-projection.js

@@ -3,21 +3,22 @@ import { compareTraceNodes, flattenTraceNodes } from "./trace-nodes.js";
 import { attachAsyncAgentRunNode, isRunStartToolNode } from "./trace-async-agent-runs.js";
 import { createRunNotificationNode, parseRunNotificationText } from "./trace-run-notifications.js";
 import { findLikelyTraceChildSession, isSubagentToolName, } from "./trace-subagent-links.js";
-export function applySingleEventToNodes(nodes, byId, piboSessionId, storedEvent, childByParent, linkedChildByToolCallId, hasPersistedTranscript, openTranscriptEventIds, sessionStatus) {
+export function applySingleEventToNodes(nodes, byId, piboSessionId, storedEvent, childByParent, linkedChildByToolCallId, historyCoverage, openTranscriptEventIds, sessionStatus) {
     const payload = storedEvent.payload;
-    const confirmedUserMessage = hasPersistedTranscript ? confirmedUserMessageEchoNode(nodes, storedEvent) : undefined;
+    const confirmedUserMessage = historyCoverage.mode !== "none" ? confirmedUserMessageEchoNode(nodes, storedEvent) : undefined;
     if (confirmedUserMessage) {
         if (payload.type === "message_steered" && payload.activeEventId) {
             confirmedUserMessage.parentId = messageTurnNodeId(payload.activeEventId);
         }
         return;
     }
-    if (hasPersistedTranscript &&
-        isTranscriptEchoEvent(payload) &&
+    if (((historyCoverage.mode === "native" && isTranscriptEchoEvent(payload))
+        || (historyCoverage.mode === "product" && isProductHistoryEchoEvent(payload))) &&
+        historyCoversEvent(payload, historyCoverage) &&
         !shouldKeepTranscriptEchoEvent(payload, openTranscriptEventIds)) {
         return;
     }
-    if (hasPersistedTranscript && isStaleToolCallEchoEvent(payload, sessionStatus)) {
+    if (isNativeHistoryToolEchoEvent(payload, historyCoverage, sessionStatus)) {
         return;
     }
     if (payload.type === "assistant_delta") {
@@ -201,16 +202,16 @@ export function contentDeltaPatchNodeId(event) {
     return undefined;
 }
 export function reconcileTranscriptUserMessages(nodes, events, turnTimings = []) {
-    const transcriptUsers = nodes.filter((node) => node.type === "user.message" && node.source === "transcript");
+    const transcriptUsers = nodes.filter((node) => node.type === "user.message" && isPersistedHistorySource(node.source));
     const timingByEventId = new Map(turnTimings.map((timing) => [timing.eventId, timing]));
     let latestTranscriptUser;
     for (const node of nodes) {
-        if (node.type === "user.message" && node.source === "transcript") {
+        if (node.type === "user.message" && isPersistedHistorySource(node.source)) {
             latestTranscriptUser = node;
             continue;
         }
         if (node.type !== "assistant.message" ||
-            node.source !== "transcript" ||
+            !isPersistedHistorySource(node.source) ||
             !node.eventId ||
             !latestTranscriptUser ||
             transcriptUserMessageHasCanonicalIdentity(latestTranscriptUser))
@@ -285,12 +286,15 @@ function confirmedUserMessageEchoNode(nodes, event) {
     const canonicalId = `event:${payload.type}:${payloadEventId ?? cryptoSafeId(payload)}`;
     const stableKey = eventStableKey(payload);
     const text = typeof payload.text === "string" ? payload.text : undefined;
-    const transcriptUsers = flattenTraceNodes([...nodes]).filter((node) => node.type === "user.message" && node.source === "transcript");
+    const transcriptUsers = flattenTraceNodes([...nodes]).filter((node) => node.type === "user.message" && isPersistedHistorySource(node.source));
     const identityMatch = transcriptUsers.find((node) => transcriptUserMessageMatchesIdentity(node, canonicalId, stableKey, eventId));
     if (identityMatch)
         return identityMatch;
     return transcriptUsers.find((node) => !transcriptUserMessageHasCanonicalIdentity(node) && Boolean(text && traceNodeText(node) === text));
 }
+function isPersistedHistorySource(source) {
+    return source === "transcript" || source === "product-history";
+}
 function transcriptUserMessageMatchesIdentity(node, canonicalId, stableKey, eventId) {
     if (node.id === canonicalId || node.stableKey === stableKey)
         return true;
@@ -652,6 +656,9 @@ function isOptimisticUserMessageEvent(event) {
 function isInternalSessionOperation(action) {
     return action === "session.fork" || action === "session.clone" || action === "session.switch";
 }
+function isProductHistoryEchoEvent(event) {
+    return event.type === "assistant_delta" || event.type === "assistant_message";
+}
 function isTranscriptEchoEvent(event) {
     return (event.type === "message_queued" ||
         event.type === "message_started" ||
@@ -666,8 +673,19 @@ function shouldKeepTranscriptEchoEvent(event, openTranscriptEventIds) {
     const eventId = "eventId" in event && typeof event.eventId === "string" ? event.eventId : undefined;
     return Boolean(eventId && openTranscriptEventIds.has(eventId));
 }
-function isStaleToolCallEchoEvent(event, sessionStatus) {
-    return sessionStatus !== "running" && event.type === "tool_call";
+function historyCoversEvent(event, coverage) {
+    const eventId = "eventId" in event && typeof event.eventId === "string" ? event.eventId : undefined;
+    return eventId === undefined || coverage.eventIds.has(eventId);
+}
+function isNativeHistoryToolEchoEvent(event, coverage, sessionStatus) {
+    if (coverage.mode !== "native" || sessionStatus === "running")
+        return false;
+    if (event.type !== "tool_call" &&
+        event.type !== "tool_execution_started" &&
+        event.type !== "tool_execution_updated" &&
+        event.type !== "tool_execution_finished")
+        return false;
+    return coverage.toolCallIds.has(event.toolCallId) || historyCoversEvent(event, coverage);
 }
 export function mergeMessageTurnTimings(...groups) {
     const byEventId = new Map();