ai 3.1.14 → 3.1.16

package/dist/index.d.mts

@@ -39,6 +39,18 @@ Warning from the model provider for this call. The call will proceed, but e.g.
 some settings might not be supported, which can lead to suboptimal results.
 */
 type CallWarning = LanguageModelV1CallWarning;
+/**
+Tool choice for the generation. It supports the following settings:
+
+- `auto` (default): the model can choose whether and which tools to call.
+- `required`: the model must call a tool. It can choose which tool to call.
+- `none`: the model must not call tools
+- `{ type: 'tool', tooName: string (typed) }`: the model must call the specified tool
+ */
+type CoreToolChoice<TOOLS extends Record<string, unknown>> = 'auto' | 'none' | 'required' | {
+    type: 'tool';
+    toolName: keyof TOOLS;
+};
 
 /**
 Embed a value using an embedding model. The type of the value is defined by the embedding model.
@@ -820,7 +832,7 @@ If set and supported by the model, calls will generate deterministic results.
 @returns
 A result object that contains the generated text, the results of the tool calls, and additional information.
  */
-declare function generateText<TOOLS extends Record<string, CoreTool>>({ model, tools, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
+declare function generateText<TOOLS extends Record<string, CoreTool>>({ model, tools, toolChoice, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
     /**
 The language model to use.
      */
@@ -829,6 +841,10 @@ The language model to use.
 The tools that the model can call. The model needs to support calling tools.
 */
     tools?: TOOLS;
+    /**
+The tool choice strategy. Default: 'auto'.
+     */
+    toolChoice?: CoreToolChoice<TOOLS>;
 }): Promise<GenerateTextResult<TOOLS>>;
 /**
 The result of a `generateText` call.
@@ -922,10 +938,13 @@ If set and supported by the model, calls will generate deterministic results.
 @param maxRetries - Maximum number of retries. Set to 0 to disable retries. Default: 2.
 @param abortSignal - An optional abort signal that can be used to cancel the call.
 
+@param onFinish - Callback that is called when the LLM response and all request tool executions
+(for tools that have an `execute` function) are finished.
+
 @return
 A result object for accessing different stream types and additional information.
  */
-declare function streamText<TOOLS extends Record<string, CoreTool>>({ model, tools, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
+declare function streamText<TOOLS extends Record<string, CoreTool>>({ model, tools, toolChoice, system, prompt, messages, maxRetries, abortSignal, onFinish, ...settings }: CallSettings & Prompt & {
     /**
 The language model to use.
      */
@@ -934,6 +953,49 @@ The language model to use.
 The tools that the model can call. The model needs to support calling tools.
     */
     tools?: TOOLS;
+    /**
+The tool choice strategy. Default: 'auto'.
+     */
+    toolChoice?: CoreToolChoice<TOOLS>;
+    /**
+Callback that is called when the LLM response and all request tool executions
+(for tools that have an `execute` function) are finished.
+     */
+    onFinish?: (event: {
+        /**
+  The reason why the generation finished.
+         */
+        finishReason: FinishReason;
+        /**
+  The token usage of the generated response.
+   */
+        usage: TokenUsage;
+        /**
+  The full text that has been generated.
+         */
+        text: string;
+        /**
+  The tool calls that have been executed.
+         */
+        toolCalls?: ToToolCall<TOOLS>[];
+        /**
+  The tool results that have been generated.
+         */
+        toolResults?: ToToolResult<TOOLS>[];
+        /**
+  Optional raw response data.
+         */
+        rawResponse?: {
+            /**
+    Response headers.
+             */
+            headers?: Record<string, string>;
+        };
+        /**
+  Warnings from the model provider (e.g. unsupported settings).
+         */
+        warnings?: CallWarning[];
+    }) => Promise<void> | void;
 }): Promise<StreamTextResult<TOOLS>>;
 type TextStreamPart<TOOLS extends Record<string, CoreTool>> = {
     type: 'text-delta';
@@ -960,12 +1022,13 @@ A result object for accessing different stream types and additional information.
  */
 declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     private originalStream;
+    private onFinish?;
     /**
   Warnings from the model provider (e.g. unsupported settings).
      */
     readonly warnings: CallWarning[] | undefined;
     /**
-  The token usage of the generated text. Resolved when the response is finished.
+  The token usage of the generated response. Resolved when the response is finished.
      */
     readonly usage: Promise<TokenUsage>;
     /**
@@ -973,20 +1036,33 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
      */
     readonly finishReason: Promise<FinishReason>;
     /**
+  The full text that has been generated. Resolved when the response is finished.
+     */
+    readonly text: Promise<string>;
+    /**
+  The tool calls that have been executed. Resolved when the response is finished.
+     */
+    readonly toolCalls: Promise<ToToolCall<TOOLS>[]>;
+    /**
+  The tool results that have been generated. Resolved when the all tool executions are finished.
+     */
+    readonly toolResults: Promise<ToToolResult<TOOLS>[]>;
+    /**
   Optional raw response data.
      */
-    rawResponse?: {
+    readonly rawResponse?: {
         /**
     Response headers.
          */
         headers?: Record<string, string>;
     };
-    constructor({ stream, warnings, rawResponse, }: {
+    constructor({ stream, warnings, rawResponse, onFinish, }: {
         stream: ReadableStream<TextStreamPart<TOOLS>>;
         warnings: CallWarning[] | undefined;
         rawResponse?: {
             headers?: Record<string, string>;
         };
+        onFinish?: Parameters<typeof streamText>[0]['onFinish'];
     });
     /**
   Split out a new stream from the original stream.
@@ -2268,4 +2344,4 @@ declare class StreamingTextResponse extends Response {
     constructor(res: ReadableStream, init?: ResponseInit, data?: StreamData);
 }
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, AssistantStatus, CallWarning, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolMessage, CoreUserMessage, CreateMessage, DataContent, DataMessage, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, Message$1 as Message, MistralStream, ObjectStreamPart, ObjectStreamPartInput, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamData, StreamObjectResult, StreamPart, StreamString, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, Tool, ToolCall, ToolCallHandler, ToolCallPart, ToolCallPayload, ToolChoice, ToolContent, ToolInvocation, ToolResultPart, UseAssistantOptions, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, generateObject, generateText, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, AssistantStatus, CallWarning, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, CreateMessage, DataContent, DataMessage, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, Message$1 as Message, MistralStream, ObjectStreamPart, ObjectStreamPartInput, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamData, StreamObjectResult, StreamPart, StreamString, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, Tool, ToolCall, ToolCallHandler, ToolCallPart, ToolCallPayload, ToolChoice, ToolContent, ToolInvocation, ToolResultPart, UseAssistantOptions, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, generateObject, generateText, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.d.ts

@@ -39,6 +39,18 @@ Warning from the model provider for this call. The call will proceed, but e.g.
 some settings might not be supported, which can lead to suboptimal results.
 */
 type CallWarning = LanguageModelV1CallWarning;
+/**
+Tool choice for the generation. It supports the following settings:
+
+- `auto` (default): the model can choose whether and which tools to call.
+- `required`: the model must call a tool. It can choose which tool to call.
+- `none`: the model must not call tools
+- `{ type: 'tool', tooName: string (typed) }`: the model must call the specified tool
+ */
+type CoreToolChoice<TOOLS extends Record<string, unknown>> = 'auto' | 'none' | 'required' | {
+    type: 'tool';
+    toolName: keyof TOOLS;
+};
 
 /**
 Embed a value using an embedding model. The type of the value is defined by the embedding model.
@@ -820,7 +832,7 @@ If set and supported by the model, calls will generate deterministic results.
 @returns
 A result object that contains the generated text, the results of the tool calls, and additional information.
  */
-declare function generateText<TOOLS extends Record<string, CoreTool>>({ model, tools, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
+declare function generateText<TOOLS extends Record<string, CoreTool>>({ model, tools, toolChoice, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
     /**
 The language model to use.
      */
@@ -829,6 +841,10 @@ The language model to use.
 The tools that the model can call. The model needs to support calling tools.
 */
     tools?: TOOLS;
+    /**
+The tool choice strategy. Default: 'auto'.
+     */
+    toolChoice?: CoreToolChoice<TOOLS>;
 }): Promise<GenerateTextResult<TOOLS>>;
 /**
 The result of a `generateText` call.
@@ -922,10 +938,13 @@ If set and supported by the model, calls will generate deterministic results.
 @param maxRetries - Maximum number of retries. Set to 0 to disable retries. Default: 2.
 @param abortSignal - An optional abort signal that can be used to cancel the call.
 
+@param onFinish - Callback that is called when the LLM response and all request tool executions
+(for tools that have an `execute` function) are finished.
+
 @return
 A result object for accessing different stream types and additional information.
  */
-declare function streamText<TOOLS extends Record<string, CoreTool>>({ model, tools, system, prompt, messages, maxRetries, abortSignal, ...settings }: CallSettings & Prompt & {
+declare function streamText<TOOLS extends Record<string, CoreTool>>({ model, tools, toolChoice, system, prompt, messages, maxRetries, abortSignal, onFinish, ...settings }: CallSettings & Prompt & {
     /**
 The language model to use.
      */
@@ -934,6 +953,49 @@ The language model to use.
 The tools that the model can call. The model needs to support calling tools.
     */
     tools?: TOOLS;
+    /**
+The tool choice strategy. Default: 'auto'.
+     */
+    toolChoice?: CoreToolChoice<TOOLS>;
+    /**
+Callback that is called when the LLM response and all request tool executions
+(for tools that have an `execute` function) are finished.
+     */
+    onFinish?: (event: {
+        /**
+  The reason why the generation finished.
+         */
+        finishReason: FinishReason;
+        /**
+  The token usage of the generated response.
+   */
+        usage: TokenUsage;
+        /**
+  The full text that has been generated.
+         */
+        text: string;
+        /**
+  The tool calls that have been executed.
+         */
+        toolCalls?: ToToolCall<TOOLS>[];
+        /**
+  The tool results that have been generated.
+         */
+        toolResults?: ToToolResult<TOOLS>[];
+        /**
+  Optional raw response data.
+         */
+        rawResponse?: {
+            /**
+    Response headers.
+             */
+            headers?: Record<string, string>;
+        };
+        /**
+  Warnings from the model provider (e.g. unsupported settings).
+         */
+        warnings?: CallWarning[];
+    }) => Promise<void> | void;
 }): Promise<StreamTextResult<TOOLS>>;
 type TextStreamPart<TOOLS extends Record<string, CoreTool>> = {
     type: 'text-delta';
@@ -960,12 +1022,13 @@ A result object for accessing different stream types and additional information.
  */
 declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     private originalStream;
+    private onFinish?;
     /**
   Warnings from the model provider (e.g. unsupported settings).
      */
     readonly warnings: CallWarning[] | undefined;
     /**
-  The token usage of the generated text. Resolved when the response is finished.
+  The token usage of the generated response. Resolved when the response is finished.
      */
     readonly usage: Promise<TokenUsage>;
     /**
@@ -973,20 +1036,33 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
      */
     readonly finishReason: Promise<FinishReason>;
     /**
+  The full text that has been generated. Resolved when the response is finished.
+     */
+    readonly text: Promise<string>;
+    /**
+  The tool calls that have been executed. Resolved when the response is finished.
+     */
+    readonly toolCalls: Promise<ToToolCall<TOOLS>[]>;
+    /**
+  The tool results that have been generated. Resolved when the all tool executions are finished.
+     */
+    readonly toolResults: Promise<ToToolResult<TOOLS>[]>;
+    /**
   Optional raw response data.
      */
-    rawResponse?: {
+    readonly rawResponse?: {
         /**
     Response headers.
          */
         headers?: Record<string, string>;
     };
-    constructor({ stream, warnings, rawResponse, }: {
+    constructor({ stream, warnings, rawResponse, onFinish, }: {
         stream: ReadableStream<TextStreamPart<TOOLS>>;
         warnings: CallWarning[] | undefined;
         rawResponse?: {
             headers?: Record<string, string>;
         };
+        onFinish?: Parameters<typeof streamText>[0]['onFinish'];
     });
     /**
   Split out a new stream from the original stream.
@@ -2268,4 +2344,4 @@ declare class StreamingTextResponse extends Response {
     constructor(res: ReadableStream, init?: ResponseInit, data?: StreamData);
 }
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, AssistantStatus, CallWarning, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolMessage, CoreUserMessage, CreateMessage, DataContent, DataMessage, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, Message$1 as Message, MistralStream, ObjectStreamPart, ObjectStreamPartInput, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamData, StreamObjectResult, StreamPart, StreamString, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, Tool, ToolCall, ToolCallHandler, ToolCallPart, ToolCallPayload, ToolChoice, ToolContent, ToolInvocation, ToolResultPart, UseAssistantOptions, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, generateObject, generateText, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, AssistantStatus, CallWarning, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, CreateMessage, DataContent, DataMessage, DeepPartial, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, Message$1 as Message, MistralStream, ObjectStreamPart, ObjectStreamPartInput, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamData, StreamObjectResult, StreamPart, StreamString, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, Tool, ToolCall, ToolCallHandler, ToolCallPart, ToolCallPayload, ToolChoice, ToolContent, ToolInvocation, ToolResultPart, UseAssistantOptions, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, generateObject, generateText, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.js

@@ -1265,6 +1265,33 @@ var StreamObjectResult = class {
 };
 var experimental_streamObject = streamObject;
 
+// core/util/is-non-empty-object.ts
+function isNonEmptyObject(object) {
+  return object != null && Object.keys(object).length > 0;
+}
+
+// core/prompt/prepare-tools-and-tool-choice.ts
+function prepareToolsAndToolChoice({
+  tools,
+  toolChoice
+}) {
+  if (!isNonEmptyObject(tools)) {
+    return {
+      tools: void 0,
+      toolChoice: void 0
+    };
+  }
+  return {
+    tools: Object.entries(tools).map(([name, tool2]) => ({
+      type: "function",
+      name,
+      description: tool2.description,
+      parameters: convertZodToJSONSchema(tool2.parameters)
+    })),
+    toolChoice: toolChoice == null ? { type: "auto" } : typeof toolChoice === "string" ? { type: toolChoice } : { type: "tool", toolName: toolChoice.toolName }
+  };
+}
+
 // core/generate-text/tool-call.ts
 var import_provider6 = require("@ai-sdk/provider");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
@@ -1306,6 +1333,7 @@ function parseToolCall({
 async function generateText({
   model,
   tools,
+  toolChoice,
   system,
   prompt,
   messages,
@@ -1320,12 +1348,7 @@ async function generateText({
     return model.doGenerate({
       mode: {
         type: "regular",
-        tools: tools == null ? void 0 : Object.entries(tools).map(([name, tool2]) => ({
-          type: "function",
-          name,
-          description: tool2.description,
-          parameters: convertZodToJSONSchema(tool2.parameters)
-        }))
+        ...prepareToolsAndToolChoice({ tools, toolChoice })
       },
       ...prepareCallSettings(settings),
       inputFormat: validatedPrompt.type,
@@ -1538,11 +1561,13 @@ function runToolsTransformation({
 async function streamText({
   model,
   tools,
+  toolChoice,
   system,
   prompt,
   messages,
   maxRetries,
   abortSignal,
+  onFinish,
   ...settings
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
@@ -1551,12 +1576,7 @@ async function streamText({
     () => model.doStream({
       mode: {
         type: "regular",
-        tools: tools == null ? void 0 : Object.entries(tools).map(([name, tool2]) => ({
-          type: "function",
-          name,
-          description: tool2.description,
-          parameters: convertZodToJSONSchema(tool2.parameters)
-        }))
+        ...prepareToolsAndToolChoice({ tools, toolChoice })
       },
       ...prepareCallSettings(settings),
       inputFormat: validatedPrompt.type,
@@ -1570,17 +1590,20 @@ async function streamText({
       generatorStream: stream
     }),
     warnings,
-    rawResponse
+    rawResponse,
+    onFinish
   });
 }
 var StreamTextResult = class {
   constructor({
     stream,
     warnings,
-    rawResponse
+    rawResponse,
+    onFinish
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
+    this.onFinish = onFinish;
     let resolveUsage;
     this.usage = new Promise((resolve) => {
       resolveUsage = resolve;
@@ -1589,13 +1612,70 @@ var StreamTextResult = class {
     this.finishReason = new Promise((resolve) => {
       resolveFinishReason = resolve;
     });
+    let resolveText;
+    this.text = new Promise((resolve) => {
+      resolveText = resolve;
+    });
+    let resolveToolCalls;
+    this.toolCalls = new Promise((resolve) => {
+      resolveToolCalls = resolve;
+    });
+    let resolveToolResults;
+    this.toolResults = new Promise((resolve) => {
+      resolveToolResults = resolve;
+    });
+    let finishReason;
+    let usage;
+    let text = "";
+    const toolCalls = [];
+    const toolResults = [];
+    const self = this;
     this.originalStream = stream.pipeThrough(
       new TransformStream({
         async transform(chunk, controller) {
           controller.enqueue(chunk);
+          if (chunk.type === "text-delta") {
+            text += chunk.textDelta;
+          }
+          if (chunk.type === "tool-call") {
+            toolCalls.push(chunk);
+          }
+          if (chunk.type === "tool-result") {
+            toolResults.push(chunk);
+          }
           if (chunk.type === "finish") {
-            resolveUsage(chunk.usage);
-            resolveFinishReason(chunk.finishReason);
+            usage = chunk.usage;
+            finishReason = chunk.finishReason;
+            resolveUsage(usage);
+            resolveFinishReason(finishReason);
+            resolveText(text);
+            resolveToolCalls(toolCalls);
+          }
+        },
+        // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
+        async flush(controller) {
+          var _a;
+          try {
+            resolveToolResults(toolResults);
+            await ((_a = self.onFinish) == null ? void 0 : _a.call(self, {
+              finishReason: finishReason != null ? finishReason : "unknown",
+              usage: usage != null ? usage : {
+                promptTokens: NaN,
+                completionTokens: NaN,
+                totalTokens: NaN
+              },
+              text,
+              toolCalls,
+              // The tool results are inferred as a never[] type, because they are
+              // optional and the execute method with an inferred result type is
+              // optional as well. Therefore we need to cast the toolResults to any.
+              // The type exposed to the users will be correctly inferred.
+              toolResults,
+              rawResponse,
+              warnings
+            }));
+          } catch (error) {
+            controller.error(error);
           }
         }
       })