Support full Google GenAI client options

Author: mcndt

.changeset/hot-crews-battle.md

@@ -0,0 +1,5 @@
+---
+"@browserbasehq/stagehand": patch
+---
+
+Support full Google GenAI client options (to e.g. use Vertex AI, or control the region)

lib/llm/GoogleClient.ts

@@ -8,11 +8,12 @@ import {
   FunctionCall,
   Schema,
   Type,
+  GoogleGenAIOptions as ClientOptions,
 } from "@google/genai";
 import zodToJsonSchema from "zod-to-json-schema";
 
 import { LogLine } from "../../types/log";
-import { AvailableModel, ClientOptions } from "../../types/model";
+import { AvailableModel } from "../../types/model";
 import { LLMCache } from "../cache/LLMCache";
 import { validateZodSchema, toGeminiSchema, loadApiKeyFromEnv } from "../utils";
 import {
@@ -75,15 +76,15 @@ export class GoogleClient extends LLMClient {
     enableCaching?: boolean;
     cache?: LLMCache;
     modelName: AvailableModel;
-    clientOptions?: ClientOptions; // Expecting { apiKey: string } here
+    clientOptions?: ClientOptions;
   }) {
     super(modelName);
     if (!clientOptions?.apiKey) {
       // Try to get the API key from the environment variable GOOGLE_API_KEY
       clientOptions.apiKey = loadApiKeyFromEnv("google_legacy", logger);
     }
     this.clientOptions = clientOptions;
-    this.client = new GoogleGenAI({ apiKey: clientOptions.apiKey });
+    this.client = new GoogleGenAI(clientOptions);
     this.cache = cache;
     this.enableCaching = enableCaching;
     this.modelName = modelName;

types/model.ts

@@ -1,4 +1,5 @@
 import type { ClientOptions as AnthropicClientOptions } from "@anthropic-ai/sdk";
+import { GoogleGenAIOptions as GoogleGenAIClientOptions } from "@google/genai";
 import type { ClientOptions as OpenAIClientOptions } from "openai";
 import { z } from "zod/v3";
 
@@ -44,7 +45,16 @@ export type ModelProvider =
   | "google"
   | "aisdk";
 
-export type ClientOptions = OpenAIClientOptions | AnthropicClientOptions;
+export type ClientOptions = (
+  | OpenAIClientOptions
+  | AnthropicClientOptions
+  | GoogleGenAIClientOptions
+) &
+  // aisdk client language model options
+  {
+    apiKey?: string;
+    baseURL?: string;
+  };
 
 export interface AnthropicJsonSchemaObject {
   definitions?: {