ai 5.0.0-alpha.10 → 5.0.0-alpha.12

package/dist/index.js

@@ -583,6 +583,15 @@ var uiMessageStreamPartSchema = import_zod.z.union([
     providerMetadata: import_zod.z.any().optional()
     // Use z.any() for generic metadata
   }),
+  import_zod.z.object({
+    type: import_zod.z.literal("source-document"),
+    sourceId: import_zod.z.string(),
+    mediaType: import_zod.z.string(),
+    title: import_zod.z.string(),
+    filename: import_zod.z.string().optional(),
+    providerMetadata: import_zod.z.any().optional()
+    // Use z.any() for generic metadata
+  }),
   import_zod.z.object({
     type: import_zod.z.literal("file"),
     url: import_zod.z.string(),
@@ -803,8 +812,137 @@ var SerialJobExecutor = class {
   }
 };
 
+// src/ui/convert-file-list-to-file-ui-parts.ts
+async function convertFileListToFileUIParts(files) {
+  if (files == null) {
+    return [];
+  }
+  if (!globalThis.FileList || !(files instanceof globalThis.FileList)) {
+    throw new Error("FileList is not supported in the current environment");
+  }
+  return Promise.all(
+    Array.from(files).map(async (file) => {
+      const { name: name17, type } = file;
+      const dataUrl = await new Promise((resolve, reject) => {
+        const reader = new FileReader();
+        reader.onload = (readerEvent) => {
+          var _a17;
+          resolve((_a17 = readerEvent.target) == null ? void 0 : _a17.result);
+        };
+        reader.onerror = (error) => reject(error);
+        reader.readAsDataURL(file);
+      });
+      return {
+        type: "file",
+        mediaType: type,
+        filename: name17,
+        url: dataUrl
+      };
+    })
+  );
+}
+
+// src/ui/default-chat-transport.ts
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var getOriginalFetch2 = () => fetch;
+async function fetchUIMessageStream({
+  api,
+  body,
+  credentials,
+  headers,
+  abortSignal,
+  fetch: fetch2 = getOriginalFetch2(),
+  requestType = "generate"
+}) {
+  var _a17;
+  const response = requestType === "resume" ? await fetch2(`${api}?id=${body.id}`, {
+    method: "GET",
+    headers: {
+      "Content-Type": "application/json",
+      ...headers
+    },
+    signal: abortSignal,
+    credentials
+  }) : await fetch2(api, {
+    method: "POST",
+    body: JSON.stringify(body),
+    headers: {
+      "Content-Type": "application/json",
+      ...headers
+    },
+    signal: abortSignal,
+    credentials
+  });
+  if (!response.ok) {
+    throw new Error(
+      (_a17 = await response.text()) != null ? _a17 : "Failed to fetch the chat response."
+    );
+  }
+  if (!response.body) {
+    throw new Error("The response body is empty.");
+  }
+  return (0, import_provider_utils2.parseJsonEventStream)({
+    stream: response.body,
+    schema: uiMessageStreamPartSchema
+  }).pipeThrough(
+    new TransformStream({
+      async transform(part, controller) {
+        if (!part.success) {
+          throw part.error;
+        }
+        controller.enqueue(part.value);
+      }
+    })
+  );
+}
+var DefaultChatTransport = class {
+  constructor({
+    api = "/api/chat",
+    credentials,
+    headers,
+    body,
+    fetch: fetch2,
+    prepareRequest
+  } = {}) {
+    this.api = api;
+    this.credentials = credentials;
+    this.headers = headers;
+    this.body = body;
+    this.fetch = fetch2;
+    this.prepareRequest = prepareRequest;
+  }
+  submitMessages({
+    chatId,
+    messages,
+    abortSignal,
+    metadata,
+    headers,
+    body,
+    requestType
+  }) {
+    var _a17, _b;
+    const preparedRequest = (_a17 = this.prepareRequest) == null ? void 0 : _a17.call(this, {
+      id: chatId,
+      messages,
+      body: { ...this.body, ...body },
+      headers: { ...this.headers, ...headers },
+      credentials: this.credentials,
+      requestMetadata: metadata
+    });
+    return fetchUIMessageStream({
+      api: this.api,
+      body: (preparedRequest == null ? void 0 : preparedRequest.body) !== void 0 ? preparedRequest.body : { ...this.body, ...body, id: chatId, messages },
+      headers: (preparedRequest == null ? void 0 : preparedRequest.headers) !== void 0 ? preparedRequest.headers : { ...this.headers, ...headers },
+      credentials: (_b = preparedRequest == null ? void 0 : preparedRequest.credentials) != null ? _b : this.credentials,
+      abortSignal,
+      fetch: this.fetch,
+      requestType
+    });
+  }
+};
+
 // src/ui/process-ui-message-stream.ts
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
 // src/util/merge-objects.ts
 function mergeObjects(base, overrides) {
@@ -840,7 +978,7 @@ function mergeObjects(base, overrides) {
 }
 
 // src/util/parse-partial-json.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // src/util/fix-json.ts
 function fixJson(input) {
@@ -1165,11 +1303,11 @@ async function parsePartialJson(jsonText) {
   if (jsonText === void 0) {
     return { value: void 0, state: "undefined-input" };
   }
-  let result = await (0, import_provider_utils2.safeParseJSON)({ text: jsonText });
+  let result = await (0, import_provider_utils3.safeParseJSON)({ text: jsonText });
   if (result.success) {
     return { value: result.value, state: "successful-parse" };
   }
-  result = await (0, import_provider_utils2.safeParseJSON)({ text: fixJson(jsonText) });
+  result = await (0, import_provider_utils3.safeParseJSON)({ text: fixJson(jsonText) });
   if (result.success) {
     return { value: result.value, state: "repaired-parse" };
   }
@@ -1230,7 +1368,7 @@ function processUIMessageStream({
             if (metadata != null) {
               const mergedMetadata = state.message.metadata != null ? mergeObjects(state.message.metadata, metadata) : metadata;
               if (messageMetadataSchema != null) {
-                await (0, import_provider_utils3.validateTypes)({
+                await (0, import_provider_utils4.validateTypes)({
                   value: mergedMetadata,
                   schema: messageMetadataSchema
                 });
@@ -1293,6 +1431,18 @@ function processUIMessageStream({
               write();
               break;
             }
+            case "source-document": {
+              state.message.parts.push({
+                type: "source-document",
+                sourceId: part.sourceId,
+                mediaType: part.mediaType,
+                title: part.title,
+                filename: part.filename,
+                providerMetadata: part.providerMetadata
+              });
+              write();
+              break;
+            }
             case "tool-call-streaming-start": {
               const toolInvocations = getToolInvocations(state.message);
               state.partialToolCalls[part.toolCallId] = {
@@ -1473,135 +1623,6 @@ function isAssistantMessageWithCompletedToolCalls(message) {
   return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every((part) => "result" in part.toolInvocation);
 }
 
-// src/ui/default-chat-transport.ts
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
-var getOriginalFetch2 = () => fetch;
-async function fetchUIMessageStream({
-  api,
-  body,
-  credentials,
-  headers,
-  abortSignal,
-  fetch: fetch2 = getOriginalFetch2(),
-  requestType = "generate"
-}) {
-  var _a17;
-  const response = requestType === "resume" ? await fetch2(`${api}?id=${body.id}`, {
-    method: "GET",
-    headers: {
-      "Content-Type": "application/json",
-      ...headers
-    },
-    signal: abortSignal,
-    credentials
-  }) : await fetch2(api, {
-    method: "POST",
-    body: JSON.stringify(body),
-    headers: {
-      "Content-Type": "application/json",
-      ...headers
-    },
-    signal: abortSignal,
-    credentials
-  });
-  if (!response.ok) {
-    throw new Error(
-      (_a17 = await response.text()) != null ? _a17 : "Failed to fetch the chat response."
-    );
-  }
-  if (!response.body) {
-    throw new Error("The response body is empty.");
-  }
-  return (0, import_provider_utils4.parseJsonEventStream)({
-    stream: response.body,
-    schema: uiMessageStreamPartSchema
-  }).pipeThrough(
-    new TransformStream({
-      async transform(part, controller) {
-        if (!part.success) {
-          throw part.error;
-        }
-        controller.enqueue(part.value);
-      }
-    })
-  );
-}
-var DefaultChatTransport = class {
-  constructor({
-    api = "/api/chat",
-    credentials,
-    headers,
-    body,
-    fetch: fetch2,
-    prepareRequest
-  } = {}) {
-    this.api = api;
-    this.credentials = credentials;
-    this.headers = headers;
-    this.body = body;
-    this.fetch = fetch2;
-    this.prepareRequest = prepareRequest;
-  }
-  submitMessages({
-    chatId,
-    messages,
-    abortSignal,
-    metadata,
-    headers,
-    body,
-    requestType
-  }) {
-    var _a17, _b;
-    const preparedRequest = (_a17 = this.prepareRequest) == null ? void 0 : _a17.call(this, {
-      id: chatId,
-      messages,
-      body: { ...this.body, ...body },
-      headers: { ...this.headers, ...headers },
-      credentials: this.credentials,
-      requestMetadata: metadata
-    });
-    return fetchUIMessageStream({
-      api: this.api,
-      body: (preparedRequest == null ? void 0 : preparedRequest.body) !== void 0 ? preparedRequest.body : { ...this.body, ...body, id: chatId, messages },
-      headers: (preparedRequest == null ? void 0 : preparedRequest.headers) !== void 0 ? preparedRequest.headers : { ...this.headers, ...headers },
-      credentials: (_b = preparedRequest == null ? void 0 : preparedRequest.credentials) != null ? _b : this.credentials,
-      abortSignal,
-      fetch: this.fetch,
-      requestType
-    });
-  }
-};
-
-// src/ui/convert-file-list-to-file-ui-parts.ts
-async function convertFileListToFileUIParts(files) {
-  if (files == null) {
-    return [];
-  }
-  if (!globalThis.FileList || !(files instanceof globalThis.FileList)) {
-    throw new Error("FileList is not supported in the current environment");
-  }
-  return Promise.all(
-    Array.from(files).map(async (file) => {
-      const { name: name17, type } = file;
-      const dataUrl = await new Promise((resolve, reject) => {
-        const reader = new FileReader();
-        reader.onload = (readerEvent) => {
-          var _a17;
-          resolve((_a17 = readerEvent.target) == null ? void 0 : _a17.result);
-        };
-        reader.onerror = (error) => reject(error);
-        reader.readAsDataURL(file);
-      });
-      return {
-        type: "file",
-        mediaType: type,
-        filename: name17,
-        url: dataUrl
-      };
-    })
-  );
-}
-
 // src/ui/chat.ts
 var AbstractChat = class {
   constructor({
@@ -1616,7 +1637,6 @@ var AbstractChat = class {
     onToolCall,
     onFinish
   }) {
-    this.subscribers = /* @__PURE__ */ new Set();
     this.activeResponse = void 0;
     this.jobExecutor = new SerialJobExecutor();
     this.removeAssistantResponse = () => {
@@ -1628,7 +1648,6 @@ var AbstractChat = class {
         throw new Error("Last message is not an assistant message");
       }
       this.state.popMessage();
-      this.emit({ type: "messages-changed" });
     };
     /**
      * Append a user message to the chat list. This triggers the API call to fetch
@@ -1653,7 +1672,6 @@ var AbstractChat = class {
         id: (_a17 = uiMessage.id) != null ? _a17 : this.generateId(),
         role: (_b = uiMessage.role) != null ? _b : "user"
       });
-      this.emit({ type: "messages-changed" });
       await this.triggerRequest({ requestType: "generate", ...options });
     };
     /**
@@ -1665,7 +1683,6 @@ var AbstractChat = class {
       }
       if (this.lastMessage.role === "assistant") {
         this.state.popMessage();
-        this.emit({ type: "messages-changed" });
       }
       await this.triggerRequest({ requestType: "generate", ...options });
     };
@@ -1739,7 +1756,6 @@ var AbstractChat = class {
       return;
     this.state.status = status;
     this.state.error = error;
-    this.emit({ type: "status-changed" });
   }
   get error() {
     return this.state.error;
@@ -1750,18 +1766,8 @@ var AbstractChat = class {
   get lastMessage() {
     return this.state.messages[this.state.messages.length - 1];
   }
-  subscribe(subscriber) {
-    this.subscribers.add(subscriber);
-    return () => this.subscribers.delete(subscriber);
-  }
   set messages(messages) {
     this.state.messages = messages;
-    this.emit({ type: "messages-changed" });
-  }
-  emit(event) {
-    for (const subscriber of this.subscribers) {
-      subscriber.onChange(event);
-    }
   }
   async triggerRequest({
     requestType,
@@ -1809,9 +1815,6 @@ var AbstractChat = class {
               } else {
                 this.state.pushMessage(activeResponse.state.message);
               }
-              this.emit({
-                type: "messages-changed"
-              });
             }
           })
         )
@@ -2214,7 +2217,10 @@ function createUIMessageStream({
                 safeEnqueue(value);
               }
             })().catch((error) => {
-              safeEnqueue({ type: "error", errorText: onError(error) });
+              safeEnqueue({
+                type: "error",
+                errorText: onError(error)
+              });
             })
           );
         },
@@ -2224,12 +2230,18 @@ function createUIMessageStream({
     if (result) {
       ongoingStreamPromises.push(
         result.catch((error) => {
-          safeEnqueue({ type: "error", errorText: onError(error) });
+          safeEnqueue({
+            type: "error",
+            errorText: onError(error)
+          });
         })
       );
     }
   } catch (error) {
-    safeEnqueue({ type: "error", errorText: onError(error) });
+    safeEnqueue({
+      type: "error",
+      errorText: onError(error)
+    });
   }
   const waitForStreams = new Promise(async (resolve) => {
     while (ongoingStreamPromises.length > 0) {
@@ -2251,16 +2263,6 @@ function createUIMessageStream({
   });
 }
 
-// src/ui-message-stream/ui-message-stream-headers.ts
-var uiMessageStreamHeaders = {
-  "content-type": "text/event-stream",
-  "cache-control": "no-cache",
-  connection: "keep-alive",
-  "x-vercel-ai-ui-message-stream": "v1",
-  "x-accel-buffering": "no"
-  // disable nginx buffering
-};
-
 // src/ui-message-stream/json-to-sse-transform-stream.ts
 var JsonToSseTransformStream = class extends TransformStream {
   constructor() {
@@ -2277,6 +2279,16 @@ var JsonToSseTransformStream = class extends TransformStream {
   }
 };
 
+// src/ui-message-stream/ui-message-stream-headers.ts
+var uiMessageStreamHeaders = {
+  "content-type": "text/event-stream",
+  "cache-control": "no-cache",
+  connection: "keep-alive",
+  "x-vercel-ai-ui-message-stream": "v1",
+  "x-accel-buffering": "no"
+  // disable nginx buffering
+};
+
 // src/ui-message-stream/create-ui-message-stream-response.ts
 function createUIMessageStreamResponse({
   status,
@@ -4918,7 +4930,8 @@ var DefaultStreamObjectResult = class {
           }),
           providerOptions,
           abortSignal,
-          headers
+          headers,
+          includeRawChunks: false
         };
         const transformer = {
           transform: (chunk, controller) => {
@@ -5957,6 +5970,14 @@ async function executeTools({
   const toolResults = await Promise.all(
     toolCalls.map(async ({ toolCallId, toolName, args }) => {
       const tool2 = tools[toolName];
+      if ((tool2 == null ? void 0 : tool2.onArgsAvailable) != null) {
+        await tool2.onArgsAvailable({
+          args,
+          toolCallId,
+          messages,
+          abortSignal
+        });
+      }
       if ((tool2 == null ? void 0 : tool2.execute) == null) {
         return void 0;
       }
@@ -6260,7 +6281,6 @@ var import_provider_utils21 = require("@ai-sdk/provider-utils");
 function runToolsTransformation({
   tools,
   generatorStream,
-  toolCallStreaming,
   tracer,
   telemetry,
   system,
@@ -6301,6 +6321,10 @@ function runToolsTransformation({
           controller.enqueue(chunk);
           break;
         }
+        case "raw": {
+          controller.enqueue(chunk);
+          break;
+        }
         case "file": {
           controller.enqueue({
             type: "file",
@@ -6312,22 +6336,20 @@ function runToolsTransformation({
           break;
         }
         case "tool-call-delta": {
-          if (toolCallStreaming) {
-            if (!activeToolCalls[chunk.toolCallId]) {
-              controller.enqueue({
-                type: "tool-call-streaming-start",
-                toolCallId: chunk.toolCallId,
-                toolName: chunk.toolName
-              });
-              activeToolCalls[chunk.toolCallId] = true;
-            }
+          if (!activeToolCalls[chunk.toolCallId]) {
             controller.enqueue({
-              type: "tool-call-delta",
+              type: "tool-call-streaming-start",
               toolCallId: chunk.toolCallId,
-              toolName: chunk.toolName,
-              argsTextDelta: chunk.argsTextDelta
+              toolName: chunk.toolName
             });
+            activeToolCalls[chunk.toolCallId] = true;
           }
+          controller.enqueue({
+            type: "tool-call-delta",
+            toolCallId: chunk.toolCallId,
+            toolName: chunk.toolName,
+            argsTextDelta: chunk.argsTextDelta
+          });
           break;
         }
         case "tool-call": {
@@ -6341,6 +6363,14 @@ function runToolsTransformation({
             });
             controller.enqueue(toolCall);
             const tool2 = tools[toolCall.toolName];
+            if (tool2.onArgsAvailable != null) {
+              await tool2.onArgsAvailable({
+                args: toolCall.args,
+                toolCallId: toolCall.toolCallId,
+                messages,
+                abortSignal
+              });
+            }
             if (tool2.execute != null) {
               const toolExecutionId = (0, import_provider_utils21.generateId)();
               outstandingToolResults.add(toolExecutionId);
@@ -6470,12 +6500,11 @@ function streamText({
   experimental_telemetry: telemetry,
   prepareStep,
   providerOptions,
-  experimental_toolCallStreaming = false,
-  toolCallStreaming = experimental_toolCallStreaming,
   experimental_activeTools,
   activeTools = experimental_activeTools,
   experimental_repairToolCall: repairToolCall,
   experimental_transform: transform,
+  includeRawChunks = false,
   onChunk,
   onError = ({ error }) => {
     console.error(error);
@@ -6501,7 +6530,6 @@ function streamText({
     messages,
     tools,
     toolChoice,
-    toolCallStreaming,
     transforms: asArray(transform),
     activeTools,
     repairToolCall,
@@ -6509,6 +6537,7 @@ function streamText({
     output,
     providerOptions,
     prepareStep,
+    includeRawChunks,
     onChunk,
     onError,
     onFinish,
@@ -6579,7 +6608,6 @@ var DefaultStreamTextResult = class {
     messages,
     tools,
     toolChoice,
-    toolCallStreaming,
     transforms,
     activeTools,
     repairToolCall,
@@ -6587,6 +6615,7 @@ var DefaultStreamTextResult = class {
     output,
     providerOptions,
     prepareStep,
+    includeRawChunks,
     now: now2,
     currentDate,
     generateId: generateId3,
@@ -6599,6 +6628,7 @@ var DefaultStreamTextResult = class {
     this._finishReason = new DelayedPromise();
     this._steps = new DelayedPromise();
     this.output = output;
+    this.includeRawChunks = includeRawChunks;
     this.generateId = generateId3;
     let stepFinish;
     let activeReasoningPart = void 0;
@@ -6806,6 +6836,7 @@ var DefaultStreamTextResult = class {
           usage
         }) {
           var _a17, _b, _c, _d;
+          const includeRawChunks2 = self.includeRawChunks;
           stepFinish = new DelayedPromise();
           const initialPrompt = await standardizePrompt({
             system,
@@ -6892,7 +6923,8 @@ var DefaultStreamTextResult = class {
                     prompt: promptMessages,
                     providerOptions,
                     abortSignal,
-                    headers
+                    headers,
+                    includeRawChunks: includeRawChunks2
                   })
                 };
               }
@@ -6901,7 +6933,6 @@ var DefaultStreamTextResult = class {
           const streamWithToolResults = runToolsTransformation({
             tools,
             generatorStream: stream2,
-            toolCallStreaming,
             tracer,
             telemetry,
             system,
@@ -7031,8 +7062,28 @@ var DefaultStreamTextResult = class {
                       controller.enqueue(chunk);
                       break;
                     }
-                    case "tool-call-streaming-start":
+                    case "tool-call-streaming-start": {
+                      const tool2 = tools == null ? void 0 : tools[chunk.toolName];
+                      if ((tool2 == null ? void 0 : tool2.onArgsStreamingStart) != null) {
+                        await tool2.onArgsStreamingStart({
+                          toolCallId: chunk.toolCallId,
+                          messages: stepInputMessages,
+                          abortSignal
+                        });
+                      }
+                      controller.enqueue(chunk);
+                      break;
+                    }
                     case "tool-call-delta": {
+                      const tool2 = tools == null ? void 0 : tools[chunk.toolName];
+                      if ((tool2 == null ? void 0 : tool2.onArgsStreamingDelta) != null) {
+                        await tool2.onArgsStreamingDelta({
+                          argsTextDelta: chunk.argsTextDelta,
+                          toolCallId: chunk.toolCallId,
+                          messages: stepInputMessages,
+                          abortSignal
+                        });
+                      }
                       controller.enqueue(chunk);
                       break;
                     }
@@ -7041,6 +7092,10 @@ var DefaultStreamTextResult = class {
                       stepFinishReason = "error";
                       break;
                     }
+                    case "raw": {
+                      controller.enqueue(chunk);
+                      break;
+                    }
                     default: {
                       const exhaustiveCheck = chunkType;
                       throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);
@@ -7312,7 +7367,7 @@ var DefaultStreamTextResult = class {
               break;
             }
             case "source": {
-              if (sendSources) {
+              if (sendSources && part.sourceType === "url") {
                 controller.enqueue({
                   type: "source-url",
                   sourceId: part.id,
@@ -7321,6 +7376,16 @@ var DefaultStreamTextResult = class {
                   providerMetadata: part.providerMetadata
                 });
               }
+              if (sendSources && part.sourceType === "document") {
+                controller.enqueue({
+                  type: "source-document",
+                  sourceId: part.id,
+                  mediaType: part.mediaType,
+                  title: part.title,
+                  filename: part.filename,
+                  providerMetadata: part.providerMetadata
+                });
+              }
               break;
             }
             case "tool-call-streaming-start": {
@@ -7400,6 +7465,9 @@ var DefaultStreamTextResult = class {
               }
               break;
             }
+            case "raw": {
+              break;
+            }
             default: {
               const exhaustiveCheck = partType;
               throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);