ai 5.0.0-canary.21 → 5.0.0-canary.22

package/dist/index.js

@@ -63,11 +63,11 @@ __export(src_exports, {
   coreToolMessageSchema: () => coreToolMessageSchema,
   coreUserMessageSchema: () => coreUserMessageSchema,
   cosineSimilarity: () => cosineSimilarity,
-  createDataStream: () => createDataStream,
-  createDataStreamResponse: () => createDataStreamResponse,
   createIdGenerator: () => import_provider_utils25.createIdGenerator,
   createProviderRegistry: () => createProviderRegistry,
   createTextStreamResponse: () => createTextStreamResponse,
+  createUIMessageStream: () => createUIMessageStream,
+  createUIMessageStreamResponse: () => createUIMessageStreamResponse,
   customProvider: () => customProvider,
   defaultSettingsMiddleware: () => defaultSettingsMiddleware,
   embed: () => embed,
@@ -90,8 +90,8 @@ __export(src_exports, {
   jsonSchema: () => import_provider_utils25.jsonSchema,
   modelMessageSchema: () => modelMessageSchema,
   parsePartialJson: () => parsePartialJson,
-  pipeDataStreamToResponse: () => pipeDataStreamToResponse,
   pipeTextStreamToResponse: () => pipeTextStreamToResponse,
+  pipeUIMessageStreamToResponse: () => pipeUIMessageStreamToResponse,
   processTextStream: () => processTextStream,
   shouldResubmitMessages: () => shouldResubmitMessages,
   simulateReadableStream: () => simulateReadableStream,
@@ -109,172 +109,6 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 var import_provider_utils25 = require("@ai-sdk/provider-utils");
 
-// src/data-stream/create-data-stream.ts
-function createDataStream({
-  execute,
-  onError = () => "An error occurred."
-  // mask error messages for safety by default
-}) {
-  let controller;
-  const ongoingStreamPromises = [];
-  const stream = new ReadableStream({
-    start(controllerArg) {
-      controller = controllerArg;
-    }
-  });
-  function safeEnqueue(data) {
-    try {
-      controller.enqueue(data);
-    } catch (error) {
-    }
-  }
-  try {
-    const result = execute({
-      write(part) {
-        safeEnqueue(part);
-      },
-      merge(streamArg) {
-        ongoingStreamPromises.push(
-          (async () => {
-            const reader = streamArg.getReader();
-            while (true) {
-              const { done, value } = await reader.read();
-              if (done)
-                break;
-              safeEnqueue(value);
-            }
-          })().catch((error) => {
-            safeEnqueue({ type: "error", value: onError(error) });
-          })
-        );
-      },
-      onError
-    });
-    if (result) {
-      ongoingStreamPromises.push(
-        result.catch((error) => {
-          safeEnqueue({ type: "error", value: onError(error) });
-        })
-      );
-    }
-  } catch (error) {
-    safeEnqueue({ type: "error", value: onError(error) });
-  }
-  const waitForStreams = new Promise(async (resolve) => {
-    while (ongoingStreamPromises.length > 0) {
-      await ongoingStreamPromises.shift();
-    }
-    resolve();
-  });
-  waitForStreams.finally(() => {
-    try {
-      controller.close();
-    } catch (error) {
-    }
-  });
-  return stream;
-}
-
-// src/util/prepare-headers.ts
-function prepareHeaders(headers, defaultHeaders) {
-  const responseHeaders = new Headers(headers != null ? headers : {});
-  for (const [key, value] of Object.entries(defaultHeaders)) {
-    if (!responseHeaders.has(key)) {
-      responseHeaders.set(key, value);
-    }
-  }
-  return responseHeaders;
-}
-
-// src/data-stream/data-stream-headers.ts
-var dataStreamHeaders = {
-  "content-type": "text/event-stream",
-  "cache-control": "no-cache",
-  connection: "keep-alive",
-  "x-vercel-ai-data-stream": "v2",
-  "x-accel-buffering": "no"
-  // disable nginx buffering
-};
-
-// src/data-stream/json-to-sse-transform-stream.ts
-var JsonToSseTransformStream = class extends TransformStream {
-  constructor() {
-    super({
-      transform(part, controller) {
-        controller.enqueue(`data: ${JSON.stringify(part)}
-
-`);
-      },
-      flush(controller) {
-        controller.enqueue("data: [DONE]\n\n");
-      }
-    });
-  }
-};
-
-// src/data-stream/create-data-stream-response.ts
-function createDataStreamResponse({
-  status,
-  statusText,
-  headers,
-  dataStream
-}) {
-  return new Response(
-    dataStream.pipeThrough(new JsonToSseTransformStream()).pipeThrough(new TextEncoderStream()),
-    {
-      status,
-      statusText,
-      headers: prepareHeaders(headers, dataStreamHeaders)
-    }
-  );
-}
-
-// src/util/write-to-server-response.ts
-function writeToServerResponse({
-  response,
-  status,
-  statusText,
-  headers,
-  stream
-}) {
-  response.writeHead(status != null ? status : 200, statusText, headers);
-  const reader = stream.getReader();
-  const read = async () => {
-    try {
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done)
-          break;
-        response.write(value);
-      }
-    } catch (error) {
-      throw error;
-    } finally {
-      response.end();
-    }
-  };
-  read();
-}
-
-// src/data-stream/pipe-data-stream-to-response.ts
-function pipeDataStreamToResponse({
-  response,
-  status,
-  statusText,
-  headers,
-  dataStream
-}) {
-  writeToServerResponse({
-    response,
-    status,
-    statusText,
-    headers: Object.fromEntries(
-      prepareHeaders(headers, dataStreamHeaders).entries()
-    ),
-    stream: dataStream.pipeThrough(new JsonToSseTransformStream()).pipeThrough(new TextEncoderStream())
-  });
-}
-
 // src/error/index.ts
 var import_provider16 = require("@ai-sdk/provider");
 
@@ -625,6 +459,17 @@ var RetryError = class extends import_provider15.AISDKError {
 };
 _a15 = symbol15;
 
+// src/util/prepare-headers.ts
+function prepareHeaders(headers, defaultHeaders) {
+  const responseHeaders = new Headers(headers != null ? headers : {});
+  for (const [key, value] of Object.entries(defaultHeaders)) {
+    if (!responseHeaders.has(key)) {
+      responseHeaders.set(key, value);
+    }
+  }
+  return responseHeaders;
+}
+
 // src/text-stream/create-text-stream-response.ts
 function createTextStreamResponse({
   status,
@@ -641,6 +486,33 @@ function createTextStreamResponse({
   });
 }
 
+// src/util/write-to-server-response.ts
+function writeToServerResponse({
+  response,
+  status,
+  statusText,
+  headers,
+  stream
+}) {
+  response.writeHead(status != null ? status : 200, statusText, headers);
+  const reader = stream.getReader();
+  const read = async () => {
+    try {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done)
+          break;
+        response.write(value);
+      }
+    } catch (error) {
+      throw error;
+    } finally {
+      response.end();
+    }
+  };
+  read();
+}
+
 // src/text-stream/pipe-text-stream-to-response.ts
 function pipeTextStreamToResponse({
   response,
@@ -676,7 +548,7 @@ function appendClientMessage({
 // src/ui/call-chat-api.ts
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
-// src/data-stream/data-stream-parts.ts
+// src/ui-message-stream/ui-message-stream-parts.ts
 var import_zod = require("zod");
 var toolCallSchema = import_zod.z.object({
   toolCallId: import_zod.z.string(),
@@ -697,7 +569,7 @@ var sourceSchema = import_zod.z.object({
   providerMetadata: import_zod.z.any().optional()
   // Use z.any() for generic metadata
 });
-var dataStreamPartSchema = import_zod.z.discriminatedUnion("type", [
+var uiMessageStreamPartSchema = import_zod.z.discriminatedUnion("type", [
   import_zod.z.object({
     type: import_zod.z.literal("text"),
     value: import_zod.z.string()
@@ -796,7 +668,7 @@ async function consumeStream({
   }
 }
 
-// src/ui/process-chat-response.ts
+// src/ui/process-ui-message-stream.ts
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/util/merge-objects.ts
@@ -1184,8 +1056,8 @@ function getToolInvocations(message) {
   ).map((part) => part.toolInvocation);
 }
 
-// src/ui/process-chat-response.ts
-function processChatResponse({
+// src/ui/process-ui-message-stream.ts
+function processUIMessageStream({
   stream,
   onUpdate,
   onToolCall,
@@ -1470,11 +1342,10 @@ var getOriginalFetch = () => fetch;
 async function callChatApi({
   api,
   body,
-  streamProtocol = "data",
+  streamProtocol = "ui-message",
   credentials,
   headers,
   abortController,
-  onResponse,
   onUpdate,
   onFinish,
   onToolCall,
@@ -1503,9 +1374,6 @@ async function callChatApi({
     signal: (_b = abortController == null ? void 0 : abortController()) == null ? void 0 : _b.signal,
     credentials
   });
-  if (onResponse != null) {
-    await onResponse(response);
-  }
   if (!response.ok) {
     throw new Error(
       (_c = await response.text()) != null ? _c : "Failed to fetch the chat response."
@@ -1524,12 +1392,12 @@ async function callChatApi({
       });
       return;
     }
-    case "data": {
+    case "ui-message": {
       await consumeStream({
-        stream: processChatResponse({
+        stream: processUIMessageStream({
           stream: (0, import_provider_utils4.parseJsonEventStream)({
             stream: response.body,
-            schema: dataStreamPartSchema
+            schema: uiMessageStreamPartSchema
           }).pipeThrough(
             new TransformStream({
               async transform(part, controller) {
@@ -1587,7 +1455,6 @@ async function callCompletionApi({
   setLoading,
   setError,
   setAbortController,
-  onResponse,
   onFinish,
   onError,
   fetch: fetch2 = getOriginalFetch2()
@@ -1614,13 +1481,6 @@ async function callCompletionApi({
     }).catch((err) => {
       throw err;
     });
-    if (onResponse) {
-      try {
-        await onResponse(response);
-      } catch (err) {
-        throw err;
-      }
-    }
     if (!response.ok) {
       throw new Error(
         (_a17 = await response.text()) != null ? _a17 : "Failed to fetch the chat response."
@@ -1645,7 +1505,7 @@ async function callCompletionApi({
         await consumeStream({
           stream: (0, import_provider_utils5.parseJsonEventStream)({
             stream: response.body,
-            schema: dataStreamPartSchema
+            schema: uiMessageStreamPartSchema
           }).pipeThrough(
             new TransformStream({
               async transform(part) {
@@ -1929,6 +1789,134 @@ function updateToolCallResult({
   };
 }
 
+// src/ui-message-stream/create-ui-message-stream.ts
+function createUIMessageStream({
+  execute,
+  onError = () => "An error occurred."
+  // mask error messages for safety by default
+}) {
+  let controller;
+  const ongoingStreamPromises = [];
+  const stream = new ReadableStream({
+    start(controllerArg) {
+      controller = controllerArg;
+    }
+  });
+  function safeEnqueue(data) {
+    try {
+      controller.enqueue(data);
+    } catch (error) {
+    }
+  }
+  try {
+    const result = execute({
+      write(part) {
+        safeEnqueue(part);
+      },
+      merge(streamArg) {
+        ongoingStreamPromises.push(
+          (async () => {
+            const reader = streamArg.getReader();
+            while (true) {
+              const { done, value } = await reader.read();
+              if (done)
+                break;
+              safeEnqueue(value);
+            }
+          })().catch((error) => {
+            safeEnqueue({ type: "error", value: onError(error) });
+          })
+        );
+      },
+      onError
+    });
+    if (result) {
+      ongoingStreamPromises.push(
+        result.catch((error) => {
+          safeEnqueue({ type: "error", value: onError(error) });
+        })
+      );
+    }
+  } catch (error) {
+    safeEnqueue({ type: "error", value: onError(error) });
+  }
+  const waitForStreams = new Promise(async (resolve) => {
+    while (ongoingStreamPromises.length > 0) {
+      await ongoingStreamPromises.shift();
+    }
+    resolve();
+  });
+  waitForStreams.finally(() => {
+    try {
+      controller.close();
+    } catch (error) {
+    }
+  });
+  return stream;
+}
+
+// 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() {
+    super({
+      transform(part, controller) {
+        controller.enqueue(`data: ${JSON.stringify(part)}
+
+`);
+      },
+      flush(controller) {
+        controller.enqueue("data: [DONE]\n\n");
+      }
+    });
+  }
+};
+
+// src/ui-message-stream/create-ui-message-stream-response.ts
+function createUIMessageStreamResponse({
+  status,
+  statusText,
+  headers,
+  stream
+}) {
+  return new Response(
+    stream.pipeThrough(new JsonToSseTransformStream()).pipeThrough(new TextEncoderStream()),
+    {
+      status,
+      statusText,
+      headers: prepareHeaders(headers, uiMessageStreamHeaders)
+    }
+  );
+}
+
+// src/ui-message-stream/pipe-ui-message-stream-to-response.ts
+function pipeUIMessageStreamToResponse({
+  response,
+  status,
+  statusText,
+  headers,
+  stream
+}) {
+  writeToServerResponse({
+    response,
+    status,
+    statusText,
+    headers: Object.fromEntries(
+      prepareHeaders(headers, uiMessageStreamHeaders).entries()
+    ),
+    stream: stream.pipeThrough(new JsonToSseTransformStream()).pipeThrough(new TextEncoderStream())
+  });
+}
+
 // src/util/data-url.ts
 function getTextFromDataUrl(dataUrl) {
   const [header, base64Content] = dataUrl.split(",");
@@ -6764,7 +6752,7 @@ var DefaultStreamTextResult = class {
       )
     );
   }
-  toDataStream({
+  toUIMessageStream({
     newMessageId,
     originalMessages = [],
     onFinish,
@@ -6916,7 +6904,7 @@ var DefaultStreamTextResult = class {
         }
       })
     );
-    return onFinish == null ? baseStream : processChatResponse({
+    return onFinish == null ? baseStream : processUIMessageStream({
       stream: baseStream,
       lastMessage,
       newMessageId: messageId != null ? messageId : this.generateId(),
@@ -6933,7 +6921,7 @@ var DefaultStreamTextResult = class {
       }
     });
   }
-  pipeDataStreamToResponse(response, {
+  pipeUIMessageStreamToResponse(response, {
     newMessageId,
     originalMessages,
     onFinish,
@@ -6945,9 +6933,9 @@ var DefaultStreamTextResult = class {
     onError,
     ...init
   } = {}) {
-    pipeDataStreamToResponse({
+    pipeUIMessageStreamToResponse({
       response,
-      dataStream: this.toDataStream({
+      stream: this.toUIMessageStream({
         newMessageId,
         originalMessages,
         onFinish,
@@ -6968,7 +6956,7 @@ var DefaultStreamTextResult = class {
       ...init
     });
   }
-  toDataStreamResponse({
+  toUIMessageStreamResponse({
     newMessageId,
     originalMessages,
     onFinish,
@@ -6980,8 +6968,8 @@ var DefaultStreamTextResult = class {
     onError,
     ...init
   } = {}) {
-    return createDataStreamResponse({
-      dataStream: this.toDataStream({
+    return createUIMessageStreamResponse({
+      stream: this.toUIMessageStream({
         newMessageId,
         originalMessages,
         onFinish,
@@ -8053,11 +8041,11 @@ var DefaultTranscriptionResult = class {
   coreToolMessageSchema,
   coreUserMessageSchema,
   cosineSimilarity,
-  createDataStream,
-  createDataStreamResponse,
   createIdGenerator,
   createProviderRegistry,
   createTextStreamResponse,
+  createUIMessageStream,
+  createUIMessageStreamResponse,
   customProvider,
   defaultSettingsMiddleware,
   embed,
@@ -8080,8 +8068,8 @@ var DefaultTranscriptionResult = class {
   jsonSchema,
   modelMessageSchema,
   parsePartialJson,
-  pipeDataStreamToResponse,
   pipeTextStreamToResponse,
+  pipeUIMessageStreamToResponse,
   processTextStream,
   shouldResubmitMessages,
   simulateReadableStream,