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

package/dist/index.js

@@ -689,17 +689,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 +723,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 +917,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