ai 4.0.7 → 4.0.9

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # ai
 
+## 4.0.9
+
+### Patch Changes
+
+- fda9695: feat (ai/core): reworked data stream management
+
+## 4.0.8
+
+### Patch Changes
+
+- a803d76: feat (ai/core): pass toolCallId option into tool execute function
+
 ## 4.0.7
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,12 +1,48 @@
-import { ToolInvocation, Attachment, Schema, DeepPartial, JSONValue as JSONValue$1, AssistantMessage, DataMessage } from '@ai-sdk/ui-utils';
+import { DataStreamString, ToolInvocation, Attachment, Schema, DeepPartial, JSONValue as JSONValue$1, AssistantMessage, DataMessage } from '@ai-sdk/ui-utils';
 export { AssistantMessage, AssistantStatus, Attachment, ChatRequest, ChatRequestOptions, CreateMessage, DataMessage, DataStreamPart, DeepPartial, IdGenerator, JSONValue, Message, RequestOptions, Schema, ToolInvocation, UseAssistantOptions, formatAssistantStreamPart, formatDataStreamPart, jsonSchema, parseAssistantStreamPart, parseDataStreamPart, processDataStream, processTextStream } from '@ai-sdk/ui-utils';
 export { ToolCall as CoreToolCall, ToolResult as CoreToolResult, generateId } from '@ai-sdk/provider-utils';
-import { AttributeValue, Tracer } from '@opentelemetry/api';
-import { EmbeddingModelV1, EmbeddingModelV1Embedding, LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning, LanguageModelV1ProviderMetadata, JSONValue, LanguageModelV1CallOptions, NoSuchModelError, AISDKError } from '@ai-sdk/provider';
+import { JSONValue, EmbeddingModelV1, EmbeddingModelV1Embedding, LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning, LanguageModelV1ProviderMetadata, LanguageModelV1CallOptions, NoSuchModelError, AISDKError } from '@ai-sdk/provider';
 export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, LanguageModelV1, LanguageModelV1CallOptions, LanguageModelV1Prompt, LanguageModelV1StreamPart, LoadAPIKeyError, NoContentGeneratedError, NoSuchModelError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
+import { ServerResponse } from 'node:http';
+import { AttributeValue, Tracer } from '@opentelemetry/api';
 import { z } from 'zod';
-import { ServerResponse } from 'http';
-import { ServerResponse as ServerResponse$1 } from 'node:http';
+import { ServerResponse as ServerResponse$1 } from 'http';
+
+interface DataStreamWriter {
+    /**
+     * Appends a data part to the stream.
+     */
+    writeData(value: JSONValue): void;
+    /**
+     * Appends a message annotation to the stream.
+     */
+    writeMessageAnnotation(value: JSONValue): void;
+    /**
+     * Merges the contents of another stream to this stream.
+     */
+    merge(stream: ReadableStream<DataStreamString>): void;
+    /**
+     * Error handler that is used by the data stream writer.
+     * This is intended for forwarding when merging streams
+     * to prevent duplicated error masking.
+     */
+    onError: ((error: unknown) => string) | undefined;
+}
+
+declare function createDataStream({ execute, onError, }: {
+    execute: (dataStream: DataStreamWriter) => Promise<void> | void;
+    onError?: (error: unknown) => string;
+}): ReadableStream<DataStreamString>;
+
+declare function createDataStreamResponse({ status, statusText, headers, execute, onError, }: ResponseInit & {
+    execute: (dataStream: DataStreamWriter) => Promise<void> | void;
+    onError?: (error: unknown) => string;
+}): Response;
+
+declare function pipeDataStreamToResponse(response: ServerResponse, { status, statusText, headers, execute, onError, }: ResponseInit & {
+    execute: (writer: DataStreamWriter) => Promise<void> | void;
+    onError?: (error: unknown) => string;
+}): void;
 
 /**
  * Telemetry configuration.
@@ -943,7 +979,7 @@ interface StreamObjectResult<PARTIAL, RESULT, ELEMENT_STREAM> {
     @param response A Node.js response-like object (ServerResponse).
     @param init Optional headers, status code, and status text.
        */
-    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): void;
     /**
     Creates a simple text stream response.
     The response has a `Content-Type` header set to `text/plain; charset=utf-8`.
@@ -1178,6 +1214,21 @@ Callback that is called when the LLM response and the final object validation ar
 
 type Parameters = z.ZodTypeAny | Schema<any>;
 type inferParameters<PARAMETERS extends Parameters> = PARAMETERS extends Schema<any> ? PARAMETERS['_type'] : PARAMETERS extends z.ZodTypeAny ? z.infer<PARAMETERS> : never;
+interface ToolExecutionOptions {
+    /**
+     * The ID of the tool call. You can use it e.g. when sending tool-call related information with stream data.
+     */
+    toolCallId: string;
+    /**
+     * Messages that were sent to the language model to initiate the response that contained the tool call.
+     * The messages **do not** include the system prompt nor the assistant response that contained the tool call.
+     */
+    messages: CoreMessage[];
+    /**
+     * An optional abort signal that indicates that the overall operation should be aborted.
+     */
+    abortSignal?: AbortSignal;
+}
 /**
 A tool contains the description and the schema of the input that the tool expects.
 This enables the language model to generate the input.
@@ -1202,17 +1253,7 @@ type CoreTool<PARAMETERS extends Parameters = any, RESULT = any> = {
   @args is the input of the tool call.
   @options.abortSignal is a signal that can be used to abort the tool call.
      */
-    execute?: (args: inferParameters<PARAMETERS>, options: {
-        /**
-         * Messages that were sent to the language model to initiate the response that contained the tool call.
-         * The messages **do not** include the system prompt nor the assistant response that contained the tool call.
-         */
-        messages: CoreMessage[];
-        /**
-         * An optional abort signal that indicates that the overall operation should be aborted.
-         */
-        abortSignal?: AbortSignal;
-    }) => PromiseLike<RESULT>;
+    execute?: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
 } & ({
     /**
 Function tool.
@@ -1240,15 +1281,9 @@ The arguments for configuring the tool. Must match the expected arguments define
 Helper function for inferring the execute args of a tool.
  */
 declare function tool<PARAMETERS extends Parameters, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: inferParameters<PARAMETERS>, options: {
-        messages: CoreMessage[];
-        abortSignal?: AbortSignal;
-    }) => PromiseLike<RESULT>;
+    execute: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
 }): CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: inferParameters<PARAMETERS>, options: {
-        messages: CoreMessage[];
-        abortSignal?: AbortSignal;
-    }) => PromiseLike<RESULT>;
+    execute: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
 };
 declare function tool<PARAMETERS extends Parameters, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
     execute?: undefined;
@@ -1606,6 +1641,8 @@ changing the tool call and result types in the result.
 
 /**
  * A stream wrapper to send custom JSON-encoded data back to the client.
+ *
+ * @deprecated Please use `createDataStream`, `createDataStreamResponse`, and `pipeDataStreamToResponse` instead.
  */
 declare class StreamData {
     private encoder;
@@ -1618,11 +1655,6 @@ declare class StreamData {
     append(value: JSONValue$1): void;
     appendMessageAnnotation(value: JSONValue$1): void;
 }
-/**
- * A TransformStream for LLMs that do not have their own transform stream handlers managing encoding (e.g. OpenAIStream has one for function call handling).
- * This assumes every chunk is a 'text' chunk.
- */
-declare function createStreamDataTransformer(): TransformStream<any, any>;
 
 /**
 A result object for accessing different stream types and additional information.
@@ -1720,6 +1752,12 @@ interface StreamTextResult<TOOLS extends Record<string, CoreTool>> {
         getErrorMessage?: (error: unknown) => string;
         sendUsage?: boolean;
     }): ReadableStream<Uint8Array>;
+    /**
+     * Merges the result as a data stream into another data stream.
+     *
+     * @param dataStream A data stream writer.
+     */
+    mergeIntoDataStream(dataStream: DataStreamWriter): void;
     /**
     Writes data stream output to a Node.js response-like object.
   
@@ -1731,7 +1769,7 @@ interface StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     @param options.getErrorMessage An optional function that converts an error to an error message.
     @param options.sendUsage Whether to send the usage information to the client. Defaults to true.
        */
-    pipeDataStreamToResponse(response: ServerResponse$1, options?: ResponseInit & {
+    pipeDataStreamToResponse(response: ServerResponse, options?: ResponseInit & {
         data?: StreamData;
         getErrorMessage?: (error: unknown) => string;
         sendUsage?: boolean;
@@ -1744,7 +1782,7 @@ interface StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     @param response A Node.js response-like object (ServerResponse).
     @param init Optional headers, status code, and status text.
        */
-    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): void;
     /**
     Converts the result to a streamed response object with a stream data part stream.
     It can be used with the `useChat` and `useCompletion` hooks.
@@ -2256,15 +2294,20 @@ The following streams are supported:
 - `LangChainAIMessageChunk` streams (LangChain `model.stream` output)
 - `string` streams (LangChain `StringOutputParser` output)
  */
-declare function toDataStream$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: StreamCallbacks): ReadableStream<any>;
+declare function toDataStream$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: StreamCallbacks): ReadableStream<Uint8Array>;
 declare function toDataStreamResponse$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, options?: {
     init?: ResponseInit;
     data?: StreamData;
     callbacks?: StreamCallbacks;
 }): Response;
+declare function mergeIntoDataStream$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, options: {
+    dataStream: DataStreamWriter;
+    callbacks?: StreamCallbacks;
+}): void;
 
 declare namespace langchainAdapter {
   export {
+    mergeIntoDataStream$1 as mergeIntoDataStream,
     toDataStream$1 as toDataStream,
     toDataStreamResponse$1 as toDataStreamResponse,
   };
@@ -2273,20 +2316,26 @@ declare namespace langchainAdapter {
 type EngineResponse = {
     delta: string;
 };
-declare function toDataStream(stream: AsyncIterable<EngineResponse>, callbacks?: StreamCallbacks): ReadableStream<any>;
+declare function toDataStream(stream: AsyncIterable<EngineResponse>, callbacks?: StreamCallbacks): ReadableStream<Uint8Array>;
 declare function toDataStreamResponse(stream: AsyncIterable<EngineResponse>, options?: {
     init?: ResponseInit;
     data?: StreamData;
     callbacks?: StreamCallbacks;
 }): Response;
+declare function mergeIntoDataStream(stream: AsyncIterable<EngineResponse>, options: {
+    dataStream: DataStreamWriter;
+    callbacks?: StreamCallbacks;
+}): void;
 
+declare const llamaindexAdapter_mergeIntoDataStream: typeof mergeIntoDataStream;
 declare const llamaindexAdapter_toDataStream: typeof toDataStream;
 declare const llamaindexAdapter_toDataStreamResponse: typeof toDataStreamResponse;
 declare namespace llamaindexAdapter {
   export {
+    llamaindexAdapter_mergeIntoDataStream as mergeIntoDataStream,
     llamaindexAdapter_toDataStream as toDataStream,
     llamaindexAdapter_toDataStreamResponse as toDataStreamResponse,
   };
 }
 
-export { AssistantContent, AssistantResponse, CallWarning, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, ToolCallUnion as CoreToolCallUnion, CoreToolChoice, CoreToolMessage, ToolResultUnion as CoreToolResultUnion, CoreUserMessage, DataContent, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, Experimental_LanguageModelV1Middleware, FilePart, FinishReason, GenerateObjectResult, GenerateTextResult, ImagePart, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidToolArgumentsError, langchainAdapter as LangChainAdapter, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, llamaindexAdapter as LlamaIndexAdapter, LogProbs, MessageConversionError, NoObjectGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, Provider, ProviderMetadata, RetryError, StepResult, StreamData, StreamObjectResult, StreamTextResult, TextPart, TextStreamPart, ToolCallPart, ToolContent, ToolResultPart, UserContent, convertToCoreMessages, cosineSimilarity, createStreamDataTransformer, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, experimental_wrapLanguageModel, generateObject, generateText, streamObject, streamText, tool };
+export { AssistantContent, AssistantResponse, CallWarning, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, ToolCallUnion as CoreToolCallUnion, CoreToolChoice, CoreToolMessage, ToolResultUnion as CoreToolResultUnion, CoreUserMessage, DataContent, DataStreamWriter, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, Experimental_LanguageModelV1Middleware, FilePart, FinishReason, GenerateObjectResult, GenerateTextResult, ImagePart, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidToolArgumentsError, langchainAdapter as LangChainAdapter, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, llamaindexAdapter as LlamaIndexAdapter, LogProbs, MessageConversionError, NoObjectGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, Provider, ProviderMetadata, RetryError, StepResult, StreamData, StreamObjectResult, StreamTextResult, TextPart, TextStreamPart, ToolCallPart, ToolContent, ToolExecutionOptions, ToolResultPart, UserContent, convertToCoreMessages, cosineSimilarity, createDataStream, createDataStreamResponse, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, experimental_wrapLanguageModel, generateObject, generateText, pipeDataStreamToResponse, streamObject, streamText, tool };

package/dist/index.d.ts

@@ -1,12 +1,48 @@
-import { ToolInvocation, Attachment, Schema, DeepPartial, JSONValue as JSONValue$1, AssistantMessage, DataMessage } from '@ai-sdk/ui-utils';
+import { DataStreamString, ToolInvocation, Attachment, Schema, DeepPartial, JSONValue as JSONValue$1, AssistantMessage, DataMessage } from '@ai-sdk/ui-utils';
 export { AssistantMessage, AssistantStatus, Attachment, ChatRequest, ChatRequestOptions, CreateMessage, DataMessage, DataStreamPart, DeepPartial, IdGenerator, JSONValue, Message, RequestOptions, Schema, ToolInvocation, UseAssistantOptions, formatAssistantStreamPart, formatDataStreamPart, jsonSchema, parseAssistantStreamPart, parseDataStreamPart, processDataStream, processTextStream } from '@ai-sdk/ui-utils';
 export { ToolCall as CoreToolCall, ToolResult as CoreToolResult, generateId } from '@ai-sdk/provider-utils';
-import { AttributeValue, Tracer } from '@opentelemetry/api';
-import { EmbeddingModelV1, EmbeddingModelV1Embedding, LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning, LanguageModelV1ProviderMetadata, JSONValue, LanguageModelV1CallOptions, NoSuchModelError, AISDKError } from '@ai-sdk/provider';
+import { JSONValue, EmbeddingModelV1, EmbeddingModelV1Embedding, LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1LogProbs, LanguageModelV1CallWarning, LanguageModelV1ProviderMetadata, LanguageModelV1CallOptions, NoSuchModelError, AISDKError } from '@ai-sdk/provider';
 export { AISDKError, APICallError, EmptyResponseBodyError, InvalidPromptError, InvalidResponseDataError, JSONParseError, LanguageModelV1, LanguageModelV1CallOptions, LanguageModelV1Prompt, LanguageModelV1StreamPart, LoadAPIKeyError, NoContentGeneratedError, NoSuchModelError, TypeValidationError, UnsupportedFunctionalityError } from '@ai-sdk/provider';
+import { ServerResponse } from 'node:http';
+import { AttributeValue, Tracer } from '@opentelemetry/api';
 import { z } from 'zod';
-import { ServerResponse } from 'http';
-import { ServerResponse as ServerResponse$1 } from 'node:http';
+import { ServerResponse as ServerResponse$1 } from 'http';
+
+interface DataStreamWriter {
+    /**
+     * Appends a data part to the stream.
+     */
+    writeData(value: JSONValue): void;
+    /**
+     * Appends a message annotation to the stream.
+     */
+    writeMessageAnnotation(value: JSONValue): void;
+    /**
+     * Merges the contents of another stream to this stream.
+     */
+    merge(stream: ReadableStream<DataStreamString>): void;
+    /**
+     * Error handler that is used by the data stream writer.
+     * This is intended for forwarding when merging streams
+     * to prevent duplicated error masking.
+     */
+    onError: ((error: unknown) => string) | undefined;
+}
+
+declare function createDataStream({ execute, onError, }: {
+    execute: (dataStream: DataStreamWriter) => Promise<void> | void;
+    onError?: (error: unknown) => string;
+}): ReadableStream<DataStreamString>;
+
+declare function createDataStreamResponse({ status, statusText, headers, execute, onError, }: ResponseInit & {
+    execute: (dataStream: DataStreamWriter) => Promise<void> | void;
+    onError?: (error: unknown) => string;
+}): Response;
+
+declare function pipeDataStreamToResponse(response: ServerResponse, { status, statusText, headers, execute, onError, }: ResponseInit & {
+    execute: (writer: DataStreamWriter) => Promise<void> | void;
+    onError?: (error: unknown) => string;
+}): void;
 
 /**
  * Telemetry configuration.
@@ -943,7 +979,7 @@ interface StreamObjectResult<PARTIAL, RESULT, ELEMENT_STREAM> {
     @param response A Node.js response-like object (ServerResponse).
     @param init Optional headers, status code, and status text.
        */
-    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): void;
     /**
     Creates a simple text stream response.
     The response has a `Content-Type` header set to `text/plain; charset=utf-8`.
@@ -1178,6 +1214,21 @@ Callback that is called when the LLM response and the final object validation ar
 
 type Parameters = z.ZodTypeAny | Schema<any>;
 type inferParameters<PARAMETERS extends Parameters> = PARAMETERS extends Schema<any> ? PARAMETERS['_type'] : PARAMETERS extends z.ZodTypeAny ? z.infer<PARAMETERS> : never;
+interface ToolExecutionOptions {
+    /**
+     * The ID of the tool call. You can use it e.g. when sending tool-call related information with stream data.
+     */
+    toolCallId: string;
+    /**
+     * Messages that were sent to the language model to initiate the response that contained the tool call.
+     * The messages **do not** include the system prompt nor the assistant response that contained the tool call.
+     */
+    messages: CoreMessage[];
+    /**
+     * An optional abort signal that indicates that the overall operation should be aborted.
+     */
+    abortSignal?: AbortSignal;
+}
 /**
 A tool contains the description and the schema of the input that the tool expects.
 This enables the language model to generate the input.
@@ -1202,17 +1253,7 @@ type CoreTool<PARAMETERS extends Parameters = any, RESULT = any> = {
   @args is the input of the tool call.
   @options.abortSignal is a signal that can be used to abort the tool call.
      */
-    execute?: (args: inferParameters<PARAMETERS>, options: {
-        /**
-         * Messages that were sent to the language model to initiate the response that contained the tool call.
-         * The messages **do not** include the system prompt nor the assistant response that contained the tool call.
-         */
-        messages: CoreMessage[];
-        /**
-         * An optional abort signal that indicates that the overall operation should be aborted.
-         */
-        abortSignal?: AbortSignal;
-    }) => PromiseLike<RESULT>;
+    execute?: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
 } & ({
     /**
 Function tool.
@@ -1240,15 +1281,9 @@ The arguments for configuring the tool. Must match the expected arguments define
 Helper function for inferring the execute args of a tool.
  */
 declare function tool<PARAMETERS extends Parameters, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: inferParameters<PARAMETERS>, options: {
-        messages: CoreMessage[];
-        abortSignal?: AbortSignal;
-    }) => PromiseLike<RESULT>;
+    execute: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
 }): CoreTool<PARAMETERS, RESULT> & {
-    execute: (args: inferParameters<PARAMETERS>, options: {
-        messages: CoreMessage[];
-        abortSignal?: AbortSignal;
-    }) => PromiseLike<RESULT>;
+    execute: (args: inferParameters<PARAMETERS>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
 };
 declare function tool<PARAMETERS extends Parameters, RESULT>(tool: CoreTool<PARAMETERS, RESULT> & {
     execute?: undefined;
@@ -1606,6 +1641,8 @@ changing the tool call and result types in the result.
 
 /**
  * A stream wrapper to send custom JSON-encoded data back to the client.
+ *
+ * @deprecated Please use `createDataStream`, `createDataStreamResponse`, and `pipeDataStreamToResponse` instead.
  */
 declare class StreamData {
     private encoder;
@@ -1618,11 +1655,6 @@ declare class StreamData {
     append(value: JSONValue$1): void;
     appendMessageAnnotation(value: JSONValue$1): void;
 }
-/**
- * A TransformStream for LLMs that do not have their own transform stream handlers managing encoding (e.g. OpenAIStream has one for function call handling).
- * This assumes every chunk is a 'text' chunk.
- */
-declare function createStreamDataTransformer(): TransformStream<any, any>;
 
 /**
 A result object for accessing different stream types and additional information.
@@ -1720,6 +1752,12 @@ interface StreamTextResult<TOOLS extends Record<string, CoreTool>> {
         getErrorMessage?: (error: unknown) => string;
         sendUsage?: boolean;
     }): ReadableStream<Uint8Array>;
+    /**
+     * Merges the result as a data stream into another data stream.
+     *
+     * @param dataStream A data stream writer.
+     */
+    mergeIntoDataStream(dataStream: DataStreamWriter): void;
     /**
     Writes data stream output to a Node.js response-like object.
   
@@ -1731,7 +1769,7 @@ interface StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     @param options.getErrorMessage An optional function that converts an error to an error message.
     @param options.sendUsage Whether to send the usage information to the client. Defaults to true.
        */
-    pipeDataStreamToResponse(response: ServerResponse$1, options?: ResponseInit & {
+    pipeDataStreamToResponse(response: ServerResponse, options?: ResponseInit & {
         data?: StreamData;
         getErrorMessage?: (error: unknown) => string;
         sendUsage?: boolean;
@@ -1744,7 +1782,7 @@ interface StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     @param response A Node.js response-like object (ServerResponse).
     @param init Optional headers, status code, and status text.
        */
-    pipeTextStreamToResponse(response: ServerResponse$1, init?: ResponseInit): void;
+    pipeTextStreamToResponse(response: ServerResponse, init?: ResponseInit): void;
     /**
     Converts the result to a streamed response object with a stream data part stream.
     It can be used with the `useChat` and `useCompletion` hooks.
@@ -2256,15 +2294,20 @@ The following streams are supported:
 - `LangChainAIMessageChunk` streams (LangChain `model.stream` output)
 - `string` streams (LangChain `StringOutputParser` output)
  */
-declare function toDataStream$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: StreamCallbacks): ReadableStream<any>;
+declare function toDataStream$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, callbacks?: StreamCallbacks): ReadableStream<Uint8Array>;
 declare function toDataStreamResponse$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, options?: {
     init?: ResponseInit;
     data?: StreamData;
     callbacks?: StreamCallbacks;
 }): Response;
+declare function mergeIntoDataStream$1(stream: ReadableStream<LangChainStreamEvent> | ReadableStream<LangChainAIMessageChunk> | ReadableStream<string>, options: {
+    dataStream: DataStreamWriter;
+    callbacks?: StreamCallbacks;
+}): void;
 
 declare namespace langchainAdapter {
   export {
+    mergeIntoDataStream$1 as mergeIntoDataStream,
     toDataStream$1 as toDataStream,
     toDataStreamResponse$1 as toDataStreamResponse,
   };
@@ -2273,20 +2316,26 @@ declare namespace langchainAdapter {
 type EngineResponse = {
     delta: string;
 };
-declare function toDataStream(stream: AsyncIterable<EngineResponse>, callbacks?: StreamCallbacks): ReadableStream<any>;
+declare function toDataStream(stream: AsyncIterable<EngineResponse>, callbacks?: StreamCallbacks): ReadableStream<Uint8Array>;
 declare function toDataStreamResponse(stream: AsyncIterable<EngineResponse>, options?: {
     init?: ResponseInit;
     data?: StreamData;
     callbacks?: StreamCallbacks;
 }): Response;
+declare function mergeIntoDataStream(stream: AsyncIterable<EngineResponse>, options: {
+    dataStream: DataStreamWriter;
+    callbacks?: StreamCallbacks;
+}): void;
 
+declare const llamaindexAdapter_mergeIntoDataStream: typeof mergeIntoDataStream;
 declare const llamaindexAdapter_toDataStream: typeof toDataStream;
 declare const llamaindexAdapter_toDataStreamResponse: typeof toDataStreamResponse;
 declare namespace llamaindexAdapter {
   export {
+    llamaindexAdapter_mergeIntoDataStream as mergeIntoDataStream,
     llamaindexAdapter_toDataStream as toDataStream,
     llamaindexAdapter_toDataStreamResponse as toDataStreamResponse,
   };
 }
 
-export { AssistantContent, AssistantResponse, CallWarning, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, ToolCallUnion as CoreToolCallUnion, CoreToolChoice, CoreToolMessage, ToolResultUnion as CoreToolResultUnion, CoreUserMessage, DataContent, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, Experimental_LanguageModelV1Middleware, FilePart, FinishReason, GenerateObjectResult, GenerateTextResult, ImagePart, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidToolArgumentsError, langchainAdapter as LangChainAdapter, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, llamaindexAdapter as LlamaIndexAdapter, LogProbs, MessageConversionError, NoObjectGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, Provider, ProviderMetadata, RetryError, StepResult, StreamData, StreamObjectResult, StreamTextResult, TextPart, TextStreamPart, ToolCallPart, ToolContent, ToolResultPart, UserContent, convertToCoreMessages, cosineSimilarity, createStreamDataTransformer, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, experimental_wrapLanguageModel, generateObject, generateText, streamObject, streamText, tool };
+export { AssistantContent, AssistantResponse, CallWarning, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, ToolCallUnion as CoreToolCallUnion, CoreToolChoice, CoreToolMessage, ToolResultUnion as CoreToolResultUnion, CoreUserMessage, DataContent, DataStreamWriter, DownloadError, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelUsage, Experimental_LanguageModelV1Middleware, FilePart, FinishReason, GenerateObjectResult, GenerateTextResult, ImagePart, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidToolArgumentsError, langchainAdapter as LangChainAdapter, LanguageModel, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, llamaindexAdapter as LlamaIndexAdapter, LogProbs, MessageConversionError, NoObjectGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, Provider, ProviderMetadata, RetryError, StepResult, StreamData, StreamObjectResult, StreamTextResult, TextPart, TextStreamPart, ToolCallPart, ToolContent, ToolExecutionOptions, ToolResultPart, UserContent, convertToCoreMessages, cosineSimilarity, createDataStream, createDataStreamResponse, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, experimental_wrapLanguageModel, generateObject, generateText, pipeDataStreamToResponse, streamObject, streamText, tool };