ai 3.2.17 → 3.2.19

package/dist/index.mjs

@@ -83,6 +83,7 @@ async function embed({
   abortSignal,
   headers
 }) {
+  var _a;
   const retry = retryWithExponentialBackoff({ maxRetries });
   const modelResponse = await retry(
     () => model.doEmbed({ values: [value], abortSignal, headers })
@@ -90,6 +91,7 @@ async function embed({
   return new EmbedResult({
     value,
     embedding: modelResponse.embeddings[0],
+    usage: (_a = modelResponse.usage) != null ? _a : { tokens: NaN },
     rawResponse: modelResponse.rawResponse
   });
 }
@@ -97,6 +99,7 @@ var EmbedResult = class {
   constructor(options) {
     this.value = options.value;
     this.embedding = options.embedding;
+    this.usage = options.usage;
     this.rawResponse = options.rawResponse;
   }
 };
@@ -121,6 +124,7 @@ async function embedMany({
   abortSignal,
   headers
 }) {
+  var _a, _b, _c;
   const retry = retryWithExponentialBackoff({ maxRetries });
   const maxEmbeddingsPerCall = model.maxEmbeddingsPerCall;
   if (maxEmbeddingsPerCall == null) {
@@ -129,23 +133,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;
   }
 };
 
@@ -153,15 +161,6 @@ var EmbedManyResult = class {
 import { NoObjectGeneratedError } from "@ai-sdk/provider";
 import { safeParseJSON } from "@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] },
@@ -512,12 +511,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
 import zodToJsonSchema from "zod-to-json-schema";
 function convertZodToJSONSchema(zodSchema) {
   return zodToJsonSchema(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.";
@@ -537,16 +555,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,
@@ -676,7 +684,7 @@ async function generateObject({
   return new GenerateObjectResult({
     object: parseResult.value,
     finishReason,
-    usage: calculateTokenUsage(usage),
+    usage: calculateCompletionTokenUsage(usage),
     warnings,
     rawResponse,
     logprobs
@@ -922,7 +930,7 @@ var StreamObjectResult = class {
                   textDelta: delta
                 });
               }
-              usage = calculateTokenUsage(chunk.usage);
+              usage = calculateCompletionTokenUsage(chunk.usage);
               controller.enqueue({ ...chunk, usage });
               resolveUsage(usage);
               const validationResult = safeValidateTypes({
@@ -1213,7 +1221,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,
@@ -1371,7 +1379,7 @@ function runToolsTransformation({
             type: "finish",
             finishReason: chunk.finishReason,
             logprobs: chunk.logprobs,
-            usage: calculateTokenUsage(chunk.usage)
+            usage: calculateCompletionTokenUsage(chunk.usage)
           });
           break;
         }