@ai-sdk/google 4.0.72 → 4.0.74

package/dist/internal/index.js

@@ -1868,9 +1868,12 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       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 generateId2 = this.config.generateId;
     let hasToolCalls = false;
     let hasEmittedResponseMetadata = false;
@@ -1916,7 +1919,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
             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;
+            var _a, _b, _c, _d, _e, _f, _g;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1936,33 +1939,35 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
             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
+                  raw: confirmedPromptBlockReason
                 };
-                providerMetadata = wrapProviderMetadata({
-                  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
-                });
               }
-              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: generateId2
@@ -1976,9 +1981,9 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
               }
             }
             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 = generateId2();
                   lastCodeExecutionToolCallId = toolCallId;
                   controller.enqueue({
@@ -1997,7 +2002,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                       toolName: toolNameMapping.toCustomToolName("code_execution"),
                       result: {
                         outcome: part.codeExecutionResult.outcome,
-                        output: (_g = part.codeExecutionResult.output) != null ? _g : ""
+                        output: (_d = part.codeExecutionResult.output) != null ? _d : ""
                       }
                     });
                   }
@@ -2097,7 +2102,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                     type: "tool-call",
                     toolCallId,
                     toolName: `server:${part.toolCall.toolType}`,
-                    input: JSON.stringify((_h = part.toolCall.args) != null ? _h : {}),
+                    input: JSON.stringify((_e = part.toolCall.args) != null ? _e : {}),
                     providerExecuted: true,
                     dynamic: true,
                     providerMetadata: serverMeta
@@ -2113,7 +2118,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                     type: "tool-result",
                     toolCallId: responseToolCallId,
                     toolName: `server:${part.toolResponse.toolType}`,
-                    result: (_i = part.toolResponse.response) != null ? _i : {},
+                    result: (_f = part.toolResponse.response) != null ? _f : {},
                     providerMetadata: serverMeta
                   });
                   lastServerToolCallId = void 0;
@@ -2180,7 +2185,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                 } else if (isCompleteCall) {
                   const toolCallId = part.functionCall.id || generateId2();
                   const toolName = part.functionCall.name;
-                  const args2 = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify((_j = part.functionCall.args) != null ? _j : {});
+                  const args2 = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify((_g = part.functionCall.args) != null ? _g : {});
                   controller.enqueue({
                     type: "tool-input-start",
                     id: toolCallId,
@@ -2231,29 +2236,18 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                 }
               }
             }
-            const promptBlockReason = (_k = value.promptFeedback) == null ? void 0 : _k.blockReason;
-            const isPromptBlocked = candidate.finishReason == null && promptBlockReason != null;
-            const rawFinishReason = (_m = (_l = candidate.finishReason) != null ? _l : promptBlockReason) != null ? _m : void 0;
-            if (rawFinishReason != null) {
+            if (candidate.finishReason != null) {
               finishReason = {
-                unified: isPromptBlocked ? "content-filter" : mapGoogleFinishReason({
-                  finishReason: rawFinishReason,
+                unified: mapGoogleFinishReason({
+                  finishReason: candidate.finishReason,
                   hasToolCalls
                 }),
-                raw: rawFinishReason
+                raw: candidate.finishReason
               };
-              providerMetadata = wrapProviderMetadata({
-                promptFeedback: (_n = value.promptFeedback) != null ? _n : null,
-                groundingMetadata: lastGroundingMetadata,
-                urlContextMetadata: lastUrlContextMetadata,
-                safetyRatings: (_o = candidate.safetyRatings) != null ? _o : null,
-                usageMetadata: usageMetadata != null ? usageMetadata : null,
-                finishMessage: (_p = candidate.finishMessage) != null ? _p : null,
-                serviceTier: (_q = usage == null ? void 0 : usage.serviceTier) != null ? _q : null
-              });
             }
           },
           flush(controller) {
+            var _a;
             if (currentTextBlockId !== null) {
               controller.enqueue({
                 type: "text-end",
@@ -2270,7 +2264,15 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
               type: "finish",
               finishReason,
               usage: convertGoogleUsage(usage),
-              providerMetadata
+              providerMetadata: wrapProviderMetadata({
+                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
+              })
             });
           }
         })
@@ -2651,6 +2653,9 @@ var chunkSchema = lazySchema3(
     })
   )
 );
+function isConfirmedPromptBlockReason(blockReason) {
+  return blockReason != null && blockReason !== "" && blockReason !== "BLOCK_REASON_UNSPECIFIED" && blockReason !== "BLOCKED_REASON_UNSPECIFIED";
+}
 
 // src/google-speech-model.ts
 import {