@mastra/react 0.0.0-dynamic-model-router-20251010230835 → 0.0.0-error-handler-fix-20251020202607

package/dist/index.js

@@ -1,7 +1,6 @@
 import { jsx, jsxs } from 'react/jsx-runtime';
 import { createContext, useContext, useState, Fragment, useLayoutEffect, useRef, useEffect } from 'react';
 import { MastraClient } from '@mastra/client-js';
-import { flushSync } from 'react-dom';
 import { ChevronDownIcon, CheckIcon, CopyIcon } from 'lucide-react';
 import { twMerge } from 'tailwind-merge';
 import { toJsxRuntime } from 'hast-util-to-jsx-runtime';
@@ -550,6 +549,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 +562,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;
@@ -1040,9 +1100,7 @@ const useChat = ({ agentId, initializeMessages }) => {
     }
     await response.processDataStream({
       onChunk: async (chunk) => {
-        flushSync(() => {
-          setMessages((prev) => toUIMessage({ chunk, conversation: prev, metadata: { mode: "stream" } }));
-        });
+        setMessages((prev) => toUIMessage({ chunk, conversation: prev, metadata: { mode: "stream" } }));
         onChunk?.(chunk);
       }
     });
@@ -1082,21 +1140,29 @@ const useChat = ({ agentId, initializeMessages }) => {
     const transformer = new AISdkNetworkTransformer();
     await response.processDataStream({
       onChunk: async (chunk) => {
-        flushSync(() => {
-          setMessages((prev) => transformer.transform({ chunk, conversation: prev, metadata: { mode: "network" } }));
-        });
+        setMessages((prev) => transformer.transform({ chunk, conversation: prev, metadata: { mode: "network" } }));
         onNetworkChunk?.(chunk);
       }
     });
     setIsRunning(false);
   };
+  const sendMessage = async ({ mode = "stream", ...args }) => {
+    const nextMessage = { role: "user", content: [{ type: "text", text: args.message }] };
+    const messages2 = args.coreUserMessages ? [nextMessage, ...args.coreUserMessages] : [nextMessage];
+    setMessages((s) => [...s, { role: "user", parts: [{ type: "text", text: args.message }] }]);
+    if (mode === "generate") {
+      await generate({ ...args, coreUserMessages: messages2 });
+    } else if (mode === "stream") {
+      await stream({ ...args, coreUserMessages: messages2 });
+    } else if (mode === "network") {
+      await network({ ...args, coreUserMessages: messages2 });
+    }
+  };
   return {
-    network,
-    stream,
-    generate,
+    setMessages,
+    sendMessage,
     isRunning,
     messages,
-    setMessages,
     cancelRun: () => setIsRunning(false)
   };
 };