ai 7.0.131 → 7.0.132

package/dist/index.js

@@ -1174,7 +1174,7 @@ function cloneValue(value) {
 }
 //#endregion
 //#region src/version.ts
-const VERSION = "7.0.131";
+const VERSION = "7.0.132";
 //#endregion
 //#region src/util/download/download.ts
 /**
@@ -7422,9 +7422,9 @@ function toUIMessageChunk(part, { tools, sendReasoning = true, sendSources = fal
 * Converts a stream of `TextStreamPart<TOOLS>` chunks (as emitted by
 * `streamText`'s `stream`) into a stream of `UIMessageChunk`s suitable for
 * UI message streaming, including response message ID injection and
-* `onEnd` handling.
+* step and stream end callbacks.
 */
-function toUIMessageStream({ stream, tools, sendReasoning = true, sendSources = false, sendStart = true, sendFinish = true, onError = () => "An error occurred.", messageMetadata, originalMessages, generateMessageId, onEnd, onFinish }) {
+function toUIMessageStream({ stream, tools, sendReasoning = true, sendSources = false, sendStart = true, sendFinish = true, onError = () => "An error occurred.", messageMetadata, originalMessages, generateMessageId, onStepEnd, onStepFinish, onEnd, onFinish }) {
 	let outcome = { status: "unknown" };
 	let hasFatalFailure = false;
 	const setSourceOutcome = (newOutcome) => {
@@ -7489,11 +7489,12 @@ function toUIMessageStream({ stream, tools, sendReasoning = true, sendSources =
 					messageMetadata: messageMetadataValue,
 					responseMessageId
 				});
-				if (uiMessageChunk != null) controller.enqueue(uiMessageChunk);
+				if (uiMessageChunk != null && part.type !== "finish-step") controller.enqueue(uiMessageChunk);
 				if (messageMetadataValue != null && part.type !== "start" && part.type !== "finish") controller.enqueue({
 					type: "message-metadata",
 					messageMetadata: messageMetadataValue
 				});
+				if (uiMessageChunk != null && part.type === "finish-step") controller.enqueue(uiMessageChunk);
 				if (part.type === "finish") setSourceOutcome({ status: "completed" });
 				else if (part.type === "abort") setSourceOutcome({ status: "aborted" });
 				else if (part.type === "error") setSourceOutcome({
@@ -7507,6 +7508,7 @@ function toUIMessageStream({ stream, tools, sendReasoning = true, sendSources =
 		} })),
 		messageId: responseMessageId ?? generateMessageId?.(),
 		originalMessages,
+		onStepEnd: onStepEnd ?? onStepFinish,
 		onEnd: onEnd ?? onFinish,
 		onError,
 		getOutcome: () => outcome
@@ -10242,12 +10244,13 @@ var DefaultStreamTextResult = class {
 	get output() {
 		return this.getOutputPromise();
 	}
-	toUIMessageStream({ originalMessages, generateMessageId, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendStart, sendFinish, onError } = {}) {
+	toUIMessageStream({ originalMessages, generateMessageId, onStepEnd, onStepFinish, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendStart, sendFinish, onError } = {}) {
 		return createAsyncIterableStream(toUIMessageStream({
 			stream: this.stream,
 			tools: this.tools,
 			originalMessages,
 			generateMessageId,
+			onStepEnd: onStepEnd ?? onStepFinish,
 			onEnd: onEnd ?? onFinish,
 			messageMetadata,
 			sendReasoning,
@@ -10257,12 +10260,13 @@ var DefaultStreamTextResult = class {
 			onError
 		}));
 	}
-	pipeUIMessageStreamToResponse(response, { originalMessages, generateMessageId, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendFinish, sendStart, onError, ...init } = {}) {
+	pipeUIMessageStreamToResponse(response, { originalMessages, generateMessageId, onStepEnd, onStepFinish, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendFinish, sendStart, onError, ...init } = {}) {
 		return pipeUIMessageStreamToResponse({
 			response,
 			stream: this.toUIMessageStream({
 				originalMessages,
 				generateMessageId,
+				onStepEnd: onStepEnd ?? onStepFinish,
 				onEnd: onEnd ?? onFinish,
 				messageMetadata,
 				sendReasoning,
@@ -10281,11 +10285,12 @@ var DefaultStreamTextResult = class {
 			...init
 		});
 	}
-	toUIMessageStreamResponse({ originalMessages, generateMessageId, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendFinish, sendStart, onError, ...init } = {}) {
+	toUIMessageStreamResponse({ originalMessages, generateMessageId, onStepEnd, onStepFinish, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendFinish, sendStart, onError, ...init } = {}) {
 		return createUIMessageStreamResponse({
 			stream: this.toUIMessageStream({
 				originalMessages,
 				generateMessageId,
+				onStepEnd: onStepEnd ?? onStepFinish,
 				onEnd: onEnd ?? onFinish,
 				messageMetadata,
 				sendReasoning,
@@ -11384,10 +11389,11 @@ async function validateUIMessagesForAgent(options) {
 * @param experimental_transform - The stream transformations. Optional.
 * @param onStepEnd - Callback that is called when each step ends. Optional.
 * @param onStepFinish - Deprecated alias for `onStepEnd`. Optional.
+* @param onUIMessageStepEnd - Callback with the accumulated UI message after each step. Optional.
 *
 * @returns The UI message stream.
 */
-async function createAgentUIStream({ agent, uiMessages, convertDataPart, options, abortSignal, timeout, experimental_sandbox: sandbox, experimental_transform, onStepEnd, onStepFinish, ...uiMessageStreamOptions }) {
+async function createAgentUIStream({ agent, uiMessages, convertDataPart, options, abortSignal, timeout, experimental_sandbox: sandbox, experimental_transform, onStepEnd, onStepFinish, onUIMessageStepEnd, ...uiMessageStreamOptions }) {
 	const validatedMessages = await validateUIMessagesForAgent({
 		messages: uiMessages,
 		tools: agent.tools
@@ -11409,6 +11415,7 @@ async function createAgentUIStream({ agent, uiMessages, convertDataPart, options
 	return createAsyncIterableStream(toUIMessageStream({
 		...uiMessageStreamOptions,
 		originalMessages,
+		onStepEnd: onUIMessageStepEnd,
 		stream: result.stream,
 		tools: agent.tools
 	}));
@@ -11428,6 +11435,7 @@ async function createAgentUIStream({ agent, uiMessages, convertDataPart, options
 * @param experimental_transform - Stream transformations. Optional.
 * @param onStepEnd - Callback that is called when each step ends. Optional.
 * @param onStepFinish - Deprecated alias for `onStepEnd`. Optional.
+* @param onUIMessageStepEnd - Callback with the accumulated UI message after each step. Optional.
 * @param headers - Additional headers for the response. Optional.
 * @param status - The status code for the response. Optional.
 * @param statusText - The status text for the response. Optional.
@@ -11462,6 +11470,7 @@ async function createAgentUIStreamResponse({ headers, status, statusText, keepAl
 * @param experimental_transform - Stream transformations. Optional.
 * @param onStepEnd - Callback that is called when each step ends. Optional.
 * @param onStepFinish - Deprecated alias for `onStepEnd`. Optional.
+* @param onUIMessageStepEnd - Callback with the accumulated UI message after each step. Optional.
 * @param headers - Additional headers for the response. Optional.
 * @param status - The status code for the response. Optional.
 * @param statusText - The status text for the response. Optional.