ai 3.2.18 → 3.2.19

package/dist/index.d.mts

@@ -8,6 +8,33 @@ import { ServerResponse as ServerResponse$1 } from 'node:http';
 import { AssistantStream } from 'openai/lib/AssistantStream';
 import { Run } from 'openai/resources/beta/threads/runs/runs';
 
+/**
+Represents the number of tokens used in a prompt and completion.
+ */
+type CompletionTokenUsage$1 = {
+    /**
+  The number of tokens used in the prompt
+     */
+    promptTokens: number;
+    /**
+  The number of tokens used in the completion.
+   */
+    completionTokens: number;
+    /**
+  The total number of tokens used (promptTokens + completionTokens).
+     */
+    totalTokens: number;
+};
+/**
+Represents the number of tokens used in an embedding.
+ */
+type EmbeddingTokenUsage = {
+    /**
+  The number of tokens used in the embedding.
+     */
+    tokens: number;
+};
+
 /**
 Embedding model that is used by the AI SDK Core functions.
 */
@@ -55,6 +82,12 @@ type CoreToolChoice<TOOLS extends Record<string, unknown>> = 'auto' | 'none' | '
     toolName: keyof TOOLS;
 };
 
+/**
+ * @deprecated Use CompletionTokenUsage instead.
+ */
+type TokenUsage = CompletionTokenUsage$1;
+type CompletionTokenUsage = CompletionTokenUsage$1;
+
 /**
 Embed a value using an embedding model. The type of the value is defined by the embedding model.
 
@@ -106,6 +139,10 @@ declare class EmbedResult<VALUE> {
     */
     readonly embedding: Embedding;
     /**
+  The embedding token usage.
+    */
+    readonly usage: EmbeddingTokenUsage;
+    /**
   Optional raw response data.
      */
     readonly rawResponse?: {
@@ -117,6 +154,7 @@ declare class EmbedResult<VALUE> {
     constructor(options: {
         value: VALUE;
         embedding: Embedding;
+        usage: EmbeddingTokenUsage;
         rawResponse?: {
             headers?: Record<string, string>;
         };
@@ -177,30 +215,17 @@ declare class EmbedManyResult<VALUE> {
   The embeddings. They are in the same order as the values.
     */
     readonly embeddings: Array<Embedding>;
+    /**
+  The embedding token usage.
+    */
+    readonly usage: EmbeddingTokenUsage;
     constructor(options: {
         values: Array<VALUE>;
         embeddings: Array<Embedding>;
+        usage: EmbeddingTokenUsage;
     });
 }
 
-/**
-Represents the number of tokens used in a prompt and completion.
- */
-type TokenUsage = {
-    /**
-  The number of tokens used in the prompt
-     */
-    promptTokens: number;
-    /**
-  The number of tokens used in the completion.
-   */
-    completionTokens: number;
-    /**
-  The total number of tokens used (promptTokens + completionTokens).
-     */
-    totalTokens: number;
-};
-
 type CallSettings = {
     /**
   Maximum number of tokens to generate.
@@ -514,7 +539,7 @@ declare class GenerateObjectResult<T> {
     /**
   The token usage of the generated text.
      */
-    readonly usage: TokenUsage;
+    readonly usage: CompletionTokenUsage$1;
     /**
   Warnings from the model provider (e.g. unsupported settings)
      */
@@ -536,7 +561,7 @@ declare class GenerateObjectResult<T> {
     constructor(options: {
         object: T;
         finishReason: FinishReason;
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         warnings: CallWarning[] | undefined;
         rawResponse?: {
             headers?: Record<string, string>;
@@ -624,7 +649,7 @@ Callback that is called when the LLM response and the final object validation ar
         /**
   The token usage of the generated response.
   */
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         /**
   The generated object (typed according to the schema). Can be undefined if the final object does not match the schema.
      */
@@ -684,7 +709,7 @@ declare class StreamObjectResult<T> {
     /**
   The token usage of the generated response. Resolved when the response is finished.
      */
-    readonly usage: Promise<TokenUsage>;
+    readonly usage: Promise<CompletionTokenUsage$1>;
     /**
   Optional raw response data.
      */
@@ -1000,7 +1025,7 @@ declare class GenerateTextResult<TOOLS extends Record<string, CoreTool>> {
     /**
   The token usage of the generated text.
      */
-    readonly usage: TokenUsage;
+    readonly usage: CompletionTokenUsage$1;
     /**
   Warnings from the model provider (e.g. unsupported settings)
      */
@@ -1032,7 +1057,7 @@ declare class GenerateTextResult<TOOLS extends Record<string, CoreTool>> {
         toolCalls: ToToolCallArray<TOOLS>;
         toolResults: ToToolResultArray<TOOLS>;
         finishReason: FinishReason;
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         warnings: CallWarning[] | undefined;
         rawResponse?: {
             headers?: Record<string, string>;
@@ -1109,7 +1134,7 @@ Callback that is called when the LLM response and all request tool executions
         /**
   The token usage of the generated response.
    */
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         /**
   The full text that has been generated.
          */
@@ -1170,7 +1195,7 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     /**
   The token usage of the generated response. Resolved when the response is finished.
      */
-    readonly usage: Promise<TokenUsage>;
+    readonly usage: Promise<CompletionTokenUsage$1>;
     /**
   The reason why the generation finished. Resolved when the response is finished.
      */
@@ -2130,4 +2155,4 @@ declare const generateId: (size?: number | undefined) => string;
  */
 declare const nanoid: (size?: number | undefined) => string;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionTokenUsage, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingTokenUsage, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.d.ts

@@ -8,6 +8,33 @@ import { ServerResponse as ServerResponse$1 } from 'node:http';
 import { AssistantStream } from 'openai/lib/AssistantStream';
 import { Run } from 'openai/resources/beta/threads/runs/runs';
 
+/**
+Represents the number of tokens used in a prompt and completion.
+ */
+type CompletionTokenUsage$1 = {
+    /**
+  The number of tokens used in the prompt
+     */
+    promptTokens: number;
+    /**
+  The number of tokens used in the completion.
+   */
+    completionTokens: number;
+    /**
+  The total number of tokens used (promptTokens + completionTokens).
+     */
+    totalTokens: number;
+};
+/**
+Represents the number of tokens used in an embedding.
+ */
+type EmbeddingTokenUsage = {
+    /**
+  The number of tokens used in the embedding.
+     */
+    tokens: number;
+};
+
 /**
 Embedding model that is used by the AI SDK Core functions.
 */
@@ -55,6 +82,12 @@ type CoreToolChoice<TOOLS extends Record<string, unknown>> = 'auto' | 'none' | '
     toolName: keyof TOOLS;
 };
 
+/**
+ * @deprecated Use CompletionTokenUsage instead.
+ */
+type TokenUsage = CompletionTokenUsage$1;
+type CompletionTokenUsage = CompletionTokenUsage$1;
+
 /**
 Embed a value using an embedding model. The type of the value is defined by the embedding model.
 
@@ -106,6 +139,10 @@ declare class EmbedResult<VALUE> {
     */
     readonly embedding: Embedding;
     /**
+  The embedding token usage.
+    */
+    readonly usage: EmbeddingTokenUsage;
+    /**
   Optional raw response data.
      */
     readonly rawResponse?: {
@@ -117,6 +154,7 @@ declare class EmbedResult<VALUE> {
     constructor(options: {
         value: VALUE;
         embedding: Embedding;
+        usage: EmbeddingTokenUsage;
         rawResponse?: {
             headers?: Record<string, string>;
         };
@@ -177,30 +215,17 @@ declare class EmbedManyResult<VALUE> {
   The embeddings. They are in the same order as the values.
     */
     readonly embeddings: Array<Embedding>;
+    /**
+  The embedding token usage.
+    */
+    readonly usage: EmbeddingTokenUsage;
     constructor(options: {
         values: Array<VALUE>;
         embeddings: Array<Embedding>;
+        usage: EmbeddingTokenUsage;
     });
 }
 
-/**
-Represents the number of tokens used in a prompt and completion.
- */
-type TokenUsage = {
-    /**
-  The number of tokens used in the prompt
-     */
-    promptTokens: number;
-    /**
-  The number of tokens used in the completion.
-   */
-    completionTokens: number;
-    /**
-  The total number of tokens used (promptTokens + completionTokens).
-     */
-    totalTokens: number;
-};
-
 type CallSettings = {
     /**
   Maximum number of tokens to generate.
@@ -514,7 +539,7 @@ declare class GenerateObjectResult<T> {
     /**
   The token usage of the generated text.
      */
-    readonly usage: TokenUsage;
+    readonly usage: CompletionTokenUsage$1;
     /**
   Warnings from the model provider (e.g. unsupported settings)
      */
@@ -536,7 +561,7 @@ declare class GenerateObjectResult<T> {
     constructor(options: {
         object: T;
         finishReason: FinishReason;
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         warnings: CallWarning[] | undefined;
         rawResponse?: {
             headers?: Record<string, string>;
@@ -624,7 +649,7 @@ Callback that is called when the LLM response and the final object validation ar
         /**
   The token usage of the generated response.
   */
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         /**
   The generated object (typed according to the schema). Can be undefined if the final object does not match the schema.
      */
@@ -684,7 +709,7 @@ declare class StreamObjectResult<T> {
     /**
   The token usage of the generated response. Resolved when the response is finished.
      */
-    readonly usage: Promise<TokenUsage>;
+    readonly usage: Promise<CompletionTokenUsage$1>;
     /**
   Optional raw response data.
      */
@@ -1000,7 +1025,7 @@ declare class GenerateTextResult<TOOLS extends Record<string, CoreTool>> {
     /**
   The token usage of the generated text.
      */
-    readonly usage: TokenUsage;
+    readonly usage: CompletionTokenUsage$1;
     /**
   Warnings from the model provider (e.g. unsupported settings)
      */
@@ -1032,7 +1057,7 @@ declare class GenerateTextResult<TOOLS extends Record<string, CoreTool>> {
         toolCalls: ToToolCallArray<TOOLS>;
         toolResults: ToToolResultArray<TOOLS>;
         finishReason: FinishReason;
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         warnings: CallWarning[] | undefined;
         rawResponse?: {
             headers?: Record<string, string>;
@@ -1109,7 +1134,7 @@ Callback that is called when the LLM response and all request tool executions
         /**
   The token usage of the generated response.
    */
-        usage: TokenUsage;
+        usage: CompletionTokenUsage$1;
         /**
   The full text that has been generated.
          */
@@ -1170,7 +1195,7 @@ declare class StreamTextResult<TOOLS extends Record<string, CoreTool>> {
     /**
   The token usage of the generated response. Resolved when the response is finished.
      */
-    readonly usage: Promise<TokenUsage>;
+    readonly usage: Promise<CompletionTokenUsage$1>;
     /**
   The reason why the generation finished. Resolved when the response is finished.
      */
@@ -2130,4 +2155,4 @@ declare const generateId: (size?: number | undefined) => string;
  */
 declare const nanoid: (size?: number | undefined) => string;
 
-export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };
+export { AIStream, AIStreamCallbacksAndOptions, AIStreamParser, AIStreamParserOptions, AWSBedrockAnthropicMessagesStream, AWSBedrockAnthropicStream, AWSBedrockCohereStream, AWSBedrockLlama2Stream, AWSBedrockStream, AnthropicStream, AssistantContent, AssistantResponse, CallWarning, CohereStream, CompletionTokenUsage, CompletionUsage, CoreAssistantMessage, CoreMessage, CoreSystemMessage, CoreTool, CoreToolChoice, CoreToolMessage, CoreUserMessage, DataContent, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingTokenUsage, ExperimentalAssistantMessage, ExperimentalMessage, ExperimentalTool, ExperimentalToolMessage, ExperimentalUserMessage, FinishReason, FunctionCallPayload, GenerateObjectResult, GenerateTextResult, GoogleGenerativeAIStream, HuggingFaceStream, ImagePart, InkeepAIStreamCallbacksAndOptions, InkeepChatResultCallbacks, InkeepOnFinalMetadata, InkeepStream, InvalidMessageRoleError, InvalidModelIdError, langchainAdapter as LangChainAdapter, LangChainStream, LanguageModel, LogProbs, MistralStream, NoSuchModelError, NoSuchProviderError, ObjectStreamInputPart, ObjectStreamPart, OpenAIStream, OpenAIStreamCallbacks, ReplicateStream, StreamData, StreamObjectResult, StreamTextResult, StreamingTextResponse, TextPart$1 as TextPart, TextStreamPart, TokenUsage, ToolCallPart, ToolCallPayload, ToolContent, ToolResultPart, UserContent, convertDataContentToBase64String, convertDataContentToUint8Array, convertToCoreMessages, cosineSimilarity, createCallbacksTransformer, createEventStreamTransformer, createStreamDataTransformer, embed, embedMany, experimental_AssistantResponse, experimental_ModelRegistry, experimental_ProviderRegistry, experimental_StreamData, experimental_createModelRegistry, experimental_createProviderRegistry, experimental_generateObject, experimental_generateText, experimental_streamObject, experimental_streamText, generateId, generateObject, generateText, nanoid, readableFromAsyncIterable, streamObject, streamText, streamToResponse, tool, trimStartOfStreamHelper };

package/dist/index.js

@@ -181,6 +181,7 @@ async function embed({
   abortSignal,
   headers
 }) {
+  var _a;
   const retry = retryWithExponentialBackoff({ maxRetries });
   const modelResponse = await retry(
     () => model.doEmbed({ values: [value], abortSignal, headers })
@@ -188,6 +189,7 @@ async function embed({
   return new EmbedResult({
     value,
     embedding: modelResponse.embeddings[0],
+    usage: (_a = modelResponse.usage) != null ? _a : { tokens: NaN },
     rawResponse: modelResponse.rawResponse
   });
 }
@@ -195,6 +197,7 @@ var EmbedResult = class {
   constructor(options) {
     this.value = options.value;
     this.embedding = options.embedding;
+    this.usage = options.usage;
     this.rawResponse = options.rawResponse;
   }
 };
@@ -219,6 +222,7 @@ async function embedMany({
   abortSignal,
   headers
 }) {
+  var _a, _b, _c;
   const retry = retryWithExponentialBackoff({ maxRetries });
   const maxEmbeddingsPerCall = model.maxEmbeddingsPerCall;
   if (maxEmbeddingsPerCall == null) {
@@ -227,23 +231,27 @@ async function embedMany({
     );
     return new EmbedManyResult({
       values,
-      embeddings: modelResponse.embeddings
+      embeddings: modelResponse.embeddings,
+      usage: (_a = modelResponse.usage) != null ? _a : { tokens: NaN }
     });
   }
   const valueChunks = splitArray(values, maxEmbeddingsPerCall);
   const embeddings = [];
+  let tokens = 0;
   for (const chunk of valueChunks) {
     const modelResponse = await retry(
       () => model.doEmbed({ values: chunk, abortSignal, headers })
     );
     embeddings.push(...modelResponse.embeddings);
+    tokens += (_c = (_b = modelResponse.usage) == null ? void 0 : _b.tokens) != null ? _c : NaN;
   }
-  return new EmbedManyResult({ values, embeddings });
+  return new EmbedManyResult({ values, embeddings, usage: { tokens } });
 }
 var EmbedManyResult = class {
   constructor(options) {
     this.values = options.values;
     this.embeddings = options.embeddings;
+    this.usage = options.usage;
   }
 };
 
@@ -251,15 +259,6 @@ var EmbedManyResult = class {
 var import_provider5 = require("@ai-sdk/provider");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
-// core/generate-text/token-usage.ts
-function calculateTokenUsage(usage) {
-  return {
-    promptTokens: usage.promptTokens,
-    completionTokens: usage.completionTokens,
-    totalTokens: usage.promptTokens + usage.completionTokens
-  };
-}
-
 // core/util/detect-image-mimetype.ts
 var mimeTypeSignatures = [
   { mimeType: "image/gif", bytes: [71, 73, 70] },
@@ -607,12 +606,31 @@ function prepareCallSettings({
   };
 }
 
+// core/types/token-usage.ts
+function calculateCompletionTokenUsage(usage) {
+  return {
+    promptTokens: usage.promptTokens,
+    completionTokens: usage.completionTokens,
+    totalTokens: usage.promptTokens + usage.completionTokens
+  };
+}
+
 // core/util/convert-zod-to-json-schema.ts
 var import_zod_to_json_schema = __toESM(require("zod-to-json-schema"));
 function convertZodToJSONSchema(zodSchema) {
   return (0, import_zod_to_json_schema.default)(zodSchema);
 }
 
+// core/util/prepare-response-headers.ts
+function prepareResponseHeaders(init, { contentType }) {
+  var _a;
+  const headers = new Headers((_a = init == null ? void 0 : init.headers) != null ? _a : {});
+  if (!headers.has("Content-Type")) {
+    headers.set("Content-Type", contentType);
+  }
+  return headers;
+}
+
 // core/generate-object/inject-json-schema-into-system.ts
 var DEFAULT_SCHEMA_PREFIX = "JSON schema:";
 var DEFAULT_SCHEMA_SUFFIX = "You MUST answer with a JSON object that matches the JSON schema above.";
@@ -632,16 +650,6 @@ function injectJsonSchemaIntoSystem({
   ].filter((line) => line != null).join("\n");
 }
 
-// core/util/prepare-response-headers.ts
-function prepareResponseHeaders(init, { contentType }) {
-  var _a;
-  const headers = new Headers((_a = init == null ? void 0 : init.headers) != null ? _a : {});
-  if (!headers.has("Content-Type")) {
-    headers.set("Content-Type", contentType);
-  }
-  return headers;
-}
-
 // core/generate-object/generate-object.ts
 async function generateObject({
   model,
@@ -771,7 +779,7 @@ async function generateObject({
   return new GenerateObjectResult({
     object: parseResult.value,
     finishReason,
-    usage: calculateTokenUsage(usage),
+    usage: calculateCompletionTokenUsage(usage),
     warnings,
     rawResponse,
     logprobs
@@ -1014,7 +1022,7 @@ var StreamObjectResult = class {
                   textDelta: delta
                 });
               }
-              usage = calculateTokenUsage(chunk.usage);
+              usage = calculateCompletionTokenUsage(chunk.usage);
               controller.enqueue({ ...chunk, usage });
               resolveUsage(usage);
               const validationResult = (0, import_provider_utils5.safeValidateTypes)({
@@ -1302,7 +1310,7 @@ async function generateText({
     toolCalls: currentToolCalls,
     toolResults: currentToolResults,
     finishReason: currentModelResponse.finishReason,
-    usage: calculateTokenUsage(currentModelResponse.usage),
+    usage: calculateCompletionTokenUsage(currentModelResponse.usage),
     warnings: currentModelResponse.warnings,
     rawResponse: currentModelResponse.rawResponse,
     logprobs: currentModelResponse.logprobs,
@@ -1460,7 +1468,7 @@ function runToolsTransformation({
             type: "finish",
             finishReason: chunk.finishReason,
             logprobs: chunk.logprobs,
-            usage: calculateTokenUsage(chunk.usage)
+            usage: calculateCompletionTokenUsage(chunk.usage)
           });
           break;
         }