@ai-sdk/google 3.0.121 → 3.0.123

package/dist/index.mjs

@@ -224,7 +224,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.121" : "0.0.0-test";
+var VERSION = true ? "3.0.123" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -2284,9 +2284,12 @@ var GoogleGenerativeAILanguageModel = class {
       raw: void 0
     };
     let usage = void 0;
-    let providerMetadata = void 0;
+    let promptFeedback = null;
     let lastGroundingMetadata = null;
     let lastUrlContextMetadata = null;
+    let lastSafetyRatings = null;
+    let lastFinishMessage = null;
+    let confirmedPromptBlockReason;
     const generateId3 = this.config.generateId;
     let hasToolCalls = false;
     let hasEmittedResponseMetadata = false;
@@ -2332,7 +2335,7 @@ var GoogleGenerativeAILanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -2352,35 +2355,35 @@ var GoogleGenerativeAILanguageModel = class {
             if (usageMetadata != null) {
               usage = usageMetadata;
             }
-            const candidate = (_a = value.candidates) == null ? void 0 : _a[0];
-            if (candidate == null) {
-              const promptBlockReason2 = (_b = value.promptFeedback) == null ? void 0 : _b.blockReason;
-              if (promptBlockReason2 != null) {
+            if (value.promptFeedback != null && confirmedPromptBlockReason == null) {
+              promptFeedback = value.promptFeedback;
+              if (isConfirmedPromptBlockReason(value.promptFeedback.blockReason)) {
+                confirmedPromptBlockReason = value.promptFeedback.blockReason;
                 finishReason = {
                   unified: "content-filter",
-                  raw: promptBlockReason2
-                };
-                providerMetadata = {
-                  [providerOptionsName]: {
-                    promptFeedback: (_c = value.promptFeedback) != null ? _c : null,
-                    groundingMetadata: lastGroundingMetadata,
-                    urlContextMetadata: lastUrlContextMetadata,
-                    safetyRatings: null,
-                    usageMetadata: usageMetadata != null ? usageMetadata : null,
-                    finishMessage: null,
-                    serviceTier: (_d = usage == null ? void 0 : usage.serviceTier) != null ? _d : null
-                  }
+                  raw: confirmedPromptBlockReason
                 };
               }
-              return;
             }
-            const content = candidate.content;
-            if (candidate.groundingMetadata != null) {
-              lastGroundingMetadata = candidate.groundingMetadata;
+            const candidate = (_a = value.candidates) == null ? void 0 : _a[0];
+            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;
+            if (confirmedPromptBlockReason != null || candidate == null) {
+              return;
             }
+            const content = candidate.content;
             const sources = extractSources({
               groundingMetadata: candidate.groundingMetadata,
               generateId: generateId3
@@ -2394,9 +2397,9 @@ var GoogleGenerativeAILanguageModel = class {
               }
             }
             if (content != null) {
-              const parts = (_e = content.parts) != null ? _e : [];
+              const parts = (_b = content.parts) != null ? _b : [];
               for (const part of parts) {
-                if ("executableCode" in part && ((_f = part.executableCode) == null ? void 0 : _f.code)) {
+                if ("executableCode" in part && ((_c = part.executableCode) == null ? void 0 : _c.code)) {
                   const toolCallId = generateId3();
                   lastCodeExecutionToolCallId = toolCallId;
                   controller.enqueue({
@@ -2415,7 +2418,7 @@ var GoogleGenerativeAILanguageModel = class {
                       toolName: "code_execution",
                       result: {
                         outcome: part.codeExecutionResult.outcome,
-                        output: (_g = part.codeExecutionResult.output) != null ? _g : ""
+                        output: (_d = part.codeExecutionResult.output) != null ? _d : ""
                       }
                     });
                   }
@@ -2509,7 +2512,7 @@ var GoogleGenerativeAILanguageModel = class {
                     providerMetadata: fileMeta
                   });
                 } else if ("toolCall" in part && part.toolCall) {
-                  const toolCallId = (_h = part.toolCall.id) != null ? _h : generateId3();
+                  const toolCallId = (_e = part.toolCall.id) != null ? _e : generateId3();
                   lastServerToolCallId = toolCallId;
                   const serverMeta = {
                     [providerOptionsName]: {
@@ -2522,13 +2525,13 @@ var GoogleGenerativeAILanguageModel = class {
                     type: "tool-call",
                     toolCallId,
                     toolName: `server:${part.toolCall.toolType}`,
-                    input: JSON.stringify((_i = part.toolCall.args) != null ? _i : {}),
+                    input: JSON.stringify((_f = part.toolCall.args) != null ? _f : {}),
                     providerExecuted: true,
                     dynamic: true,
                     providerMetadata: serverMeta
                   });
                 } else if ("toolResponse" in part && part.toolResponse) {
-                  const responseToolCallId = (_j = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _j : generateId3();
+                  const responseToolCallId = (_g = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _g : generateId3();
                   const serverMeta = {
                     [providerOptionsName]: {
                       ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
@@ -2540,7 +2543,7 @@ var GoogleGenerativeAILanguageModel = class {
                     type: "tool-result",
                     toolCallId: responseToolCallId,
                     toolName: `server:${part.toolResponse.toolType}`,
-                    result: (_k = part.toolResponse.response) != null ? _k : {},
+                    result: (_h = part.toolResponse.response) != null ? _h : {},
                     providerMetadata: serverMeta
                   });
                   lastServerToolCallId = void 0;
@@ -2559,7 +2562,7 @@ var GoogleGenerativeAILanguageModel = class {
                 const isNoArgsCompleteCall = part.functionCall.name != null && part.functionCall.args == null && part.functionCall.partialArgs == null && part.functionCall.willContinue !== true;
                 if (isStreamingChunk) {
                   if (part.functionCall.name != null) {
-                    const toolCallId = (_l = part.functionCall.id) != null ? _l : generateId3();
+                    const toolCallId = (_i = part.functionCall.id) != null ? _i : generateId3();
                     const accumulator = new GoogleJSONAccumulator();
                     activeStreamingToolCalls.push({
                       toolCallId,
@@ -2607,9 +2610,9 @@ var GoogleGenerativeAILanguageModel = class {
                 } else if (isTerminalChunk && activeStreamingToolCalls.length > 0) {
                   finishActiveStreamingToolCall(controller);
                 } else if (isCompleteCall) {
-                  const toolCallId = (_m = part.functionCall.id) != null ? _m : generateId3();
+                  const toolCallId = (_j = part.functionCall.id) != null ? _j : generateId3();
                   const toolName = part.functionCall.name;
-                  const args2 = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify((_n = part.functionCall.args) != null ? _n : {});
+                  const args2 = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify((_k = part.functionCall.args) != null ? _k : {});
                   controller.enqueue({
                     type: "tool-input-start",
                     id: toolCallId,
@@ -2636,7 +2639,7 @@ var GoogleGenerativeAILanguageModel = class {
                   });
                   hasToolCalls = true;
                 } else if (isNoArgsCompleteCall) {
-                  const toolCallId = (_o = part.functionCall.id) != null ? _o : generateId3();
+                  const toolCallId = (_l = part.functionCall.id) != null ? _l : generateId3();
                   const toolName = part.functionCall.name;
                   controller.enqueue({
                     type: "tool-input-start",
@@ -2660,31 +2663,18 @@ var GoogleGenerativeAILanguageModel = class {
                 }
               }
             }
-            const promptBlockReason = (_p = value.promptFeedback) == null ? void 0 : _p.blockReason;
-            const isPromptBlocked = candidate.finishReason == null && promptBlockReason != null;
-            const rawFinishReason = (_r = (_q = candidate.finishReason) != null ? _q : promptBlockReason) != null ? _r : void 0;
-            if (rawFinishReason != null) {
+            if (candidate.finishReason != null) {
               finishReason = {
-                unified: isPromptBlocked ? "content-filter" : mapGoogleGenerativeAIFinishReason({
-                  finishReason: rawFinishReason,
+                unified: mapGoogleGenerativeAIFinishReason({
+                  finishReason: candidate.finishReason,
                   hasToolCalls
                 }),
-                raw: rawFinishReason
-              };
-              providerMetadata = {
-                [providerOptionsName]: {
-                  promptFeedback: (_s = value.promptFeedback) != null ? _s : null,
-                  groundingMetadata: lastGroundingMetadata,
-                  urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_t = candidate.safetyRatings) != null ? _t : null,
-                  usageMetadata: usageMetadata != null ? usageMetadata : null,
-                  finishMessage: (_u = candidate.finishMessage) != null ? _u : null,
-                  serviceTier: (_v = usage == null ? void 0 : usage.serviceTier) != null ? _v : null
-                }
+                raw: candidate.finishReason
               };
             }
           },
           flush(controller) {
+            var _a;
             if (currentTextBlockId !== null) {
               controller.enqueue({
                 type: "text-end",
@@ -2701,7 +2691,17 @@ var GoogleGenerativeAILanguageModel = class {
               type: "finish",
               finishReason,
               usage: convertGoogleGenerativeAIUsage(usage),
-              providerMetadata
+              providerMetadata: {
+                [providerOptionsName]: {
+                  promptFeedback,
+                  groundingMetadata: lastGroundingMetadata,
+                  urlContextMetadata: lastUrlContextMetadata,
+                  safetyRatings: lastSafetyRatings,
+                  usageMetadata: usage != null ? usage : null,
+                  finishMessage: lastFinishMessage,
+                  serviceTier: (_a = usage == null ? void 0 : usage.serviceTier) != null ? _a : null
+                }
+              }
             });
           }
         })
@@ -2998,6 +2998,9 @@ var chunkSchema = lazySchema5(
     })
   )
 );
+function isConfirmedPromptBlockReason(blockReason) {
+  return blockReason != null && blockReason !== "" && blockReason !== "BLOCK_REASON_UNSPECIFIED" && blockReason !== "BLOCKED_REASON_UNSPECIFIED";
+}
 
 // src/tool/code-execution.ts
 import { createProviderToolFactoryWithOutputSchema } from "@ai-sdk/provider-utils";
@@ -3294,14 +3297,20 @@ var GoogleGenerativeAIImageModel = class {
       abortSignal,
       fetch: this.config.fetch
     });
+    const imagePredictions = response.predictions.filter(
+      (prediction) => "bytesBase64Encoded" in prediction
+    );
     return {
-      images: response.predictions.map(
-        (p) => 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
           }))
         }
@@ -3415,6 +3424,7 @@ var GoogleGenerativeAIImageModel = class {
     const languageModelGoogleMetadata = (_h = (_g = result.providerMetadata) == null ? void 0 : _g.google) != null ? _h : {};
     return {
       images,
+      ...result.finishReason.unified === "content-filter" ? { isRetryable: false } : {},
       warnings,
       providerMetadata: {
         google: {
@@ -3441,7 +3451,12 @@ function isGeminiModel(modelId) {
 var googleImageResponseSchema = lazySchema11(
   () => zodSchema11(
     z15.object({
-      predictions: z15.array(z15.object({ bytesBase64Encoded: z15.string() })).default([])
+      predictions: z15.array(
+        z15.union([
+          z15.object({ bytesBase64Encoded: z15.string() }),
+          z15.object({ raiFilteredReason: z15.string() })
+        ])
+      ).default([])
     })
   )
 );