@mastra/react 1.4.13-alpha.0 → 1.4.13-alpha.3

package/dist/agent/hooks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../../src/agent/hooks.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAY,eAAe,EAA4B,MAAM,iCAAiC,CAAC;AAE3G,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAEnE,OAAO,KAAK,EAAE,SAAS,EAAiB,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAkBtF,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AA0I/D,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,eAAe,EAAE,CAAC;IACpC,6FAA6F;IAC7F,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;;;OAIG;IACH,WAAW,CAAC,EAAE,gBAAgB,CAAC;IAC/B,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3D,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IAC1C,0BAA0B,CAAC,EAAE,MAAM,IAAI,CAAC;IACxC;;;OAGG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,UAAU,UAAU;IAClB,gBAAgB,EAAE,eAAe,EAAE,CAAC;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,MAAM,eAAe,GAAG;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,gBAAgB,CAAC,EAAE,eAAe,EAAE,CAAA;CAAE,GAAG,CACtF,CAAC;IAAE,IAAI,EAAE,UAAU,CAAA;CAAE,GAAG,IAAI,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC,GAC/D,CAAC;IAAE,IAAI,EAAE,QAAQ,CAAA;CAAE,GAAG,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,GAC3D,CAAC;IAAE,IAAI,EAAE,SAAS,CAAA;CAAE,GAAG,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC,GAC7D,CAAC;IAAE,IAAI,CAAC,EAAE,SAAS,CAAA;CAAE,GAAG,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,CAChE,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG,UAAU,GAAG;IACtC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,WAAW,CAAC,EAAE,gBAAgB,CAAC;CAChC,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG;IACpC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9C,WAAW,CAAC,EAAE,gBAAgB,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG,UAAU,GAAG;IACrC,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,gBAAgB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7D,CAAC;AAoDF,eAAO,MAAM,OAAO,wMAWjB,eAAe;;qCAm4ByC,eAAe;;;;;;;;;;kCA/O7B,MAAM,eAAe,OAAO;kCAqE5B,MAAM;0CAkDE,MAAM;0CA2BN,MAAM;;;;oBA/wBvB,UAAU,GAAG,UAAU;;;uCA0yBT,MAAM,UAAU,MAAM;uCAiCtB,MAAM,UAAU,MAAM;;;oBAx0BtC,UAAU,GAAG,UAAU;;;CA66BxD,CAAC"}
+{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../../src/agent/hooks.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAY,eAAe,EAA4B,MAAM,iCAAiC,CAAC;AAE3G,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAEnE,OAAO,KAAK,EAAE,SAAS,EAAiB,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAkBtF,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AA0J/D,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,eAAe,EAAE,CAAC;IACpC,6FAA6F;IAC7F,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC;;;;OAIG;IACH,WAAW,CAAC,EAAE,gBAAgB,CAAC;IAC/B,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IAC3D,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;IAC1C,0BAA0B,CAAC,EAAE,MAAM,IAAI,CAAC;IACxC;;;OAGG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AAED,UAAU,UAAU;IAClB,gBAAgB,EAAE,eAAe,EAAE,CAAC;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,MAAM,eAAe,GAAG;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,gBAAgB,CAAC,EAAE,eAAe,EAAE,CAAA;CAAE,GAAG,CACtF,CAAC;IAAE,IAAI,EAAE,UAAU,CAAA;CAAE,GAAG,IAAI,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC,GAC/D,CAAC;IAAE,IAAI,EAAE,QAAQ,CAAA;CAAE,GAAG,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,GAC3D,CAAC;IAAE,IAAI,EAAE,SAAS,CAAA;CAAE,GAAG,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC,GAC7D,CAAC;IAAE,IAAI,CAAC,EAAE,SAAS,CAAA;CAAE,GAAG,IAAI,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,CAChE,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG,UAAU,GAAG;IACtC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,eAAe,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1D,WAAW,CAAC,EAAE,gBAAgB,CAAC;CAChC,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG;IACpC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9C,WAAW,CAAC,EAAE,gBAAgB,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG,UAAU,GAAG;IACrC,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,gBAAgB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7D,CAAC;AAoDF,eAAO,MAAM,OAAO,wMAWjB,eAAe;;qCAu8ByC,eAAe;;;;;;;;;;kCAnP7B,MAAM,eAAe,OAAO;kCAuE5B,MAAM;0CAoDE,MAAM;0CA2BN,MAAM;;;;oBAl1BvB,UAAU,GAAG,UAAU;;;uCA62BT,MAAM,UAAU,MAAM;uCAiCtB,MAAM,UAAU,MAAM;;;oBA34BtC,UAAU,GAAG,UAAU;;;CAg/BxD,CAAC"}

package/dist/index.cjs

@@ -1796,7 +1796,7 @@ const extractRunIdFromMessages = (messages) => {
 };
 //#endregion
 //#region src/agent/hooks.ts
-const extractPendingToolApprovalIdsFromMessages = (messages) => {
+const extractPendingToolApprovalIdsFromMessages = (messages, runId) => {
 	const pendingToolApprovalIds = /* @__PURE__ */ new Set();
 	for (const message of messages) {
 		const metadata = message.content?.metadata;
@@ -1809,6 +1809,7 @@ const extractPendingToolApprovalIdsFromMessages = (messages) => {
 		for (const source of metadataSources) {
 			if (!source || typeof source !== "object") continue;
 			for (const suspensionData of Object.values(source)) {
+				if (runId && suspensionData?.runId !== runId) continue;
 				const toolCallId = suspensionData?.toolCallId;
 				if (typeof toolCallId === "string" && toolCallId.length > 0) pendingToolApprovalIds.add(toolCallId);
 			}
@@ -1822,6 +1823,11 @@ const toolCallHasOutput = (parts, toolCallId) => parts.some((part) => {
 	if (invocation.toolCallId !== toolCallId) return false;
 	return invocation.state === "result" || invocation.result != null;
 });
+const filterUnresolvedApprovals = (entries, parts) => {
+	if (!entries || typeof entries !== "object") return void 0;
+	const pending = Object.fromEntries(Object.entries(entries).filter(([, approval]) => approval && typeof approval === "object" && typeof approval.toolCallId === "string" && !toolCallHasOutput(parts, approval.toolCallId)));
+	return Object.keys(pending).length ? pending : void 0;
+};
 /**
 * Normalize persisted initial messages back into the stream-friendly shape the
 * UI renders from. Mirrors `main`'s `resolveInitialMessages`:
@@ -1851,22 +1857,25 @@ const resolveInitialMessages = (messages) => messages.filter((message) => {
 		};
 	})() : message;
 	const normalizedMetadata = normalizedMessage.content?.metadata;
-	const pendingToolApprovals = normalizedMetadata?.pendingToolApprovals;
-	if (!pendingToolApprovals || typeof pendingToolApprovals !== "object") return normalizedMessage;
-	const stillPending = Object.fromEntries(Object.entries(pendingToolApprovals).filter(([, approval]) => approval && typeof approval === "object" && typeof approval.toolCallId === "string" && !toolCallHasOutput(normalizedMessage.content.parts, approval.toolCallId)));
-	const { pendingToolApprovals: _omit, ...restMetadata } = normalizedMetadata;
-	const hasStillPending = Object.keys(stillPending).length > 0;
+	if (!normalizedMetadata?.pendingToolApprovals && !normalizedMetadata?.requireApprovalMetadata && !normalizedMetadata?.suspendedTools) return normalizedMessage;
+	const { pendingToolApprovals, requireApprovalMetadata, suspendedTools, ...restMetadata } = normalizedMetadata;
+	const parts = normalizedMessage.content.parts;
+	const pending = filterUnresolvedApprovals(pendingToolApprovals, parts);
+	const required = {
+		...filterUnresolvedApprovals(requireApprovalMetadata, parts),
+		...pending
+	};
+	const suspended = filterUnresolvedApprovals(suspendedTools, parts);
 	return {
 		...normalizedMessage,
 		content: {
 			...normalizedMessage.content,
 			metadata: {
 				...restMetadata,
-				mode: "stream",
-				...hasStillPending ? {
-					pendingToolApprovals: stillPending,
-					requireApprovalMetadata: stillPending
-				} : {}
+				...pendingToolApprovals ? { mode: "stream" } : {},
+				...pending ? { pendingToolApprovals: pending } : {},
+				...Object.keys(required).length ? { requireApprovalMetadata: required } : {},
+				...suspended ? { suspendedTools: suspended } : {}
 			}
 		}
 	};
@@ -1915,20 +1924,64 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 	const _threadSignalsUnsupportedRef = (0, react.useRef)(false);
 	const [messages, setMessages] = (0, react.useState)([]);
 	const [tasks, setTasks] = (0, react.useState)([]);
+	const liveTasks = (0, react.useRef)(void 0);
 	const [toolCallApprovals, setToolCallApprovals] = (0, react.useState)({});
 	const [networkToolCallApprovals, setNetworkToolCallApprovals] = (0, react.useState)({});
 	const pendingToolApprovalIdsRef = (0, react.useRef)(/* @__PURE__ */ new Set());
+	const liveApprovalIds = (0, react.useRef)(/* @__PURE__ */ new Set());
+	const liveRunId = (0, react.useRef)(void 0);
+	const liveRunFinished = (0, react.useRef)(false);
 	const [isAwaitingToolApproval, setIsAwaitingToolApproval] = (0, react.useState)(false);
 	const baseClient = useMastraClient();
 	const [isRunning, setIsRunning] = (0, react.useState)(false);
+	const lastHydration = (0, react.useRef)(void 0);
 	(0, react.useEffect)(() => {
+		const previous = lastHydration.current;
+		const sameThread = previous?.agentId === agentId && previous.resourceId === resourceId && previous.threadId === threadId;
+		if (sameThread && previous.initialMessages === initialMessages) return;
 		const formattedMessages = resolveInitialMessages(initialMessages ?? []);
-		setMessages(formattedMessages);
-		setTasks(extractLatestTasksFromMessages(formattedMessages));
-		pendingToolApprovalIdsRef.current = extractPendingToolApprovalIdsFromMessages(formattedMessages);
-		setIsAwaitingToolApproval(pendingToolApprovalIdsRef.current.size > 0);
-		_currentRunId.current = extractRunIdFromMessages(formattedMessages);
-	}, [initialMessages]);
+		lastHydration.current = {
+			agentId,
+			resourceId,
+			threadId,
+			initialMessages,
+			formattedMessages
+		};
+		if (sameThread) {
+			const previousById = new Map(previous.formattedMessages.map((message) => [message.id, message]));
+			setMessages((current) => {
+				const live = current.filter((message) => previousById.get(message.id) !== message);
+				const liveById = new Map(live.map((message) => [message.id, message]));
+				const historyIds = new Set(formattedMessages.map((message) => message.id));
+				return [...formattedMessages.map((message) => liveById.get(message.id) ?? message), ...live.filter((message) => !historyIds.has(message.id))];
+			});
+			setTasks(liveTasks.current ?? extractLatestTasksFromMessages(formattedMessages));
+			const historyRunId = extractRunIdFromMessages(formattedMessages);
+			if (liveRunFinished.current) return;
+			if (!liveRunId.current && isRunning && historyRunId !== _currentRunId.current) return;
+		} else {
+			liveTasks.current = void 0;
+			liveApprovalIds.current.clear();
+			liveRunId.current = void 0;
+			liveRunFinished.current = false;
+			if (previous) setIsRunning(false);
+			setMessages(formattedMessages);
+			setTasks(extractLatestTasksFromMessages(formattedMessages));
+		}
+		const pendingApprovals = extractPendingToolApprovalIdsFromMessages(formattedMessages, liveRunId.current);
+		for (const toolCallId of liveApprovalIds.current) if (pendingToolApprovalIdsRef.current.has(toolCallId)) pendingApprovals.add(toolCallId);
+		else pendingApprovals.delete(toolCallId);
+		pendingToolApprovalIdsRef.current = pendingApprovals;
+		setIsAwaitingToolApproval(pendingApprovals.size > 0);
+		_currentRunId.current = liveRunId.current ?? extractRunIdFromMessages(formattedMessages);
+	}, [
+		agentId,
+		resourceId,
+		threadId,
+		initialMessages,
+		isRunning,
+		isAwaitingToolApproval
+	]);
 	(0, react.useEffect)(() => {
 		_activeContinuation.current = {
 			..._activeContinuation.current,
@@ -1991,29 +2044,40 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 		_threadSubscriptionPromiseRef.current = null;
 	}, []);
 	const processStreamChunk = (0, react.useCallback)(async (chunk, onChunk) => {
+		const isTerminal = chunk.type === "finish" || chunk.type === "abort" || chunk.type === "error";
+		if (isTerminal && liveRunId.current && chunk.runId !== liveRunId.current) return;
 		setMessages((prev) => accumulateChunk({
 			chunk,
 			conversation: prev,
 			metadata: { mode: "stream" }
 		}));
-		const signalTasks = extractTasksFromSignalChunk(chunk);
-		if (signalTasks !== void 0) setTasks(signalTasks);
-		const toolTasks = extractTasksFromToolResultChunk(chunk);
-		if (toolTasks !== void 0) setTasks(toolTasks);
+		const streamedTasks = extractTasksFromToolResultChunk(chunk) ?? extractTasksFromSignalChunk(chunk);
+		if (streamedTasks !== void 0) {
+			liveTasks.current = streamedTasks;
+			setTasks(streamedTasks);
+		}
 		if (chunk.type === "data-user-message" && isDataChunk(chunk) && (chunk.data?.type === "user-message" || chunk.data?.type === "user") && typeof chunk.data?.id === "string") onSignalEcho?.(chunk.data.id);
 		if (chunk.type === "start") {
 			setIsRunning(true);
-			if ("runId" in chunk && typeof chunk.runId === "string") _currentRunId.current = chunk.runId;
+			if ("runId" in chunk && typeof chunk.runId === "string") {
+				if (liveRunId.current !== chunk.runId) liveApprovalIds.current.clear();
+				liveRunFinished.current = false;
+				liveRunId.current = chunk.runId;
+				_currentRunId.current = chunk.runId;
+			}
 		}
 		if (chunk.type === "tool-call-approval" || chunk.type === "tool-call-suspended") {
 			const toolCallId = chunk.payload?.toolCallId;
 			if (typeof toolCallId === "string") {
+				liveApprovalIds.current.add(toolCallId);
 				pendingToolApprovalIdsRef.current.add(toolCallId);
 				setIsAwaitingToolApproval(true);
 			}
 			setIsRunning(false);
 		}
-		if (chunk.type === "finish" || chunk.type === "abort" || chunk.type === "error") {
+		if (isTerminal) {
+			if (chunk.runId === liveRunId.current) liveRunFinished.current = true;
+			for (const toolCallId of pendingToolApprovalIdsRef.current) liveApprovalIds.current.add(toolCallId);
 			pendingToolApprovalIdsRef.current.clear();
 			setIsAwaitingToolApproval(false);
 			setIsRunning(false);
@@ -2294,30 +2358,28 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 			onSignalSent?.(echoedSignalId, getSignalPreview(coreUserMessages));
 			if (pendingToolApprovalIdsRef.current.size > 0) setIsRunning(false);
 		} catch (error) {
-			if (isThreadSignalUnsupportedError(error)) {
+			if (isThreadSignalUnsupportedError(error)) try {
+				await agent.sendSignal({
+					signal: {
+						id: resolvedSignalId,
+						type: "user-message",
+						contents: messageContents
+					},
+					resourceId: resourceId || agentId,
+					threadId,
+					ifIdle: { streamOptions }
+				});
 				onSignalSent?.(resolvedSignalId, getSignalPreview(coreUserMessages));
-				try {
-					await agent.sendSignal({
-						signal: {
-							id: resolvedSignalId,
-							type: "user-message",
-							contents: messageContents
-						},
-						resourceId: resourceId || agentId,
-						threadId,
-						ifIdle: { streamOptions }
-					});
+				return;
+			} catch (signalError) {
+				onSignalEcho?.(resolvedSignalId);
+				if (isThreadSignalUnsupportedError(signalError)) {
+					markThreadSignalsUnsupported();
+					setMessages((prev) => [...prev, fromCoreUserMessagesToMastraDBMessage(coreUserMessages)]);
+					await streamWithLegacyRoute();
 					return;
-				} catch (signalError) {
-					onSignalEcho?.(resolvedSignalId);
-					if (isThreadSignalUnsupportedError(signalError)) {
-						markThreadSignalsUnsupported();
-						setMessages((prev) => [...prev, fromCoreUserMessagesToMastraDBMessage(coreUserMessages)]);
-						await streamWithLegacyRoute();
-						return;
-					}
-					throw signalError;
 				}
+				throw signalError;
 			}
 			throw error;
 		}
@@ -2378,6 +2440,7 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 		});
 		closeThreadSubscription();
 		setMessages((prev) => finishStreamingAssistantMessage(prev));
+		liveRunFinished.current = true;
 		pendingToolApprovalIdsRef.current.clear();
 		setIsAwaitingToolApproval(false);
 		setIsRunning(false);
@@ -2409,6 +2472,8 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 					...resumeData !== void 0 ? { resumeData } : {},
 					requestContext: continuation.requestContext
 				});
+				liveRunId.current ??= currentRunId;
+				liveApprovalIds.current.add(toolCallId);
 				pendingToolApprovalIdsRef.current.delete(toolCallId);
 				setIsAwaitingToolApproval(pendingToolApprovalIdsRef.current.size > 0);
 				setIsRunning(false);
@@ -2467,6 +2532,8 @@ const useChat = ({ agentId, resourceId, threadId, initialMessages, requestContex
 					...continuation.model !== void 0 ? { streamOptions: { model: continuation.model } } : {},
 					requestContext: continuation.requestContext
 				});
+				liveRunId.current ??= currentRunId;
+				liveApprovalIds.current.add(toolCallId);
 				pendingToolApprovalIdsRef.current.delete(toolCallId);
 				setIsAwaitingToolApproval(pendingToolApprovalIdsRef.current.size > 0);
 				setIsRunning(false);