ai 5.0.0-beta.12 → 5.0.0-beta.13

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # ai
 
+## 5.0.0-beta.13
+
+### Patch Changes
+
+- 377bbcf: fix (ui): tool input can be undefined during input-streaming
+- ce1d1f3: feat (ai): export mock image, speech, and transcription models
+- c040e2f: fix (ui): inject generated response message id
+- c808e4d: fix (ui): do not send changing assistant message ids when onFinish is provided
+
 ## 5.0.0-beta.12
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1324,7 +1324,7 @@ type ToolUIPart<TOOLS extends UITools = UITools> = ValueOf<{
         toolCallId: string;
     } & ({
         state: 'input-streaming';
-        input: DeepPartial<TOOLS[NAME]['input']>;
+        input: DeepPartial<TOOLS[NAME]['input']> | undefined;
         providerExecuted?: boolean;
         output?: never;
         errorText?: never;
@@ -1462,6 +1462,13 @@ type UIMessageStreamOptions<UI_MESSAGE extends UIMessage> = {
      * and a message ID is provided for the response message.
      */
     originalMessages?: UI_MESSAGE[];
+    /**
+     * Generate a message ID for the response message.
+     *
+     * If not provided, no message ID will be set for the response message (unless
+     * the original messages are provided and the last message is an assistant message).
+     */
+    generateMessageId?: IdGenerator;
     onFinish?: (options: {
         /**
          * The updates list of UI messages.

package/dist/index.d.ts

@@ -1324,7 +1324,7 @@ type ToolUIPart<TOOLS extends UITools = UITools> = ValueOf<{
         toolCallId: string;
     } & ({
         state: 'input-streaming';
-        input: DeepPartial<TOOLS[NAME]['input']>;
+        input: DeepPartial<TOOLS[NAME]['input']> | undefined;
         providerExecuted?: boolean;
         output?: never;
         errorText?: never;
@@ -1462,6 +1462,13 @@ type UIMessageStreamOptions<UI_MESSAGE extends UIMessage> = {
      * and a message ID is provided for the response message.
      */
     originalMessages?: UI_MESSAGE[];
+    /**
+     * Generate a message ID for the response message.
+     *
+     * If not provided, no message ID will be set for the response message (unless
+     * the original messages are provided and the last message is an assistant message).
+     */
+    generateMessageId?: IdGenerator;
     onFinish?: (options: {
         /**
          * The updates list of UI messages.

package/dist/index.js

@@ -3281,13 +3281,31 @@ function handleUIMessageStreamFinish({
   onError,
   stream
 }) {
+  let lastMessage = originalMessages == null ? void 0 : originalMessages[originalMessages.length - 1];
+  if ((lastMessage == null ? void 0 : lastMessage.role) !== "assistant") {
+    lastMessage = void 0;
+  } else {
+    messageId = lastMessage.id;
+  }
+  const idInjectedStream = stream.pipeThrough(
+    new TransformStream({
+      transform(chunk, controller) {
+        if (chunk.type === "start") {
+          const startChunk = chunk;
+          if (startChunk.messageId == null && messageId != null) {
+            startChunk.messageId = messageId;
+          }
+        }
+        controller.enqueue(chunk);
+      }
+    })
+  );
   if (onFinish == null) {
-    return stream;
+    return idInjectedStream;
   }
-  const lastMessage = originalMessages == null ? void 0 : originalMessages[originalMessages.length - 1];
   const state = createStreamingUIMessageState({
     lastMessage: lastMessage ? structuredClone(lastMessage) : void 0,
-    messageId
+    messageId: messageId != null ? messageId : ""
     // will be overridden by the stream
   });
   const runUpdateMessageJob = async (job) => {
@@ -3295,19 +3313,7 @@ function handleUIMessageStreamFinish({
     } });
   };
   return processUIMessageStream({
-    stream: stream.pipeThrough(
-      new TransformStream({
-        transform(chunk, controller) {
-          if (chunk.type === "start") {
-            const startChunk = chunk;
-            if (startChunk.messageId == null) {
-              startChunk.messageId = messageId;
-            }
-          }
-          controller.enqueue(chunk);
-        }
-      })
-    ),
+    stream: idInjectedStream,
     runUpdateMessageJob,
     onError
   }).pipeThrough(
@@ -4671,6 +4677,7 @@ var DefaultStreamTextResult = class {
   }
   toUIMessageStream({
     originalMessages,
+    generateMessageId,
     onFinish,
     messageMetadata,
     sendReasoning = true,
@@ -4863,7 +4870,7 @@ var DefaultStreamTextResult = class {
     );
     return handleUIMessageStreamFinish({
       stream: baseStream,
-      messageId: responseMessageId != null ? responseMessageId : this.generateId(),
+      messageId: responseMessageId != null ? responseMessageId : generateMessageId == null ? void 0 : generateMessageId(),
       originalMessages,
       onFinish,
       onError
@@ -4871,6 +4878,7 @@ var DefaultStreamTextResult = class {
   }
   pipeUIMessageStreamToResponse(response, {
     originalMessages,
+    generateMessageId,
     onFinish,
     messageMetadata,
     sendReasoning,
@@ -4884,6 +4892,7 @@ var DefaultStreamTextResult = class {
       response,
       stream: this.toUIMessageStream({
         originalMessages,
+        generateMessageId,
         onFinish,
         messageMetadata,
         sendReasoning,
@@ -4904,6 +4913,7 @@ var DefaultStreamTextResult = class {
   }
   toUIMessageStreamResponse({
     originalMessages,
+    generateMessageId,
     onFinish,
     messageMetadata,
     sendReasoning,
@@ -4916,6 +4926,7 @@ var DefaultStreamTextResult = class {
     return createUIMessageStreamResponse({
       stream: this.toUIMessageStream({
         originalMessages,
+        generateMessageId,
         onFinish,
         messageMetadata,
         sendReasoning,