@ai-sdk/google 1.0.9 → 1.0.11

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @ai-sdk/google
 
+## 1.0.11
+
+### Patch Changes
+
+- db31e74: feat: adding audioTimestamp support to GoogleGenerativeAISettings
+
+## 1.0.10
+
+### Patch Changes
+
+- e07439a: feat (provider/google): Include safety ratings response detail.
+- 4017b0f: feat (provider/google-vertex): Enhance grounding metadata response detail.
+- a9df182: feat (provider/google): Add support for search grounding.
+
 ## 1.0.9
 
 ### Patch Changes

package/dist/index.d.mts

@@ -26,6 +26,18 @@ interface GoogleGenerativeAISettings {
         category: 'HARM_CATEGORY_UNSPECIFIED' | 'HARM_CATEGORY_HATE_SPEECH' | 'HARM_CATEGORY_DANGEROUS_CONTENT' | 'HARM_CATEGORY_HARASSMENT' | 'HARM_CATEGORY_SEXUALLY_EXPLICIT' | 'HARM_CATEGORY_CIVIC_INTEGRITY';
         threshold: 'HARM_BLOCK_THRESHOLD_UNSPECIFIED' | 'BLOCK_LOW_AND_ABOVE' | 'BLOCK_MEDIUM_AND_ABOVE' | 'BLOCK_ONLY_HIGH' | 'BLOCK_NONE';
     }>;
+    /**
+     * Optional. Enables timestamp understanding for audio-only files.
+     *
+     * https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/audio-understanding
+     */
+    audioTimestamp?: boolean;
+    /**
+  Optional. When enabled, the model will use Google search to ground the response.
+  
+  @see https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/overview
+     */
+    useSearchGrounding?: boolean;
 }
 
 type GoogleGenerativeAIEmbeddingModelId = 'text-embedding-004' | (string & {});
@@ -91,6 +103,7 @@ declare const google: GoogleGenerativeAIProvider;
 
 declare const groundingMetadataSchema: z.ZodObject<{
     webSearchQueries: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
+    retrievalQueries: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
     searchEntryPoint: z.ZodOptional<z.ZodNullable<z.ZodObject<{
         renderedContent: z.ZodString;
     }, "strip", z.ZodTypeAny, {
@@ -98,6 +111,46 @@ declare const groundingMetadataSchema: z.ZodObject<{
     }, {
         renderedContent: string;
     }>>>;
+    groundingChunks: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
+        web: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+            uri: z.ZodString;
+            title: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            uri: string;
+            title: string;
+        }, {
+            uri: string;
+            title: string;
+        }>>>;
+        retrievedContext: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+            uri: z.ZodString;
+            title: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            uri: string;
+            title: string;
+        }, {
+            uri: string;
+            title: string;
+        }>>>;
+    }, "strip", z.ZodTypeAny, {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }, {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }>, "many">>>;
     groundingSupports: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
         segment: z.ZodObject<{
             startIndex: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
@@ -112,24 +165,33 @@ declare const groundingMetadataSchema: z.ZodObject<{
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         }>;
-        groundingChunkIndices: z.ZodArray<z.ZodNumber, "many">;
-        confidenceScores: z.ZodArray<z.ZodNumber, "many">;
+        segment_text: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+        groundingChunkIndices: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
+        supportChunkIndices: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
+        confidenceScores: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
+        confidenceScore: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
     }, "strip", z.ZodTypeAny, {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }, {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }>, "many">>>;
     retrievalMetadata: z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodObject<{
         webDynamicRetrievalScore: z.ZodNumber;
@@ -140,43 +202,95 @@ declare const groundingMetadataSchema: z.ZodObject<{
     }>, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>>;
 }, "strip", z.ZodTypeAny, {
     webSearchQueries?: string[] | null | undefined;
+    retrievalQueries?: string[] | null | undefined;
     searchEntryPoint?: {
         renderedContent: string;
     } | null | undefined;
+    groundingChunks?: {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }[] | null | undefined;
     groundingSupports?: {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }[] | null | undefined;
     retrievalMetadata?: {
         webDynamicRetrievalScore: number;
     } | {} | null | undefined;
 }, {
     webSearchQueries?: string[] | null | undefined;
+    retrievalQueries?: string[] | null | undefined;
     searchEntryPoint?: {
         renderedContent: string;
     } | null | undefined;
+    groundingChunks?: {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }[] | null | undefined;
     groundingSupports?: {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }[] | null | undefined;
     retrievalMetadata?: {
         webDynamicRetrievalScore: number;
     } | {} | null | undefined;
 }>;
+declare const safetyRatingSchema: z.ZodObject<{
+    category: z.ZodString;
+    probability: z.ZodString;
+    probabilityScore: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+    severity: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    severityScore: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+    blocked: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
+}, "strip", z.ZodTypeAny, {
+    category: string;
+    probability: string;
+    probabilityScore?: number | null | undefined;
+    severity?: string | null | undefined;
+    severityScore?: number | null | undefined;
+    blocked?: boolean | null | undefined;
+}, {
+    category: string;
+    probability: string;
+    probabilityScore?: number | null | undefined;
+    severity?: string | null | undefined;
+    severityScore?: number | null | undefined;
+    blocked?: boolean | null | undefined;
+}>;
 
 type GoogleGenerativeAIGroundingMetadata = z.infer<typeof groundingMetadataSchema>;
+type GoogleGenerativeAISafetyRating = z.infer<typeof safetyRatingSchema>;
 interface GoogleGenerativeAIProviderMetadata {
     groundingMetadata: GoogleGenerativeAIGroundingMetadata | null;
+    safetyRatings: GoogleGenerativeAISafetyRating[] | null;
 }
 
 export { type GoogleGenerativeAIProvider, type GoogleGenerativeAIProviderMetadata, type GoogleGenerativeAIProviderSettings, createGoogleGenerativeAI, google };

package/dist/index.d.ts

@@ -26,6 +26,18 @@ interface GoogleGenerativeAISettings {
         category: 'HARM_CATEGORY_UNSPECIFIED' | 'HARM_CATEGORY_HATE_SPEECH' | 'HARM_CATEGORY_DANGEROUS_CONTENT' | 'HARM_CATEGORY_HARASSMENT' | 'HARM_CATEGORY_SEXUALLY_EXPLICIT' | 'HARM_CATEGORY_CIVIC_INTEGRITY';
         threshold: 'HARM_BLOCK_THRESHOLD_UNSPECIFIED' | 'BLOCK_LOW_AND_ABOVE' | 'BLOCK_MEDIUM_AND_ABOVE' | 'BLOCK_ONLY_HIGH' | 'BLOCK_NONE';
     }>;
+    /**
+     * Optional. Enables timestamp understanding for audio-only files.
+     *
+     * https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/audio-understanding
+     */
+    audioTimestamp?: boolean;
+    /**
+  Optional. When enabled, the model will use Google search to ground the response.
+  
+  @see https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/overview
+     */
+    useSearchGrounding?: boolean;
 }
 
 type GoogleGenerativeAIEmbeddingModelId = 'text-embedding-004' | (string & {});
@@ -91,6 +103,7 @@ declare const google: GoogleGenerativeAIProvider;
 
 declare const groundingMetadataSchema: z.ZodObject<{
     webSearchQueries: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
+    retrievalQueries: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString, "many">>>;
     searchEntryPoint: z.ZodOptional<z.ZodNullable<z.ZodObject<{
         renderedContent: z.ZodString;
     }, "strip", z.ZodTypeAny, {
@@ -98,6 +111,46 @@ declare const groundingMetadataSchema: z.ZodObject<{
     }, {
         renderedContent: string;
     }>>>;
+    groundingChunks: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
+        web: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+            uri: z.ZodString;
+            title: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            uri: string;
+            title: string;
+        }, {
+            uri: string;
+            title: string;
+        }>>>;
+        retrievedContext: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+            uri: z.ZodString;
+            title: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            uri: string;
+            title: string;
+        }, {
+            uri: string;
+            title: string;
+        }>>>;
+    }, "strip", z.ZodTypeAny, {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }, {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }>, "many">>>;
     groundingSupports: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
         segment: z.ZodObject<{
             startIndex: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
@@ -112,24 +165,33 @@ declare const groundingMetadataSchema: z.ZodObject<{
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         }>;
-        groundingChunkIndices: z.ZodArray<z.ZodNumber, "many">;
-        confidenceScores: z.ZodArray<z.ZodNumber, "many">;
+        segment_text: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+        groundingChunkIndices: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
+        supportChunkIndices: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
+        confidenceScores: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
+        confidenceScore: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodNumber, "many">>>;
     }, "strip", z.ZodTypeAny, {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }, {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }>, "many">>>;
     retrievalMetadata: z.ZodOptional<z.ZodNullable<z.ZodUnion<[z.ZodObject<{
         webDynamicRetrievalScore: z.ZodNumber;
@@ -140,43 +202,95 @@ declare const groundingMetadataSchema: z.ZodObject<{
     }>, z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>]>>>;
 }, "strip", z.ZodTypeAny, {
     webSearchQueries?: string[] | null | undefined;
+    retrievalQueries?: string[] | null | undefined;
     searchEntryPoint?: {
         renderedContent: string;
     } | null | undefined;
+    groundingChunks?: {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }[] | null | undefined;
     groundingSupports?: {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }[] | null | undefined;
     retrievalMetadata?: {
         webDynamicRetrievalScore: number;
     } | {} | null | undefined;
 }, {
     webSearchQueries?: string[] | null | undefined;
+    retrievalQueries?: string[] | null | undefined;
     searchEntryPoint?: {
         renderedContent: string;
     } | null | undefined;
+    groundingChunks?: {
+        web?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+        retrievedContext?: {
+            uri: string;
+            title: string;
+        } | null | undefined;
+    }[] | null | undefined;
     groundingSupports?: {
         segment: {
             startIndex?: number | null | undefined;
             endIndex?: number | null | undefined;
             text?: string | null | undefined;
         };
-        groundingChunkIndices: number[];
-        confidenceScores: number[];
+        segment_text?: string | null | undefined;
+        groundingChunkIndices?: number[] | null | undefined;
+        supportChunkIndices?: number[] | null | undefined;
+        confidenceScores?: number[] | null | undefined;
+        confidenceScore?: number[] | null | undefined;
     }[] | null | undefined;
     retrievalMetadata?: {
         webDynamicRetrievalScore: number;
     } | {} | null | undefined;
 }>;
+declare const safetyRatingSchema: z.ZodObject<{
+    category: z.ZodString;
+    probability: z.ZodString;
+    probabilityScore: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+    severity: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    severityScore: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+    blocked: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
+}, "strip", z.ZodTypeAny, {
+    category: string;
+    probability: string;
+    probabilityScore?: number | null | undefined;
+    severity?: string | null | undefined;
+    severityScore?: number | null | undefined;
+    blocked?: boolean | null | undefined;
+}, {
+    category: string;
+    probability: string;
+    probabilityScore?: number | null | undefined;
+    severity?: string | null | undefined;
+    severityScore?: number | null | undefined;
+    blocked?: boolean | null | undefined;
+}>;
 
 type GoogleGenerativeAIGroundingMetadata = z.infer<typeof groundingMetadataSchema>;
+type GoogleGenerativeAISafetyRating = z.infer<typeof safetyRatingSchema>;
 interface GoogleGenerativeAIProviderMetadata {
     groundingMetadata: GoogleGenerativeAIGroundingMetadata | null;
+    safetyRatings: GoogleGenerativeAISafetyRating[] | null;
 }
 
 export { type GoogleGenerativeAIProvider, type GoogleGenerativeAIProviderMetadata, type GoogleGenerativeAIProviderSettings, createGoogleGenerativeAI, google };

package/dist/index.js

@@ -251,13 +251,13 @@ var googleFailedResponseHandler = (0, import_provider_utils2.createJsonErrorResp
 
 // src/google-prepare-tools.ts
 var import_provider2 = require("@ai-sdk/provider");
-function prepareTools(mode, useSearchGrounding) {
+function prepareTools(mode, useSearchGrounding, isGemini2) {
   var _a, _b;
   const tools = ((_a = mode.tools) == null ? void 0 : _a.length) ? mode.tools : void 0;
   const toolWarnings = [];
   if (useSearchGrounding) {
     return {
-      tools: { googleSearchRetrieval: {} },
+      tools: isGemini2 ? { googleSearch: {} } : { googleSearchRetrieval: {} },
       toolConfig: void 0,
       toolWarnings
     };
@@ -398,14 +398,18 @@ var GoogleGenerativeAILanguageModel = class {
       responseMimeType: (responseFormat == null ? void 0 : responseFormat.type) === "json" ? "application/json" : void 0,
       responseSchema: (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && // Google GenAI does not support all OpenAPI Schema features,
       // so this is needed as an escape hatch:
-      this.supportsStructuredOutputs ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0
+      this.supportsStructuredOutputs ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
+      ...this.settings.audioTimestamp && {
+        audioTimestamp: this.settings.audioTimestamp
+      }
     };
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(prompt);
     switch (type) {
       case "regular": {
         const { tools, toolConfig, toolWarnings } = prepareTools(
           mode,
-          (_a = this.settings.useSearchGrounding) != null ? _a : false
+          (_a = this.settings.useSearchGrounding) != null ? _a : false,
+          this.modelId.includes("gemini-2")
         );
         return {
           args: {
@@ -471,7 +475,7 @@ var GoogleGenerativeAILanguageModel = class {
     return url.toString().startsWith("https://generativelanguage.googleapis.com/v1beta/files/");
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g;
+    var _a, _b, _c, _d, _e, _f, _g, _h;
     const { args, warnings } = await this.getArgs(options);
     const body = JSON.stringify(args);
     const mergedHeaders = (0, import_provider_utils3.combineHeaders)(
@@ -512,7 +516,8 @@ var GoogleGenerativeAILanguageModel = class {
       warnings,
       providerMetadata: {
         google: {
-          groundingMetadata: (_g = candidate.groundingMetadata) != null ? _g : null
+          groundingMetadata: (_g = candidate.groundingMetadata) != null ? _g : null,
+          safetyRatings: (_h = candidate.safetyRatings) != null ? _h : null
         }
       },
       request: { body }
@@ -549,7 +554,7 @@ var GoogleGenerativeAILanguageModel = class {
       stream: response.pipeThrough(
         new TransformStream({
           transform(chunk, controller) {
-            var _a, _b, _c, _d;
+            var _a, _b, _c, _d, _e;
             if (!chunk.success) {
               controller.enqueue({ type: "error", error: chunk.error });
               return;
@@ -573,7 +578,8 @@ var GoogleGenerativeAILanguageModel = class {
               });
               providerMetadata = {
                 google: {
-                  groundingMetadata: (_d = candidate.groundingMetadata) != null ? _d : null
+                  groundingMetadata: (_d = candidate.groundingMetadata) != null ? _d : null,
+                  safetyRatings: (_e = candidate.safetyRatings) != null ? _e : null
                 }
               };
             }
@@ -665,9 +671,22 @@ var contentSchema = import_zod2.z.object({
 });
 var groundingMetadataSchema = import_zod2.z.object({
   webSearchQueries: import_zod2.z.array(import_zod2.z.string()).nullish(),
+  retrievalQueries: import_zod2.z.array(import_zod2.z.string()).nullish(),
   searchEntryPoint: import_zod2.z.object({
     renderedContent: import_zod2.z.string()
   }).nullish(),
+  groundingChunks: import_zod2.z.array(
+    import_zod2.z.object({
+      web: import_zod2.z.object({
+        uri: import_zod2.z.string(),
+        title: import_zod2.z.string()
+      }).nullish(),
+      retrievedContext: import_zod2.z.object({
+        uri: import_zod2.z.string(),
+        title: import_zod2.z.string()
+      }).nullish()
+    })
+  ).nullish(),
   groundingSupports: import_zod2.z.array(
     import_zod2.z.object({
       segment: import_zod2.z.object({
@@ -675,8 +694,11 @@ var groundingMetadataSchema = import_zod2.z.object({
         endIndex: import_zod2.z.number().nullish(),
         text: import_zod2.z.string().nullish()
       }),
-      groundingChunkIndices: import_zod2.z.array(import_zod2.z.number()),
-      confidenceScores: import_zod2.z.array(import_zod2.z.number())
+      segment_text: import_zod2.z.string().nullish(),
+      groundingChunkIndices: import_zod2.z.array(import_zod2.z.number()).nullish(),
+      supportChunkIndices: import_zod2.z.array(import_zod2.z.number()).nullish(),
+      confidenceScores: import_zod2.z.array(import_zod2.z.number()).nullish(),
+      confidenceScore: import_zod2.z.array(import_zod2.z.number()).nullish()
     })
   ).nullish(),
   retrievalMetadata: import_zod2.z.union([
@@ -686,11 +708,20 @@ var groundingMetadataSchema = import_zod2.z.object({
     import_zod2.z.object({})
   ]).nullish()
 });
+var safetyRatingSchema = import_zod2.z.object({
+  category: import_zod2.z.string(),
+  probability: import_zod2.z.string(),
+  probabilityScore: import_zod2.z.number().nullish(),
+  severity: import_zod2.z.string().nullish(),
+  severityScore: import_zod2.z.number().nullish(),
+  blocked: import_zod2.z.boolean().nullish()
+});
 var responseSchema = import_zod2.z.object({
   candidates: import_zod2.z.array(
     import_zod2.z.object({
       content: contentSchema.nullish(),
       finishReason: import_zod2.z.string().nullish(),
+      safetyRatings: import_zod2.z.array(safetyRatingSchema).nullish(),
       groundingMetadata: groundingMetadataSchema.nullish()
     })
   ),
@@ -705,6 +736,7 @@ var chunkSchema = import_zod2.z.object({
     import_zod2.z.object({
       content: contentSchema.nullish(),
       finishReason: import_zod2.z.string().nullish(),
+      safetyRatings: import_zod2.z.array(safetyRatingSchema).nullish(),
       groundingMetadata: groundingMetadataSchema.nullish()
     })
   ).nullish(),