@ai-sdk/google 2.0.101 → 2.0.103

package/dist/internal/index.mjs

@@ -12,6 +12,100 @@ import {
 } from "@ai-sdk/provider-utils";
 import { z as z3 } from "zod/v4";
 
+// src/add-google-generative-ai-usage-metadata.ts
+function addGoogleGenerativeAIUsageMetadata(a, b) {
+  if (a == null || b == null) {
+    return b != null ? b : a;
+  }
+  return {
+    ...a,
+    ...b,
+    promptTokenCount: addCounts(a.promptTokenCount, b.promptTokenCount),
+    candidatesTokenCount: addCounts(
+      a.candidatesTokenCount,
+      b.candidatesTokenCount
+    ),
+    totalTokenCount: addCounts(a.totalTokenCount, b.totalTokenCount),
+    cachedContentTokenCount: addCounts(
+      a.cachedContentTokenCount,
+      b.cachedContentTokenCount
+    ),
+    thoughtsTokenCount: addCounts(a.thoughtsTokenCount, b.thoughtsTokenCount),
+    promptTokensDetails: addTokenDetails(
+      a.promptTokensDetails,
+      b.promptTokensDetails
+    ),
+    candidatesTokensDetails: addTokenDetails(
+      a.candidatesTokensDetails,
+      b.candidatesTokensDetails
+    )
+  };
+}
+function addCounts(a, b) {
+  return a == null && b == null ? b != null ? b : a : (a != null ? a : 0) + (b != null ? b : 0);
+}
+function addTokenDetails(a, b) {
+  var _a;
+  if (a == null || b == null) {
+    return b != null ? b : a;
+  }
+  const tokenCounts = /* @__PURE__ */ new Map();
+  for (const { modality, tokenCount } of [...a, ...b]) {
+    tokenCounts.set(modality, ((_a = tokenCounts.get(modality)) != null ? _a : 0) + tokenCount);
+  }
+  return Array.from(tokenCounts, ([modality, tokenCount]) => ({
+    modality,
+    tokenCount
+  }));
+}
+
+// src/continue-google-stream.ts
+function continueGoogleStream({
+  stream,
+  getContinuationToken,
+  continueStream
+}) {
+  let reader = stream.getReader();
+  let pendingContinuationToken;
+  return new ReadableStream({
+    async pull(controller) {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (!done) {
+          if (value.success) {
+            const token = getContinuationToken(value.value);
+            if (token != null) {
+              pendingContinuationToken = token;
+            }
+          }
+          controller.enqueue(value);
+          return;
+        }
+        if (pendingContinuationToken == null) {
+          controller.close();
+          return;
+        }
+        const continuationToken = pendingContinuationToken;
+        pendingContinuationToken = void 0;
+        try {
+          reader = (await continueStream(continuationToken)).getReader();
+        } catch (error) {
+          controller.enqueue({
+            success: false,
+            error,
+            rawValue: void 0
+          });
+          controller.close();
+          return;
+        }
+      }
+    },
+    async cancel(reason) {
+      await reader.cancel(reason);
+    }
+  });
+}
+
 // src/convert-json-schema-to-openapi-schema.ts
 import {
   UnsupportedFunctionalityError
@@ -698,7 +792,17 @@ var googleGenerativeAIProviderOptions = lazySchema2(
       /**
        * Optional. The service tier to use for the request.
        */
-      serviceTier: z2.enum(["standard", "flex", "priority"]).optional()
+      serviceTier: z2.enum(["standard", "flex", "priority"]).optional(),
+      /**
+       * Optional. Resumes a response that stopped with `MAX_TOKENS` because it
+       * used up its `maxOutputTokens` budget (Gemini 4 long decode). Pass the
+       * `continuationToken` from the previous response's provider metadata
+       * together with the original prompt and a higher `maxOutputTokens` (the
+       * budget is cumulative across requests). Requests that reach the
+       * per-request cap while budget remains (`CONTINUATION`) are continued
+       * automatically. Sent as the `continuationToken` body field.
+       */
+      continuationToken: z2.string().optional()
     })
   )
 );
@@ -998,6 +1102,9 @@ function mapGoogleGenerativeAIFinishReason({
     case "STOP":
       return hasToolCalls ? "tool-calls" : "stop";
     case "MAX_TOKENS":
+    // Gemini 4 long decode: the per-request cap was reached while budget
+    // remains. Continued automatically; only surfaced if continuing fails.
+    case "CONTINUATION":
       return "length";
     case "IMAGE_SAFETY":
     case "RECITATION":
@@ -1172,43 +1279,70 @@ var GoogleGenerativeAILanguageModel = class {
         } : googleToolConfig,
         cachedContent: googleOptions == null ? void 0 : googleOptions.cachedContent,
         labels: googleOptions == null ? void 0 : googleOptions.labels,
-        serviceTier: sanitizedServiceTier
+        serviceTier: sanitizedServiceTier,
+        ...(googleOptions == null ? void 0 : googleOptions.continuationToken) != null && {
+          continuationToken: googleOptions.continuationToken
+        }
       },
       warnings: [...warnings, ...toolWarnings]
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w;
     const { args, warnings } = await this.getArgs(options);
     const body = JSON.stringify(args);
     const mergedHeaders = combineHeaders(
       await resolve(this.config.headers),
       options.headers
     );
-    const {
-      responseHeaders,
-      value: response,
-      rawValue: rawResponse
-    } = await postJsonToApi({
+    const generateContent = (body2) => postJsonToApi({
       url: `${this.config.baseURL}/${getModelPath(
         this.modelId
       )}:generateContent`,
       headers: mergedHeaders,
-      body: args,
+      body: body2,
       failedResponseHandler: googleFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(responseSchema),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const candidate = (_a = response.candidates) == null ? void 0 : _a[0];
-    const promptBlockReason = (_b = response.promptFeedback) == null ? void 0 : _b.blockReason;
+    let {
+      responseHeaders,
+      value: response,
+      rawValue: rawResponse
+    } = await generateContent(args);
+    const continuedParts = [];
+    let continuedUsageMetadata;
+    const firstPromptTokenCount = (_a = response.usageMetadata) == null ? void 0 : _a.promptTokenCount;
+    while (true) {
+      const candidate2 = (_b = response.candidates) == null ? void 0 : _b[0];
+      const continuationToken = (candidate2 == null ? void 0 : candidate2.finishReason) === "CONTINUATION" ? candidate2.continuationToken : void 0;
+      if (continuationToken == null) {
+        break;
+      }
+      continuedParts.push(...(_d = (_c = candidate2 == null ? void 0 : candidate2.content) == null ? void 0 : _c.parts) != null ? _d : []);
+      continuedUsageMetadata = addGoogleGenerativeAIUsageMetadata(
+        continuedUsageMetadata,
+        (_e = response.usageMetadata) != null ? _e : void 0
+      );
+      ({
+        responseHeaders,
+        value: response,
+        rawValue: rawResponse
+      } = await generateContent({ ...args, continuationToken }));
+    }
+    const candidate = (_f = response.candidates) == null ? void 0 : _f[0];
+    const promptBlockReason = (_g = response.promptFeedback) == null ? void 0 : _g.blockReason;
     const isPromptBlocked = (candidate == null ? void 0 : candidate.finishReason) == null && promptBlockReason != null;
     const content = [];
-    const parts = (_d = (_c = candidate == null ? void 0 : candidate.content) == null ? void 0 : _c.parts) != null ? _d : [];
-    const usageMetadata = response.usageMetadata;
+    const parts = [...continuedParts, ...(_i = (_h = candidate == null ? void 0 : candidate.content) == null ? void 0 : _h.parts) != null ? _i : []];
+    const usageMetadata = addGoogleGenerativeAIUsageMetadata(
+      continuedUsageMetadata,
+      (_j = response.usageMetadata) != null ? _j : void 0
+    );
     let lastCodeExecutionToolCallId;
     for (const part of parts) {
-      if ("executableCode" in part && ((_e = part.executableCode) == null ? void 0 : _e.code)) {
+      if ("executableCode" in part && ((_k = part.executableCode) == null ? void 0 : _k.code)) {
         const toolCallId = this.config.generateId();
         lastCodeExecutionToolCallId = toolCallId;
         content.push({
@@ -1249,7 +1383,7 @@ var GoogleGenerativeAILanguageModel = class {
       } else if ("functionCall" in part) {
         content.push({
           type: "tool-call",
-          toolCallId: (_f = part.functionCall.id) != null ? _f : this.config.generateId(),
+          toolCallId: (_l = part.functionCall.id) != null ? _l : this.config.generateId(),
           toolName: part.functionCall.name,
           input: JSON.stringify(part.functionCall.args),
           providerMetadata: part.thoughtSignature ? { google: { thoughtSignature: part.thoughtSignature } } : void 0
@@ -1262,10 +1396,10 @@ var GoogleGenerativeAILanguageModel = class {
         });
       }
     }
-    const sources = (_g = extractSources({
+    const sources = (_m = extractSources({
       groundingMetadata: candidate == null ? void 0 : candidate.groundingMetadata,
       generateId: this.config.generateId
-    })) != null ? _g : [];
+    })) != null ? _m : [];
     for (const source of sources) {
       content.push(source);
     }
@@ -1276,20 +1410,22 @@ var GoogleGenerativeAILanguageModel = class {
         hasToolCalls: content.some((part) => part.type === "tool-call")
       }),
       usage: {
-        inputTokens: (_h = usageMetadata == null ? void 0 : usageMetadata.promptTokenCount) != null ? _h : void 0,
-        outputTokens: (_i = usageMetadata == null ? void 0 : usageMetadata.candidatesTokenCount) != null ? _i : void 0,
-        totalTokens: (_j = usageMetadata == null ? void 0 : usageMetadata.totalTokenCount) != null ? _j : void 0,
-        reasoningTokens: (_k = usageMetadata == null ? void 0 : usageMetadata.thoughtsTokenCount) != null ? _k : void 0,
-        cachedInputTokens: (_l = usageMetadata == null ? void 0 : usageMetadata.cachedContentTokenCount) != null ? _l : void 0
+        inputTokens: (_n = usageMetadata == null ? void 0 : usageMetadata.promptTokenCount) != null ? _n : void 0,
+        outputTokens: (_o = usageMetadata == null ? void 0 : usageMetadata.candidatesTokenCount) != null ? _o : void 0,
+        totalTokens: (_p = usageMetadata == null ? void 0 : usageMetadata.totalTokenCount) != null ? _p : void 0,
+        reasoningTokens: (_q = usageMetadata == null ? void 0 : usageMetadata.thoughtsTokenCount) != null ? _q : void 0,
+        cachedInputTokens: (_r = usageMetadata == null ? void 0 : usageMetadata.cachedContentTokenCount) != null ? _r : void 0
       },
       warnings,
       providerMetadata: {
         google: {
-          promptFeedback: (_m = response.promptFeedback) != null ? _m : null,
-          groundingMetadata: (_n = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _n : null,
-          urlContextMetadata: (_o = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _o : null,
-          safetyRatings: (_p = candidate == null ? void 0 : candidate.safetyRatings) != null ? _p : null,
-          serviceTier: (_q = response.serviceTier) != null ? _q : null,
+          promptFeedback: (_s = response.promptFeedback) != null ? _s : null,
+          groundingMetadata: (_t = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _t : null,
+          urlContextMetadata: (_u = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _u : null,
+          safetyRatings: (_v = candidate == null ? void 0 : candidate.safetyRatings) != null ? _v : null,
+          serviceTier: (_w = response.serviceTier) != null ? _w : null,
+          ...(candidate == null ? void 0 : candidate.continuationToken) != null ? { continuationToken: candidate.continuationToken } : {},
+          ...continuedUsageMetadata != null && firstPromptTokenCount != null ? { initialPromptTokenCount: firstPromptTokenCount } : {},
           usageMetadata: usageMetadata != null ? usageMetadata : null
         }
       },
@@ -1308,17 +1444,27 @@ var GoogleGenerativeAILanguageModel = class {
       await resolve(this.config.headers),
       options.headers
     );
-    const { responseHeaders, value: response } = await postJsonToApi({
+    const streamGenerateContent = (body2) => postJsonToApi({
       url: `${this.config.baseURL}/${getModelPath(
         this.modelId
       )}:streamGenerateContent?alt=sse`,
       headers,
-      body: args,
+      body: body2,
       failedResponseHandler: googleFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(chunkSchema),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
+    const { responseHeaders, value: firstResponse } = await streamGenerateContent(args);
+    const response = continueGoogleStream({
+      stream: firstResponse,
+      getContinuationToken: (chunk) => {
+        var _a, _b;
+        const candidate = (_a = chunk.candidates) == null ? void 0 : _a[0];
+        return (candidate == null ? void 0 : candidate.finishReason) === "CONTINUATION" ? (_b = candidate.continuationToken) != null ? _b : void 0 : void 0;
+      },
+      continueStream: async (continuationToken) => (await streamGenerateContent({ ...args, continuationToken })).value
+    });
     let finishReason = "unknown";
     const usage = {
       inputTokens: void 0,
@@ -1329,6 +1475,10 @@ var GoogleGenerativeAILanguageModel = class {
     let lastGroundingMetadata = null;
     let lastUrlContextMetadata = null;
     let serviceTier = null;
+    let requestUsageMetadata;
+    let continuedUsageMetadata;
+    let lastContinuationToken = null;
+    let initialPromptTokenCount = void 0;
     const generateId2 = this.config.generateId;
     let hasToolCalls = false;
     let currentTextBlockId = null;
@@ -1352,7 +1502,13 @@ var GoogleGenerativeAILanguageModel = class {
               return;
             }
             const value = chunk.value;
-            const usageMetadata = value.usageMetadata;
+            if (value.usageMetadata != null) {
+              requestUsageMetadata = value.usageMetadata;
+            }
+            const usageMetadata = value.usageMetadata == null ? void 0 : addGoogleGenerativeAIUsageMetadata(
+              continuedUsageMetadata,
+              requestUsageMetadata
+            );
             if (usageMetadata != null) {
               usage.inputTokens = (_a = usageMetadata.promptTokenCount) != null ? _a : void 0;
               usage.outputTokens = (_b = usageMetadata.candidatesTokenCount) != null ? _b : void 0;
@@ -1364,6 +1520,19 @@ var GoogleGenerativeAILanguageModel = class {
               serviceTier = value.serviceTier;
             }
             const candidate = (_f = value.candidates) == null ? void 0 : _f[0];
+            if ((candidate == null ? void 0 : candidate.continuationToken) != null) {
+              lastContinuationToken = candidate.continuationToken;
+            }
+            if ((candidate == null ? void 0 : candidate.finishReason) === "CONTINUATION") {
+              if (continuedUsageMetadata == null) {
+                initialPromptTokenCount = requestUsageMetadata == null ? void 0 : requestUsageMetadata.promptTokenCount;
+              }
+              continuedUsageMetadata = addGoogleGenerativeAIUsageMetadata(
+                continuedUsageMetadata,
+                requestUsageMetadata
+              );
+              requestUsageMetadata = void 0;
+            }
             if (candidate == null) {
               const promptBlockReason2 = (_g = value.promptFeedback) == null ? void 0 : _g.blockReason;
               if (promptBlockReason2 != null) {
@@ -1546,7 +1715,12 @@ var GoogleGenerativeAILanguageModel = class {
                   groundingMetadata: lastGroundingMetadata,
                   urlContextMetadata: lastUrlContextMetadata,
                   safetyRatings: (_o = candidate.safetyRatings) != null ? _o : null,
-                  serviceTier
+                  serviceTier,
+                  // Only surface the token when the response stopped early
+                  // (budget used up, or continuing failed), not for the
+                  // checkpoint tokens of a stream that finished normally.
+                  ...lastContinuationToken != null && (rawFinishReason === "MAX_TOKENS" || rawFinishReason === "CONTINUATION") ? { continuationToken: lastContinuationToken } : {},
+                  ...initialPromptTokenCount != null ? { initialPromptTokenCount } : {}
                 }
               };
               if (usageMetadata != null) {
@@ -1817,6 +1991,7 @@ var responseSchema = lazySchema3(
         z3.object({
           content: getContentSchema().nullish().or(z3.object({}).strict()),
           finishReason: z3.string().nullish(),
+          continuationToken: z3.string().nullish(),
           safetyRatings: z3.array(getSafetyRatingSchema()).nullish(),
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()
@@ -1838,6 +2013,7 @@ var chunkSchema = lazySchema3(
         z3.object({
           content: getContentSchema().nullish(),
           finishReason: z3.string().nullish(),
+          continuationToken: z3.string().nullish(),
           safetyRatings: z3.array(getSafetyRatingSchema()).nullish(),
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()