@mastra/react 1.4.4-alpha.6 → 1.4.4-alpha.8

package/dist/index.js

@@ -2418,14 +2418,28 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 			}
 			return;
 		}
-		await (await agent.approveToolCall({
-			runId: currentRunId,
-			toolCallId,
-			...continuation
-		})).processDataStream({ onChunk: async (chunk) => {
-			await processStreamChunk(chunk, onChunk);
-		} });
-		setIsRunning(false);
+		try {
+			await (resumeData !== void 0 ? await agent.resumeStream(resumeData, {
+				runId: currentRunId,
+				toolCallId,
+				...continuation
+			}) : await agent.approveToolCall({
+				runId: currentRunId,
+				toolCallId,
+				...continuation
+			})).processDataStream({ onChunk: async (chunk) => {
+				await processStreamChunk(chunk, onChunk);
+			} });
+			setIsRunning(false);
+		} catch (error) {
+			setToolCallApprovals((prev) => {
+				const next = { ...prev };
+				delete next[toolCallId];
+				return next;
+			});
+			setIsRunning(false);
+			throw error;
+		}
 	};
 	const declineToolCall = async (toolCallId) => {
 		const onChunk = _onChunk.current;