chore(js/plugins/googleai): added google search options to gemini config schema (#2938)

Author: pavelgj

js/plugins/googleai/src/gemini.ts

@@ -28,6 +28,7 @@ import {
   GenerationConfig,
   GenerativeModel,
   GoogleGenerativeAI,
+  GoogleSearchRetrievalTool,
   InlineDataPart,
   RequestOptions,
   Schema,
@@ -132,6 +133,12 @@ export const GeminiConfigSchema = GenerationCommonConfigSchema.extend({
         "'gemini-2.0-flash-exp' model at present."
     )
     .optional(),
+  googleSearchRetrieval: z
+    .union([z.boolean(), z.object({}).passthrough()])
+    .describe(
+      'Retrieve public web data for grounding, powered by Google Search.'
+    )
+    .optional(),
 }).passthrough();
 export type GeminiConfig = z.infer<typeof GeminiConfigSchema>;
 
@@ -916,6 +923,7 @@ export function defineGoogleAIModel({
         codeExecution: codeExecutionFromConfig,
         version: versionFromConfig,
         functionCallingConfig,
+        googleSearchRetrieval,
         ...restOfConfigOptions
       } = requestConfig;
 
@@ -928,6 +936,13 @@ export function defineGoogleAIModel({
         });
       }
 
+      if (googleSearchRetrieval) {
+        tools.push({
+          googleSearch:
+            googleSearchRetrieval === true ? {} : googleSearchRetrieval,
+        } as GoogleSearchRetrievalTool);
+      }
+
       let toolConfig: ToolConfig | undefined;
       if (functionCallingConfig) {
         toolConfig = {