ai 4.0.8 → 4.0.10

package/dist/index.js

@@ -48,32 +48,202 @@ __export(streams_exports, {
   UnsupportedFunctionalityError: () => import_provider13.UnsupportedFunctionalityError,
   convertToCoreMessages: () => convertToCoreMessages,
   cosineSimilarity: () => cosineSimilarity,
-  createStreamDataTransformer: () => createStreamDataTransformer,
+  createDataStream: () => createDataStream,
+  createDataStreamResponse: () => createDataStreamResponse,
   embed: () => embed,
   embedMany: () => embedMany,
   experimental_createProviderRegistry: () => experimental_createProviderRegistry,
   experimental_customProvider: () => experimental_customProvider,
   experimental_wrapLanguageModel: () => experimental_wrapLanguageModel,
-  formatAssistantStreamPart: () => import_ui_utils11.formatAssistantStreamPart,
-  formatDataStreamPart: () => import_ui_utils11.formatDataStreamPart,
+  formatAssistantStreamPart: () => import_ui_utils14.formatAssistantStreamPart,
+  formatDataStreamPart: () => import_ui_utils14.formatDataStreamPart,
   generateId: () => import_provider_utils12.generateId,
   generateObject: () => generateObject,
   generateText: () => generateText,
-  jsonSchema: () => import_ui_utils8.jsonSchema,
-  parseAssistantStreamPart: () => import_ui_utils11.parseAssistantStreamPart,
-  parseDataStreamPart: () => import_ui_utils11.parseDataStreamPart,
-  processDataStream: () => import_ui_utils11.processDataStream,
-  processTextStream: () => import_ui_utils11.processTextStream,
+  jsonSchema: () => import_ui_utils9.jsonSchema,
+  parseAssistantStreamPart: () => import_ui_utils14.parseAssistantStreamPart,
+  parseDataStreamPart: () => import_ui_utils14.parseDataStreamPart,
+  pipeDataStreamToResponse: () => pipeDataStreamToResponse,
+  processDataStream: () => import_ui_utils14.processDataStream,
+  processTextStream: () => import_ui_utils14.processTextStream,
   streamObject: () => streamObject,
   streamText: () => streamText,
   tool: () => tool
 });
 module.exports = __toCommonJS(streams_exports);
-var import_ui_utils11 = require("@ai-sdk/ui-utils");
+var import_ui_utils14 = require("@ai-sdk/ui-utils");
 var import_provider_utils12 = require("@ai-sdk/provider-utils");
 
 // core/index.ts
-var import_ui_utils8 = require("@ai-sdk/ui-utils");
+var import_ui_utils9 = require("@ai-sdk/ui-utils");
+
+// core/data-stream/create-data-stream.ts
+var import_ui_utils = require("@ai-sdk/ui-utils");
+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;
+    }
+  });
+  try {
+    const result = execute({
+      writeData(data) {
+        controller.enqueue((0, import_ui_utils.formatDataStreamPart)("data", [data]));
+      },
+      writeMessageAnnotation(annotation) {
+        controller.enqueue(
+          (0, import_ui_utils.formatDataStreamPart)("message_annotations", [annotation])
+        );
+      },
+      merge(streamArg) {
+        ongoingStreamPromises.push(
+          (async () => {
+            const reader = streamArg.getReader();
+            while (true) {
+              const { done, value } = await reader.read();
+              if (done)
+                break;
+              controller.enqueue(value);
+            }
+          })().catch((error) => {
+            controller.enqueue((0, import_ui_utils.formatDataStreamPart)("error", onError(error)));
+          })
+        );
+      },
+      onError
+    });
+    if (result) {
+      ongoingStreamPromises.push(
+        result.catch((error) => {
+          controller.enqueue((0, import_ui_utils.formatDataStreamPart)("error", onError(error)));
+        })
+      );
+    }
+  } catch (error) {
+    controller.enqueue((0, import_ui_utils.formatDataStreamPart)("error", onError(error)));
+  }
+  const waitForStreams = new Promise(async (resolve) => {
+    while (ongoingStreamPromises.length > 0) {
+      await ongoingStreamPromises.shift();
+    }
+    resolve();
+  });
+  waitForStreams.finally(() => {
+    controller.close();
+  });
+  return stream;
+}
+
+// core/util/prepare-response-headers.ts
+function prepareResponseHeaders(headers, {
+  contentType,
+  dataStreamVersion
+}) {
+  const responseHeaders = new Headers(headers != null ? headers : {});
+  if (!responseHeaders.has("Content-Type")) {
+    responseHeaders.set("Content-Type", contentType);
+  }
+  if (dataStreamVersion !== void 0) {
+    responseHeaders.set("X-Vercel-AI-Data-Stream", dataStreamVersion);
+  }
+  return responseHeaders;
+}
+
+// core/data-stream/create-data-stream-response.ts
+function createDataStreamResponse({
+  status,
+  statusText,
+  headers,
+  execute,
+  onError
+}) {
+  return new Response(
+    createDataStream({ execute, onError }).pipeThrough(new TextEncoderStream()),
+    {
+      status,
+      statusText,
+      headers: prepareResponseHeaders(headers, {
+        contentType: "text/plain; charset=utf-8",
+        dataStreamVersion: "v1"
+      })
+    }
+  );
+}
+
+// core/util/prepare-outgoing-http-headers.ts
+function prepareOutgoingHttpHeaders(headers, {
+  contentType,
+  dataStreamVersion
+}) {
+  const outgoingHeaders = {};
+  if (headers != null) {
+    for (const [key, value] of Object.entries(headers)) {
+      outgoingHeaders[key] = value;
+    }
+  }
+  if (outgoingHeaders["Content-Type"] == null) {
+    outgoingHeaders["Content-Type"] = contentType;
+  }
+  if (dataStreamVersion !== void 0) {
+    outgoingHeaders["X-Vercel-AI-Data-Stream"] = dataStreamVersion;
+  }
+  return outgoingHeaders;
+}
+
+// core/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();
+}
+
+// core/data-stream/pipe-data-stream-to-response.ts
+function pipeDataStreamToResponse(response, {
+  status,
+  statusText,
+  headers,
+  execute,
+  onError
+}) {
+  writeToServerResponse({
+    response,
+    status,
+    statusText,
+    headers: prepareOutgoingHttpHeaders(headers, {
+      contentType: "text/plain; charset=utf-8",
+      dataStreamVersion: "v1"
+    }),
+    stream: createDataStream({ execute, onError }).pipeThrough(
+      new TextEncoderStream()
+    )
+  });
+}
 
 // errors/invalid-argument-error.ts
 var import_provider = require("@ai-sdk/provider");
@@ -1566,21 +1736,6 @@ function calculateLanguageModelUsage({
   };
 }
 
-// core/util/prepare-response-headers.ts
-function prepareResponseHeaders(headers, {
-  contentType,
-  dataStreamVersion
-}) {
-  const responseHeaders = new Headers(headers != null ? headers : {});
-  if (!responseHeaders.has("Content-Type")) {
-    responseHeaders.set("Content-Type", contentType);
-  }
-  if (dataStreamVersion !== void 0) {
-    responseHeaders.set("X-Vercel-AI-Data-Stream", dataStreamVersion);
-  }
-  return responseHeaders;
-}
-
 // core/generate-object/inject-json-instruction.ts
 var DEFAULT_SCHEMA_PREFIX = "JSON schema:";
 var DEFAULT_SCHEMA_SUFFIX = "You MUST answer with a JSON object that matches the JSON schema above.";
@@ -1622,7 +1777,7 @@ _a7 = symbol7;
 // core/generate-object/output-strategy.ts
 var import_provider10 = require("@ai-sdk/provider");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
-var import_ui_utils = require("@ai-sdk/ui-utils");
+var import_ui_utils2 = require("@ai-sdk/ui-utils");
 
 // core/util/async-iterable-stream.ts
 function createAsyncIterableStream(source, transformer) {
@@ -1839,9 +1994,9 @@ function getOutputStrategy({
 }) {
   switch (output) {
     case "object":
-      return objectOutputStrategy((0, import_ui_utils.asSchema)(schema));
+      return objectOutputStrategy((0, import_ui_utils2.asSchema)(schema));
     case "array":
-      return arrayOutputStrategy((0, import_ui_utils.asSchema)(schema));
+      return arrayOutputStrategy((0, import_ui_utils2.asSchema)(schema));
     case "enum":
       return enumOutputStrategy(enumValues);
     case "no-schema":
@@ -2348,7 +2503,7 @@ var DefaultGenerateObjectResult = class {
 
 // core/generate-object/stream-object.ts
 var import_provider_utils6 = require("@ai-sdk/provider-utils");
-var import_ui_utils2 = require("@ai-sdk/ui-utils");
+var import_ui_utils3 = require("@ai-sdk/ui-utils");
 
 // util/delayed-promise.ts
 var DelayedPromise = class {
@@ -2476,53 +2631,6 @@ function now() {
   return (_b = (_a11 = globalThis == null ? void 0 : globalThis.performance) == null ? void 0 : _a11.now()) != null ? _b : Date.now();
 }
 
-// core/util/prepare-outgoing-http-headers.ts
-function prepareOutgoingHttpHeaders(headers, {
-  contentType,
-  dataStreamVersion
-}) {
-  const outgoingHeaders = {};
-  if (headers != null) {
-    for (const [key, value] of Object.entries(headers)) {
-      outgoingHeaders[key] = value;
-    }
-  }
-  if (outgoingHeaders["Content-Type"] == null) {
-    outgoingHeaders["Content-Type"] = contentType;
-  }
-  if (dataStreamVersion !== void 0) {
-    outgoingHeaders["X-Vercel-AI-Data-Stream"] = dataStreamVersion;
-  }
-  return outgoingHeaders;
-}
-
-// core/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();
-}
-
 // core/generate-object/stream-object.ts
 var originalGenerateId2 = (0, import_provider_utils6.createIdGenerator)({ prefix: "aiobj", size: 24 });
 function streamObject({
@@ -2823,8 +2931,8 @@ var DefaultStreamObjectResult = class {
               if (typeof chunk === "string") {
                 accumulatedText += chunk;
                 textDelta += chunk;
-                const { value: currentObjectJson, state: parseState } = (0, import_ui_utils2.parsePartialJson)(accumulatedText);
-                if (currentObjectJson !== void 0 && !(0, import_ui_utils2.isDeepEqualData)(latestObjectJson, currentObjectJson)) {
+                const { value: currentObjectJson, state: parseState } = (0, import_ui_utils3.parsePartialJson)(accumulatedText);
+                if (currentObjectJson !== void 0 && !(0, import_ui_utils3.isDeepEqualData)(latestObjectJson, currentObjectJson)) {
                   const validationResult = outputStrategy.validatePartialResult({
                     value: currentObjectJson,
                     textDelta,
@@ -2832,7 +2940,7 @@ var DefaultStreamObjectResult = class {
                     isFirstDelta,
                     isFinalDelta: parseState === "successful-parse"
                   });
-                  if (validationResult.success && !(0, import_ui_utils2.isDeepEqualData)(
+                  if (validationResult.success && !(0, import_ui_utils3.isDeepEqualData)(
                     latestObject,
                     validationResult.value.partial
                   )) {
@@ -3118,7 +3226,7 @@ var NoSuchToolError = class extends import_provider12.AISDKError {
 _a9 = symbol9;
 
 // core/prompt/prepare-tools-and-tool-choice.ts
-var import_ui_utils3 = require("@ai-sdk/ui-utils");
+var import_ui_utils4 = require("@ai-sdk/ui-utils");
 
 // core/util/is-non-empty-object.ts
 function isNonEmptyObject(object2) {
@@ -3150,7 +3258,7 @@ function prepareToolsAndToolChoice({
             type: "function",
             name: name11,
             description: tool2.description,
-            parameters: (0, import_ui_utils3.asSchema)(tool2.parameters).jsonSchema
+            parameters: (0, import_ui_utils4.asSchema)(tool2.parameters).jsonSchema
           };
         case "provider-defined":
           return {
@@ -3184,7 +3292,7 @@ function removeTextAfterLastWhitespace(text2) {
 
 // core/generate-text/parse-tool-call.ts
 var import_provider_utils7 = require("@ai-sdk/provider-utils");
-var import_ui_utils4 = require("@ai-sdk/ui-utils");
+var import_ui_utils5 = require("@ai-sdk/ui-utils");
 function parseToolCall({
   toolCall,
   tools
@@ -3200,7 +3308,7 @@ function parseToolCall({
       availableTools: Object.keys(tools)
     });
   }
-  const schema = (0, import_ui_utils4.asSchema)(tool2.parameters);
+  const schema = (0, import_ui_utils5.asSchema)(tool2.parameters);
   const parseResult = toolCall.args.trim() === "" ? (0, import_provider_utils7.safeValidateTypes)({ value: {}, schema }) : (0, import_provider_utils7.safeParseJSON)({ text: toolCall.args, schema });
   if (parseResult.success === false) {
     throw new InvalidToolArgumentsError({
@@ -3639,7 +3747,7 @@ var DefaultGenerateTextResult = class {
 
 // core/generate-text/stream-text.ts
 var import_provider_utils9 = require("@ai-sdk/provider-utils");
-var import_ui_utils6 = require("@ai-sdk/ui-utils");
+var import_ui_utils7 = require("@ai-sdk/ui-utils");
 
 // core/util/merge-streams.ts
 function mergeStreams(stream1, stream2) {
@@ -3730,7 +3838,7 @@ function mergeStreams(stream1, stream2) {
 }
 
 // core/generate-text/run-tools-transformation.ts
-var import_ui_utils5 = require("@ai-sdk/ui-utils");
+var import_ui_utils6 = require("@ai-sdk/ui-utils");
 function runToolsTransformation({
   tools,
   generatorStream,
@@ -3814,7 +3922,7 @@ function runToolsTransformation({
             });
             controller.enqueue(toolCall);
             if (tool2.execute != null) {
-              const toolExecutionId = (0, import_ui_utils5.generateId)();
+              const toolExecutionId = (0, import_ui_utils6.generateId)();
               outstandingToolResults.add(toolExecutionId);
               recordSpan({
                 name: "ai.toolCall",
@@ -4578,12 +4686,12 @@ var DefaultStreamTextResult = class {
         const chunkType = chunk.type;
         switch (chunkType) {
           case "text-delta": {
-            controller.enqueue((0, import_ui_utils6.formatDataStreamPart)("text", chunk.textDelta));
+            controller.enqueue((0, import_ui_utils7.formatDataStreamPart)("text", chunk.textDelta));
             break;
           }
           case "tool-call-streaming-start": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("tool_call_streaming_start", {
+              (0, import_ui_utils7.formatDataStreamPart)("tool_call_streaming_start", {
                 toolCallId: chunk.toolCallId,
                 toolName: chunk.toolName
               })
@@ -4592,7 +4700,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-call-delta": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("tool_call_delta", {
+              (0, import_ui_utils7.formatDataStreamPart)("tool_call_delta", {
                 toolCallId: chunk.toolCallId,
                 argsTextDelta: chunk.argsTextDelta
               })
@@ -4601,7 +4709,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-call": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("tool_call", {
+              (0, import_ui_utils7.formatDataStreamPart)("tool_call", {
                 toolCallId: chunk.toolCallId,
                 toolName: chunk.toolName,
                 args: chunk.args
@@ -4611,7 +4719,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-result": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("tool_result", {
+              (0, import_ui_utils7.formatDataStreamPart)("tool_result", {
                 toolCallId: chunk.toolCallId,
                 result: chunk.result
               })
@@ -4620,13 +4728,13 @@ var DefaultStreamTextResult = class {
           }
           case "error": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("error", getErrorMessage3(chunk.error))
+              (0, import_ui_utils7.formatDataStreamPart)("error", getErrorMessage3(chunk.error))
             );
             break;
           }
           case "step-finish": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("finish_step", {
+              (0, import_ui_utils7.formatDataStreamPart)("finish_step", {
                 finishReason: chunk.finishReason,
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
@@ -4639,7 +4747,7 @@ var DefaultStreamTextResult = class {
           }
           case "finish": {
             controller.enqueue(
-              (0, import_ui_utils6.formatDataStreamPart)("finish_message", {
+              (0, import_ui_utils7.formatDataStreamPart)("finish_message", {
                 finishReason: chunk.finishReason,
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
@@ -4656,7 +4764,7 @@ var DefaultStreamTextResult = class {
         }
       }
     });
-    return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
+    return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer);
   }
   pipeDataStreamToResponse(response, {
     status,
@@ -4688,13 +4796,21 @@ var DefaultStreamTextResult = class {
       stream: this.textStream.pipeThrough(new TextEncoderStream())
     });
   }
+  // TODO breaking change 5.0: remove pipeThrough(new TextEncoderStream())
   toDataStream(options) {
     const stream = this.toDataStreamInternal({
       getErrorMessage: options == null ? void 0 : options.getErrorMessage,
       sendUsage: options == null ? void 0 : options.sendUsage
-    });
+    }).pipeThrough(new TextEncoderStream());
     return (options == null ? void 0 : options.data) ? mergeStreams(options == null ? void 0 : options.data.stream, stream) : stream;
   }
+  mergeIntoDataStream(writer) {
+    writer.merge(
+      this.toDataStreamInternal({
+        getErrorMessage: writer.onError
+      })
+    );
+  }
   toDataStreamResponse({
     headers,
     status,
@@ -4733,7 +4849,7 @@ __export(output_exports, {
   text: () => text
 });
 var import_provider_utils10 = require("@ai-sdk/provider-utils");
-var import_ui_utils7 = require("@ai-sdk/ui-utils");
+var import_ui_utils8 = require("@ai-sdk/ui-utils");
 var text = () => ({
   type: "text",
   responseFormat: () => ({ type: "text" }),
@@ -4747,7 +4863,7 @@ var text = () => ({
 var object = ({
   schema: inputSchema
 }) => {
-  const schema = (0, import_ui_utils7.asSchema)(inputSchema);
+  const schema = (0, import_ui_utils8.asSchema)(inputSchema);
   return {
     type: "object",
     responseFormat: ({ model }) => ({
@@ -4948,7 +5064,7 @@ function magnitude(vector) {
 }
 
 // streams/assistant-response.ts
-var import_ui_utils9 = require("@ai-sdk/ui-utils");
+var import_ui_utils10 = require("@ai-sdk/ui-utils");
 function AssistantResponse({ threadId, messageId }, process2) {
   const stream = new ReadableStream({
     async start(controller) {
@@ -4957,20 +5073,20 @@ function AssistantResponse({ threadId, messageId }, process2) {
       const sendMessage = (message) => {
         controller.enqueue(
           textEncoder.encode(
-            (0, import_ui_utils9.formatAssistantStreamPart)("assistant_message", message)
+            (0, import_ui_utils10.formatAssistantStreamPart)("assistant_message", message)
           )
         );
       };
       const sendDataMessage = (message) => {
         controller.enqueue(
           textEncoder.encode(
-            (0, import_ui_utils9.formatAssistantStreamPart)("data_message", message)
+            (0, import_ui_utils10.formatAssistantStreamPart)("data_message", message)
           )
         );
       };
       const sendError = (errorMessage) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils9.formatAssistantStreamPart)("error", errorMessage))
+          textEncoder.encode((0, import_ui_utils10.formatAssistantStreamPart)("error", errorMessage))
         );
       };
       const forwardStream = async (stream2) => {
@@ -4981,7 +5097,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
             case "thread.message.created": {
               controller.enqueue(
                 textEncoder.encode(
-                  (0, import_ui_utils9.formatAssistantStreamPart)("assistant_message", {
+                  (0, import_ui_utils10.formatAssistantStreamPart)("assistant_message", {
                     id: value.data.id,
                     role: "assistant",
                     content: [{ type: "text", text: { value: "" } }]
@@ -4995,7 +5111,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
               if ((content == null ? void 0 : content.type) === "text" && ((_b = content.text) == null ? void 0 : _b.value) != null) {
                 controller.enqueue(
                   textEncoder.encode(
-                    (0, import_ui_utils9.formatAssistantStreamPart)("text", content.text.value)
+                    (0, import_ui_utils10.formatAssistantStreamPart)("text", content.text.value)
                   )
                 );
               }
@@ -5012,7 +5128,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
       };
       controller.enqueue(
         textEncoder.encode(
-          (0, import_ui_utils9.formatAssistantStreamPart)("assistant_control_data", {
+          (0, import_ui_utils10.formatAssistantStreamPart)("assistant_control_data", {
             threadId,
             messageId
           })
@@ -5046,9 +5162,11 @@ function AssistantResponse({ threadId, messageId }, process2) {
 // streams/langchain-adapter.ts
 var langchain_adapter_exports = {};
 __export(langchain_adapter_exports, {
+  mergeIntoDataStream: () => mergeIntoDataStream,
   toDataStream: () => toDataStream,
   toDataStreamResponse: () => toDataStreamResponse
 });
+var import_ui_utils11 = require("@ai-sdk/ui-utils");
 
 // streams/stream-callbacks.ts
 function createCallbacksTransformer(callbacks = {}) {
@@ -5079,87 +5197,8 @@ function createCallbacksTransformer(callbacks = {}) {
   });
 }
 
-// streams/stream-data.ts
-var import_ui_utils10 = require("@ai-sdk/ui-utils");
-
-// util/constants.ts
-var HANGING_STREAM_WARNING_TIME_MS = 15 * 1e3;
-
-// streams/stream-data.ts
-var StreamData = class {
-  constructor() {
-    this.encoder = new TextEncoder();
-    this.controller = null;
-    this.isClosed = false;
-    this.warningTimeout = null;
-    const self = this;
-    this.stream = new ReadableStream({
-      start: async (controller) => {
-        self.controller = controller;
-        if (process.env.NODE_ENV === "development") {
-          self.warningTimeout = setTimeout(() => {
-            console.warn(
-              "The data stream is hanging. Did you forget to close it with `data.close()`?"
-            );
-          }, HANGING_STREAM_WARNING_TIME_MS);
-        }
-      },
-      pull: (controller) => {
-      },
-      cancel: (reason) => {
-        this.isClosed = true;
-      }
-    });
-  }
-  async close() {
-    if (this.isClosed) {
-      throw new Error("Data Stream has already been closed.");
-    }
-    if (!this.controller) {
-      throw new Error("Stream controller is not initialized.");
-    }
-    this.controller.close();
-    this.isClosed = true;
-    if (this.warningTimeout) {
-      clearTimeout(this.warningTimeout);
-    }
-  }
-  append(value) {
-    if (this.isClosed) {
-      throw new Error("Data Stream has already been closed.");
-    }
-    if (!this.controller) {
-      throw new Error("Stream controller is not initialized.");
-    }
-    this.controller.enqueue(
-      this.encoder.encode((0, import_ui_utils10.formatDataStreamPart)("data", [value]))
-    );
-  }
-  appendMessageAnnotation(value) {
-    if (this.isClosed) {
-      throw new Error("Data Stream has already been closed.");
-    }
-    if (!this.controller) {
-      throw new Error("Stream controller is not initialized.");
-    }
-    this.controller.enqueue(
-      this.encoder.encode((0, import_ui_utils10.formatDataStreamPart)("message_annotations", [value]))
-    );
-  }
-};
-function createStreamDataTransformer() {
-  const encoder = new TextEncoder();
-  const decoder = new TextDecoder();
-  return new TransformStream({
-    transform: async (chunk, controller) => {
-      const message = decoder.decode(chunk);
-      controller.enqueue(encoder.encode((0, import_ui_utils10.formatDataStreamPart)("text", message)));
-    }
-  });
-}
-
 // streams/langchain-adapter.ts
-function toDataStream(stream, callbacks) {
+function toDataStreamInternal(stream, callbacks) {
   return stream.pipeThrough(
     new TransformStream({
       transform: async (value, controller) => {
@@ -5180,11 +5219,25 @@ function toDataStream(stream, callbacks) {
         forwardAIMessageChunk(value, controller);
       }
     })
-  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
+  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(new TextDecoderStream()).pipeThrough(
+    new TransformStream({
+      transform: async (chunk, controller) => {
+        controller.enqueue((0, import_ui_utils11.formatDataStreamPart)("text", chunk));
+      }
+    })
+  );
+}
+function toDataStream(stream, callbacks) {
+  return toDataStreamInternal(stream, callbacks).pipeThrough(
+    new TextEncoderStream()
+  );
 }
 function toDataStreamResponse(stream, options) {
   var _a11;
-  const dataStream = toDataStream(stream, options == null ? void 0 : options.callbacks);
+  const dataStream = toDataStreamInternal(
+    stream,
+    options == null ? void 0 : options.callbacks
+  ).pipeThrough(new TextEncoderStream());
   const data = options == null ? void 0 : options.data;
   const init = options == null ? void 0 : options.init;
   const responseStream = data ? mergeStreams(data.stream, dataStream) : dataStream;
@@ -5197,6 +5250,9 @@ function toDataStreamResponse(stream, options) {
     })
   });
 }
+function mergeIntoDataStream(stream, options) {
+  options.dataStream.merge(toDataStreamInternal(stream, options.callbacks));
+}
 function forwardAIMessageChunk(chunk, controller) {
   if (typeof chunk.content === "string") {
     controller.enqueue(chunk.content);
@@ -5213,11 +5269,13 @@ function forwardAIMessageChunk(chunk, controller) {
 // streams/llamaindex-adapter.ts
 var llamaindex_adapter_exports = {};
 __export(llamaindex_adapter_exports, {
+  mergeIntoDataStream: () => mergeIntoDataStream2,
   toDataStream: () => toDataStream2,
   toDataStreamResponse: () => toDataStreamResponse2
 });
 var import_provider_utils11 = require("@ai-sdk/provider-utils");
-function toDataStream2(stream, callbacks) {
+var import_ui_utils12 = require("@ai-sdk/ui-utils");
+function toDataStreamInternal2(stream, callbacks) {
   const trimStart = trimStartOfStream();
   return (0, import_provider_utils11.convertAsyncIteratorToReadableStream)(stream[Symbol.asyncIterator]()).pipeThrough(
     new TransformStream({
@@ -5225,12 +5283,25 @@ function toDataStream2(stream, callbacks) {
         controller.enqueue(trimStart(message.delta));
       }
     })
-  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
+  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(new TextDecoderStream()).pipeThrough(
+    new TransformStream({
+      transform: async (chunk, controller) => {
+        controller.enqueue((0, import_ui_utils12.formatDataStreamPart)("text", chunk));
+      }
+    })
+  );
+}
+function toDataStream2(stream, callbacks) {
+  return toDataStreamInternal2(stream, callbacks).pipeThrough(
+    new TextEncoderStream()
+  );
 }
 function toDataStreamResponse2(stream, options = {}) {
   var _a11;
   const { init, data, callbacks } = options;
-  const dataStream = toDataStream2(stream, callbacks);
+  const dataStream = toDataStreamInternal2(stream, callbacks).pipeThrough(
+    new TextEncoderStream()
+  );
   const responseStream = data ? mergeStreams(data.stream, dataStream) : dataStream;
   return new Response(responseStream, {
     status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
@@ -5241,6 +5312,9 @@ function toDataStreamResponse2(stream, options = {}) {
     })
   });
 }
+function mergeIntoDataStream2(stream, options) {
+  options.dataStream.merge(toDataStreamInternal2(stream, options.callbacks));
+}
 function trimStartOfStream() {
   let isStreamStart = true;
   return (text2) => {
@@ -5252,6 +5326,75 @@ function trimStartOfStream() {
     return text2;
   };
 }
+
+// streams/stream-data.ts
+var import_ui_utils13 = require("@ai-sdk/ui-utils");
+
+// util/constants.ts
+var HANGING_STREAM_WARNING_TIME_MS = 15 * 1e3;
+
+// streams/stream-data.ts
+var StreamData = class {
+  constructor() {
+    this.encoder = new TextEncoder();
+    this.controller = null;
+    this.isClosed = false;
+    this.warningTimeout = null;
+    const self = this;
+    this.stream = new ReadableStream({
+      start: async (controller) => {
+        self.controller = controller;
+        if (process.env.NODE_ENV === "development") {
+          self.warningTimeout = setTimeout(() => {
+            console.warn(
+              "The data stream is hanging. Did you forget to close it with `data.close()`?"
+            );
+          }, HANGING_STREAM_WARNING_TIME_MS);
+        }
+      },
+      pull: (controller) => {
+      },
+      cancel: (reason) => {
+        this.isClosed = true;
+      }
+    });
+  }
+  async close() {
+    if (this.isClosed) {
+      throw new Error("Data Stream has already been closed.");
+    }
+    if (!this.controller) {
+      throw new Error("Stream controller is not initialized.");
+    }
+    this.controller.close();
+    this.isClosed = true;
+    if (this.warningTimeout) {
+      clearTimeout(this.warningTimeout);
+    }
+  }
+  append(value) {
+    if (this.isClosed) {
+      throw new Error("Data Stream has already been closed.");
+    }
+    if (!this.controller) {
+      throw new Error("Stream controller is not initialized.");
+    }
+    this.controller.enqueue(
+      this.encoder.encode((0, import_ui_utils13.formatDataStreamPart)("data", [value]))
+    );
+  }
+  appendMessageAnnotation(value) {
+    if (this.isClosed) {
+      throw new Error("Data Stream has already been closed.");
+    }
+    if (!this.controller) {
+      throw new Error("Stream controller is not initialized.");
+    }
+    this.controller.enqueue(
+      this.encoder.encode((0, import_ui_utils13.formatDataStreamPart)("message_annotations", [value]))
+    );
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AISDKError,
@@ -5282,7 +5425,8 @@ function trimStartOfStream() {
   UnsupportedFunctionalityError,
   convertToCoreMessages,
   cosineSimilarity,
-  createStreamDataTransformer,
+  createDataStream,
+  createDataStreamResponse,
   embed,
   embedMany,
   experimental_createProviderRegistry,
@@ -5296,6 +5440,7 @@ function trimStartOfStream() {
   jsonSchema,
   parseAssistantStreamPart,
   parseDataStreamPart,
+  pipeDataStreamToResponse,
   processDataStream,
   processTextStream,
   streamObject,