@ai-sdk/google 3.0.121 → 3.0.123

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/google",
-  "version": "3.0.121",
+  "version": "3.0.123",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -36,8 +36,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "3.0.15",
-    "@ai-sdk/provider-utils": "4.0.50"
+    "@ai-sdk/provider": "3.0.16",
+    "@ai-sdk/provider-utils": "4.0.51"
   },
   "devDependencies": {
     "@types/node": "20.17.24",

package/src/google-generative-ai-image-model.ts

@@ -159,7 +159,9 @@ export class GoogleGenerativeAIImageModel implements ImageModelV3 {
     };
 
     const { responseHeaders, value: response } = await postJsonToApi<{
-      predictions: Array<{ bytesBase64Encoded: string }>;
+      predictions: Array<
+        { bytesBase64Encoded: string } | { raiFilteredReason: string }
+      >;
     }>({
       url: `${this.config.baseURL}/models/${this.modelId}:predict`,
       headers: combineHeaders(await resolve(this.config.headers), headers),
@@ -171,14 +173,22 @@ export class GoogleGenerativeAIImageModel implements ImageModelV3 {
       abortSignal,
       fetch: this.config.fetch,
     });
+    const imagePredictions = response.predictions.filter(
+      (prediction): prediction is { bytesBase64Encoded: string } =>
+        'bytesBase64Encoded' in prediction,
+    );
+
     return {
-      images: response.predictions.map(
-        (p: { bytesBase64Encoded: string }) => p.bytesBase64Encoded,
+      images: imagePredictions.map(
+        ({ bytesBase64Encoded }) => bytesBase64Encoded,
       ),
+      ...(response.predictions.some(
+        prediction => 'raiFilteredReason' in prediction,
+      ) && { isRetryable: false }),
       warnings,
       providerMetadata: {
         google: {
-          images: response.predictions.map(() => ({
+          images: imagePredictions.map(() => ({
             // Add any prediction-specific metadata here
           })),
         },
@@ -341,6 +351,9 @@ export class GoogleGenerativeAIImageModel implements ImageModelV3 {
 
     return {
       images,
+      ...(result.finishReason.unified === 'content-filter'
+        ? { isRetryable: false }
+        : {}),
       warnings,
       providerMetadata: {
         google: {
@@ -375,7 +388,12 @@ const googleImageResponseSchema = lazySchema(() =>
   zodSchema(
     z.object({
       predictions: z
-        .array(z.object({ bytesBase64Encoded: z.string() }))
+        .array(
+          z.union([
+            z.object({ bytesBase64Encoded: z.string() }),
+            z.object({ raiFilteredReason: z.string() }),
+          ]),
+        )
         .default([]),
     }),
   ),

package/src/google-generative-ai-language-model.ts

@@ -615,9 +615,12 @@ export class GoogleGenerativeAILanguageModel implements LanguageModelV3 {
       raw: undefined,
     };
     let usage: GoogleGenerativeAIUsageMetadata | undefined = undefined;
-    let providerMetadata: SharedV3ProviderMetadata | undefined = undefined;
+    let promptFeedback: PromptFeedbackSchema | null = null;
     let lastGroundingMetadata: GroundingMetadataSchema | null = null;
     let lastUrlContextMetadata: UrlContextMetadataSchema | null = null;
+    let lastSafetyRatings: SafetyRatingSchema[] | null = null;
+    let lastFinishMessage: string | null = null;
+    let confirmedPromptBlockReason: string | undefined;
 
     const generateId = this.config.generateId;
     let hasToolCalls = false;
@@ -714,40 +717,48 @@ export class GoogleGenerativeAILanguageModel implements LanguageModelV3 {
               usage = usageMetadata;
             }
 
-            const candidate = value.candidates?.[0];
+            if (
+              value.promptFeedback != null &&
+              confirmedPromptBlockReason == null
+            ) {
+              promptFeedback = value.promptFeedback;
 
-            // sometimes the API returns an empty candidates array
-            if (candidate == null) {
-              const promptBlockReason = value.promptFeedback?.blockReason;
-              if (promptBlockReason != null) {
+              if (
+                isConfirmedPromptBlockReason(value.promptFeedback.blockReason)
+              ) {
+                confirmedPromptBlockReason = value.promptFeedback.blockReason;
                 finishReason = {
                   unified: 'content-filter',
-                  raw: promptBlockReason,
-                };
-                providerMetadata = {
-                  [providerOptionsName]: {
-                    promptFeedback: value.promptFeedback ?? null,
-                    groundingMetadata: lastGroundingMetadata,
-                    urlContextMetadata: lastUrlContextMetadata,
-                    safetyRatings: null,
-                    usageMetadata: usageMetadata ?? null,
-                    finishMessage: null,
-                    serviceTier: usage?.serviceTier ?? null,
-                  } satisfies GoogleGenerativeAIProviderMetadata,
+                  raw: confirmedPromptBlockReason,
                 };
               }
-              return;
             }
 
-            const content = candidate.content;
+            const candidate = value.candidates?.[0];
 
-            if (candidate.groundingMetadata != null) {
-              lastGroundingMetadata = candidate.groundingMetadata;
+            if (candidate != null) {
+              if (candidate.groundingMetadata != null) {
+                lastGroundingMetadata = candidate.groundingMetadata;
+              }
+              if (candidate.urlContextMetadata != null) {
+                lastUrlContextMetadata = candidate.urlContextMetadata;
+              }
+              if (candidate.safetyRatings != null) {
+                lastSafetyRatings = candidate.safetyRatings;
+              }
+              if (candidate.finishMessage != null) {
+                lastFinishMessage = candidate.finishMessage;
+              }
             }
-            if (candidate.urlContextMetadata != null) {
-              lastUrlContextMetadata = candidate.urlContextMetadata;
+
+            // A confirmed prompt block is terminal for generated content, but
+            // later chunks can still contribute usage and provider metadata.
+            if (confirmedPromptBlockReason != null || candidate == null) {
+              return;
             }
 
+            const content = candidate.content;
+
             const sources = extractSources({
               groundingMetadata: candidate.groundingMetadata,
               generateId,
@@ -1125,33 +1136,13 @@ export class GoogleGenerativeAILanguageModel implements LanguageModelV3 {
               }
             }
 
-            const promptBlockReason = value.promptFeedback?.blockReason;
-            const isPromptBlocked =
-              candidate.finishReason == null && promptBlockReason != null;
-            const rawFinishReason =
-              candidate.finishReason ?? promptBlockReason ?? undefined;
-
-            if (rawFinishReason != null) {
+            if (candidate.finishReason != null) {
               finishReason = {
-                unified: isPromptBlocked
-                  ? 'content-filter'
-                  : mapGoogleGenerativeAIFinishReason({
-                      finishReason: rawFinishReason,
-                      hasToolCalls,
-                    }),
-                raw: rawFinishReason,
-              };
-
-              providerMetadata = {
-                [providerOptionsName]: {
-                  promptFeedback: value.promptFeedback ?? null,
-                  groundingMetadata: lastGroundingMetadata,
-                  urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: candidate.safetyRatings ?? null,
-                  usageMetadata: usageMetadata ?? null,
-                  finishMessage: candidate.finishMessage ?? null,
-                  serviceTier: usage?.serviceTier ?? null,
-                } satisfies GoogleGenerativeAIProviderMetadata,
+                unified: mapGoogleGenerativeAIFinishReason({
+                  finishReason: candidate.finishReason,
+                  hasToolCalls,
+                }),
+                raw: candidate.finishReason,
               };
             }
           },
@@ -1174,7 +1165,17 @@ export class GoogleGenerativeAILanguageModel implements LanguageModelV3 {
               type: 'finish',
               finishReason,
               usage: convertGoogleGenerativeAIUsage(usage),
-              providerMetadata,
+              providerMetadata: {
+                [providerOptionsName]: {
+                  promptFeedback,
+                  groundingMetadata: lastGroundingMetadata,
+                  urlContextMetadata: lastUrlContextMetadata,
+                  safetyRatings: lastSafetyRatings,
+                  usageMetadata: usage ?? null,
+                  finishMessage: lastFinishMessage,
+                  serviceTier: usage?.serviceTier ?? null,
+                } satisfies GoogleGenerativeAIProviderMetadata,
+              },
             });
           },
         }),
@@ -1609,3 +1610,14 @@ const chunkSchema = lazySchema(() =>
 );
 
 type ChunkSchema = InferSchema<typeof chunkSchema>;
+
+function isConfirmedPromptBlockReason(
+  blockReason: string | null | undefined,
+): blockReason is string {
+  return (
+    blockReason != null &&
+    blockReason !== '' &&
+    blockReason !== 'BLOCK_REASON_UNSPECIFIED' &&
+    blockReason !== 'BLOCKED_REASON_UNSPECIFIED'
+  );
+}