@mastra/react 0.2.36-alpha.0 → 0.2.36-alpha.10

package/dist/index.js

@@ -383,23 +383,21 @@ var toUIMessage = ({ chunk, conversation, metadata }) => {
         return [...result, newMessage];
       }
       const parts = [...lastMessage.parts];
-      let reasoningPartIndex = parts.findIndex((part) => part.type === "reasoning");
-      if (reasoningPartIndex === -1) {
+      const lastPartIndex = parts.length - 1;
+      const lastPart = parts[lastPartIndex];
+      if (lastPart?.type === "reasoning") {
+        parts[lastPartIndex] = {
+          ...lastPart,
+          text: lastPart.text + chunk.payload.text,
+          state: "streaming"
+        };
+      } else {
         parts.push({
           type: "reasoning",
           text: chunk.payload.text,
           state: "streaming",
           providerMetadata: chunk.payload.providerMetadata
         });
-      } else {
-        const reasoningPart = parts[reasoningPartIndex];
-        if (reasoningPart.type === "reasoning") {
-          parts[reasoningPartIndex] = {
-            ...reasoningPart,
-            text: reasoningPart.text + chunk.payload.text,
-            state: "streaming"
-          };
-        }
       }
       return [
         ...result.slice(0, -1),
@@ -689,17 +687,29 @@ var toUIMessage = ({ chunk, conversation, metadata }) => {
         }
       ];
     }
-    case "tool-call-suspended": {
-      const lastMessage = result[result.length - 1];
-      if (!lastMessage || lastMessage.role !== "assistant") return result;
-      const lastSuspendedTools = lastMessage.metadata?.mode === "stream" ? lastMessage.metadata?.suspendedTools : {};
-      return [
-        ...result.slice(0, -1),
-        {
-          ...lastMessage,
-          metadata: {
-            ...lastMessage.metadata,
-            mode: "stream",
+    case "tool-call-suspended":
+    case "background-task-suspended": {
+      const isBgTaskEvent = chunk.type === "background-task-suspended";
+      const location = isBgTaskEvent ? locateToolPart(result, chunk.payload.toolCallId, isBgTaskEvent) : { messageIndex: result.length - 1 };
+      if (!location) return result;
+      const { messageIndex } = location;
+      const targetMessage = result[messageIndex];
+      if (!targetMessage || targetMessage.role !== "assistant") return result;
+      const lastSuspendedTools = targetMessage.metadata?.mode === "stream" ? targetMessage.metadata?.suspendedTools : {};
+      const nextMessage = {
+        ...targetMessage,
+        metadata: mergeBgTaskMetadata(
+          targetMessage.metadata,
+          "stream",
+          {
+            resetRunningCount: isBgTaskEvent,
+            perTaskEntry: isBgTaskEvent ? {
+              toolCallId: chunk.payload.toolCallId,
+              suspendedAt: chunk.payload.suspendedAt,
+              taskId: chunk.payload.taskId
+            } : void 0
+          },
+          {
             suspendedTools: {
               ...lastSuspendedTools,
               [chunk.payload.toolName]: {
@@ -711,8 +721,9 @@ var toUIMessage = ({ chunk, conversation, metadata }) => {
               }
             }
           }
-        }
-      ];
+        )
+      };
+      return [...result.slice(0, messageIndex), nextMessage, ...result.slice(messageIndex + 1)];
     }
     case "finish": {
       const lastMessage = result[result.length - 1];
@@ -904,22 +915,24 @@ var locateToolPart = (messages, toolCallId, allowMetadataOnlyMatch) => {
   }
   return { messageIndex, toolPartIndex };
 };
-var mergeBgTaskMetadata = (existing, mode, args) => {
+var mergeBgTaskMetadata = (existing, mode, args, otherMetadata) => {
   const existingAny = existing ?? {};
   const existingBgTasks = existingAny.backgroundTasks ?? {};
   const nextBgTasks = { ...existingBgTasks };
   if (args.perTaskEntry) {
-    const { toolCallId, startedAt, completedAt, taskId } = args.perTaskEntry;
+    const { toolCallId, startedAt, completedAt, taskId, suspendedAt } = args.perTaskEntry;
     const prev = existingBgTasks[toolCallId] ?? { taskId };
     nextBgTasks[toolCallId] = {
       ...prev,
       taskId,
       ...startedAt !== void 0 ? { startedAt } : {},
-      ...completedAt !== void 0 ? { completedAt } : {}
+      ...completedAt !== void 0 ? { completedAt } : {},
+      ...suspendedAt !== void 0 ? { suspendedAt } : {}
     };
   }
   return {
     ...existingAny,
+    ...otherMetadata ?? {},
     mode,
     ...args.resetRunningCount ? { runningBackgroundTasksCount: void 0 } : {},
     backgroundTasks: nextBgTasks