@mastra/react 0.0.0-monorepo-binary-20251013210052 → 0.0.0-usechat-duplicate-20251016110554

package/dist/index.js

@@ -550,6 +550,9 @@ const toAssistantUIMessage = (message) => {
 class AISdkNetworkTransformer {
   transform({ chunk, conversation, metadata }) {
     const newConversation = [...conversation];
+    if (chunk.type === "routing-agent-text-delta") {
+      return this.handleRoutingAgentConversation(chunk, newConversation);
+    }
     if (chunk.type.startsWith("agent-execution-")) {
       return this.handleAgentConversation(chunk, newConversation, metadata);
     }
@@ -560,22 +563,80 @@ class AISdkNetworkTransformer {
       return this.handleToolConversation(chunk, newConversation, metadata);
     }
     if (chunk.type === "network-execution-event-step-finish") {
-      const newMessage = {
-        id: `network-execution-event-step-finish-${chunk.runId}-${Date.now()}`,
-        role: "assistant",
-        parts: [
+      const lastMessage = newConversation[newConversation.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+      const agentChunk = chunk.payload;
+      const parts = [...lastMessage.parts];
+      const textPartIndex = parts.findIndex((part) => part.type === "text");
+      if (textPartIndex === -1) {
+        parts.push({
+          type: "text",
+          text: agentChunk.result,
+          state: "done"
+        });
+        return [
+          ...newConversation.slice(0, -1),
           {
-            type: "text",
-            text: chunk.payload?.result || "",
-            state: "done"
+            ...lastMessage,
+            parts
           }
-        ],
-        metadata
-      };
-      return [...newConversation, newMessage];
+        ];
+      }
+      const textPart = parts[textPartIndex];
+      if (textPart.type === "text") {
+        parts[textPartIndex] = {
+          ...textPart,
+          state: "done"
+        };
+        return [
+          ...newConversation.slice(0, -1),
+          {
+            ...lastMessage,
+            parts
+          }
+        ];
+      }
+      return newConversation;
     }
     return newConversation;
   }
+  handleRoutingAgentConversation = (chunk, newConversation) => {
+    const lastMessage = newConversation[newConversation.length - 1];
+    if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+    const agentChunk = chunk.payload;
+    const parts = [...lastMessage.parts];
+    const textPartIndex = parts.findIndex((part) => part.type === "text");
+    if (textPartIndex === -1) {
+      parts.push({
+        type: "text",
+        text: agentChunk.text,
+        state: "streaming"
+      });
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          parts
+        }
+      ];
+    }
+    const textPart = parts[textPartIndex];
+    if (textPart.type === "text") {
+      parts[textPartIndex] = {
+        ...textPart,
+        text: textPart.text + agentChunk.text,
+        state: "streaming"
+      };
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          parts
+        }
+      ];
+    }
+    return newConversation;
+  };
   handleAgentConversation = (chunk, newConversation, metadata) => {
     if (chunk.type === "agent-execution-start") {
       const primitiveId = chunk.payload?.args?.primitiveId;