From 3511e84dc3ecdc121ca657e77859f302d514d271 Mon Sep 17 00:00:00 2001
From: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com>
Date: Tue, 27 May 2025 10:00:07 -0700
Subject: fix: default to Gemini API if GEMINI_API_KEY is set and when
 GOOGLE_GENAI_USE_VERTEXAI is set to True (#566)

---
 packages/server/src/core/client.ts | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'packages/server/src/core')

diff --git a/packages/server/src/core/client.ts b/packages/server/src/core/client.ts
index 8823012e..341ce021 100644
--- a/packages/server/src/core/client.ts
+++ b/packages/server/src/core/client.ts
@@ -36,8 +36,11 @@ export class GeminiClient {
   constructor(private config: Config) {
     const userAgent = config.getUserAgent();
     const apiKeyFromConfig = config.getApiKey();
+    const vertexaiFlag = config.getVertexAI();
+
     this.client = new GoogleGenAI({
       apiKey: apiKeyFromConfig === '' ? undefined : apiKeyFromConfig,
+      vertexai: vertexaiFlag,
       httpOptions: {
         headers: {
           'User-Agent': userAgent,
-- 
cgit v1.2.3