ai 4.0.0-canary.3 → 4.0.0-canary.5

package/dist/index.js

@@ -21,7 +21,6 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var streams_exports = {};
 __export(streams_exports, {
   AISDKError: () => import_provider13.AISDKError,
-  AIStream: () => AIStream,
   APICallError: () => import_provider13.APICallError,
   AssistantResponse: () => AssistantResponse,
   DownloadError: () => DownloadError,
@@ -43,14 +42,11 @@ __export(streams_exports, {
   NoSuchProviderError: () => NoSuchProviderError,
   NoSuchToolError: () => NoSuchToolError,
   RetryError: () => RetryError,
-  StreamData: () => StreamData2,
-  StreamingTextResponse: () => StreamingTextResponse,
+  StreamData: () => StreamData,
   TypeValidationError: () => import_provider13.TypeValidationError,
   UnsupportedFunctionalityError: () => import_provider13.UnsupportedFunctionalityError,
   convertToCoreMessages: () => convertToCoreMessages,
   cosineSimilarity: () => cosineSimilarity,
-  createCallbacksTransformer: () => createCallbacksTransformer,
-  createEventStreamTransformer: () => createEventStreamTransformer,
   createStreamDataTransformer: () => createStreamDataTransformer,
   embed: () => embed,
   embedMany: () => embedMany,
@@ -59,32 +55,25 @@ __export(streams_exports, {
   experimental_createModelRegistry: () => experimental_createModelRegistry,
   experimental_createProviderRegistry: () => experimental_createProviderRegistry,
   experimental_customProvider: () => experimental_customProvider,
-  experimental_generateObject: () => experimental_generateObject,
-  experimental_generateText: () => experimental_generateText,
-  experimental_streamObject: () => experimental_streamObject,
-  experimental_streamText: () => experimental_streamText,
   experimental_wrapLanguageModel: () => experimental_wrapLanguageModel,
-  formatStreamPart: () => import_ui_utils9.formatStreamPart,
-  generateId: () => generateId2,
+  formatStreamPart: () => import_ui_utils10.formatStreamPart,
+  generateId: () => import_provider_utils11.generateId,
   generateObject: () => generateObject,
   generateText: () => generateText,
-  jsonSchema: () => import_ui_utils6.jsonSchema,
-  parseStreamPart: () => import_ui_utils9.parseStreamPart,
-  processDataProtocolResponse: () => import_ui_utils9.processDataProtocolResponse,
-  readDataStream: () => import_ui_utils9.readDataStream,
-  readableFromAsyncIterable: () => readableFromAsyncIterable,
+  jsonSchema: () => import_ui_utils7.jsonSchema,
+  parseStreamPart: () => import_ui_utils10.parseStreamPart,
+  processDataProtocolResponse: () => import_ui_utils10.processDataProtocolResponse,
+  readDataStream: () => import_ui_utils10.readDataStream,
   streamObject: () => streamObject,
   streamText: () => streamText,
-  streamToResponse: () => streamToResponse,
-  tool: () => tool,
-  trimStartOfStreamHelper: () => trimStartOfStreamHelper
+  tool: () => tool
 });
 module.exports = __toCommonJS(streams_exports);
-var import_ui_utils9 = require("@ai-sdk/ui-utils");
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_ui_utils10 = require("@ai-sdk/ui-utils");
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
 
 // core/index.ts
-var import_ui_utils6 = require("@ai-sdk/ui-utils");
+var import_ui_utils7 = require("@ai-sdk/ui-utils");
 
 // util/retry-with-exponential-backoff.ts
 var import_provider2 = require("@ai-sdk/provider");
@@ -2147,9 +2136,6 @@ async function generateObject({
                       "ai.response.timestamp": responseData.timestamp.toISOString(),
                       "ai.usage.promptTokens": result2.usage.promptTokens,
                       "ai.usage.completionTokens": result2.usage.completionTokens,
-                      // deprecated:
-                      "ai.finishReason": result2.finishReason,
-                      "ai.result.object": { output: () => result2.text },
                       // standardized gen-ai llm span attributes:
                       "gen_ai.response.finish_reasons": [result2.finishReason],
                       "gen_ai.response.id": responseData.id,
@@ -2254,9 +2240,6 @@ async function generateObject({
                       "ai.response.timestamp": responseData.timestamp.toISOString(),
                       "ai.usage.promptTokens": result2.usage.promptTokens,
                       "ai.usage.completionTokens": result2.usage.completionTokens,
-                      // deprecated:
-                      "ai.finishReason": result2.finishReason,
-                      "ai.result.object": { output: () => objectText },
                       // standardized gen-ai llm span attributes:
                       "gen_ai.response.finish_reasons": [result2.finishReason],
                       "gen_ai.response.id": responseData.id,
@@ -2310,12 +2293,7 @@ async function generateObject({
               output: () => JSON.stringify(validationResult.value)
             },
             "ai.usage.promptTokens": usage.promptTokens,
-            "ai.usage.completionTokens": usage.completionTokens,
-            // deprecated:
-            "ai.finishReason": finishReason,
-            "ai.result.object": {
-              output: () => JSON.stringify(validationResult.value)
-            }
+            "ai.usage.completionTokens": usage.completionTokens
           }
         })
       );
@@ -2344,9 +2322,6 @@ var DefaultGenerateObjectResult = class {
     this.experimental_providerMetadata = options.providerMetadata;
     this.response = options.response;
     this.request = options.request;
-    this.rawResponse = {
-      headers: options.response.headers
-    };
     this.logprobs = options.logprobs;
   }
   toJsonResponse(init) {
@@ -2359,7 +2334,6 @@ var DefaultGenerateObjectResult = class {
     });
   }
 };
-var experimental_generateObject = generateObject;
 
 // core/generate-object/stream-object.ts
 var import_provider_utils6 = require("@ai-sdk/provider-utils");
@@ -2722,7 +2696,6 @@ var DefaultStreamObjectResult = class {
     generateId: generateId3
   }) {
     this.warnings = warnings;
-    this.rawResponse = rawResponse;
     this.outputStrategy = outputStrategy;
     this.request = Promise.resolve(request);
     this.objectPromise = new DelayedPromise();
@@ -2855,9 +2828,6 @@ var DefaultStreamObjectResult = class {
                   "ai.response.timestamp": response.timestamp.toISOString(),
                   "ai.usage.promptTokens": finalUsage.promptTokens,
                   "ai.usage.completionTokens": finalUsage.completionTokens,
-                  // deprecated
-                  "ai.finishReason": finishReason,
-                  "ai.result.object": { output: () => JSON.stringify(object) },
                   // standardized gen-ai llm span attributes:
                   "gen_ai.response.finish_reasons": [finishReason],
                   "gen_ai.response.id": response.id,
@@ -2876,9 +2846,7 @@ var DefaultStreamObjectResult = class {
                   "ai.usage.completionTokens": finalUsage.completionTokens,
                   "ai.response.object": {
                     output: () => JSON.stringify(object)
-                  },
-                  // deprecated
-                  "ai.result.object": { output: () => JSON.stringify(object) }
+                  }
                 }
               })
             );
@@ -2886,7 +2854,6 @@ var DefaultStreamObjectResult = class {
               usage: finalUsage,
               object,
               error,
-              rawResponse,
               response: {
                 ...response,
                 headers: rawResponse == null ? void 0 : rawResponse.headers
@@ -2979,7 +2946,6 @@ var DefaultStreamObjectResult = class {
     });
   }
 };
-var experimental_streamObject = streamObject;
 
 // core/generate-text/generate-text.ts
 var import_provider_utils8 = require("@ai-sdk/provider-utils");
@@ -3010,12 +2976,6 @@ var InvalidToolArgumentsError = class extends import_provider11.AISDKError {
   static isInstance(error) {
     return import_provider11.AISDKError.hasMarker(error, marker8);
   }
-  /**
-   * @deprecated use `isInstance` instead
-   */
-  static isInvalidToolArgumentsError(error) {
-    return error instanceof Error && error.name === name8 && typeof error.toolName === "string" && typeof error.toolArgs === "string";
-  }
 };
 _a8 = symbol8;
 
@@ -3191,9 +3151,7 @@ async function generateText({
   maxRetries,
   abortSignal,
   headers,
-  maxAutomaticRoundtrips = 0,
-  maxToolRoundtrips = maxAutomaticRoundtrips,
-  maxSteps = maxToolRoundtrips != null ? maxToolRoundtrips + 1 : 1,
+  maxSteps = 1,
   experimental_continuationSteps,
   experimental_continueSteps: continueSteps = experimental_continuationSteps != null ? experimental_continuationSteps : false,
   experimental_telemetry: telemetry,
@@ -3347,14 +3305,6 @@ async function generateText({
                     "ai.response.timestamp": responseData.timestamp.toISOString(),
                     "ai.usage.promptTokens": result.usage.promptTokens,
                     "ai.usage.completionTokens": result.usage.completionTokens,
-                    // deprecated:
-                    "ai.finishReason": result.finishReason,
-                    "ai.result.text": {
-                      output: () => result.text
-                    },
-                    "ai.result.toolCalls": {
-                      output: () => JSON.stringify(result.toolCalls)
-                    },
                     // standardized gen-ai llm span attributes:
                     "gen_ai.response.finish_reasons": [result.finishReason],
                     "gen_ai.response.id": responseData.id,
@@ -3457,15 +3407,7 @@ async function generateText({
               output: () => JSON.stringify(currentModelResponse.toolCalls)
             },
             "ai.usage.promptTokens": currentModelResponse.usage.promptTokens,
-            "ai.usage.completionTokens": currentModelResponse.usage.completionTokens,
-            // deprecated:
-            "ai.finishReason": currentModelResponse.finishReason,
-            "ai.result.text": {
-              output: () => currentModelResponse.text
-            },
-            "ai.result.toolCalls": {
-              output: () => JSON.stringify(currentModelResponse.toolCalls)
-            }
+            "ai.usage.completionTokens": currentModelResponse.usage.completionTokens
           }
         })
       );
@@ -3561,19 +3503,15 @@ var DefaultGenerateTextResult = class {
     this.request = options.request;
     this.response = options.response;
     this.responseMessages = options.responseMessages;
-    this.roundtrips = options.steps;
     this.steps = options.steps;
     this.experimental_providerMetadata = options.providerMetadata;
-    this.rawResponse = {
-      headers: options.response.headers
-    };
     this.logprobs = options.logprobs;
   }
 };
-var experimental_generateText = generateText;
 
 // core/generate-text/stream-text.ts
 var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_ui_utils6 = require("@ai-sdk/ui-utils");
 
 // core/util/create-stitchable-stream.ts
 function createStitchableStream() {
@@ -3930,8 +3868,7 @@ async function streamText({
   maxRetries,
   abortSignal,
   headers,
-  maxToolRoundtrips = 0,
-  maxSteps = maxToolRoundtrips != null ? maxToolRoundtrips + 1 : 1,
+  maxSteps = 1,
   experimental_continueSteps: continueSteps = false,
   experimental_telemetry: telemetry,
   experimental_providerMetadata: providerMetadata,
@@ -4224,14 +4161,10 @@ var DefaultStreamTextResult = class {
                 const msToFirstChunk = now2() - startTimestamp;
                 stepFirstChunk = false;
                 doStreamSpan2.addEvent("ai.stream.firstChunk", {
-                  "ai.response.msToFirstChunk": msToFirstChunk,
-                  // deprecated:
-                  "ai.stream.msToFirstChunk": msToFirstChunk
+                  "ai.response.msToFirstChunk": msToFirstChunk
                 });
                 doStreamSpan2.setAttributes({
-                  "ai.response.msToFirstChunk": msToFirstChunk,
-                  // deprecated:
-                  "ai.stream.msToFirstChunk": msToFirstChunk
+                  "ai.response.msToFirstChunk": msToFirstChunk
                 });
               }
               if (chunk.type === "text-delta" && chunk.textDelta.length === 0) {
@@ -4356,12 +4289,6 @@ var DefaultStreamTextResult = class {
                       "ai.response.timestamp": stepResponse.timestamp.toISOString(),
                       "ai.usage.promptTokens": stepUsage.promptTokens,
                       "ai.usage.completionTokens": stepUsage.completionTokens,
-                      // deprecated
-                      "ai.finishReason": stepFinishReason,
-                      "ai.result.text": { output: () => stepText },
-                      "ai.result.toolCalls": {
-                        output: () => stepToolCallsJson
-                      },
                       // standardized gen-ai llm span attributes:
                       "gen_ai.response.finish_reasons": [stepFinishReason],
                       "gen_ai.response.id": stepResponse.id,
@@ -4416,7 +4343,6 @@ var DefaultStreamTextResult = class {
                 warnings: self.warnings,
                 logprobs: stepLogProbs,
                 request: stepRequest,
-                rawResponse: self.rawResponse,
                 response: {
                   ...stepResponse,
                   headers: (_a11 = self.rawResponse) == null ? void 0 : _a11.headers,
@@ -4477,13 +4403,7 @@ var DefaultStreamTextResult = class {
                         output: () => stepToolCallsJson
                       },
                       "ai.usage.promptTokens": combinedUsage.promptTokens,
-                      "ai.usage.completionTokens": combinedUsage.completionTokens,
-                      // deprecated
-                      "ai.finishReason": stepFinishReason,
-                      "ai.result.text": { output: () => fullStepText },
-                      "ai.result.toolCalls": {
-                        output: () => stepToolCallsJson
-                      }
+                      "ai.usage.completionTokens": combinedUsage.completionTokens
                     }
                   })
                 );
@@ -4513,7 +4433,6 @@ var DefaultStreamTextResult = class {
                   // The type exposed to the users will be correctly inferred.
                   toolResults: stepToolResults,
                   request: stepRequest,
-                  rawResponse,
                   response: {
                     ...stepResponse,
                     headers: rawResponse == null ? void 0 : rawResponse.headers,
@@ -4577,37 +4496,18 @@ var DefaultStreamTextResult = class {
       }
     });
   }
-  toAIStream(callbacks = {}) {
-    return this.toDataStreamInternal({ callbacks });
-  }
   toDataStreamInternal({
-    callbacks = {},
     getErrorMessage: getErrorMessage3 = () => "",
     // mask error messages for safety by default
     sendUsage = true
   } = {}) {
     let aggregatedResponse = "";
     const callbackTransformer = new TransformStream({
-      async start() {
-        if (callbacks.onStart)
-          await callbacks.onStart();
-      },
       async transform(chunk, controller) {
         controller.enqueue(chunk);
         if (chunk.type === "text-delta") {
-          const textDelta = chunk.textDelta;
-          aggregatedResponse += textDelta;
-          if (callbacks.onToken)
-            await callbacks.onToken(textDelta);
-          if (callbacks.onText)
-            await callbacks.onText(textDelta);
+          aggregatedResponse += chunk.textDelta;
         }
-      },
-      async flush() {
-        if (callbacks.onCompletion)
-          await callbacks.onCompletion(aggregatedResponse);
-        if (callbacks.onFinal)
-          await callbacks.onFinal(aggregatedResponse);
       }
     });
     const streamPartsTransformer = new TransformStream({
@@ -4615,12 +4515,12 @@ var DefaultStreamTextResult = class {
         const chunkType = chunk.type;
         switch (chunkType) {
           case "text-delta": {
-            controller.enqueue((0, import_ui_utils9.formatStreamPart)("text", chunk.textDelta));
+            controller.enqueue((0, import_ui_utils6.formatStreamPart)("text", chunk.textDelta));
             break;
           }
           case "tool-call-streaming-start": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_call_streaming_start", {
+              (0, import_ui_utils6.formatStreamPart)("tool_call_streaming_start", {
                 toolCallId: chunk.toolCallId,
                 toolName: chunk.toolName
               })
@@ -4629,7 +4529,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-call-delta": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_call_delta", {
+              (0, import_ui_utils6.formatStreamPart)("tool_call_delta", {
                 toolCallId: chunk.toolCallId,
                 argsTextDelta: chunk.argsTextDelta
               })
@@ -4638,7 +4538,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-call": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_call", {
+              (0, import_ui_utils6.formatStreamPart)("tool_call", {
                 toolCallId: chunk.toolCallId,
                 toolName: chunk.toolName,
                 args: chunk.args
@@ -4648,7 +4548,7 @@ var DefaultStreamTextResult = class {
           }
           case "tool-result": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("tool_result", {
+              (0, import_ui_utils6.formatStreamPart)("tool_result", {
                 toolCallId: chunk.toolCallId,
                 result: chunk.result
               })
@@ -4657,13 +4557,13 @@ var DefaultStreamTextResult = class {
           }
           case "error": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("error", getErrorMessage3(chunk.error))
+              (0, import_ui_utils6.formatStreamPart)("error", getErrorMessage3(chunk.error))
             );
             break;
           }
           case "step-finish": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("finish_step", {
+              (0, import_ui_utils6.formatStreamPart)("finish_step", {
                 finishReason: chunk.finishReason,
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
@@ -4676,7 +4576,7 @@ var DefaultStreamTextResult = class {
           }
           case "finish": {
             controller.enqueue(
-              (0, import_ui_utils9.formatStreamPart)("finish_message", {
+              (0, import_ui_utils6.formatStreamPart)("finish_message", {
                 finishReason: chunk.finishReason,
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
@@ -4695,9 +4595,6 @@ var DefaultStreamTextResult = class {
     });
     return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
   }
-  pipeAIStreamToResponse(response, init) {
-    return this.pipeDataStreamToResponse(response, init);
-  }
   pipeDataStreamToResponse(response, options) {
     const init = options == null ? void 0 : "init" in options ? options.init : {
       headers: "headers" in options ? options.headers : void 0,
@@ -4729,9 +4626,6 @@ var DefaultStreamTextResult = class {
       stream: this.textStream.pipeThrough(new TextEncoderStream())
     });
   }
-  toAIStreamResponse(options) {
-    return this.toDataStreamResponse(options);
-  }
   toDataStream(options) {
     const stream = this.toDataStreamInternal({
       getErrorMessage: options == null ? void 0 : options.getErrorMessage,
@@ -4771,7 +4665,6 @@ var DefaultStreamTextResult = class {
     });
   }
 };
-var experimental_streamText = streamText;
 
 // core/middleware/wrap-language-model.ts
 var experimental_wrapLanguageModel = ({
@@ -4958,123 +4851,8 @@ function magnitude(vector) {
   return Math.sqrt(dotProduct(vector, vector));
 }
 
-// streams/ai-stream.ts
-var import_eventsource_parser = require("eventsource-parser");
-function createEventStreamTransformer(customParser) {
-  const textDecoder = new TextDecoder();
-  let eventSourceParser;
-  return new TransformStream({
-    async start(controller) {
-      eventSourceParser = (0, import_eventsource_parser.createParser)(
-        (event) => {
-          if ("data" in event && event.type === "event" && event.data === "[DONE]" || // Replicate doesn't send [DONE] but does send a 'done' event
-          // @see https://replicate.com/docs/streaming
-          event.event === "done") {
-            controller.terminate();
-            return;
-          }
-          if ("data" in event) {
-            const parsedMessage = customParser ? customParser(event.data, {
-              event: event.event
-            }) : event.data;
-            if (parsedMessage)
-              controller.enqueue(parsedMessage);
-          }
-        }
-      );
-    },
-    transform(chunk) {
-      eventSourceParser.feed(textDecoder.decode(chunk));
-    }
-  });
-}
-function createCallbacksTransformer(cb) {
-  const textEncoder = new TextEncoder();
-  let aggregatedResponse = "";
-  const callbacks = cb || {};
-  return new TransformStream({
-    async start() {
-      if (callbacks.onStart)
-        await callbacks.onStart();
-    },
-    async transform(message, controller) {
-      const content = typeof message === "string" ? message : message.content;
-      controller.enqueue(textEncoder.encode(content));
-      aggregatedResponse += content;
-      if (callbacks.onToken)
-        await callbacks.onToken(content);
-      if (callbacks.onText && typeof message === "string") {
-        await callbacks.onText(message);
-      }
-    },
-    async flush() {
-      if (callbacks.onCompletion) {
-        await callbacks.onCompletion(aggregatedResponse);
-      }
-    }
-  });
-}
-function trimStartOfStreamHelper() {
-  let isStreamStart = true;
-  return (text) => {
-    if (isStreamStart) {
-      text = text.trimStart();
-      if (text)
-        isStreamStart = false;
-    }
-    return text;
-  };
-}
-function AIStream(response, customParser, callbacks) {
-  if (!response.ok) {
-    if (response.body) {
-      const reader = response.body.getReader();
-      return new ReadableStream({
-        async start(controller) {
-          const { done, value } = await reader.read();
-          if (!done) {
-            const errorText = new TextDecoder().decode(value);
-            controller.error(new Error(`Response error: ${errorText}`));
-          }
-        }
-      });
-    } else {
-      return new ReadableStream({
-        start(controller) {
-          controller.error(new Error("Response error: No response body"));
-        }
-      });
-    }
-  }
-  const responseBodyStream = response.body || createEmptyReadableStream();
-  return responseBodyStream.pipeThrough(createEventStreamTransformer(customParser)).pipeThrough(createCallbacksTransformer(callbacks));
-}
-function createEmptyReadableStream() {
-  return new ReadableStream({
-    start(controller) {
-      controller.close();
-    }
-  });
-}
-function readableFromAsyncIterable(iterable) {
-  let it = iterable[Symbol.asyncIterator]();
-  return new ReadableStream({
-    async pull(controller) {
-      const { done, value } = await it.next();
-      if (done)
-        controller.close();
-      else
-        controller.enqueue(value);
-    },
-    async cancel(reason) {
-      var _a11;
-      await ((_a11 = it.return) == null ? void 0 : _a11.call(it, reason));
-    }
-  });
-}
-
 // streams/assistant-response.ts
-var import_ui_utils7 = require("@ai-sdk/ui-utils");
+var import_ui_utils8 = require("@ai-sdk/ui-utils");
 function AssistantResponse({ threadId, messageId }, process2) {
   const stream = new ReadableStream({
     async start(controller) {
@@ -5082,17 +4860,17 @@ function AssistantResponse({ threadId, messageId }, process2) {
       const textEncoder = new TextEncoder();
       const sendMessage = (message) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils7.formatStreamPart)("assistant_message", message))
+          textEncoder.encode((0, import_ui_utils8.formatStreamPart)("assistant_message", message))
         );
       };
       const sendDataMessage = (message) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils7.formatStreamPart)("data_message", message))
+          textEncoder.encode((0, import_ui_utils8.formatStreamPart)("data_message", message))
         );
       };
       const sendError = (errorMessage) => {
         controller.enqueue(
-          textEncoder.encode((0, import_ui_utils7.formatStreamPart)("error", errorMessage))
+          textEncoder.encode((0, import_ui_utils8.formatStreamPart)("error", errorMessage))
         );
       };
       const forwardStream = async (stream2) => {
@@ -5103,7 +4881,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
             case "thread.message.created": {
               controller.enqueue(
                 textEncoder.encode(
-                  (0, import_ui_utils7.formatStreamPart)("assistant_message", {
+                  (0, import_ui_utils8.formatStreamPart)("assistant_message", {
                     id: value.data.id,
                     role: "assistant",
                     content: [{ type: "text", text: { value: "" } }]
@@ -5117,7 +4895,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_utils7.formatStreamPart)("text", content.text.value)
+                    (0, import_ui_utils8.formatStreamPart)("text", content.text.value)
                   )
                 );
               }
@@ -5134,7 +4912,7 @@ function AssistantResponse({ threadId, messageId }, process2) {
       };
       controller.enqueue(
         textEncoder.encode(
-          (0, import_ui_utils7.formatStreamPart)("assistant_control_data", {
+          (0, import_ui_utils8.formatStreamPart)("assistant_control_data", {
             threadId,
             messageId
           })
@@ -5176,14 +4954,40 @@ __export(langchain_adapter_exports, {
   toDataStreamResponse: () => toDataStreamResponse
 });
 
+// streams/stream-callbacks.ts
+function createCallbacksTransformer(callbacks = {}) {
+  const textEncoder = new TextEncoder();
+  let aggregatedResponse = "";
+  return new TransformStream({
+    async start() {
+      if (callbacks.onStart)
+        await callbacks.onStart();
+    },
+    async transform(message, controller) {
+      controller.enqueue(textEncoder.encode(message));
+      aggregatedResponse += message;
+      if (callbacks.onToken)
+        await callbacks.onToken(message);
+      if (callbacks.onText && typeof message === "string") {
+        await callbacks.onText(message);
+      }
+    },
+    async flush() {
+      if (callbacks.onCompletion) {
+        await callbacks.onCompletion(aggregatedResponse);
+      }
+    }
+  });
+}
+
 // streams/stream-data.ts
-var import_ui_utils8 = require("@ai-sdk/ui-utils");
+var import_ui_utils9 = require("@ai-sdk/ui-utils");
 
 // util/constants.ts
 var HANGING_STREAM_WARNING_TIME_MS = 15 * 1e3;
 
 // streams/stream-data.ts
-var StreamData2 = class {
+var StreamData = class {
   constructor() {
     this.encoder = new TextEncoder();
     this.controller = null;
@@ -5229,7 +5033,7 @@ var StreamData2 = class {
       throw new Error("Stream controller is not initialized.");
     }
     this.controller.enqueue(
-      this.encoder.encode((0, import_ui_utils8.formatStreamPart)("data", [value]))
+      this.encoder.encode((0, import_ui_utils9.formatStreamPart)("data", [value]))
     );
   }
   appendMessageAnnotation(value) {
@@ -5240,7 +5044,7 @@ var StreamData2 = class {
       throw new Error("Stream controller is not initialized.");
     }
     this.controller.enqueue(
-      this.encoder.encode((0, import_ui_utils8.formatStreamPart)("message_annotations", [value]))
+      this.encoder.encode((0, import_ui_utils9.formatStreamPart)("message_annotations", [value]))
     );
   }
 };
@@ -5250,11 +5054,11 @@ function createStreamDataTransformer() {
   return new TransformStream({
     transform: async (chunk, controller) => {
       const message = decoder.decode(chunk);
-      controller.enqueue(encoder.encode((0, import_ui_utils8.formatStreamPart)("text", message)));
+      controller.enqueue(encoder.encode((0, import_ui_utils9.formatStreamPart)("text", message)));
     }
   });
 }
-var experimental_StreamData = class extends StreamData2 {
+var experimental_StreamData = class extends StreamData {
 };
 
 // streams/langchain-adapter.ts
@@ -5318,8 +5122,16 @@ __export(llamaindex_adapter_exports, {
   toDataStream: () => toDataStream2,
   toDataStreamResponse: () => toDataStreamResponse2
 });
+var import_provider_utils10 = require("@ai-sdk/provider-utils");
 function toDataStream2(stream, callbacks) {
-  return toReadableStream(stream).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
+  const trimStart = trimStartOfStream();
+  return (0, import_provider_utils10.convertAsyncIteratorToReadableStream)(stream[Symbol.asyncIterator]()).pipeThrough(
+    new TransformStream({
+      async transform(message, controller) {
+        controller.enqueue(trimStart(message.delta));
+      }
+    })
+  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 function toDataStreamResponse2(stream, options = {}) {
   var _a11;
@@ -5335,73 +5147,20 @@ function toDataStreamResponse2(stream, options = {}) {
     })
   });
 }
-function toReadableStream(res) {
-  const it = res[Symbol.asyncIterator]();
-  const trimStartOfStream = trimStartOfStreamHelper();
-  return new ReadableStream({
-    async pull(controller) {
-      var _a11;
-      const { value, done } = await it.next();
-      if (done) {
-        controller.close();
-        return;
-      }
-      const text = trimStartOfStream((_a11 = value.delta) != null ? _a11 : "");
-      if (text) {
-        controller.enqueue(text);
-      }
+function trimStartOfStream() {
+  let isStreamStart = true;
+  return (text) => {
+    if (isStreamStart) {
+      text = text.trimStart();
+      if (text)
+        isStreamStart = false;
     }
-  });
-}
-
-// streams/stream-to-response.ts
-function streamToResponse(res, response, init, data) {
-  var _a11;
-  response.writeHead((_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200, {
-    "Content-Type": "text/plain; charset=utf-8",
-    ...init == null ? void 0 : init.headers
-  });
-  let processedStream = res;
-  if (data) {
-    processedStream = mergeStreams(data.stream, res);
-  }
-  const reader = processedStream.getReader();
-  function read() {
-    reader.read().then(({ done, value }) => {
-      if (done) {
-        response.end();
-        return;
-      }
-      response.write(value);
-      read();
-    });
-  }
-  read();
+    return text;
+  };
 }
-
-// streams/streaming-text-response.ts
-var StreamingTextResponse = class extends Response {
-  constructor(res, init, data) {
-    let processedStream = res;
-    if (data) {
-      processedStream = mergeStreams(data.stream, res);
-    }
-    super(processedStream, {
-      ...init,
-      status: 200,
-      headers: prepareResponseHeaders(init, {
-        contentType: "text/plain; charset=utf-8"
-      })
-    });
-  }
-};
-
-// streams/index.ts
-var generateId2 = import_provider_utils10.generateId;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AISDKError,
-  AIStream,
   APICallError,
   AssistantResponse,
   DownloadError,
@@ -5424,13 +5183,10 @@ var generateId2 = import_provider_utils10.generateId;
   NoSuchToolError,
   RetryError,
   StreamData,
-  StreamingTextResponse,
   TypeValidationError,
   UnsupportedFunctionalityError,
   convertToCoreMessages,
   cosineSimilarity,
-  createCallbacksTransformer,
-  createEventStreamTransformer,
   createStreamDataTransformer,
   embed,
   embedMany,
@@ -5439,10 +5195,6 @@ var generateId2 = import_provider_utils10.generateId;
   experimental_createModelRegistry,
   experimental_createProviderRegistry,
   experimental_customProvider,
-  experimental_generateObject,
-  experimental_generateText,
-  experimental_streamObject,
-  experimental_streamText,
   experimental_wrapLanguageModel,
   formatStreamPart,
   generateId,
@@ -5452,11 +5204,8 @@ var generateId2 = import_provider_utils10.generateId;
   parseStreamPart,
   processDataProtocolResponse,
   readDataStream,
-  readableFromAsyncIterable,
   streamObject,
   streamText,
-  streamToResponse,
-  tool,
-  trimStartOfStreamHelper
+  tool
 });
 //# sourceMappingURL=index.js.map