@pasko70/pibo 3.0.1 → 3.1.0

package/dist/apps/chat/stream.js

@@ -58,7 +58,7 @@ export function chatStreamFramesFromOutputEvent(event, state, options = {}) {
             }
             break;
         case "tool_call":
-            ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, event.args, event.intent, eventId);
+            ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, event.args, event.intent, eventId, event.toolInvocationOrdinal);
             frames.push({
                 type: "TOOL_CALL_ARGS",
                 toolCallId: event.toolCallId,
@@ -71,7 +71,7 @@ export function chatStreamFramesFromOutputEvent(event, state, options = {}) {
             });
             break;
         case "tool_execution_started": {
-            const started = ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, event.args, event.intent, eventId);
+            const started = ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, event.args, event.intent, eventId, event.toolInvocationOrdinal);
             if (!started && event.intent) {
                 frames.push({
                     type: "TOOL_CALL_START",
@@ -85,7 +85,7 @@ export function chatStreamFramesFromOutputEvent(event, state, options = {}) {
             break;
         }
         case "tool_execution_updated":
-            ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, event.args, event.intent, eventId);
+            ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, event.args, event.intent, eventId, event.toolInvocationOrdinal);
             frames.push({
                 type: "TOOL_CALL_ARGS",
                 toolCallId: event.toolCallId,
@@ -99,7 +99,7 @@ export function chatStreamFramesFromOutputEvent(event, state, options = {}) {
             });
             break;
         case "tool_execution_finished":
-            ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, undefined, event.intent, eventId);
+            ensureToolCallStarted(frames, state, event.toolCallId, event.toolName, undefined, event.intent, eventId, event.toolInvocationOrdinal);
             frames.push({ type: "TOOL_CALL_RESULT", toolCallId: event.toolCallId, toolName: event.toolName, result: event.result, isError: event.isError, ...(event.intent ? { intent: event.intent } : {}), runId: eventId });
             break;
         case "subagent_session":
@@ -135,6 +135,14 @@ export function chatStreamFramesFromOutputEvent(event, state, options = {}) {
     }
     if (options.includeRawEvent ?? true)
         frames.push({ type: "RAW_EVENT", event });
+    if (event.renderSequence !== undefined) {
+        for (const frame of frames)
+            frame.renderSequence = event.renderSequence;
+    }
+    if (event.toolInvocationOrdinal !== undefined) {
+        for (const frame of frames)
+            frame.toolInvocationOrdinal = event.toolInvocationOrdinal;
+    }
     return frames;
 }
 function reasoningIdFromOutputEvent(event) {
@@ -162,10 +170,11 @@ function ensureReasoningStarted(frames, state, messageId, runId) {
     state.reasoningMessageIds.add(messageId);
     frames.push({ type: "REASONING_MESSAGE_START", messageId, runId });
 }
-function ensureToolCallStarted(frames, state, toolCallId, toolName, args, intent, runId) {
-    if (state.toolCallIds.has(toolCallId))
+function ensureToolCallStarted(frames, state, toolCallId, toolName, args, intent, runId, toolInvocationOrdinal) {
+    const invocationKey = JSON.stringify([runId ?? "", toolCallId, toolInvocationOrdinal ?? 0]);
+    if (state.toolCallIds.has(invocationKey))
         return false;
-    state.toolCallIds.add(toolCallId);
+    state.toolCallIds.add(invocationKey);
     frames.push({ type: "TOOL_CALL_START", toolCallId, toolName, args, ...(intent ? { intent } : {}), runId });
     return true;
 }

package/dist/apps/chat/trace-v2.js

@@ -255,6 +255,7 @@ function compactTraceNode(node, payloadStore, piboSessionId, depth) {
         payloadRefs: Object.keys(payloadRefs).length ? payloadRefs : undefined,
         linkedPiboSessionId: node.linkedPiboSessionId,
         toolCallId: node.toolCallId,
+        toolInvocationOrdinal: node.toolInvocationOrdinal,
         runId: node.runId,
         intent: node.intent,
         eventId: node.eventId,