@ai-sdk/google 4.0.99 → 4.0.101

package/docs/15-google.mdx

@@ -395,6 +395,14 @@ and long coding or agentic tasks.
 - Complex requests can run for several minutes. Make sure your HTTP client
   allows read timeouts of at least 15 to 20 minutes.
 
+**Grounding across requests.** When a response is continued automatically,
+`providerMetadata.google.groundingMetadataPerRequest` contains the grounding
+metadata from each request, in order. An entry is `null` if that request did
+not return grounding metadata. This preserves earlier Search results even if
+the final request returns none, and keeps citation indices scoped to their
+original request. The existing `groundingMetadata` field keeps its current
+behavior; use the per-request array when inspecting a continued response.
+
 **Usage across requests.** Each continued request re-sends the prompt plus all
 output so far as input, so its prompt token count includes everything
 generated before it. `usage` is the sum of all requests, which matches what

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/google",
-  "version": "4.0.99",
+  "version": "4.0.101",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -35,13 +35,14 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "4.0.29",
-    "@ai-sdk/provider-utils": "5.0.62"
+    "@ai-sdk/provider": "4.0.31",
+    "@ai-sdk/provider-utils": "5.0.64"
   },
   "devDependencies": {
     "@ai-sdk/test-server": "2.0.4",
     "@types/node": "22.19.19",
     "@vercel/ai-tsconfig": "0.0.0",
+    "@vercel/ai-tsdown-config": "0.0.0",
     "tsdown": "^0.23.0",
     "typescript": "6.0.3",
     "zod": "3.25.76"

package/src/google-language-model.ts

@@ -520,6 +520,7 @@ export class GoogleLanguageModel implements LanguageModelV4 {
     toolNameMapping,
     jsonResponseToolName,
     initialPromptTokenCount,
+    groundingMetadataPerRequest,
   }: {
     config: GoogleLanguageModelConfig;
     response: InferSchema<typeof responseSchema>;
@@ -528,6 +529,7 @@ export class GoogleLanguageModel implements LanguageModelV4 {
     toolNameMapping?: ReturnType<typeof createToolNameMapping>;
     jsonResponseToolName?: string;
     initialPromptTokenCount?: number | null;
+    groundingMetadataPerRequest?: Array<GroundingMetadataSchema | null>;
   }): LanguageModelV4GenerateResult {
     const wrapProviderMetadata = (payload: Record<string, unknown>) =>
       Object.fromEntries(
@@ -718,6 +720,9 @@ export class GoogleLanguageModel implements LanguageModelV4 {
       providerMetadata: wrapProviderMetadata({
         promptFeedback: response.promptFeedback ?? null,
         groundingMetadata: candidate?.groundingMetadata ?? null,
+        ...(groundingMetadataPerRequest != null
+          ? { groundingMetadataPerRequest }
+          : {}),
         urlContextMetadata: candidate?.urlContextMetadata ?? null,
         safetyRatings: candidate?.safetyRatings ?? null,
         usageMetadata: usageMetadata ?? null,
@@ -781,12 +786,15 @@ export class GoogleLanguageModel implements LanguageModelV4 {
     // the prompt plus all output so far as input, so the first request's
     // prompt token count is kept separately for context window checks.
     const content: LanguageModelV4Content[] = [];
+    const groundingMetadataPerRequest: Array<GroundingMetadataSchema | null> =
+      [];
     let usageMetadata: GoogleUsageMetadata | undefined;
     const firstPromptTokenCount = response.usageMetadata?.promptTokenCount;
     let isContinued = false;
 
     while (true) {
       const candidate = response.candidates?.[0];
+      groundingMetadataPerRequest.push(candidate?.groundingMetadata ?? null);
       const continuationToken =
         candidate?.finishReason === 'CONTINUATION'
           ? candidate.continuationToken
@@ -828,6 +836,9 @@ export class GoogleLanguageModel implements LanguageModelV4 {
       toolNameMapping,
       jsonResponseToolName,
       initialPromptTokenCount: isContinued ? firstPromptTokenCount : undefined,
+      groundingMetadataPerRequest: isContinued
+        ? groundingMetadataPerRequest
+        : undefined,
     });
 
     const result = {
@@ -885,6 +896,10 @@ export class GoogleLanguageModel implements LanguageModelV4 {
     const { responseHeaders, value: firstResponse } =
       await streamGenerateContent(args);
 
+    const groundingMetadataPerRequest: Array<GroundingMetadataSchema | null> =
+      [];
+    let requestGroundingMetadata: GroundingMetadataSchema | null = null;
+
     // Gemini 4 long decode: when a stream reaches the per-request cap while
     // `maxOutputTokens` budget remains, its last chunk finishes with
     // `CONTINUATION`. Resend the same request with the continuation token and
@@ -893,12 +908,22 @@ export class GoogleLanguageModel implements LanguageModelV4 {
       stream: firstResponse,
       getContinuationToken: chunk => {
         const candidate = chunk.candidates?.[0];
+        if (candidate?.groundingMetadata != null) {
+          requestGroundingMetadata = candidate.groundingMetadata;
+        }
         return candidate?.finishReason === 'CONTINUATION'
           ? (candidate.continuationToken ?? undefined)
           : undefined;
       },
-      continueStream: async continuationToken =>
-        (await streamGenerateContent({ ...args, continuationToken })).value,
+      continueStream: async continuationToken => {
+        const { value } = await streamGenerateContent({
+          ...args,
+          continuationToken,
+        });
+        groundingMetadataPerRequest.push(requestGroundingMetadata);
+        requestGroundingMetadata = null;
+        return value;
+      },
     });
 
     let finishReason: LanguageModelV4FinishReason = {
@@ -1489,6 +1514,14 @@ export class GoogleLanguageModel implements LanguageModelV4 {
               providerMetadata: wrapProviderMetadata({
                 promptFeedback,
                 groundingMetadata: lastGroundingMetadata,
+                ...(groundingMetadataPerRequest.length > 0
+                  ? {
+                      groundingMetadataPerRequest: [
+                        ...groundingMetadataPerRequest,
+                        requestGroundingMetadata,
+                      ],
+                    }
+                  : {}),
                 urlContextMetadata: lastUrlContextMetadata,
                 safetyRatings: lastSafetyRatings,
                 usageMetadata: usage ?? null,

package/src/google-prompt.ts

@@ -94,6 +94,8 @@ export type GoogleUsageMetadata = UsageMetadataSchema;
 export interface GoogleProviderMetadata {
   promptFeedback: GooglePromptFeedback | null;
   groundingMetadata: GoogleGroundingMetadata | null;
+  /** Grounding from each automatically continued request, in request order. */
+  groundingMetadataPerRequest?: Array<GoogleGroundingMetadata | null>;
   urlContextMetadata: GoogleUrlContextMetadata | null;
   safetyRatings: GoogleSafetyRating[] | null;
   usageMetadata: GoogleUsageMetadata | null;