bs-agent 0.0.20 → 0.0.21

package/dist/react/index.js

@@ -171,11 +171,16 @@ var createDebugHandlers = (setDebugData) => {
           if (existingItemIndex === -1) {
             currentData.push({ itemType: "reasoning", reasoning: delta, index });
           } else {
-            currentData[existingItemIndex] = {
-              itemType: "reasoning",
-              reasoning: currentData[existingItemIndex].reasoning + delta,
-              index
-            };
+            const hasInterleavedItems = currentData.slice(existingItemIndex + 1).some((item) => item.itemType !== "reasoning");
+            if (hasInterleavedItems) {
+              currentData.push({ itemType: "reasoning", reasoning: delta, index });
+            } else {
+              currentData[existingItemIndex] = {
+                itemType: "reasoning",
+                reasoning: currentData[existingItemIndex].reasoning + delta,
+                index
+              };
+            }
           }
           return {
             ...prev,
@@ -330,7 +335,8 @@ function buildStreamCallbacks(deps, debugKey) {
     messagesRef,
     toolContext,
     agentId,
-    textDeltaModifier
+    textDeltaModifier,
+    fullTextModifier
   } = deps;
   return {
     onComplete: () => {
@@ -349,7 +355,14 @@ function buildStreamCallbacks(deps, debugKey) {
     },
     onEvent: (event) => {
       if (event.type === "text_delta") {
-        handleTextDelta(event, setMessages, syncSessionRef, textDeltaModifier, agentId);
+        handleTextDelta(
+          event,
+          setMessages,
+          syncSessionRef,
+          textDeltaModifier,
+          fullTextModifier,
+          agentId
+        );
       } else if (event.type === "tool_call_start" && event.data.toolType === "client") {
         handleClientToolCall(event, setMessages, syncSessionRef, toolContext, agentId);
         if (debugKey) {
@@ -379,20 +392,24 @@ function buildStreamCallbacks(deps, debugKey) {
     }
   };
 }
-function handleTextDelta(event, setMessages, syncSessionRef, modifier, agentId) {
+function handleTextDelta(event, setMessages, syncSessionRef, modifier, fullTextModifier, agentId) {
   const sequence = event.meta.sequence;
   const originalText = event.data;
   const eventExecutionId = event.meta.executionId;
   setMessages((prev) => {
     const lastMessage = prev[prev.length - 1];
+    const meta = {
+      executionId: eventExecutionId,
+      sequence,
+      agentId: event.meta.agentId || agentId
+    };
     let text = originalText;
     if (modifier) {
-      const currentFullText = lastMessage?.role === "agent" ? lastMessage.content : "";
-      text = modifier(originalText, currentFullText, {
-        executionId: eventExecutionId,
-        sequence,
-        agentId: event.meta.agentId || agentId
-      });
+      const lastTextPart = lastMessage?.role === "agent" ? lastMessage.parts?.findLast(
+        (p) => p.type === "text"
+      ) : void 0;
+      const currentFullText = lastTextPart?._rawText ?? lastTextPart?.text ?? "";
+      text = modifier(originalText, currentFullText, meta);
     }
     const newPart = {
       type: "text",
@@ -402,17 +419,53 @@ function handleTextDelta(event, setMessages, syncSessionRef, modifier, agentId)
     };
     let updatedMessages;
     if (lastMessage?.role === "agent") {
+      const lastTextPart = lastMessage.parts?.findLast(
+        (p) => p.type === "text"
+      );
+      const prevRaw = lastTextPart?._rawText ?? lastTextPart?.text ?? "";
+      const rawText = prevRaw + text;
+      let updatedParts = updateAgentMessageParts(lastMessage.parts || [], newPart);
+      let displayContent = rawText;
+      if (fullTextModifier) {
+        displayContent = fullTextModifier(rawText, meta);
+        const textParts = updatedParts.filter(
+          (p) => p.type === "text"
+        );
+        const nonTextParts = updatedParts.filter((p) => p.type !== "text");
+        const modifiedTextPart = {
+          type: "text",
+          text: displayContent,
+          _rawText: rawText,
+          firstSequence: textParts[0]?.firstSequence ?? sequence,
+          lastSequence: textParts[textParts.length - 1]?.lastSequence ?? sequence
+        };
+        updatedParts = [...nonTextParts, modifiedTextPart];
+      }
       const updatedMessage = {
         ...lastMessage,
-        content: lastMessage.content + text,
-        parts: updateAgentMessageParts(lastMessage.parts || [], newPart)
+        content: displayContent,
+        parts: updatedParts
       };
       updatedMessages = [...prev.slice(0, -1), updatedMessage];
     } else {
+      let displayContent = text;
+      let parts = [newPart];
+      if (fullTextModifier) {
+        displayContent = fullTextModifier(text, meta);
+        parts = [
+          {
+            type: "text",
+            text: displayContent,
+            _rawText: text,
+            firstSequence: sequence,
+            lastSequence: sequence
+          }
+        ];
+      }
       const updatedMessage = {
         role: "agent",
-        content: text,
-        parts: [newPart],
+        content: displayContent,
+        parts,
         executionId: eventExecutionId
       };
       updatedMessages = [...prev, updatedMessage];
@@ -425,7 +478,7 @@ function handleTextDelta(event, setMessages, syncSessionRef, modifier, agentId)
 }
 function handleClientToolCall(event, setMessages, syncSessionRef, toolContext, agentId) {
   const tool = toolContext?.getTool(agentId, event.data.toolName);
-  if (!tool?.render) return;
+  if (!tool?.render && !event.data.paused) return;
   setMessages((prev) => {
     const lastMessage = prev[prev.length - 1];
     const newPart = {
@@ -524,7 +577,8 @@ function useAgent(agent, options) {
         syncSessionRef: sessionUtils.syncSessionRef,
         debugHandlers,
         toolContext,
-        textDeltaModifier: optionsRef.current?.textDeltaModifier
+        textDeltaModifier: optionsRef.current?.textDeltaModifier,
+        fullTextModifier: optionsRef.current?.fullTextModifier
       };
       const callbacks = buildStreamCallbacks(deps, debugKey);
       const executeOptions = {