@mastra/react 0.1.0-beta.21 → 0.1.0-beta.22

package/dist/{index-Bk0KrCu7.js → index-DFYpOuhE.js}

@@ -163,38 +163,57 @@ const toUIMessage = ({ chunk, conversation, metadata }) => {
       };
       return [...result, newMessage];
     }
-    case "text-start":
-    case "text-delta": {
+    case "text-start": {
       const lastMessage = result[result.length - 1];
       if (!lastMessage || lastMessage.role !== "assistant") return result;
       const parts = [...lastMessage.parts];
-      let textPartIndex = parts.findIndex((part) => part.type === "text");
-      if (chunk.type === "text-start") {
-        if (textPartIndex === -1) {
-          parts.push({
-            type: "text",
-            text: "",
-            state: "streaming",
-            providerMetadata: chunk.payload.providerMetadata
-          });
+      const textId = chunk.payload.id || `text-${Date.now()}`;
+      const newTextPart = {
+        type: "text",
+        text: "",
+        state: "streaming",
+        textId,
+        providerMetadata: chunk.payload.providerMetadata
+      };
+      parts.push(newTextPart);
+      return [
+        ...result.slice(0, -1),
+        {
+          ...lastMessage,
+          parts
         }
+      ];
+    }
+    case "text-delta": {
+      const lastMessage = result[result.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return result;
+      const parts = [...lastMessage.parts];
+      const textId = chunk.payload.id;
+      let textPartIndex = textId ? parts.findLastIndex((part) => part.type === "text" && part.textId === textId) : -1;
+      if (textPartIndex === -1) {
+        textPartIndex = parts.findLastIndex(
+          (part) => part.type === "text" && part.state === "streaming"
+        );
+      }
+      if (textPartIndex === -1) {
+        const newTextPart = {
+          type: "text",
+          text: chunk.payload.text,
+          state: "streaming",
+          textId,
+          providerMetadata: chunk.payload.providerMetadata
+        };
+        parts.push(newTextPart);
       } else {
-        if (textPartIndex === -1) {
-          parts.push({
-            type: "text",
-            text: chunk.payload.text,
-            state: "streaming",
-            providerMetadata: chunk.payload.providerMetadata
-          });
-        } else {
-          const textPart = parts[textPartIndex];
-          if (textPart.type === "text") {
-            parts[textPartIndex] = {
-              ...textPart,
-              text: textPart.text + chunk.payload.text,
-              state: "streaming"
-            };
-          }
+        const textPart = parts[textPartIndex];
+        if (textPart.type === "text") {
+          const extendedTextPart = textPart;
+          const updatedTextPart = {
+            ...extendedTextPart,
+            text: extendedTextPart.text + chunk.payload.text,
+            state: "streaming"
+          };
+          parts[textPartIndex] = updatedTextPart;
         }
       }
       return [
@@ -792,7 +811,8 @@ const toAssistantUIMessage = (message) => {
 };
 
 const resolveInitialMessages = (messages) => {
-  return messages.map((message) => {
+  const messagesLength = messages.length;
+  return messages.map((message, index) => {
     const networkPart = message.parts.find(
       (part) => typeof part === "object" && part !== null && "type" in part && part.type === "text" && "text" in part && typeof part.text === "string" && part.text.includes('"isNetwork":true')
     );
@@ -840,7 +860,7 @@ const resolveInitialMessages = (messages) => {
               content: finalResult.text
             });
           }
-          const result = {
+          const result = primitiveType === "tool" ? finalResult?.result : {
             childMessages,
             result: finalResult?.text || ""
           };
@@ -862,6 +882,7 @@ const resolveInitialMessages = (messages) => {
               mode: "network",
               selectionReason,
               agentInput: json.input,
+              hasMoreMessages: index < messagesLength - 1,
               from: primitiveType === "agent" ? "AGENT" : primitiveType === "tool" ? "TOOL" : "WORKFLOW"
             }
           };
@@ -3108,8 +3129,8 @@ var require_get_vercel_oidc_token = __commonJS$2({
       }
       try {
         const [{ getTokenPayload, isExpired }, { refreshToken }] = await Promise.all([
-          await import('./token-util-NEHG7TUY-JRJTGTAB-CrKHYQe8.js'),
-          await import('./token-6GSAFR2W-SPYPLMBM-Dw0-CxcM.js')
+          await import('./token-util-NEHG7TUY-JRJTGTAB-N8sYk0L_.js'),
+          await import('./token-6GSAFR2W-SPYPLMBM-BeIpRVyM.js')
         ]);
         if (!token || isExpired(getTokenPayload(token))) {
           await refreshToken();
@@ -4941,20 +4962,6 @@ createIdGenerator$3({
   prefix: "aitxt",
   size: 24
 });
-(class extends TransformStream {
-  constructor() {
-    super({
-      transform(part, controller) {
-        controller.enqueue(`data: ${JSON.stringify(part)}
-
-`);
-      },
-      flush(controller) {
-        controller.enqueue("data: [DONE]\n\n");
-      }
-    });
-  }
-});
 function fixJson$2(input) {
   const stack = ["ROOT"];
   let lastValidIndex = -1;
@@ -10069,15 +10076,16 @@ var AIV4Adapter = class {
    */
   static toUIMessage(m) {
     const experimentalAttachments = m.content.experimental_attachments ? [...m.content.experimental_attachments] : [];
-    const contentString = typeof m.content.content === `string` && m.content.content !== "" ? m.content.content : m.content.parts.reduce((prev, part) => {
+    const contentString = typeof m.content.content === `string` && m.content.content !== "" ? m.content.content : (m.content.parts ?? []).reduce((prev, part) => {
       if (part.type === `text`) {
         return part.text;
       }
       return prev;
     }, "");
     const parts = [];
-    if (m.content.parts.length) {
-      for (const part of m.content.parts) {
+    const sourceParts = m.content.parts ?? [];
+    if (sourceParts.length) {
+      for (const part of sourceParts) {
         if (part.type === `file`) {
           let normalizedUrl;
           if (typeof part.data === "string") {
@@ -10100,7 +10108,7 @@ var AIV4Adapter = class {
           const toolInvocation = { ...part.toolInvocation };
           let currentStep = -1;
           let toolStep = -1;
-          for (const innerPart of m.content.parts) {
+          for (const innerPart of sourceParts) {
             if (innerPart.type === `step-start`) currentStep++;
             if (innerPart.type === `tool-invocation` && innerPart.toolInvocation.toolCallId === part.toolInvocation.toolCallId) {
               toolStep = currentStep;
@@ -13483,9 +13491,15 @@ var MessageList$1 = class MessageList {
     this.lastCreatedAt = nowTime;
     return now;
   }
-  newMessageId() {
+  newMessageId(role) {
     if (this.generateMessageId) {
-      return this.generateMessageId();
+      return this.generateMessageId({
+        idType: "message",
+        source: "agent",
+        threadId: this.memoryInfo?.threadId,
+        resourceId: this.memoryInfo?.resourceId,
+        role
+      });
     }
     return v4();
   }
@@ -19200,9 +19214,9 @@ z$1.object({
   steps: z$1.custom().optional()
 });
 var languageModelUsageSchema = z10.object({
-  inputTokens: z10.number(),
-  outputTokens: z10.number(),
-  totalTokens: z10.number(),
+  inputTokens: z10.number().optional(),
+  outputTokens: z10.number().optional(),
+  totalTokens: z10.number().optional(),
   reasoningTokens: z10.number().optional(),
   cachedInputTokens: z10.number().optional()
 });
@@ -19326,7 +19340,7 @@ var coreToolSchema = z$1.object({
     // Zod schema or other schema types - validated at tool execution
   ]),
   outputSchema: z$1.union([z$1.record(z$1.string(), z$1.any()), z$1.any()]).optional(),
-  execute: z$1.function(z$1.tuple([z$1.any(), z$1.any()]), z$1.promise(z$1.any())).optional(),
+  execute: z$1.optional(z$1.function(z$1.tuple([z$1.any(), z$1.any()]), z$1.promise(z$1.any()))),
   type: z$1.union([z$1.literal("function"), z$1.literal("provider-defined"), z$1.undefined()]).optional(),
   args: z$1.record(z$1.string(), z$1.any()).optional()
 });
@@ -19505,6 +19519,54 @@ class AISdkNetworkTransformer {
       };
       return [...newConversation, newMessage];
     }
+    if (chunk.type === "agent-execution-approval") {
+      const lastMessage = newConversation[newConversation.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+      const lastRequireApprovalMetadata = lastMessage.metadata?.mode === "network" ? lastMessage.metadata?.requireApprovalMetadata : {};
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          metadata: {
+            ...lastMessage.metadata,
+            mode: "network",
+            requireApprovalMetadata: {
+              ...lastRequireApprovalMetadata,
+              [chunk.payload.toolName]: {
+                toolCallId: chunk.payload.toolCallId,
+                toolName: chunk.payload.toolName,
+                args: chunk.payload.args,
+                runId: chunk.payload.runId
+              }
+            }
+          }
+        }
+      ];
+    }
+    if (chunk.type === "agent-execution-suspended") {
+      const lastMessage = newConversation[newConversation.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+      const lastSuspendedTools = lastMessage.metadata?.mode === "network" ? lastMessage.metadata?.suspendedTools : {};
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          metadata: {
+            ...lastMessage.metadata,
+            mode: "network",
+            suspendedTools: {
+              ...lastSuspendedTools,
+              [chunk.payload.toolName]: {
+                toolCallId: chunk.payload.toolCallId,
+                toolName: chunk.payload.toolName,
+                args: chunk.payload.args,
+                suspendPayload: chunk.payload.suspendPayload
+              }
+            }
+          }
+        }
+      ];
+    }
     if (chunk.type === "agent-execution-end") {
       const lastMessage = newConversation[newConversation.length - 1];
       if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
@@ -19660,6 +19722,30 @@ class AISdkNetworkTransformer {
       };
       return [...newConversation, newMessage];
     }
+    if (chunk.type === "workflow-execution-suspended") {
+      const lastMessage = newConversation[newConversation.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+      const lastSuspendedTools = lastMessage.metadata?.mode === "network" ? lastMessage.metadata?.suspendedTools : {};
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          metadata: {
+            ...lastMessage.metadata,
+            mode: "network",
+            suspendedTools: {
+              ...lastSuspendedTools,
+              [chunk.payload.toolName]: {
+                toolCallId: chunk.payload.toolCallId,
+                toolName: chunk.payload.toolName,
+                args: chunk.payload.args,
+                suspendPayload: chunk.payload.suspendPayload
+              }
+            }
+          }
+        }
+      ];
+    }
     if (chunk.type.startsWith("workflow-execution-event-")) {
       const lastMessage = newConversation[newConversation.length - 1];
       if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
@@ -19727,6 +19813,54 @@ class AISdkNetworkTransformer {
         }
       ];
     }
+    if (chunk.type === "tool-execution-approval") {
+      const lastMessage = newConversation[newConversation.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+      const lastRequireApprovalMetadata = lastMessage.metadata?.mode === "network" ? lastMessage.metadata?.requireApprovalMetadata : {};
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          metadata: {
+            ...lastMessage.metadata,
+            mode: "network",
+            requireApprovalMetadata: {
+              ...lastRequireApprovalMetadata,
+              [chunk.payload.toolName]: {
+                toolCallId: chunk.payload.toolCallId,
+                toolName: chunk.payload.toolName,
+                args: chunk.payload.args,
+                runId: chunk.payload.runId
+              }
+            }
+          }
+        }
+      ];
+    }
+    if (chunk.type === "tool-execution-suspended") {
+      const lastMessage = newConversation[newConversation.length - 1];
+      if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
+      const lastSuspendedTools = lastMessage.metadata?.mode === "network" ? lastMessage.metadata?.suspendedTools : {};
+      return [
+        ...newConversation.slice(0, -1),
+        {
+          ...lastMessage,
+          metadata: {
+            ...lastMessage.metadata,
+            mode: "network",
+            suspendedTools: {
+              ...lastSuspendedTools,
+              [chunk.payload.toolName]: {
+                toolCallId: chunk.payload.toolCallId,
+                toolName: chunk.payload.toolName,
+                args: chunk.payload.args,
+                suspendPayload: chunk.payload.suspendPayload
+              }
+            }
+          }
+        }
+      ];
+    }
     if (chunk.type === "tool-execution-end") {
       const lastMessage = newConversation[newConversation.length - 1];
       if (!lastMessage || lastMessage.role !== "assistant") return newConversation;
@@ -19822,8 +19956,11 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
   const initialRunId = extractRunIdFromMessages(initialMessages);
   const _currentRunId = useRef(initialRunId);
   const _onChunk = useRef(void 0);
+  const _networkRunId = useRef(void 0);
+  const _onNetworkChunk = useRef(void 0);
   const [messages, setMessages] = useState(() => resolveInitialMessages(initialMessages));
   const [toolCallApprovals, setToolCallApprovals] = useState({});
+  const [networkToolCallApprovals, setNetworkToolCallApprovals] = useState({});
   const baseClient = useMastraClient();
   const [isRunning, setIsRunning] = useState(false);
   const generate = async ({
@@ -19868,7 +20005,7 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
       },
       instructions,
       requestContext,
-      ...threadId ? { threadId, resourceId: resourceId || agentId } : {},
+      ...threadId ? { memory: { thread: threadId, resource: resourceId || agentId } } : {},
       providerOptions,
       tracingOptions
     });
@@ -19928,7 +20065,7 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
       },
       instructions,
       requestContext,
-      ...threadId ? { threadId, resourceId: resourceId || agentId } : {},
+      ...threadId ? { memory: { thread: threadId, resource: resourceId || agentId } } : {},
       providerOptions,
       requireToolApproval,
       tracingOptions
@@ -19974,9 +20111,11 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
       },
       runId,
       requestContext,
-      ...threadId ? { thread: threadId, resourceId: resourceId || agentId } : {},
+      ...threadId ? { memory: { thread: threadId, resource: resourceId || agentId } } : {},
       tracingOptions
     });
+    _onNetworkChunk.current = onNetworkChunk;
+    _networkRunId.current = runId;
     const transformer = new AISdkNetworkTransformer();
     await response.processDataStream({
       onChunk: async (chunk) => {
@@ -19990,6 +20129,8 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
     setIsRunning(false);
     _currentRunId.current = void 0;
     _onChunk.current = void 0;
+    _networkRunId.current = void 0;
+    _onNetworkChunk.current = void 0;
   };
   const approveToolCall = async (toolCallId) => {
     const onChunk = _onChunk.current;
@@ -20025,6 +20166,52 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
     });
     setIsRunning(false);
   };
+  const approveNetworkToolCall = async (toolName, runId) => {
+    const onNetworkChunk = _onNetworkChunk.current;
+    const networkRunId = runId || _networkRunId.current;
+    if (!networkRunId)
+      return console.info(
+        "[approveNetworkToolCall] approveNetworkToolCall can only be called after a network stream has started"
+      );
+    setIsRunning(true);
+    setNetworkToolCallApprovals((prev) => ({
+      ...prev,
+      [runId ? `${runId}-${toolName}` : toolName]: { status: "approved" }
+    }));
+    const agent = baseClient.getAgent(agentId);
+    const response = await agent.approveNetworkToolCall({ runId: networkRunId });
+    const transformer = new AISdkNetworkTransformer();
+    await response.processDataStream({
+      onChunk: async (chunk) => {
+        setMessages((prev) => transformer.transform({ chunk, conversation: prev, metadata: { mode: "network" } }));
+        onNetworkChunk?.(chunk);
+      }
+    });
+    setIsRunning(false);
+  };
+  const declineNetworkToolCall = async (toolName, runId) => {
+    const onNetworkChunk = _onNetworkChunk.current;
+    const networkRunId = runId || _networkRunId.current;
+    if (!networkRunId)
+      return console.info(
+        "[declineNetworkToolCall] declineNetworkToolCall can only be called after a network stream has started"
+      );
+    setIsRunning(true);
+    setNetworkToolCallApprovals((prev) => ({
+      ...prev,
+      [runId ? `${runId}-${toolName}` : toolName]: { status: "declined" }
+    }));
+    const agent = baseClient.getAgent(agentId);
+    const response = await agent.declineNetworkToolCall({ runId: networkRunId });
+    const transformer = new AISdkNetworkTransformer();
+    await response.processDataStream({
+      onChunk: async (chunk) => {
+        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 coreUserMessages = [nextMessage];
@@ -20049,7 +20236,10 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
     approveToolCall,
     declineToolCall,
     cancelRun: handleCancelRun,
-    toolCallApprovals
+    toolCallApprovals,
+    approveNetworkToolCall,
+    declineNetworkToolCall,
+    networkToolCallApprovals
   };
 };
 
@@ -20369,4 +20559,4 @@ const MessageStreaming = ({ className, ...props }) => {
 };
 
 export { MessageUsages as $, EntryTitleClass as A, EntryTitle as B, CodeBlockClass as C, CodeBlock as D, Entity as E, CodeCopyButton as F, Icon as G, IconButtonClass as H, IconSizes as I, IconButton as J, AgentIcon as K, ToolsIcon as L, MastraReactProvider as M, Tooltip as N, TooltipContentClass as O, TooltipContent as P, TooltipTrigger as Q, MessageClass as R, Message as S, ToolApprovalClass as T, MessageContentClass as U, MessageContent as V, WorkflowIcon as W, MessageActionsClass as X, MessageActions as Y, MessageUsagesClass as Z, __commonJS$2 as _, __require2 as a, MessageUsageClass as a0, MessageUsage as a1, MessageUsageEntryClass as a2, MessageUsageEntry as a3, MessageUsageValueClass as a4, MessageUsageValue as a5, MessageListClass as a6, MessageList as a7, MessageStreamingClass as a8, MessageStreaming as a9, useChat as b, resolveToChildMessages as c, toAssistantUIMessage as d, useEntity as e, EntityTriggerClass as f, EntityTriggerVariantClasses as g, EntityTrigger as h, EntityContentClass as i, EntityContent as j, EntityCaret as k, ToolApproval as l, mapWorkflowStreamChunkToWatchResult as m, ToolApprovalTitleClass as n, ToolApprovalTitle as o, ToolApprovalHeaderClass as p, ToolApprovalHeader as q, require_token_error as r, ToolApprovalContentClass as s, toUIMessage as t, useMastraClient as u, ToolApprovalContent as v, ToolApprovalActionsClass as w, ToolApprovalActions as x, EntryClass as y, Entry as z };
-//# sourceMappingURL=index-Bk0KrCu7.js.map
+//# sourceMappingURL=index-DFYpOuhE.js.map