@mastra/react 1.0.0-beta.24 → 1.0.0-beta.25

package/dist/{index-CIU9zuI1.js → index-D3JtF_Zl.js}

@@ -3129,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-BLZ0BA54.js'),
-          await import('./token-6GSAFR2W-SPYPLMBM-Bmb7aObX.js')
+          await import('./token-util-NEHG7TUY-JRJTGTAB-hEay2AHp.js'),
+          await import('./token-6GSAFR2W-SPYPLMBM-ChURikIE.js')
         ]);
         if (!token || isExpired(getTokenPayload(token))) {
           await refreshToken();
@@ -19982,7 +19982,8 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
       topP,
       instructions,
       providerOptions,
-      maxSteps
+      maxSteps,
+      requireToolApproval
     } = modelSettings || {};
     setIsRunning(true);
     const clientWithAbort = new MastraClient({
@@ -19990,8 +19991,10 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
       abortSignal: signal
     });
     const agent = clientWithAbort.getAgent(agentId);
+    const runId = v4();
+    _currentRunId.current = runId;
     const response = await agent.generate(coreUserMessages, {
-      runId: v4(),
+      runId,
       maxSteps,
       modelSettings: {
         frequencyPenalty,
@@ -20006,8 +20009,30 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
       requestContext,
       ...threadId ? { memory: { thread: threadId, resource: resourceId || agentId } } : {},
       providerOptions,
-      tracingOptions
+      tracingOptions,
+      requireToolApproval
     });
+    if (response.finishReason === "suspended" && response.suspendPayload) {
+      const { toolCallId, toolName, args } = response.suspendPayload;
+      if (response.response?.uiMessages) {
+        const mastraUIMessages = (response.response.uiMessages || []).map((message) => ({
+          ...message,
+          metadata: {
+            mode: "generate",
+            requireApprovalMetadata: {
+              [toolName]: {
+                toolCallId,
+                toolName,
+                args
+              }
+            }
+          }
+        }));
+        setMessages((prev) => [...prev, ...mastraUIMessages]);
+      }
+      setIsRunning(false);
+      return;
+    }
     setIsRunning(false);
     if (response && "uiMessages" in response.response && response.response.uiMessages) {
       onFinish?.(response.response.uiMessages);
@@ -20163,6 +20188,48 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
     });
     setIsRunning(false);
   };
+  const approveToolCallGenerate = async (toolCallId) => {
+    const currentRunId = _currentRunId.current;
+    if (!currentRunId)
+      return console.info(
+        "[approveToolCallGenerate] approveToolCallGenerate can only be called after a generate has started"
+      );
+    setIsRunning(true);
+    setToolCallApprovals((prev) => ({ ...prev, [toolCallId]: { status: "approved" } }));
+    const agent = baseClient.getAgent(agentId);
+    const response = await agent.approveToolCallGenerate({ runId: currentRunId, toolCallId });
+    if (response && "uiMessages" in response.response && response.response.uiMessages) {
+      const mastraUIMessages = (response.response.uiMessages || []).map((message) => ({
+        ...message,
+        metadata: {
+          mode: "generate"
+        }
+      }));
+      setMessages((prev) => [...prev, ...mastraUIMessages]);
+    }
+    setIsRunning(false);
+  };
+  const declineToolCallGenerate = async (toolCallId) => {
+    const currentRunId = _currentRunId.current;
+    if (!currentRunId)
+      return console.info(
+        "[declineToolCallGenerate] declineToolCallGenerate can only be called after a generate has started"
+      );
+    setIsRunning(true);
+    setToolCallApprovals((prev) => ({ ...prev, [toolCallId]: { status: "declined" } }));
+    const agent = baseClient.getAgent(agentId);
+    const response = await agent.declineToolCallGenerate({ runId: currentRunId, toolCallId });
+    if (response && "uiMessages" in response.response && response.response.uiMessages) {
+      const mastraUIMessages = (response.response.uiMessages || []).map((message) => ({
+        ...message,
+        metadata: {
+          mode: "generate"
+        }
+      }));
+      setMessages((prev) => [...prev, ...mastraUIMessages]);
+    }
+    setIsRunning(false);
+  };
   const approveNetworkToolCall = async (toolName, runId) => {
     const onNetworkChunk = _onNetworkChunk.current;
     const networkRunId = runId || _networkRunId.current;
@@ -20232,6 +20299,8 @@ const useChat = ({ agentId, resourceId, initializeMessages }) => {
     messages,
     approveToolCall,
     declineToolCall,
+    approveToolCallGenerate,
+    declineToolCallGenerate,
     cancelRun: handleCancelRun,
     toolCallApprovals,
     approveNetworkToolCall,
@@ -20556,4 +20625,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-CIU9zuI1.js.map
+//# sourceMappingURL=index-D3JtF_Zl.js.map