@mastra/react 1.5.1-alpha.3 → 1.6.0-alpha.4

package/dist/index.cjs

@@ -1,5 +1,4 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_ui = require("./ui-C979LXC_.cjs");
 let _mastra_client_js = require("@mastra/client-js");
 let react = require("react");
 let react_jsx_runtime = require("react/jsx-runtime");
@@ -268,39 +267,47 @@ const mergeBgTaskMetadata = (existing, mode, args, otherMetadata) => {
 * `mapWorkflowStreamChunkToWatchResult` from the previous accumulator.
 */
 const mapWorkflowStreamChunkToWatchResult = (prev, chunk) => {
+	const previous = prev ?? {
+		status: "running",
+		input: void 0,
+		steps: {}
+	};
 	if (chunk.type === "workflow-start") return {
-		input: prev?.input,
+		input: previous.input,
 		status: "running",
-		steps: prev?.steps || {}
+		steps: previous.steps
 	};
 	if (chunk.type === "workflow-canceled") return {
-		...prev,
+		...previous,
 		status: "canceled"
 	};
+	if (chunk.type === "workflow-paused") return {
+		...previous,
+		status: "paused"
+	};
 	if (chunk.type === "workflow-finish") {
 		const finalStatus = chunk.payload.workflowStatus;
-		const prevSteps = prev?.steps ?? {};
-		const lastStep = Object.values(prevSteps).pop();
+		const lastStep = Object.values(previous.steps).pop();
 		return {
-			...prev,
+			...previous,
 			status: chunk.payload.workflowStatus,
 			...finalStatus === "success" && lastStep?.status === "success" ? { result: lastStep?.output } : finalStatus === "failed" && lastStep?.status === "failed" ? { error: lastStep?.error } : finalStatus === "tripwire" && chunk.payload.tripwire ? { tripwire: chunk.payload.tripwire } : {}
 		};
 	}
 	const { stepCallId: _stepCallId, stepName: _stepName, ...newPayload } = chunk.payload ?? {};
 	const newSteps = {
-		...prev?.steps,
+		...previous.steps,
 		[chunk.payload.id]: {
-			...prev?.steps?.[chunk.payload.id],
+			...previous.steps[chunk.payload.id],
 			...newPayload
 		}
 	};
 	if (chunk.type === "workflow-step-start") return {
-		...prev,
+		...previous,
 		steps: newSteps
 	};
 	if (chunk.type === "workflow-step-suspended") {
-		const suspendedStepIds = Object.entries(newSteps).flatMap(([stepId, stepResult]) => {
+		const suspended = Object.entries(newSteps).flatMap(([stepId, stepResult]) => {
 			if (stepResult?.status === "suspended") {
 				const nestedPath = stepResult?.suspendPayload?.__workflow_meta?.path;
 				return nestedPath ? [[stepId, ...nestedPath]] : [[stepId]];
@@ -308,24 +315,24 @@ const mapWorkflowStreamChunkToWatchResult = (prev, chunk) => {
 			return [];
 		});
 		return {
-			...prev,
+			...previous,
 			status: "suspended",
 			steps: newSteps,
 			suspendPayload: chunk.payload.suspendPayload,
-			suspended: suspendedStepIds
+			suspended
 		};
 	}
 	if (chunk.type === "workflow-step-waiting") return {
-		...prev,
+		...previous,
 		status: "waiting",
 		steps: newSteps
 	};
 	if (chunk.type === "workflow-step-progress") return {
-		...prev,
+		...previous,
 		steps: {
-			...prev?.steps,
+			...previous.steps,
 			[chunk.payload.id]: {
-				...prev?.steps?.[chunk.payload.id],
+				...previous.steps[chunk.payload.id],
 				foreachProgress: {
 					completedCount: chunk.payload.completedCount,
 					totalCount: chunk.payload.totalCount,
@@ -337,10 +344,10 @@ const mapWorkflowStreamChunkToWatchResult = (prev, chunk) => {
 		}
 	};
 	if (chunk.type === "workflow-step-result") return {
-		...prev,
+		...previous,
 		steps: newSteps
 	};
-	return prev;
+	return previous;
 };
 const signalContentsToUserMessages = (contents, metadata) => {
 	const makeUserMessage = (parts) => ({
@@ -917,7 +924,7 @@ const accumulateChunk = ({ chunk, conversation, metadata }) => {
 			const { toolName, toolCallId, args } = toolPart.toolInvocation;
 			const payloadOutput = chunk.type === "background-task-output" ? chunk.payload.payload.payload.output : chunk.payload.output;
 			if (payloadOutput?.type?.startsWith("workflow-")) {
-				const existingWorkflowState = toolPart.toolInvocation.result || {};
+				const existingWorkflowState = toolPart.toolInvocation.result;
 				const updated = mapWorkflowStreamChunkToWatchResult(existingWorkflowState, payloadOutput);
 				parts[toolPartIndex] = {
 					...toolPart,
@@ -1147,8 +1154,7 @@ const accumulateChunk = ({ chunk, conversation, metadata }) => {
 		case "network-validation-start":
 		case "network-validation-end":
 		case "network-object":
-		case "network-object-result":
-		case "tool-output-denied": return result;
+		case "network-object-result": return result;
 		default: return assertExhaustive(chunk, result);
 	}
 };
@@ -2817,275 +2823,133 @@ function useMutation(mutationFn) {
 }
 //#endregion
 //#region src/workflows/use-stream-workflow.ts
-/**
-* Hook for streaming workflow execution with support for observing, resuming, and time-travel.
-*
-* @example
-* ```tsx
-* const {
-*   streamWorkflow,
-*   streamResult,
-*   isStreaming,
-*   observeWorkflowStream,
-*   closeStreamsAndReset,
-*   resumeWorkflowStream,
-*   timeTravelWorkflowStream,
-* } = useStreamWorkflow({
-*   debugMode: true,
-*   tracingOptions: { enabled: true },
-*   onError: (error, defaultMessage) => console.error(defaultMessage, error),
-* });
-*
-* // Start streaming a workflow
-* await streamWorkflow.mutateAsync({
-*   workflowId: 'my-workflow',
-*   runId: 'run-123',
-*   inputData: { key: 'value' },
-*   requestContext: {},
-* });
-* ```
-*/
 function useStreamWorkflow({ debugMode, tracingOptions, onError }) {
 	const client = useMastraClient();
-	const [streamResult, setStreamResult] = (0, react.useState)({});
+	const [streamResult, setStreamResult] = (0, react.useState)();
 	const [isStreaming, setIsStreaming] = (0, react.useState)(false);
-	const readerRef = (0, react.useRef)(null);
-	const observerRef = (0, react.useRef)(null);
-	const resumeStreamRef = (0, react.useRef)(null);
-	const timeTravelStreamRef = (0, react.useRef)(null);
+	const activeStream = (0, react.useRef)(void 0);
 	const isMountedRef = (0, react.useRef)(true);
+	const closeActiveStream = (0, react.useCallback)(() => {
+		const operation = activeStream.current;
+		activeStream.current = void 0;
+		if (!operation) return;
+		operation.controller.abort();
+		operation.reader?.cancel().catch(() => void 0);
+		operation.reader?.releaseLock();
+		operation.reader = void 0;
+	}, []);
 	(0, react.useEffect)(() => {
 		isMountedRef.current = true;
 		return () => {
 			isMountedRef.current = false;
-			if (readerRef.current) {
-				try {
-					readerRef.current.releaseLock();
-				} catch {}
-				readerRef.current = null;
-			}
-			if (observerRef.current) {
-				try {
-					observerRef.current.releaseLock();
-				} catch {}
-				observerRef.current = null;
-			}
-			if (resumeStreamRef.current) {
-				try {
-					resumeStreamRef.current.releaseLock();
-				} catch {}
-				resumeStreamRef.current = null;
-			}
-			if (timeTravelStreamRef.current) {
-				try {
-					timeTravelStreamRef.current.releaseLock();
-				} catch {}
-				timeTravelStreamRef.current = null;
-			}
+			closeActiveStream();
 		};
-	}, []);
-	const handleStreamError = (0, react.useCallback)((err, defaultMessage, setStreamingState) => {
-		if (err instanceof TypeError) return;
-		const error = err instanceof Error ? err : new Error(defaultMessage);
-		onError?.(error, defaultMessage);
-		setStreamingState?.(false);
-	}, [onError]);
-	const handleWorkflowFinish = (0, react.useCallback)((value) => {
-		if (value.type === "workflow-finish") {
-			const streamStatus = value.payload?.workflowStatus;
-			const metadata = value.payload?.metadata;
-			setStreamResult((prev) => ({
-				...prev,
-				status: streamStatus
-			}));
-			if (streamStatus === "failed") throw new Error(metadata?.errorMessage || "Workflow execution failed");
-		}
-	}, []);
-	const streamWorkflow = useMutation(async ({ workflowId, runId, inputData, initialState, requestContext: playgroundRequestContext, perStep }) => {
-		if (readerRef.current) readerRef.current.releaseLock();
+	}, [closeActiveStream]);
+	async function consumeStream(openStream, defaultMessage) {
 		if (!isMountedRef.current) return;
+		closeActiveStream();
+		const operation = { controller: new AbortController() };
+		activeStream.current = operation;
+		const { signal } = operation.controller;
 		setIsStreaming(true);
-		setStreamResult({ input: inputData });
-		const stream = await (await client.getWorkflow(workflowId).createRun({ runId })).stream({
-			inputData,
-			initialState,
-			requestContext: playgroundRequestContext,
-			closeOnSuspend: true,
-			tracingOptions,
-			perStep: perStep ?? debugMode
-		});
-		if (!stream) return handleStreamError(/* @__PURE__ */ new Error("No stream returned"), "No stream returned", setIsStreaming);
-		const reader = stream.getReader();
-		readerRef.current = reader;
 		try {
-			while (true) {
-				if (!isMountedRef.current) break;
+			const stream = await openStream(signal);
+			if (signal.aborted) {
+				await stream?.cancel();
+				return;
+			}
+			if (!stream) {
+				onError?.(/* @__PURE__ */ new Error("No stream returned"), "No stream returned");
+				return;
+			}
+			const reader = stream.getReader();
+			operation.reader = reader;
+			while (!signal.aborted) {
 				const { done, value } = await reader.read();
-				if (done) break;
-				if (isMountedRef.current) {
-					setStreamResult((prev) => {
-						return mapWorkflowStreamChunkToWatchResult(prev, value);
-					});
-					if (value.type === "workflow-step-start") setIsStreaming(true);
-					if (value.type === "workflow-step-suspended") setIsStreaming(false);
-					if (value.type === "workflow-finish") handleWorkflowFinish(value);
-				}
+				if (done || signal.aborted) break;
+				setStreamResult((previous) => mapWorkflowStreamChunkToWatchResult(previous, value));
+				if (value.type === "workflow-step-start") setIsStreaming(true);
+				if (value.type === "workflow-step-suspended") setIsStreaming(false);
+				if (value.type === "workflow-finish" && value.payload.workflowStatus === "failed") throw new Error(value.payload.metadata?.errorMessage || "Workflow execution failed");
+			}
+		} catch (error) {
+			if (!signal.aborted) {
+				if (!operation.reader) throw error;
+				onError?.(error instanceof Error ? error : new Error(defaultMessage), defaultMessage);
 			}
-		} catch (err) {
-			handleStreamError(err, "Error streaming workflow");
 		} finally {
-			if (isMountedRef.current) setIsStreaming(false);
-			if (readerRef.current) {
-				readerRef.current.releaseLock();
-				readerRef.current = null;
+			operation.reader?.releaseLock();
+			operation.reader = void 0;
+			if (activeStream.current === operation) {
+				activeStream.current = void 0;
+				setIsStreaming(false);
 			}
 		}
+	}
+	const streamWorkflow = useMutation(({ workflowId, runId, inputData, initialState, requestContext, perStep }) => {
+		setStreamResult({
+			status: "pending",
+			input: inputData,
+			steps: {}
+		});
+		return consumeStream(async (signal) => {
+			const run = await client.getWorkflow(workflowId).createRun({ runId });
+			if (signal.aborted) return;
+			return run.stream({
+				inputData,
+				initialState,
+				requestContext,
+				closeOnSuspend: true,
+				tracingOptions,
+				perStep: perStep ?? debugMode
+			});
+		}, "Error streaming workflow");
 	});
 	const observeWorkflowStream = useMutation(async ({ workflowId, runId, storeRunResult }) => {
-		if (observerRef.current) observerRef.current.releaseLock();
 		if (!isMountedRef.current) return;
-		setIsStreaming(true);
-		setStreamResult(storeRunResult || {});
+		setStreamResult(storeRunResult ?? void 0);
 		if (storeRunResult?.status === "suspended") {
+			closeActiveStream();
 			setIsStreaming(false);
 			return;
 		}
-		const stream = await (await client.getWorkflow(workflowId).createRun({ runId })).observeStream();
-		if (!stream) return handleStreamError(/* @__PURE__ */ new Error("No stream returned"), "No stream returned", setIsStreaming);
-		const reader = stream.getReader();
-		observerRef.current = reader;
-		try {
-			while (true) {
-				if (!isMountedRef.current) break;
-				const { done, value } = await reader.read();
-				if (done) break;
-				if (isMountedRef.current) {
-					setStreamResult((prev) => {
-						return mapWorkflowStreamChunkToWatchResult(prev, value);
-					});
-					if (value.type === "workflow-step-start") setIsStreaming(true);
-					if (value.type === "workflow-step-suspended") setIsStreaming(false);
-					if (value.type === "workflow-finish") handleWorkflowFinish(value);
-				}
-			}
-		} catch (err) {
-			handleStreamError(err, "Error observing workflow");
-		} finally {
-			if (isMountedRef.current) setIsStreaming(false);
-			if (observerRef.current) {
-				observerRef.current.releaseLock();
-				observerRef.current = null;
-			}
-		}
+		await consumeStream(async (signal) => {
+			const run = await client.getWorkflow(workflowId).createRun({ runId });
+			if (signal.aborted) return;
+			return run.observeStream();
+		}, "Error observing workflow");
 	});
-	const resumeWorkflowStream = useMutation(async ({ workflowId, runId, step, resumeData, requestContext: playgroundRequestContext, perStep }) => {
-		if (resumeStreamRef.current) resumeStreamRef.current.releaseLock();
-		if (!isMountedRef.current) return;
-		setIsStreaming(true);
-		const stream = await (await client.getWorkflow(workflowId).createRun({ runId })).resumeStream({
+	const resumeWorkflowStream = useMutation(({ workflowId, runId, step, resumeData, requestContext, perStep }) => consumeStream(async (signal) => {
+		const run = await client.getWorkflow(workflowId).createRun({ runId });
+		if (signal.aborted) return;
+		return run.resumeStream({
 			step,
 			resumeData,
-			requestContext: playgroundRequestContext,
+			requestContext,
 			tracingOptions,
 			perStep: perStep ?? debugMode
 		});
-		if (!stream) return handleStreamError(/* @__PURE__ */ new Error("No stream returned"), "No stream returned", setIsStreaming);
-		const reader = stream.getReader();
-		resumeStreamRef.current = reader;
-		try {
-			while (true) {
-				if (!isMountedRef.current) break;
-				const { done, value } = await reader.read();
-				if (done) break;
-				if (isMountedRef.current) {
-					setStreamResult((prev) => {
-						return mapWorkflowStreamChunkToWatchResult(prev, value);
-					});
-					if (value.type === "workflow-step-start") setIsStreaming(true);
-					if (value.type === "workflow-step-suspended") setIsStreaming(false);
-					if (value.type === "workflow-finish") handleWorkflowFinish(value);
-				}
-			}
-		} catch (err) {
-			handleStreamError(err, "Error resuming workflow stream");
-		} finally {
-			if (isMountedRef.current) setIsStreaming(false);
-			if (resumeStreamRef.current) {
-				resumeStreamRef.current.releaseLock();
-				resumeStreamRef.current = null;
-			}
-		}
-	});
-	const timeTravelWorkflowStream = useMutation(async ({ workflowId, requestContext: playgroundRequestContext, runId, perStep, ...params }) => {
-		if (timeTravelStreamRef.current) timeTravelStreamRef.current.releaseLock();
-		if (!isMountedRef.current) return;
-		setIsStreaming(true);
-		const stream = await (await client.getWorkflow(workflowId).createRun({ runId })).timeTravelStream({
+	}, "Error resuming workflow stream"));
+	const timeTravelWorkflowStream = useMutation(({ workflowId, requestContext, runId, perStep, ...params }) => consumeStream(async (signal) => {
+		const run = await client.getWorkflow(workflowId).createRun({ runId });
+		if (signal.aborted) return;
+		return run.timeTravelStream({
 			...params,
 			perStep: perStep ?? debugMode,
-			requestContext: playgroundRequestContext,
+			requestContext,
 			tracingOptions
 		});
-		if (!stream) return handleStreamError(/* @__PURE__ */ new Error("No stream returned"), "No stream returned", setIsStreaming);
-		const reader = stream.getReader();
-		timeTravelStreamRef.current = reader;
-		try {
-			while (true) {
-				if (!isMountedRef.current) break;
-				const { done, value } = await reader.read();
-				if (done) break;
-				if (isMountedRef.current) {
-					setStreamResult((prev) => {
-						return mapWorkflowStreamChunkToWatchResult(prev, value);
-					});
-					if (value.type === "workflow-step-start") setIsStreaming(true);
-					if (value.type === "workflow-step-suspended") setIsStreaming(false);
-					if (value.type === "workflow-finish") handleWorkflowFinish(value);
-				}
-			}
-		} catch (err) {
-			handleStreamError(err, "Error time traveling workflow stream");
-		} finally {
-			if (isMountedRef.current) setIsStreaming(false);
-			if (timeTravelStreamRef.current) {
-				timeTravelStreamRef.current.releaseLock();
-				timeTravelStreamRef.current = null;
-			}
-		}
-	});
+	}, "Error time traveling workflow stream"));
 	return {
 		streamWorkflow,
 		streamResult,
 		isStreaming,
 		observeWorkflowStream,
 		closeStreamsAndReset: (0, react.useCallback)(() => {
+			closeActiveStream();
 			setIsStreaming(false);
-			setStreamResult({});
-			if (readerRef.current) {
-				try {
-					readerRef.current.releaseLock();
-				} catch {}
-				readerRef.current = null;
-			}
-			if (observerRef.current) {
-				try {
-					observerRef.current.releaseLock();
-				} catch {}
-				observerRef.current = null;
-			}
-			if (resumeStreamRef.current) {
-				try {
-					resumeStreamRef.current.releaseLock();
-				} catch {}
-				resumeStreamRef.current = null;
-			}
-			if (timeTravelStreamRef.current) {
-				try {
-					timeTravelStreamRef.current.releaseLock();
-				} catch {}
-				timeTravelStreamRef.current = null;
-			}
-		}, []),
+			setStreamResult(void 0);
+		}, [closeActiveStream]),
 		resumeWorkflowStream,
 		timeTravelWorkflowStream
 	};
@@ -3422,62 +3286,8 @@ const useMastraSpeechToText = ({ agent, language }) => {
 	};
 };
 //#endregion
-exports.AgentIcon = require_ui.AgentIcon;
 exports.CLIENT_MESSAGE_ID_KEY = CLIENT_MESSAGE_ID_KEY;
-exports.CodeBlock = require_ui.CodeBlock;
-exports.CodeBlockClass = require_ui.CodeBlockClass;
-exports.CodeCopyButton = require_ui.CodeCopyButton;
-exports.Entity = require_ui.Entity;
-exports.EntityCaret = require_ui.EntityCaret;
-exports.EntityContent = require_ui.EntityContent;
-exports.EntityContentClass = require_ui.EntityContentClass;
-exports.EntityTrigger = require_ui.EntityTrigger;
-exports.EntityTriggerClass = require_ui.EntityTriggerClass;
-exports.EntityTriggerVariantClasses = require_ui.EntityTriggerVariantClasses;
-exports.Entry = require_ui.Entry;
-exports.EntryClass = require_ui.EntryClass;
-exports.EntryTitle = require_ui.EntryTitle;
-exports.EntryTitleClass = require_ui.EntryTitleClass;
-exports.Icon = require_ui.Icon;
-exports.IconButton = require_ui.IconButton;
-exports.IconButtonClass = require_ui.IconButtonClass;
-exports.IconSizes = require_ui.IconSizes;
 exports.MastraReactProvider = MastraReactProvider;
-exports.Message = require_ui.Message;
-exports.MessageActions = require_ui.MessageActions;
-exports.MessageActionsClass = require_ui.MessageActionsClass;
-exports.MessageClass = require_ui.MessageClass;
-exports.MessageContent = require_ui.MessageContent;
-exports.MessageContentClass = require_ui.MessageContentClass;
-exports.MessageFactory = require_ui.MessageFactory;
-exports.MessageList = require_ui.MessageList;
-exports.MessageListClass = require_ui.MessageListClass;
-exports.MessageStreaming = require_ui.MessageStreaming;
-exports.MessageStreamingClass = require_ui.MessageStreamingClass;
-exports.MessageUsage = require_ui.MessageUsage;
-exports.MessageUsageClass = require_ui.MessageUsageClass;
-exports.MessageUsageEntry = require_ui.MessageUsageEntry;
-exports.MessageUsageEntryClass = require_ui.MessageUsageEntryClass;
-exports.MessageUsageValue = require_ui.MessageUsageValue;
-exports.MessageUsageValueClass = require_ui.MessageUsageValueClass;
-exports.MessageUsages = require_ui.MessageUsages;
-exports.MessageUsagesClass = require_ui.MessageUsagesClass;
-exports.ToolApproval = require_ui.ToolApproval;
-exports.ToolApprovalActions = require_ui.ToolApprovalActions;
-exports.ToolApprovalActionsClass = require_ui.ToolApprovalActionsClass;
-exports.ToolApprovalClass = require_ui.ToolApprovalClass;
-exports.ToolApprovalContent = require_ui.ToolApprovalContent;
-exports.ToolApprovalContentClass = require_ui.ToolApprovalContentClass;
-exports.ToolApprovalHeader = require_ui.ToolApprovalHeader;
-exports.ToolApprovalHeaderClass = require_ui.ToolApprovalHeaderClass;
-exports.ToolApprovalTitle = require_ui.ToolApprovalTitle;
-exports.ToolApprovalTitleClass = require_ui.ToolApprovalTitleClass;
-exports.ToolsIcon = require_ui.ToolsIcon;
-exports.Tooltip = require_ui.Tooltip;
-exports.TooltipContent = require_ui.TooltipContent;
-exports.TooltipContentClass = require_ui.TooltipContentClass;
-exports.TooltipTrigger = require_ui.TooltipTrigger;
-exports.WorkflowIcon = require_ui.WorkflowIcon;
 exports.WorkflowStepFactory = WorkflowStepFactory;
 exports.accumulateChunk = accumulateChunk;
 exports.accumulateNetworkChunk = accumulateNetworkChunk;
@@ -3490,7 +3300,6 @@ exports.recordMicrophoneToFile = recordMicrophoneToFile;
 exports.useCancelWorkflowRun = useCancelWorkflowRun;
 exports.useChat = useChat;
 exports.useCreateWorkflowRun = useCreateWorkflowRun;
-exports.useEntity = require_ui.useEntity;
 exports.useMastraClient = useMastraClient;
 exports.useSpeechRecognition = useSpeechRecognition;
 exports.useStreamWorkflow = useStreamWorkflow;