ai 3.0.27 → 3.0.29

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1CallWarning } from '@ai-sdk/provider';
+import { LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1CallWarning, LanguageModelV1LogProbs } from '@ai-sdk/provider';
 import { z } from 'zod';
 import { ServerResponse } from 'node:http';
 import { AssistantStream } from 'openai/lib/AssistantStream';
@@ -286,11 +286,29 @@ declare class GenerateObjectResult<T> {
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+     */
+        headers?: Record<string, string>;
+    };
+    /**
+  Logprobs for the completion.
+  `undefined` if the mode does not support logprobs or if was not enabled
+     */
+    readonly logprobs: LanguageModelV1LogProbs | undefined;
     constructor(options: {
         object: T;
         finishReason: LanguageModelV1FinishReason;
         usage: TokenUsage;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
+        logprobs: LanguageModelV1LogProbs | undefined;
     });
 }
 
@@ -364,6 +382,23 @@ Default and recommended: 'auto' (best mode for the model).
  */
     mode?: 'auto' | 'json' | 'tool' | 'grammar';
 }): Promise<StreamObjectResult<T>>;
+type ObjectStreamPartInput = {
+    type: 'error';
+    error: unknown;
+} | {
+    type: 'finish';
+    finishReason: LanguageModelV1FinishReason;
+    logprobs?: LanguageModelV1LogProbs;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+};
+type ObjectStreamPart<T> = ObjectStreamPartInput | {
+    type: 'object';
+    object: DeepPartial<T>;
+};
 /**
 The result of a `streamObject` call that contains the partial object stream and additional information.
  */
@@ -373,16 +408,25 @@ declare class StreamObjectResult<T> {
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
-    constructor({ stream, warnings, }: {
-        stream: ReadableStream<string | ErrorStreamPart>;
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+     */
+        headers?: Record<string, string>;
+    };
+    constructor({ stream, warnings, rawResponse, }: {
+        stream: ReadableStream<string | ObjectStreamPartInput>;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
     });
     get partialObjectStream(): AsyncIterableStream<DeepPartial<T>>;
+    get fullStream(): AsyncIterableStream<ObjectStreamPart<T>>;
 }
-type ErrorStreamPart = {
-    type: 'error';
-    error: unknown;
-};
 
 /**
 A tool contains the description and the schema of the input that the tool expects.
@@ -565,6 +609,20 @@ declare class GenerateTextResult<TOOLS extends Record<string, ExperimentalTool>>
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+       */
+        headers?: Record<string, string>;
+    };
+    /**
+  Logprobs for the completion.
+  `undefined` if the mode does not support logprobs or if was not enabled
+     */
+    readonly logprobs: LanguageModelV1LogProbs | undefined;
     constructor(options: {
         text: string;
         toolCalls: ToToolCallArray<TOOLS>;
@@ -572,6 +630,10 @@ declare class GenerateTextResult<TOOLS extends Record<string, ExperimentalTool>>
         finishReason: LanguageModelV1FinishReason;
         usage: TokenUsage;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
+        logprobs: LanguageModelV1LogProbs | undefined;
     });
 }
 
@@ -632,6 +694,7 @@ type TextStreamPart<TOOLS extends Record<string, ExperimentalTool>> = {
 } & ToToolResult<TOOLS>) | {
     type: 'finish';
     finishReason: LanguageModelV1FinishReason;
+    logprobs?: LanguageModelV1LogProbs;
     usage: {
         promptTokens: number;
         completionTokens: number;
@@ -647,9 +710,21 @@ declare class StreamTextResult<TOOLS extends Record<string, ExperimentalTool>> {
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
-    constructor({ stream, warnings, }: {
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+         */
+        headers?: Record<string, string>;
+    };
+    constructor({ stream, warnings, rawResponse, }: {
         stream: ReadableStream<TextStreamPart<TOOLS>>;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
     });
     /**
   A text stream that returns only the generated text deltas. You can use it
@@ -1795,4 +1870,4 @@ declare function streamToResponse(res: ReadableStream, response: ServerResponse,
     status?: number;
 }): void;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, DataContent, DataMessage, DeepPartial, ErrorStreamPart, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, LangChainStream, Message$1 as Message, MistralStream, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamData, StreamObjectResult, StreamPart, StreamString, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, Tool, ToolCall, ToolCallHandler, ToolCallPart, ToolCallPayload, ToolChoice, ToolContent, ToolResultPart, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, DataContent, DataMessage, DeepPartial, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, LangChainStream, 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, ToolResultPart, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1CallWarning } from '@ai-sdk/provider';
+import { LanguageModelV1, LanguageModelV1FinishReason, LanguageModelV1CallWarning, LanguageModelV1LogProbs } from '@ai-sdk/provider';
 import { z } from 'zod';
 import { ServerResponse } from 'node:http';
 import { AssistantStream } from 'openai/lib/AssistantStream';
@@ -286,11 +286,29 @@ declare class GenerateObjectResult<T> {
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+     */
+        headers?: Record<string, string>;
+    };
+    /**
+  Logprobs for the completion.
+  `undefined` if the mode does not support logprobs or if was not enabled
+     */
+    readonly logprobs: LanguageModelV1LogProbs | undefined;
     constructor(options: {
         object: T;
         finishReason: LanguageModelV1FinishReason;
         usage: TokenUsage;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
+        logprobs: LanguageModelV1LogProbs | undefined;
     });
 }
 
@@ -364,6 +382,23 @@ Default and recommended: 'auto' (best mode for the model).
  */
     mode?: 'auto' | 'json' | 'tool' | 'grammar';
 }): Promise<StreamObjectResult<T>>;
+type ObjectStreamPartInput = {
+    type: 'error';
+    error: unknown;
+} | {
+    type: 'finish';
+    finishReason: LanguageModelV1FinishReason;
+    logprobs?: LanguageModelV1LogProbs;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+};
+type ObjectStreamPart<T> = ObjectStreamPartInput | {
+    type: 'object';
+    object: DeepPartial<T>;
+};
 /**
 The result of a `streamObject` call that contains the partial object stream and additional information.
  */
@@ -373,16 +408,25 @@ declare class StreamObjectResult<T> {
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
-    constructor({ stream, warnings, }: {
-        stream: ReadableStream<string | ErrorStreamPart>;
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+     */
+        headers?: Record<string, string>;
+    };
+    constructor({ stream, warnings, rawResponse, }: {
+        stream: ReadableStream<string | ObjectStreamPartInput>;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
     });
     get partialObjectStream(): AsyncIterableStream<DeepPartial<T>>;
+    get fullStream(): AsyncIterableStream<ObjectStreamPart<T>>;
 }
-type ErrorStreamPart = {
-    type: 'error';
-    error: unknown;
-};
 
 /**
 A tool contains the description and the schema of the input that the tool expects.
@@ -565,6 +609,20 @@ declare class GenerateTextResult<TOOLS extends Record<string, ExperimentalTool>>
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+       */
+        headers?: Record<string, string>;
+    };
+    /**
+  Logprobs for the completion.
+  `undefined` if the mode does not support logprobs or if was not enabled
+     */
+    readonly logprobs: LanguageModelV1LogProbs | undefined;
     constructor(options: {
         text: string;
         toolCalls: ToToolCallArray<TOOLS>;
@@ -572,6 +630,10 @@ declare class GenerateTextResult<TOOLS extends Record<string, ExperimentalTool>>
         finishReason: LanguageModelV1FinishReason;
         usage: TokenUsage;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
+        logprobs: LanguageModelV1LogProbs | undefined;
     });
 }
 
@@ -632,6 +694,7 @@ type TextStreamPart<TOOLS extends Record<string, ExperimentalTool>> = {
 } & ToToolResult<TOOLS>) | {
     type: 'finish';
     finishReason: LanguageModelV1FinishReason;
+    logprobs?: LanguageModelV1LogProbs;
     usage: {
         promptTokens: number;
         completionTokens: number;
@@ -647,9 +710,21 @@ declare class StreamTextResult<TOOLS extends Record<string, ExperimentalTool>> {
   Warnings from the model provider (e.g. unsupported settings)
      */
     readonly warnings: LanguageModelV1CallWarning[] | undefined;
-    constructor({ stream, warnings, }: {
+    /**
+  Optional raw response data.
+     */
+    rawResponse?: {
+        /**
+    Response headers.
+         */
+        headers?: Record<string, string>;
+    };
+    constructor({ stream, warnings, rawResponse, }: {
         stream: ReadableStream<TextStreamPart<TOOLS>>;
         warnings: LanguageModelV1CallWarning[] | undefined;
+        rawResponse?: {
+            headers?: Record<string, string>;
+        };
     });
     /**
   A text stream that returns only the generated text deltas. You can use it
@@ -1795,4 +1870,4 @@ declare function streamToResponse(res: ReadableStream, response: ServerResponse,
     status?: number;
 }): void;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, DataContent, DataMessage, DeepPartial, ErrorStreamPart, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, LangChainStream, Message$1 as Message, MistralStream, OpenAIStream, OpenAIStreamCallbacks, ReactResponseRow, ReplicateStream, RequestOptions, StreamData, StreamObjectResult, StreamPart, StreamString, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, Tool, ToolCall, ToolCallHandler, ToolCallPart, ToolCallPayload, ToolChoice, ToolContent, ToolResultPart, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantMessage, AssistantResponse, ChatRequest, ChatRequestOptions, CohereStream, CompletionUsage, CreateMessage, DataContent, DataMessage, DeepPartial, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, Function, FunctionCall$1 as FunctionCall, FunctionCallHandler, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, IdGenerator, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, JSONValue, LangChainStream, 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, ToolResultPart, UseChatOptions, UseCompletionOptions, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, createCallbacksTransformer, createChunkDecoder, createEventStreamTransformer, createStreamDataTransformer, experimental_AssistantResponse, experimental_StreamData, experimental_StreamingReactResponse, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, formatStreamPart, generateId, isStreamStringEqualToType, generateId as nanoid, parseStreamPart, readDataStream, readableFromAsyncIterable, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.js

@@ -450,6 +450,8 @@ async function experimental_generateObject({
   let finishReason;
   let usage;
   let warnings;
+  let rawResponse;
+  let logprobs;
   switch (mode) {
     case "json": {
       const validatedPrompt = getValidatedPrompt({
@@ -473,6 +475,8 @@ async function experimental_generateObject({
       finishReason = generateResult.finishReason;
       usage = generateResult.usage;
       warnings = generateResult.warnings;
+      rawResponse = generateResult.rawResponse;
+      logprobs = generateResult.logprobs;
       break;
     }
     case "grammar": {
@@ -497,6 +501,8 @@ async function experimental_generateObject({
       finishReason = generateResult.finishReason;
       usage = generateResult.usage;
       warnings = generateResult.warnings;
+      rawResponse = generateResult.rawResponse;
+      logprobs = generateResult.logprobs;
       break;
     }
     case "tool": {
@@ -530,6 +536,8 @@ async function experimental_generateObject({
       finishReason = generateResult.finishReason;
       usage = generateResult.usage;
       warnings = generateResult.warnings;
+      rawResponse = generateResult.rawResponse;
+      logprobs = generateResult.logprobs;
       break;
     }
     case void 0: {
@@ -548,7 +556,9 @@ async function experimental_generateObject({
     object: parseResult.value,
     finishReason,
     usage: calculateTokenUsage(usage),
-    warnings
+    warnings,
+    rawResponse,
+    logprobs
   });
 }
 var GenerateObjectResult = class {
@@ -557,6 +567,8 @@ var GenerateObjectResult = class {
     this.finishReason = options.finishReason;
     this.usage = options.usage;
     this.warnings = options.warnings;
+    this.rawResponse = options.rawResponse;
+    this.logprobs = options.logprobs;
   }
 };
 
@@ -988,6 +1000,7 @@ async function experimental_streamObject({
             case "text-delta":
               controller.enqueue(chunk.textDelta);
               break;
+            case "finish":
             case "error":
               controller.enqueue(chunk);
               break;
@@ -1015,6 +1028,7 @@ async function experimental_streamObject({
             case "text-delta":
               controller.enqueue(chunk.textDelta);
               break;
+            case "finish":
             case "error":
               controller.enqueue(chunk);
               break;
@@ -1050,6 +1064,7 @@ async function experimental_streamObject({
             case "tool-call-delta":
               controller.enqueue(chunk.argsTextDelta);
               break;
+            case "finish":
             case "error":
               controller.enqueue(chunk);
               break;
@@ -1069,16 +1084,19 @@ async function experimental_streamObject({
   const result = await retry(() => model.doStream(callOptions));
   return new StreamObjectResult({
     stream: result.stream.pipeThrough(new TransformStream(transformer)),
-    warnings: result.warnings
+    warnings: result.warnings,
+    rawResponse: result.rawResponse
   });
 }
 var StreamObjectResult = class {
   constructor({
     stream,
-    warnings
+    warnings,
+    rawResponse
   }) {
     this.originalStream = stream;
     this.warnings = warnings;
+    this.rawResponse = rawResponse;
   }
   get partialObjectStream() {
     let accumulatedText = "";
@@ -1094,13 +1112,42 @@ var StreamObjectResult = class {
             latestObject = currentObject;
             controller.enqueue(currentObject);
           }
-        }
-        if (typeof chunk === "object" && chunk.type === "error") {
+        } else if (chunk.type === "error") {
           throw chunk.error;
         }
       }
     });
   }
+  get fullStream() {
+    let accumulatedText = "";
+    let latestObject = void 0;
+    return createAsyncIterableStream(this.originalStream, {
+      transform(chunk, controller) {
+        if (typeof chunk === "string") {
+          accumulatedText += chunk;
+          const currentObject = parsePartialJson(
+            accumulatedText
+          );
+          if (!isDeepEqualData(latestObject, currentObject)) {
+            latestObject = currentObject;
+            controller.enqueue({ type: "object", object: currentObject });
+          }
+        } else {
+          switch (chunk.type) {
+            case "finish":
+              controller.enqueue({
+                ...chunk,
+                usage: calculateTokenUsage(chunk.usage)
+              });
+              break;
+            default:
+              controller.enqueue(chunk);
+              break;
+          }
+        }
+      }
+    });
+  }
 };
 
 // core/generate-text/tool-call.ts
@@ -1185,7 +1232,9 @@ async function experimental_generateText({
     toolResults,
     finishReason: modelResponse.finishReason,
     usage: calculateTokenUsage(modelResponse.usage),
-    warnings: modelResponse.warnings
+    warnings: modelResponse.warnings,
+    rawResponse: modelResponse.rawResponse,
+    logprobs: modelResponse.logprobs
   });
 }
 async function executeTools({
@@ -1219,6 +1268,8 @@ var GenerateTextResult = class {
     this.finishReason = options.finishReason;
     this.usage = options.usage;
     this.warnings = options.warnings;
+    this.rawResponse = options.rawResponse;
+    this.logprobs = options.logprobs;
   }
 };
 
@@ -1319,11 +1370,8 @@ function runToolsTransformation({
           controller.enqueue({
             type: "finish",
             finishReason: chunk.finishReason,
-            usage: {
-              promptTokens: chunk.usage.promptTokens,
-              completionTokens: chunk.usage.completionTokens,
-              totalTokens: chunk.usage.promptTokens + chunk.usage.completionTokens
-            }
+            logprobs: chunk.logprobs,
+            usage: calculateTokenUsage(chunk.usage)
           });
           break;
         }
@@ -1381,7 +1429,7 @@ async function experimental_streamText({
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
   const validatedPrompt = getValidatedPrompt({ system, prompt, messages });
-  const { stream, warnings } = await retry(
+  const { stream, warnings, rawResponse } = await retry(
     () => model.doStream({
       mode: {
         type: "regular",
@@ -1403,16 +1451,19 @@ async function experimental_streamText({
       tools,
       generatorStream: stream
     }),
-    warnings
+    warnings,
+    rawResponse
   });
 }
 var StreamTextResult = class {
   constructor({
     stream,
-    warnings
+    warnings,
+    rawResponse
   }) {
     this.originalStream = stream;
     this.warnings = warnings;
+    this.rawResponse = rawResponse;
   }
   /**
   A text stream that returns only the generated text deltas. You can use it