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)

packages/core/lib/v3/llm/GoogleClient.ts

@@ -8,10 +8,11 @@ import {
   FunctionCall,
   Schema,
   Type,
+  GoogleGenAIOptions as ClientOptions,
 } from "@google/genai";
 
 import { LogLine } from "../types/public/logs";
-import { AvailableModel, ClientOptions } from "../types/public/model";
+import { AvailableModel } from "../types/public/model";
 import {
   validateZodSchema,
   toGeminiSchema,
@@ -71,15 +72,15 @@ export class GoogleClient extends LLMClient {
   }: {
     logger: (message: LogLine) => void; // Added logger type
     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.modelName = modelName;
     this.logger = logger;
     // Determine vision capability based on model name (adjust as needed)

packages/core/lib/v3/types/public/model.ts

@@ -1,5 +1,6 @@
 import type { ClientOptions as AnthropicClientOptions } from "@anthropic-ai/sdk";
 import type { LanguageModelV2 } from "@ai-sdk/provider";
+import { GoogleGenAIOptions as GoogleGenAIClientOptions } from "@google/genai";
 import type { ClientOptions as OpenAIClientOptions } from "openai";
 
 export type AnthropicJsonSchemaObject = {
@@ -66,7 +67,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 type ModelConfiguration =
   | AvailableModel