@ai-sdk/google 3.0.116 → 3.0.119

package/dist/internal/index.d.mts

@@ -82,7 +82,8 @@ declare const getUrlContextMetadataSchema: () => z.ZodObject<{
     }, z.core.$strip>>>>;
 }, z.core.$strip>;
 declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
-    candidates: {
+    responseId?: string | null | undefined;
+    candidates?: {
         content?: Record<string, never> | {
             parts?: ({
                 functionCall: {
@@ -198,21 +199,34 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
                 urlRetrievalStatus: string;
             }[] | null | undefined;
         } | null | undefined;
-    }[];
-    responseId?: string | null | undefined;
+    }[] | null | undefined;
     usageMetadata?: {
+        [x: string]: z.core.util.JSONType;
         cachedContentTokenCount?: number | null | undefined;
         thoughtsTokenCount?: number | null | undefined;
         promptTokenCount?: number | null | undefined;
         candidatesTokenCount?: number | null | undefined;
+        toolUsePromptTokenCount?: number | null | undefined;
         totalTokenCount?: number | null | undefined;
         trafficType?: string | null | undefined;
         serviceTier?: string | null | undefined;
         promptTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
+            modality: string;
+            tokenCount: number;
+        }[] | null | undefined;
+        cacheTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
             modality: string;
             tokenCount: number;
         }[] | null | undefined;
         candidatesTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
+            modality: string;
+            tokenCount: number;
+        }[] | null | undefined;
+        toolUsePromptTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
             modality: string;
             tokenCount: number;
         }[] | null | undefined;
@@ -229,9 +243,10 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
         }[] | null | undefined;
     } | null | undefined;
 }>;
-type GroundingMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['groundingMetadata']>;
-type UrlContextMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['urlContextMetadata']>;
-type SafetyRatingSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['safetyRatings']>[number];
+type CandidateSchema = NonNullable<InferSchema<typeof responseSchema>['candidates']>[number];
+type GroundingMetadataSchema = NonNullable<CandidateSchema['groundingMetadata']>;
+type UrlContextMetadataSchema = NonNullable<CandidateSchema['urlContextMetadata']>;
+type SafetyRatingSchema = NonNullable<CandidateSchema['safetyRatings']>[number];
 type PromptFeedbackSchema = NonNullable<InferSchema<typeof responseSchema>['promptFeedback']>;
 type UsageMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['usageMetadata']>;
 

package/dist/internal/index.d.ts

@@ -82,7 +82,8 @@ declare const getUrlContextMetadataSchema: () => z.ZodObject<{
     }, z.core.$strip>>>>;
 }, z.core.$strip>;
 declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
-    candidates: {
+    responseId?: string | null | undefined;
+    candidates?: {
         content?: Record<string, never> | {
             parts?: ({
                 functionCall: {
@@ -198,21 +199,34 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
                 urlRetrievalStatus: string;
             }[] | null | undefined;
         } | null | undefined;
-    }[];
-    responseId?: string | null | undefined;
+    }[] | null | undefined;
     usageMetadata?: {
+        [x: string]: z.core.util.JSONType;
         cachedContentTokenCount?: number | null | undefined;
         thoughtsTokenCount?: number | null | undefined;
         promptTokenCount?: number | null | undefined;
         candidatesTokenCount?: number | null | undefined;
+        toolUsePromptTokenCount?: number | null | undefined;
         totalTokenCount?: number | null | undefined;
         trafficType?: string | null | undefined;
         serviceTier?: string | null | undefined;
         promptTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
+            modality: string;
+            tokenCount: number;
+        }[] | null | undefined;
+        cacheTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
             modality: string;
             tokenCount: number;
         }[] | null | undefined;
         candidatesTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
+            modality: string;
+            tokenCount: number;
+        }[] | null | undefined;
+        toolUsePromptTokensDetails?: {
+            [x: string]: z.core.util.JSONType;
             modality: string;
             tokenCount: number;
         }[] | null | undefined;
@@ -229,9 +243,10 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
         }[] | null | undefined;
     } | null | undefined;
 }>;
-type GroundingMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['groundingMetadata']>;
-type UrlContextMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['urlContextMetadata']>;
-type SafetyRatingSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['safetyRatings']>[number];
+type CandidateSchema = NonNullable<InferSchema<typeof responseSchema>['candidates']>[number];
+type GroundingMetadataSchema = NonNullable<CandidateSchema['groundingMetadata']>;
+type UrlContextMetadataSchema = NonNullable<CandidateSchema['urlContextMetadata']>;
+type SafetyRatingSchema = NonNullable<CandidateSchema['safetyRatings']>[number];
 type PromptFeedbackSchema = NonNullable<InferSchema<typeof responseSchema>['promptFeedback']>;
 type UsageMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['usageMetadata']>;
 

package/dist/internal/index.js

@@ -1473,6 +1473,7 @@ var configurableSafetySettingCategories = [
   "HARM_CATEGORY_HARASSMENT",
   "HARM_CATEGORY_SEXUALLY_EXPLICIT"
 ];
+var gemini25ModelPattern2 = /(^|\/)gemini-2\.5(?:[.-]|$)/i;
 var GoogleGenerativeAILanguageModel = class {
   constructor(modelId, config) {
     this.specificationVersion = "v3";
@@ -1576,6 +1577,19 @@ var GoogleGenerativeAILanguageModel = class {
       }
     }
     const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
+    const isGemini25DeveloperApiModel = !isVertexProvider && gemini25ModelPattern2.test(this.modelId);
+    if (isGemini25DeveloperApiModel && frequencyPenalty != null) {
+      warnings.push({
+        type: "unsupported",
+        feature: "frequencyPenalty"
+      });
+    }
+    if (isGemini25DeveloperApiModel && presencePenalty != null) {
+      warnings.push({
+        type: "unsupported",
+        feature: "presencePenalty"
+      });
+    }
     const { usesGemini3Features } = getGoogleModelCapabilities(this.modelId);
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(
       prompt,
@@ -1624,8 +1638,8 @@ var GoogleGenerativeAILanguageModel = class {
           temperature,
           topK,
           topP,
-          frequencyPenalty,
-          presencePenalty,
+          frequencyPenalty: isGemini25DeveloperApiModel ? void 0 : frequencyPenalty,
+          presencePenalty: isGemini25DeveloperApiModel ? void 0 : presencePenalty,
           stopSequences,
           seed,
           // response format:
@@ -1660,7 +1674,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v;
     const { args, warnings, providerOptionsName, extraHeaders } = await this.getArgs(options);
     const mergedHeaders = (0, import_provider_utils4.combineHeaders)(
       await (0, import_provider_utils4.resolve)(this.config.headers),
@@ -1682,14 +1696,17 @@ var GoogleGenerativeAILanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const candidate = response.candidates[0];
+    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 content = [];
-    const parts = (_b = (_a = candidate.content) == null ? void 0 : _a.parts) != null ? _b : [];
+    const parts = (_f = (_e = candidate == null ? void 0 : candidate.content) == null ? void 0 : _e.parts) != null ? _f : [];
     const usageMetadata = response.usageMetadata;
     let lastCodeExecutionToolCallId;
     let lastServerToolCallId;
     for (const part of parts) {
-      if ("executableCode" in part && ((_c = part.executableCode) == null ? void 0 : _c.code)) {
+      if ("executableCode" in part && ((_g = part.executableCode) == null ? void 0 : _g.code)) {
         const toolCallId = this.config.generateId();
         lastCodeExecutionToolCallId = toolCallId;
         content.push({
@@ -1707,7 +1724,7 @@ var GoogleGenerativeAILanguageModel = class {
           toolName: "code_execution",
           result: {
             outcome: part.codeExecutionResult.outcome,
-            output: (_d = part.codeExecutionResult.output) != null ? _d : ""
+            output: (_h = part.codeExecutionResult.output) != null ? _h : ""
           }
         });
       } else if ("text" in part && part.text != null) {
@@ -1731,9 +1748,9 @@ var GoogleGenerativeAILanguageModel = class {
       } else if ("functionCall" in part && part.functionCall.name != null) {
         content.push({
           type: "tool-call",
-          toolCallId: (_e = part.functionCall.id) != null ? _e : this.config.generateId(),
+          toolCallId: (_i = part.functionCall.id) != null ? _i : this.config.generateId(),
           toolName: part.functionCall.name,
-          input: JSON.stringify((_f = part.functionCall.args) != null ? _f : {}),
+          input: JSON.stringify((_j = part.functionCall.args) != null ? _j : {}),
           providerMetadata: part.thoughtSignature ? {
             [providerOptionsName]: {
               thoughtSignature: part.thoughtSignature
@@ -1755,13 +1772,13 @@ var GoogleGenerativeAILanguageModel = class {
           } : void 0
         });
       } else if ("toolCall" in part && part.toolCall) {
-        const toolCallId = (_g = part.toolCall.id) != null ? _g : this.config.generateId();
+        const toolCallId = (_k = part.toolCall.id) != null ? _k : this.config.generateId();
         lastServerToolCallId = toolCallId;
         content.push({
           type: "tool-call",
           toolCallId,
           toolName: `server:${part.toolCall.toolType}`,
-          input: JSON.stringify((_h = part.toolCall.args) != null ? _h : {}),
+          input: JSON.stringify((_l = part.toolCall.args) != null ? _l : {}),
           providerExecuted: true,
           dynamic: true,
           providerMetadata: part.thoughtSignature ? {
@@ -1778,12 +1795,12 @@ var GoogleGenerativeAILanguageModel = class {
           }
         });
       } else if ("toolResponse" in part && part.toolResponse) {
-        const responseToolCallId = (_i = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _i : this.config.generateId();
+        const responseToolCallId = (_m = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _m : this.config.generateId();
         content.push({
           type: "tool-result",
           toolCallId: responseToolCallId,
           toolName: `server:${part.toolResponse.toolType}`,
-          result: (_j = part.toolResponse.response) != null ? _j : {},
+          result: (_n = part.toolResponse.response) != null ? _n : {},
           providerMetadata: part.thoughtSignature ? {
             [providerOptionsName]: {
               thoughtSignature: part.thoughtSignature,
@@ -1800,42 +1817,42 @@ var GoogleGenerativeAILanguageModel = class {
         lastServerToolCallId = void 0;
       }
     }
-    const sources = (_k = extractSources({
-      groundingMetadata: candidate.groundingMetadata,
+    const sources = (_o = extractSources({
+      groundingMetadata: candidate == null ? void 0 : candidate.groundingMetadata,
       generateId: this.config.generateId
-    })) != null ? _k : [];
+    })) != null ? _o : [];
     for (const source of sources) {
       content.push(source);
     }
     return {
       content,
       finishReason: {
-        unified: mapGoogleGenerativeAIFinishReason({
-          finishReason: candidate.finishReason,
+        unified: isPromptBlocked ? "content-filter" : mapGoogleGenerativeAIFinishReason({
+          finishReason: rawFinishReason,
           // Only count client-executed tool calls for finish reason determination.
           hasToolCalls: content.some(
             (part) => part.type === "tool-call" && !part.providerExecuted
           )
         }),
-        raw: (_l = candidate.finishReason) != null ? _l : void 0
+        raw: rawFinishReason
       },
       usage: convertGoogleGenerativeAIUsage(usageMetadata),
       warnings,
       providerMetadata: {
         [providerOptionsName]: {
-          promptFeedback: (_m = response.promptFeedback) != null ? _m : null,
-          groundingMetadata: (_n = candidate.groundingMetadata) != null ? _n : null,
-          urlContextMetadata: (_o = candidate.urlContextMetadata) != null ? _o : null,
-          safetyRatings: (_p = candidate.safetyRatings) != null ? _p : null,
+          promptFeedback: (_p = response.promptFeedback) != null ? _p : null,
+          groundingMetadata: (_q = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _q : null,
+          urlContextMetadata: (_r = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _r : null,
+          safetyRatings: (_s = candidate == null ? void 0 : candidate.safetyRatings) != null ? _s : null,
           usageMetadata: usageMetadata != null ? usageMetadata : null,
-          finishMessage: (_q = candidate.finishMessage) != null ? _q : null,
-          serviceTier: (_r = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _r : null
+          finishMessage: (_t = candidate == null ? void 0 : candidate.finishMessage) != null ? _t : null,
+          serviceTier: (_u = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _u : null
         }
       },
       request: { body: args },
       response: {
         // TODO timestamp, model id
-        id: (_s = response.responseId) != null ? _s : void 0,
+        id: (_v = response.responseId) != null ? _v : void 0,
         headers: responseHeaders,
         body: rawResponse
       }
@@ -1912,7 +1929,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;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1934,6 +1951,24 @@ var GoogleGenerativeAILanguageModel = class {
             }
             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) {
+                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
+                  }
+                };
+              }
               return;
             }
             const content = candidate.content;
@@ -1956,9 +1991,9 @@ var GoogleGenerativeAILanguageModel = class {
               }
             }
             if (content != null) {
-              const parts = (_b = content.parts) != null ? _b : [];
+              const parts = (_e = content.parts) != null ? _e : [];
               for (const part of parts) {
-                if ("executableCode" in part && ((_c = part.executableCode) == null ? void 0 : _c.code)) {
+                if ("executableCode" in part && ((_f = part.executableCode) == null ? void 0 : _f.code)) {
                   const toolCallId = generateId2();
                   lastCodeExecutionToolCallId = toolCallId;
                   controller.enqueue({
@@ -1977,7 +2012,7 @@ var GoogleGenerativeAILanguageModel = class {
                       toolName: "code_execution",
                       result: {
                         outcome: part.codeExecutionResult.outcome,
-                        output: (_d = part.codeExecutionResult.output) != null ? _d : ""
+                        output: (_g = part.codeExecutionResult.output) != null ? _g : ""
                       }
                     });
                   }
@@ -2071,7 +2106,7 @@ var GoogleGenerativeAILanguageModel = class {
                     providerMetadata: fileMeta
                   });
                 } else if ("toolCall" in part && part.toolCall) {
-                  const toolCallId = (_e = part.toolCall.id) != null ? _e : generateId2();
+                  const toolCallId = (_h = part.toolCall.id) != null ? _h : generateId2();
                   lastServerToolCallId = toolCallId;
                   const serverMeta = {
                     [providerOptionsName]: {
@@ -2084,13 +2119,13 @@ var GoogleGenerativeAILanguageModel = class {
                     type: "tool-call",
                     toolCallId,
                     toolName: `server:${part.toolCall.toolType}`,
-                    input: JSON.stringify((_f = part.toolCall.args) != null ? _f : {}),
+                    input: JSON.stringify((_i = part.toolCall.args) != null ? _i : {}),
                     providerExecuted: true,
                     dynamic: true,
                     providerMetadata: serverMeta
                   });
                 } else if ("toolResponse" in part && part.toolResponse) {
-                  const responseToolCallId = (_g = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _g : generateId2();
+                  const responseToolCallId = (_j = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _j : generateId2();
                   const serverMeta = {
                     [providerOptionsName]: {
                       ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
@@ -2102,7 +2137,7 @@ var GoogleGenerativeAILanguageModel = class {
                     type: "tool-result",
                     toolCallId: responseToolCallId,
                     toolName: `server:${part.toolResponse.toolType}`,
-                    result: (_h = part.toolResponse.response) != null ? _h : {},
+                    result: (_k = part.toolResponse.response) != null ? _k : {},
                     providerMetadata: serverMeta
                   });
                   lastServerToolCallId = void 0;
@@ -2121,7 +2156,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 = (_i = part.functionCall.id) != null ? _i : generateId2();
+                    const toolCallId = (_l = part.functionCall.id) != null ? _l : generateId2();
                     const accumulator = new GoogleJSONAccumulator();
                     activeStreamingToolCalls.push({
                       toolCallId,
@@ -2169,9 +2204,9 @@ var GoogleGenerativeAILanguageModel = class {
                 } else if (isTerminalChunk && activeStreamingToolCalls.length > 0) {
                   finishActiveStreamingToolCall(controller);
                 } else if (isCompleteCall) {
-                  const toolCallId = (_j = part.functionCall.id) != null ? _j : generateId2();
+                  const toolCallId = (_m = part.functionCall.id) != null ? _m : generateId2();
                   const toolName = part.functionCall.name;
-                  const args2 = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify((_k = part.functionCall.args) != null ? _k : {});
+                  const args2 = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify((_n = part.functionCall.args) != null ? _n : {});
                   controller.enqueue({
                     type: "tool-input-start",
                     id: toolCallId,
@@ -2198,7 +2233,7 @@ var GoogleGenerativeAILanguageModel = class {
                   });
                   hasToolCalls = true;
                 } else if (isNoArgsCompleteCall) {
-                  const toolCallId = (_l = part.functionCall.id) != null ? _l : generateId2();
+                  const toolCallId = (_o = part.functionCall.id) != null ? _o : generateId2();
                   const toolName = part.functionCall.name;
                   controller.enqueue({
                     type: "tool-input-start",
@@ -2222,23 +2257,26 @@ var GoogleGenerativeAILanguageModel = class {
                 }
               }
             }
-            if (candidate.finishReason != null) {
+            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) {
               finishReason = {
-                unified: mapGoogleGenerativeAIFinishReason({
-                  finishReason: candidate.finishReason,
+                unified: isPromptBlocked ? "content-filter" : mapGoogleGenerativeAIFinishReason({
+                  finishReason: rawFinishReason,
                   hasToolCalls
                 }),
-                raw: candidate.finishReason
+                raw: rawFinishReason
               };
               providerMetadata = {
                 [providerOptionsName]: {
-                  promptFeedback: (_m = value.promptFeedback) != null ? _m : null,
+                  promptFeedback: (_s = value.promptFeedback) != null ? _s : null,
                   groundingMetadata: lastGroundingMetadata,
                   urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_n = candidate.safetyRatings) != null ? _n : null,
+                  safetyRatings: (_t = candidate.safetyRatings) != null ? _t : null,
                   usageMetadata: usageMetadata != null ? usageMetadata : null,
-                  finishMessage: (_o = candidate.finishMessage) != null ? _o : null,
-                  serviceTier: (_p = usage == null ? void 0 : usage.serviceTier) != null ? _p : null
+                  finishMessage: (_u = candidate.finishMessage) != null ? _u : null,
+                  serviceTier: (_v = usage == null ? void 0 : usage.serviceTier) != null ? _v : null
                 }
               };
             }
@@ -2487,21 +2525,24 @@ var tokenDetailsSchema = import_v43.z.array(
   import_v43.z.object({
     modality: import_v43.z.string(),
     tokenCount: import_v43.z.number()
-  })
+  }).catchall(import_v43.z.json())
 ).nullish();
 var usageSchema = import_v43.z.object({
   cachedContentTokenCount: import_v43.z.number().nullish(),
   thoughtsTokenCount: import_v43.z.number().nullish(),
   promptTokenCount: import_v43.z.number().nullish(),
   candidatesTokenCount: import_v43.z.number().nullish(),
+  toolUsePromptTokenCount: import_v43.z.number().nullish(),
   totalTokenCount: import_v43.z.number().nullish(),
   // https://cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1/GenerateContentResponse#TrafficType
   trafficType: import_v43.z.string().nullish(),
   serviceTier: import_v43.z.string().nullish(),
   // https://ai.google.dev/api/generate-content#Modality
   promptTokensDetails: tokenDetailsSchema,
-  candidatesTokensDetails: tokenDetailsSchema
-});
+  cacheTokensDetails: tokenDetailsSchema,
+  candidatesTokensDetails: tokenDetailsSchema,
+  toolUsePromptTokensDetails: tokenDetailsSchema
+}).catchall(import_v43.z.json());
 var getUrlContextMetadataSchema = () => import_v43.z.object({
   urlMetadata: import_v43.z.array(
     import_v43.z.object({
@@ -2523,7 +2564,7 @@ var responseSchema = (0, import_provider_utils4.lazySchema)(
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()
         })
-      ),
+      ).nullish(),
       usageMetadata: usageSchema.nullish(),
       promptFeedback: import_v43.z.object({
         blockReason: import_v43.z.string().nullish(),