From a3a559265474c2b81c538c4e78ad2bfef0582fdc Mon Sep 17 00:00:00 2001
From: dongqing <dongqing@qianxin.com>
Date: Tue, 11 Mar 2025 16:46:40 +0800
Subject: [PATCH] add config for gemini custom base url

---
 src/api/providers/gemini.ts | 1 -
 src/shared/api.ts           | 1 +
 src/shared/globalState.ts   | 1 +
 3 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/api/providers/gemini.ts b/src/api/providers/gemini.ts
index 12cd6903f3..98117e99a9 100644
--- a/src/api/providers/gemini.ts
+++ b/src/api/providers/gemini.ts
@@ -28,7 +28,6 @@ export class GeminiHandler extends BaseProvider implements SingleCompletionHandl
 				baseUrl: this.options.googleGeminiBaseUrl || undefined,
 			},
 		)
-
 		const result = await model.generateContentStream({
 			contents: messages.map(convertAnthropicMessageToGemini),
 			generationConfig: {
diff --git a/src/shared/api.ts b/src/shared/api.ts
index c92bb32286..368ac710f9 100644
--- a/src/shared/api.ts
+++ b/src/shared/api.ts
@@ -116,6 +116,7 @@ export const API_CONFIG_KEYS: GlobalStateKey[] = [
 	"lmStudioBaseUrl",
 	"lmStudioDraftModelId",
 	"lmStudioSpeculativeDecodingEnabled",
+	"googleGeminiBaseUrl",
 	"mistralCodestralUrl",
 	"azureApiVersion",
 	"openRouterUseMiddleOutTransform",
diff --git a/src/shared/globalState.ts b/src/shared/globalState.ts
index 40973b3350..8dc79eb895 100644
--- a/src/shared/globalState.ts
+++ b/src/shared/globalState.ts
@@ -59,6 +59,7 @@ export const GLOBAL_STATE_KEYS = [
 	"openRouterModelInfo",
 	"openRouterBaseUrl",
 	"openRouterUseMiddleOutTransform",
+	"googleGeminiBaseUrl",
 	"allowedCommands",
 	"soundEnabled",
 	"soundVolume",