@mastra/react 0.2.34-alpha.0 → 0.2.34-alpha.2

package/dist/index.js

@@ -226,30 +226,24 @@ var mapWorkflowStreamChunkToWatchResult = (prev, chunk) => {
 var toUIMessage = ({ chunk, conversation, metadata }) => {
   const result = [...conversation];
   if (chunk.type.startsWith("data-")) {
+    const dataPart = {
+      type: chunk.type,
+      data: "data" in chunk ? chunk.data : void 0,
+      ..."id" in chunk && typeof chunk.id === "string" ? { id: chunk.id } : {}
+    };
     const lastMessage = result[result.length - 1];
     if (!lastMessage || lastMessage.role !== "assistant") {
       const newMessage = {
         id: `data-${chunk.runId}-${Date.now()}`,
         role: "assistant",
-        parts: [
-          {
-            type: chunk.type,
-            data: "data" in chunk ? chunk.data : void 0
-          }
-        ],
+        parts: [dataPart],
         metadata
       };
       return [...result, newMessage];
     }
     const updatedMessage = {
       ...lastMessage,
-      parts: [
-        ...lastMessage.parts,
-        {
-          type: chunk.type,
-          data: "data" in chunk ? chunk.data : void 0
-        }
-      ]
+      parts: [...lastMessage.parts, dataPart]
     };
     return [...result.slice(0, -1), updatedMessage];
   }