@depup/ai-sdk__google 4.0.72-depup.0 → 4.0.75-depup.0

package/dist/internal/index.js

@@ -22,18 +22,20 @@ import { z as z3 } from "zod/v4";
 // src/convert-google-usage.ts
 import { createNullLanguageModelUsage } from "@ai-sdk/provider-utils";
 function convertGoogleUsage(usage) {
-  var _a, _b, _c, _d;
+  var _a, _b, _c, _d, _e;
   if (usage == null) {
     return createNullLanguageModelUsage();
   }
   const promptTokens = (_a = usage.promptTokenCount) != null ? _a : 0;
   const candidatesTokens = (_b = usage.candidatesTokenCount) != null ? _b : 0;
-  const cachedContentTokens = (_c = usage.cachedContentTokenCount) != null ? _c : 0;
-  const thoughtsTokens = (_d = usage.thoughtsTokenCount) != null ? _d : 0;
+  const toolUsePromptTokens = (_c = usage.toolUsePromptTokenCount) != null ? _c : 0;
+  const cachedContentTokens = (_d = usage.cachedContentTokenCount) != null ? _d : 0;
+  const thoughtsTokens = (_e = usage.thoughtsTokenCount) != null ? _e : 0;
+  const inputTokens = promptTokens + toolUsePromptTokens;
   return {
     inputTokens: {
-      total: promptTokens,
-      noCache: promptTokens - cachedContentTokens,
+      total: inputTokens,
+      noCache: inputTokens - cachedContentTokens,
       cacheRead: cachedContentTokens,
       cacheWrite: void 0
     },
@@ -1645,27 +1647,30 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     providerOptionsNames,
     toolNameMapping
   }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t;
     const wrapProviderMetadata = (payload) => Object.fromEntries(
       providerOptionsNames.map((name) => [name, payload])
     );
     const candidate = (_a = response.candidates) == null ? void 0 : _a[0];
     const promptBlockReason = (_b = response.promptFeedback) == null ? void 0 : _b.blockReason;
-    const isPromptBlocked = (candidate == null ? void 0 : candidate.finishReason) == null && promptBlockReason != null;
-    const rawFinishReason = (_d = (_c = candidate == null ? void 0 : candidate.finishReason) != null ? _c : promptBlockReason) != null ? _d : void 0;
+    const confirmedPromptBlockReason = isConfirmedPromptBlockReason(
+      promptBlockReason
+    ) ? promptBlockReason : void 0;
+    const isPromptBlocked = (candidate == null ? void 0 : candidate.finishReason) == null && confirmedPromptBlockReason != null;
+    const rawFinishReason = (_c = candidate == null ? void 0 : candidate.finishReason) != null ? _c : confirmedPromptBlockReason;
     const content = [];
-    const parts = (_f = (_e = candidate == null ? void 0 : candidate.content) == null ? void 0 : _e.parts) != null ? _f : [];
+    const parts = (_e = (_d = candidate == null ? void 0 : candidate.content) == null ? void 0 : _d.parts) != null ? _e : [];
     const usageMetadata = response.usageMetadata;
     let lastCodeExecutionToolCallId;
     let lastServerToolCallId;
     for (const part of parts) {
-      if ("executableCode" in part && ((_g = part.executableCode) == null ? void 0 : _g.code)) {
+      if ("executableCode" in part && ((_f = part.executableCode) == null ? void 0 : _f.code)) {
         const toolCallId = config.generateId();
         lastCodeExecutionToolCallId = toolCallId;
         content.push({
           type: "tool-call",
           toolCallId,
-          toolName: (_h = toolNameMapping == null ? void 0 : toolNameMapping.toCustomToolName("code_execution")) != null ? _h : "code_execution",
+          toolName: (_g = toolNameMapping == null ? void 0 : toolNameMapping.toCustomToolName("code_execution")) != null ? _g : "code_execution",
           input: JSON.stringify(part.executableCode),
           providerExecuted: true
         });
@@ -1674,10 +1679,10 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           type: "tool-result",
           // Results correspond to the most recent executable code part.
           toolCallId: lastCodeExecutionToolCallId,
-          toolName: (_i = toolNameMapping == null ? void 0 : toolNameMapping.toCustomToolName("code_execution")) != null ? _i : "code_execution",
+          toolName: (_h = toolNameMapping == null ? void 0 : toolNameMapping.toCustomToolName("code_execution")) != null ? _h : "code_execution",
           result: {
             outcome: part.codeExecutionResult.outcome,
-            output: (_j = part.codeExecutionResult.output) != null ? _j : ""
+            output: (_i = part.codeExecutionResult.output) != null ? _i : ""
           }
         });
       } else if ("text" in part && part.text != null) {
@@ -1701,7 +1706,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           type: "tool-call",
           toolCallId: part.functionCall.id || config.generateId(),
           toolName: part.functionCall.name,
-          input: JSON.stringify((_k = part.functionCall.args) != null ? _k : {}),
+          input: JSON.stringify((_j = part.functionCall.args) != null ? _j : {}),
           providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
             thoughtSignature: part.thoughtSignature
           }) : void 0
@@ -1724,7 +1729,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           type: "tool-call",
           toolCallId,
           toolName: `server:${part.toolCall.toolType}`,
-          input: JSON.stringify((_l = part.toolCall.args) != null ? _l : {}),
+          input: JSON.stringify((_k = part.toolCall.args) != null ? _k : {}),
           providerExecuted: true,
           dynamic: true,
           providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
@@ -1742,7 +1747,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           type: "tool-result",
           toolCallId: responseToolCallId,
           toolName: `server:${part.toolResponse.toolType}`,
-          result: (_m = part.toolResponse.response) != null ? _m : {},
+          result: (_l = part.toolResponse.response) != null ? _l : {},
           providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
             thoughtSignature: part.thoughtSignature,
             serverToolCallId: responseToolCallId,
@@ -1755,10 +1760,10 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         lastServerToolCallId = void 0;
       }
     }
-    const sources = (_n = extractSources({
+    const sources = (_m = extractSources({
       groundingMetadata: candidate == null ? void 0 : candidate.groundingMetadata,
       generateId: config.generateId
-    })) != null ? _n : [];
+    })) != null ? _m : [];
     for (const source of sources) {
       content.push(source);
     }
@@ -1777,17 +1782,17 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       usage: convertGoogleUsage(usageMetadata),
       warnings,
       providerMetadata: wrapProviderMetadata({
-        promptFeedback: (_o = response.promptFeedback) != null ? _o : null,
-        groundingMetadata: (_p = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _p : null,
-        urlContextMetadata: (_q = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _q : null,
-        safetyRatings: (_r = candidate == null ? void 0 : candidate.safetyRatings) != null ? _r : null,
+        promptFeedback: (_n = response.promptFeedback) != null ? _n : null,
+        groundingMetadata: (_o = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _o : null,
+        urlContextMetadata: (_p = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _p : null,
+        safetyRatings: (_q = candidate == null ? void 0 : candidate.safetyRatings) != null ? _q : null,
         usageMetadata: usageMetadata != null ? usageMetadata : null,
-        finishMessage: (_s = candidate == null ? void 0 : candidate.finishMessage) != null ? _s : null,
-        serviceTier: (_t = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _t : null
+        finishMessage: (_r = candidate == null ? void 0 : candidate.finishMessage) != null ? _r : null,
+        serviceTier: (_s = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _s : null
       }),
       response: {
         // TODO timestamp, model id
-        id: (_u = response.responseId) != null ? _u : void 0
+        id: (_t = response.responseId) != null ? _t : void 0
       }
     };
   }
@@ -1868,9 +1873,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 +1924,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 +1944,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 +1986,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 +2007,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 +2107,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 +2123,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 +2190,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 +2241,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 +2269,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 +2658,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 {