@mastra/react 0.2.36-alpha.3 → 0.2.36-alpha.4

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # @mastra/react
 
+## 0.2.36-alpha.4
+
+### Patch Changes
+
+- Handle `background-task-suspended` chunks in `toUIMessage` so suspend metadata is restored after resume. ([#16260](https://github.com/mastra-ai/mastra/pull/16260))
+
+- Updated dependencies [[`9f17410`](https://github.com/mastra-ai/mastra/commit/9f1741080def23d42ee50b39887a385ae316a3c6), [`c6eb39e`](https://github.com/mastra-ai/mastra/commit/c6eb39ea6dca381c6563cb240237fbe608e02f93), [`c6eb39e`](https://github.com/mastra-ai/mastra/commit/c6eb39ea6dca381c6563cb240237fbe608e02f93), [`900d086`](https://github.com/mastra-ai/mastra/commit/900d086bb737b9cf2fcf68f11b0389b801a2738c), [`4c0e286`](https://github.com/mastra-ai/mastra/commit/4c0e28637c9cfb4f416549b55e97ebfa13319dfc), [`25184ff`](https://github.com/mastra-ai/mastra/commit/25184ffaf1293ec95119426eb1a1f8d38831b96c), [`25184ff`](https://github.com/mastra-ai/mastra/commit/25184ffaf1293ec95119426eb1a1f8d38831b96c), [`aebde9c`](https://github.com/mastra-ai/mastra/commit/aebde9cfacf56592c6b6350cae721740fe090b8a)]:
+  - @mastra/core@1.33.0-alpha.4
+  - @mastra/client-js@1.18.0-alpha.4
+
 ## 0.2.36-alpha.3
 
 ### Patch Changes

package/dist/index.cjs

@@ -691,17 +691,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]: {
@@ -713,8 +725,9 @@ var toUIMessage = ({ chunk, conversation, metadata }) => {
               }
             }
           }
-        }
-      ];
+        )
+      };
+      return [...result.slice(0, messageIndex), nextMessage, ...result.slice(messageIndex + 1)];
     }
     case "finish": {
       const lastMessage = result[result.length - 1];
@@ -906,22 +919,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