@mastra/react 1.7.0 → 1.7.1-alpha.1

package/dist/index.js

@@ -297,7 +297,7 @@ const mergeBgTaskMetadata = (existing, mode, args, otherMetadata) => {
 * `mapWorkflowStreamChunkToWatchResult` from the previous accumulator.
 */
 const mapWorkflowStreamChunkToWatchResult = (prev, chunk) => {
-	const previous = prev ?? {
+	const previous = prev && typeof prev === "object" && prev.steps && typeof prev.steps === "object" ? prev : {
 		status: "running",
 		input: void 0,
 		steps: {}
@@ -1124,6 +1124,29 @@ const accumulateChunk = ({ chunk, conversation, metadata }) => {
 			} });
 			return replaceAt(result, messageIndex, withMetadata(targetMessage, nextMetadata));
 		}
+		case "tool-call-resumed": {
+			const resumedToolCallId = chunk.payload.toolCallId;
+			for (let i = result.length - 1; i >= 0; i--) {
+				const message = result[i];
+				if (!message || message.role !== "assistant") continue;
+				const meta = message.content.metadata;
+				const metadataKeys = chunk.payload.kind === "approval" ? ["pendingToolApprovals", "requireApprovalMetadata"] : ["suspendedTools"];
+				let nextMeta;
+				for (const metadataKey of metadataKeys) {
+					const entries = meta?.[metadataKey];
+					if (!entries) continue;
+					const key = Object.keys(entries).find((k) => (entries[k]?.toolCallId ?? k) === resumedToolCallId);
+					if (!key) continue;
+					const { [key]: _removed, ...remaining } = entries;
+					nextMeta = {
+						...nextMeta ?? meta,
+						[metadataKey]: remaining
+					};
+				}
+				if (nextMeta) return replaceAt(result, i, withMetadata(message, nextMeta));
+			}
+			return result;
+		}
 		case "finish":
 		case "abort": return finishStreamingAssistantMessage(result);
 		case "error": {
@@ -2217,6 +2240,14 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 			}
 			setIsRunning(false);
 		}
+		if (chunk.type === "tool-call-resumed") {
+			const toolCallId = chunk.payload?.toolCallId;
+			if (typeof toolCallId === "string") {
+				liveApprovalIds.current.add(toolCallId);
+				pendingToolApprovalIdsRef.current.delete(toolCallId);
+				setIsAwaitingToolApproval(pendingToolApprovalIdsRef.current.size > 0);
+			}
+		}
 		if (isTerminal) {
 			if (chunk.runId === liveRunId.current) liveRunFinished.current = true;
 			for (const toolCallId of pendingToolApprovalIdsRef.current) liveApprovalIds.current.add(toolCallId);