diff --git a/packages/ai/src/providers/google.ts b/packages/ai/src/providers/google.ts index 3b1dc894..3c5a6c8c 100644 --- a/packages/ai/src/providers/google.ts +++ b/packages/ai/src/providers/google.ts @@ -265,9 +265,18 @@ function createClient(model: Model<"google-generative-ai">, apiKey?: string): Go } apiKey = process.env.GEMINI_API_KEY; } + + const httpOptions: { baseUrl?: string; headers?: Record } = {}; + if (model.baseUrl) { + httpOptions.baseUrl = model.baseUrl; + } + if (model.headers) { + httpOptions.headers = model.headers; + } + return new GoogleGenAI({ apiKey, - httpOptions: model.headers ? { headers: model.headers } : undefined, + httpOptions: Object.keys(httpOptions).length > 0 ? httpOptions : undefined, }); }