@ai-sdk/google 4.0.101 → 4.0.103

package/docs/15-google.mdx

@@ -878,7 +878,7 @@ import { generateText } from 'ai';
 const { text, sources, providerMetadata } = await generateText({
   model: google('gemini-4-argon'),
   prompt: `Based on the document: https://ai.google.dev/gemini-api/docs/url-context.
-          Answer this question: How many links we can consume in one request?`,
+          Answer this question: How many links can we consume in one request?`,
   tools: {
     url_context: google.tools.urlContext({}),
   },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/google",
-  "version": "4.0.101",
+  "version": "4.0.103",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -35,8 +35,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "4.0.31",
-    "@ai-sdk/provider-utils": "5.0.64"
+    "@ai-sdk/provider": "4.0.32",
+    "@ai-sdk/provider-utils": "5.0.66"
   },
   "devDependencies": {
     "@ai-sdk/test-server": "2.0.4",

package/src/continue-google-stream.ts

@@ -1,5 +1,9 @@
 import type { ParseResult } from '@ai-sdk/provider-utils';
 
+export type GoogleStreamPart<T> =
+  | ParseResult<T>
+  | { type: 'continuation-start' };
+
 /**
  * Concatenates the event streams of a long decode response that spans several
  * requests (Gemini 4 long decode).
@@ -7,7 +11,9 @@ import type { ParseResult } from '@ai-sdk/provider-utils';
  * Events of the current stream are passed through unchanged. When an event
  * signals that the per-request cap was reached while budget remains
  * (`getContinuationToken` returns a token), the next request is sent with that
- * token once the current stream ends, and its events are appended. If the
+ * token once the current stream ends, and its events are appended after a
+ * `continuation-start` marker. The marker orders per-response state changes
+ * with buffered events, including when the new stream is empty. If the
  * continuation request fails, the error is emitted as a failed parse result
  * and the stream ends.
  */
@@ -21,11 +27,11 @@ export function continueGoogleStream<T>({
   continueStream: (
     continuationToken: string,
   ) => Promise<ReadableStream<ParseResult<T>>>;
-}): ReadableStream<ParseResult<T>> {
+}): ReadableStream<GoogleStreamPart<T>> {
   let reader = stream.getReader();
   let pendingContinuationToken: string | undefined;
 
-  return new ReadableStream<ParseResult<T>>({
+  return new ReadableStream<GoogleStreamPart<T>>({
     async pull(controller) {
       while (true) {
         const { done, value } = await reader.read();
@@ -51,6 +57,8 @@ export function continueGoogleStream<T>({
 
         try {
           reader = (await continueStream(continuationToken)).getReader();
+          controller.enqueue({ type: 'continuation-start' });
+          return;
         } catch (error) {
           // Surfaced as an `error` stream part, followed by the `finish` part
           // with the last continuation token so the caller can resume.

package/src/google-batch.ts

@@ -78,35 +78,38 @@ function assertSupportedBatchRequests(
   }
 }
 
-const googleRpcStatusSchema = z.object({
-  code: z.union([z.number(), z.string()]).nullish(),
-  message: z.string().nullish(),
-  status: z.string().nullish(),
-});
-
-const googleBatchStatsSchema = z.object({
-  requestCount: z.union([z.string(), z.number()]).nullish(),
-  successfulRequestCount: z.union([z.string(), z.number()]).nullish(),
-  failedRequestCount: z.union([z.string(), z.number()]).nullish(),
-  pendingRequestCount: z.union([z.string(), z.number()]).nullish(),
-});
-
-const googleBatchOutputSchema = z.object({
-  responsesFile: z.string().nullish(),
-  inlinedResponses: z
-    .object({
-      inlinedResponses: z.array(
-        z.object({
-          metadata: z.object({
-            key: z.string(),
+const googleRpcStatusSchema = () =>
+  z.object({
+    code: z.union([z.number(), z.string()]).nullish(),
+    message: z.string().nullish(),
+    status: z.string().nullish(),
+  });
+
+const googleBatchStatsSchema = () =>
+  z.object({
+    requestCount: z.union([z.string(), z.number()]).nullish(),
+    successfulRequestCount: z.union([z.string(), z.number()]).nullish(),
+    failedRequestCount: z.union([z.string(), z.number()]).nullish(),
+    pendingRequestCount: z.union([z.string(), z.number()]).nullish(),
+  });
+
+const googleBatchOutputSchema = () =>
+  z.object({
+    responsesFile: z.string().nullish(),
+    inlinedResponses: z
+      .object({
+        inlinedResponses: z.array(
+          z.object({
+            metadata: z.object({
+              key: z.string(),
+            }),
+            response: z.unknown().nullish(),
+            error: googleRpcStatusSchema().nullish(),
           }),
-          response: z.unknown().nullish(),
-          error: googleRpcStatusSchema.nullish(),
-        }),
-      ),
-    })
-    .nullish(),
-});
+        ),
+      })
+      .nullish(),
+  });
 
 const googleBatchOperationZodSchema = () =>
   z.object({
@@ -115,13 +118,13 @@ const googleBatchOperationZodSchema = () =>
       .object({
         state: z.string().nullish(),
         createTime: z.string().nullish(),
-        batchStats: googleBatchStatsSchema.nullish(),
-        output: googleBatchOutputSchema.nullish(),
+        batchStats: googleBatchStatsSchema().nullish(),
+        output: googleBatchOutputSchema().nullish(),
       })
       .nullish(),
     done: z.boolean().nullish(),
-    error: googleRpcStatusSchema.nullish(),
-    response: googleBatchOutputSchema.nullish(),
+    error: googleRpcStatusSchema().nullish(),
+    response: googleBatchOutputSchema().nullish(),
   });
 
 const googleBatchOperationSchema = lazySchema(() =>
@@ -159,7 +162,7 @@ const googleBatchResultLineSchema = lazySchema(() =>
     z.object({
       key: z.string(),
       response: z.unknown().nullish(),
-      error: googleRpcStatusSchema.nullish(),
+      error: googleRpcStatusSchema().nullish(),
     }),
   ),
 );
@@ -872,7 +875,7 @@ function parseCount(value: string | number | null | undefined) {
 }
 
 function convertGoogleRpcError(
-  error: InferSchema<typeof googleRpcStatusSchema>,
+  error: InferSchema<ReturnType<typeof googleRpcStatusSchema>>,
   fallbackMessage: string,
 ): BatchV4Error {
   return {

package/src/google-embedding-model-options.ts

@@ -11,21 +11,22 @@ export type GoogleEmbeddingModelId =
   | 'gemini-embedding-2-preview'
   | (string & {});
 
-const googleEmbeddingContentPartSchema = z.union([
-  z.object({ text: z.string() }),
-  z.object({
-    inlineData: z.object({
-      mimeType: z.string(),
-      data: z.string(),
+const googleEmbeddingContentPartSchema = () =>
+  z.union([
+    z.object({ text: z.string() }),
+    z.object({
+      inlineData: z.object({
+        mimeType: z.string(),
+        data: z.string(),
+      }),
     }),
-  }),
-  z.object({
-    fileData: z.object({
-      fileUri: z.string(),
-      mimeType: z.string(),
+    z.object({
+      fileData: z.object({
+        fileUri: z.string(),
+        mimeType: z.string(),
+      }),
     }),
-  }),
-]);
+  ]);
 
 export const googleEmbeddingModelOptions = lazySchema(() =>
   zodSchema(
@@ -71,7 +72,7 @@ export const googleEmbeddingModelOptions = lazySchema(() =>
        * the case of a single embedding, the array length must be 1.
        */
       content: z
-        .array(z.array(googleEmbeddingContentPartSchema).min(1).nullable())
+        .array(z.array(googleEmbeddingContentPartSchema()).min(1).nullable())
         .optional(),
     }),
   ),

package/src/google-files.ts

@@ -13,6 +13,7 @@ import {
   lazySchema,
   parseProviderOptions,
   zodSchema,
+  encodePathSegment,
   getFromApi,
   type FetchFunction,
 } from '@ai-sdk/provider-utils';
@@ -34,17 +35,6 @@ interface GoogleFilesConfig {
   fetch?: FetchFunction;
 }
 
-function encodePathSegment(value: string): string {
-  const encodedValue = encodeURIComponent(value);
-
-  // URL parsing normalizes both literal and percent-encoded dot segments.
-  return encodedValue === '.'
-    ? '%252E'
-    : encodedValue === '..'
-      ? '%252E%252E'
-      : encodedValue;
-}
-
 export class GoogleFiles implements FilesV4 {
   readonly specificationVersion = 'v4';
 

package/src/google-image-model-options.ts

@@ -18,7 +18,7 @@ export const googleImageModelOptionsSchema = lazySchema(() =>
        * dedicated escape hatch for grounding image generation the same way
        * `generateText` does.
        */
-      googleSearch: googleSearchToolArgsBaseSchema.optional(),
+      googleSearch: googleSearchToolArgsBaseSchema().optional(),
     }),
   ),
 );

package/src/google-language-model.ts

@@ -33,11 +33,13 @@ import {
   zodSchema,
   type FetchFunction,
   type InferSchema,
-  type ParseResult,
   type Resolvable,
 } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
-import { continueGoogleStream } from './continue-google-stream';
+import {
+  continueGoogleStream,
+  type GoogleStreamPart,
+} from './continue-google-stream';
 import {
   addGoogleUsageMetadata,
   convertGoogleUsage,
@@ -926,8 +928,9 @@ export class GoogleLanguageModel implements LanguageModelV4 {
       },
     });
 
+    // A clean end without the provider's completion signal is interrupted.
     let finishReason: LanguageModelV4FinishReason = {
-      unified: 'other',
+      unified: 'interrupted',
       raw: undefined,
     };
     let usage: GoogleUsageMetadata | undefined = undefined;
@@ -1005,7 +1008,7 @@ export class GoogleLanguageModel implements LanguageModelV4 {
     const result = {
       stream: response.pipeThrough(
         new TransformStream<
-          ParseResult<ChunkSchema>,
+          GoogleStreamPart<ChunkSchema>,
           LanguageModelV4StreamPart
         >({
           start(controller) {
@@ -1013,11 +1016,21 @@ export class GoogleLanguageModel implements LanguageModelV4 {
           },
 
           transform(chunk, controller) {
+            if ('type' in chunk) {
+              // Reset in stream order, after the prior response's buffered
+              // CONTINUATION chunk, rather than when the request starts.
+              finishReason = { unified: 'interrupted', raw: undefined };
+              return;
+            }
+
             if (options.includeRawChunks) {
               controller.enqueue({ type: 'raw', rawValue: chunk.rawValue });
             }
 
             if (!chunk.success) {
+              if (finishReason.unified === 'interrupted') {
+                finishReason = { unified: 'error', raw: undefined };
+              }
               controller.enqueue({ type: 'error', error: chunk.error });
               return;
             }
@@ -1889,14 +1902,15 @@ export const getGroundingMetadataSchema = () =>
       .nullish(),
   });
 
-const partialArgSchema = z.object({
-  jsonPath: z.string(),
-  stringValue: z.string().nullish(),
-  numberValue: z.number().nullish(),
-  boolValue: z.boolean().nullish(),
-  nullValue: z.unknown().nullish(),
-  willContinue: z.boolean().nullish(),
-});
+const partialArgSchema = () =>
+  z.object({
+    jsonPath: z.string(),
+    stringValue: z.string().nullish(),
+    numberValue: z.number().nullish(),
+    boolValue: z.boolean().nullish(),
+    nullValue: z.unknown().nullish(),
+    willContinue: z.boolean().nullish(),
+  });
 
 const getContentSchema = () =>
   z.object({
@@ -1909,7 +1923,7 @@ const getContentSchema = () =>
               id: z.string().nullish(),
               name: z.string().nullish(),
               args: z.unknown().nullish(),
-              partialArgs: z.array(partialArgSchema).nullish(),
+              partialArgs: z.array(partialArgSchema()).nullish(),
               willContinue: z.boolean().nullish(),
             }),
             thoughtSignature: z.string().nullish(),
@@ -1971,35 +1985,37 @@ const getSafetyRatingSchema = () =>
     blocked: z.boolean().nullish(),
   });
 
-const tokenDetailsSchema = z
-  .array(
-    z
-      .object({
-        modality: z.string(),
-        tokenCount: z.number(),
-      })
-      .loose(),
-  )
-  .nullish();
-
-const usageSchema = z
-  .object({
-    cachedContentTokenCount: z.number().nullish(),
-    thoughtsTokenCount: z.number().nullish(),
-    promptTokenCount: z.number().nullish(),
-    candidatesTokenCount: z.number().nullish(),
-    toolUsePromptTokenCount: z.number().nullish(),
-    totalTokenCount: z.number().nullish(),
-    // https://cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1/GenerateContentResponse#TrafficType
-    trafficType: z.string().nullish(),
-    serviceTier: z.string().nullish(),
-    // https://ai.google.dev/api/generate-content#Modality
-    promptTokensDetails: tokenDetailsSchema,
-    cacheTokensDetails: tokenDetailsSchema,
-    candidatesTokensDetails: tokenDetailsSchema,
-    toolUsePromptTokensDetails: tokenDetailsSchema,
-  })
-  .loose();
+const tokenDetailsSchema = () =>
+  z
+    .array(
+      z
+        .object({
+          modality: z.string(),
+          tokenCount: z.number(),
+        })
+        .loose(),
+    )
+    .nullish();
+
+const usageSchema = () =>
+  z
+    .object({
+      cachedContentTokenCount: z.number().nullish(),
+      thoughtsTokenCount: z.number().nullish(),
+      promptTokenCount: z.number().nullish(),
+      candidatesTokenCount: z.number().nullish(),
+      toolUsePromptTokenCount: z.number().nullish(),
+      totalTokenCount: z.number().nullish(),
+      // https://cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1/GenerateContentResponse#TrafficType
+      trafficType: z.string().nullish(),
+      serviceTier: z.string().nullish(),
+      // https://ai.google.dev/api/generate-content#Modality
+      promptTokensDetails: tokenDetailsSchema(),
+      cacheTokensDetails: tokenDetailsSchema(),
+      candidatesTokensDetails: tokenDetailsSchema(),
+      toolUsePromptTokensDetails: tokenDetailsSchema(),
+    })
+    .loose();
 
 // https://ai.google.dev/api/generate-content#UrlRetrievalMetadata
 export const getUrlContextMetadataSchema = () =>
@@ -2031,7 +2047,7 @@ export const responseSchema = lazySchema(() =>
           }),
         )
         .nullish(),
-      usageMetadata: usageSchema.nullish(),
+      usageMetadata: usageSchema().nullish(),
       promptFeedback: z
         .object({
           blockReason: z.string().nullish(),
@@ -2086,7 +2102,7 @@ const chunkSchema = lazySchema(() =>
           }),
         )
         .nullish(),
-      usageMetadata: usageSchema.nullish(),
+      usageMetadata: usageSchema().nullish(),
       promptFeedback: z
         .object({
           blockReason: z.string().nullish(),

package/src/google-speech-model-options.ts

@@ -13,25 +13,28 @@ export type GoogleSpeechModelId =
   | 'gemini-3.8-flash-lite-tts'
   | (string & {});
 
-const prebuiltVoiceConfigSchema = z.object({
-  voiceName: z.string(),
-});
+const prebuiltVoiceConfigSchema = () =>
+  z.object({
+    voiceName: z.string(),
+  });
 
-const voiceConfigSchema = z.object({
-  prebuiltVoiceConfig: prebuiltVoiceConfigSchema,
-  voice: z.never().optional(),
-});
+const voiceConfigSchema = () =>
+  z.object({
+    prebuiltVoiceConfig: prebuiltVoiceConfigSchema(),
+    voice: z.never().optional(),
+  });
 
-const speechMetadataSchema = z.object({
-  speaker: z.string().min(1).optional(),
-  style: z.string().optional(),
-});
+const speechMetadataSchema = () =>
+  z.object({
+    speaker: z.string().min(1).optional(),
+    style: z.string().optional(),
+  });
 
 export const googleSpeechProviderOptionsSchema = lazySchema(() =>
   zodSchema(
     z.object({
       /** Turn-level directions for the top-level text, for Gemini 3.8 TTS. */
-      speechMetadata: speechMetadataSchema.optional(),
+      speechMetadata: speechMetadataSchema().optional(),
 
       /**
        * Structured transcript for Gemini 3.8 TTS. Replaces the top-level text;
@@ -42,7 +45,7 @@ export const googleSpeechProviderOptionsSchema = lazySchema(() =>
         .array(
           z.object({
             text: z.string(),
-            speechMetadata: speechMetadataSchema.optional(),
+            speechMetadata: speechMetadataSchema().optional(),
           }),
         )
         .min(1)
@@ -61,7 +64,7 @@ export const googleSpeechProviderOptionsSchema = lazySchema(() =>
           speakerVoiceConfigs: z.array(
             z.object({
               speaker: z.string(),
-              voiceConfig: voiceConfigSchema,
+              voiceConfig: voiceConfigSchema(),
             }),
           ),
         })

package/src/google-video-model.ts

@@ -9,6 +9,9 @@ import {
   type SharedV4Warning,
 } from '@ai-sdk/provider';
 import {
+  lazySchema,
+  zodSchema,
+  type InferSchema,
   combineHeaders,
   convertUint8ArrayToBase64,
   createJsonResponseHandler,
@@ -265,7 +268,7 @@ export class GoogleVideoModel implements VideoModelV4 {
   }
 
   private async buildCompletedResult(
-    finalOperation: z.infer<typeof googleOperationSchema>,
+    finalOperation: InferSchema<typeof googleOperationSchema>,
     responseHeaders: Record<string, string> | undefined,
     warnings: SharedV4Warning[],
     currentDate: Date,
@@ -441,33 +444,37 @@ export class GoogleVideoModel implements VideoModelV4 {
   }
 }
 
-const googleOperationSchema = z.object({
-  name: z.string().nullish(),
-  done: z.boolean().nullish(),
-  error: z
-    .object({
-      code: z.number().nullish(),
-      message: z.string(),
-      status: z.string().nullish(),
-    })
-    .nullish(),
-  response: z
-    .object({
-      generateVideoResponse: z
+const googleOperationSchema = lazySchema(() =>
+  zodSchema(
+    z.object({
+      name: z.string().nullish(),
+      done: z.boolean().nullish(),
+      error: z
         .object({
-          generatedSamples: z
-            .array(
-              z.object({
-                video: z
-                  .object({
-                    uri: z.string().nullish(),
-                  })
-                  .nullish(),
-              }),
-            )
+          code: z.number().nullish(),
+          message: z.string(),
+          status: z.string().nullish(),
+        })
+        .nullish(),
+      response: z
+        .object({
+          generateVideoResponse: z
+            .object({
+              generatedSamples: z
+                .array(
+                  z.object({
+                    video: z
+                      .object({
+                        uri: z.string().nullish(),
+                      })
+                      .nullish(),
+                  }),
+                )
+                .nullish(),
+            })
             .nullish(),
         })
         .nullish(),
-    })
-    .nullish(),
-});
+    }),
+  ),
+);

package/src/interactions/build-google-interactions-stream-transform.ts

@@ -152,6 +152,7 @@ export function buildGoogleInteractionsStreamTransform({
   let serviceTier: string | undefined = headerServiceTier;
   let finishStatus: GoogleInteractionsStatus | string | undefined;
   let hasFunctionCall = false;
+  let hasCompletionEvent = false;
 
   /*
    * Per-index open step slots. The Interactions API frames concurrent steps
@@ -838,6 +839,7 @@ export function buildGoogleInteractionsStreamTransform({
         }
 
         case 'interaction.completed': {
+          hasCompletionEvent = true;
           const event = value as Extract<
             GoogleInteractionsEvent,
             { event_type: 'interaction.completed' }
@@ -893,12 +895,17 @@ export function buildGoogleInteractionsStreamTransform({
     },
 
     flush(controller) {
+      const isInterrupted =
+        !hasCompletionEvent &&
+        (finishStatus == null || finishStatus === 'in_progress');
       const finishReason: LanguageModelV4FinishReason = {
-        unified: mapGoogleInteractionsFinishReason({
-          status: finishStatus,
-          hasFunctionCall,
-        }),
-        raw: finishStatus,
+        unified: isInterrupted
+          ? 'interrupted'
+          : mapGoogleInteractionsFinishReason({
+              status: finishStatus,
+              hasFunctionCall,
+            }),
+        raw: isInterrupted ? undefined : finishStatus,
       };
 
       const outputTokensByModality =