Browse Source

add config for gemini custom base url

dongqing 11 months ago
parent
commit
a3a5592654
3 changed files with 2 additions and 1 deletions
  1. 0 1
      src/api/providers/gemini.ts
  2. 1 0
      src/shared/api.ts
  3. 1 0
      src/shared/globalState.ts

+ 0 - 1
src/api/providers/gemini.ts

@@ -28,7 +28,6 @@ export class GeminiHandler extends BaseProvider implements SingleCompletionHandl
 				baseUrl: this.options.googleGeminiBaseUrl || undefined,
 				baseUrl: this.options.googleGeminiBaseUrl || undefined,
 			},
 			},
 		)
 		)
-
 		const result = await model.generateContentStream({
 		const result = await model.generateContentStream({
 			contents: messages.map(convertAnthropicMessageToGemini),
 			contents: messages.map(convertAnthropicMessageToGemini),
 			generationConfig: {
 			generationConfig: {

+ 1 - 0
src/shared/api.ts

@@ -116,6 +116,7 @@ export const API_CONFIG_KEYS: GlobalStateKey[] = [
 	"lmStudioBaseUrl",
 	"lmStudioBaseUrl",
 	"lmStudioDraftModelId",
 	"lmStudioDraftModelId",
 	"lmStudioSpeculativeDecodingEnabled",
 	"lmStudioSpeculativeDecodingEnabled",
+	"googleGeminiBaseUrl",
 	"mistralCodestralUrl",
 	"mistralCodestralUrl",
 	"azureApiVersion",
 	"azureApiVersion",
 	"openRouterUseMiddleOutTransform",
 	"openRouterUseMiddleOutTransform",

+ 1 - 0
src/shared/globalState.ts

@@ -59,6 +59,7 @@ export const GLOBAL_STATE_KEYS = [
 	"openRouterModelInfo",
 	"openRouterModelInfo",
 	"openRouterBaseUrl",
 	"openRouterBaseUrl",
 	"openRouterUseMiddleOutTransform",
 	"openRouterUseMiddleOutTransform",
+	"googleGeminiBaseUrl",
 	"allowedCommands",
 	"allowedCommands",
 	"soundEnabled",
 	"soundEnabled",
 	"soundVolume",
 	"soundVolume",