From 9b97dca601b5f159bf44d09720f82cc865cc77b7 Mon Sep 17 00:00:00 2001
From: lloydzhou <lloydzhou@qq.com>
Date: Fri, 12 Jul 2024 21:27:30 +0800
Subject: [PATCH] hotfix: custom comfig for Gemini api. #4944

---
 app/client/platforms/google.ts | 15 ++++++---------
 app/constant.ts                |  1 +
 app/store/access.ts            | 10 +++++-----
 3 files changed, 12 insertions(+), 14 deletions(-)

diff --git a/app/client/platforms/google.ts b/app/client/platforms/google.ts
index 828b28a0d..6054c7a47 100644
--- a/app/client/platforms/google.ts
+++ b/app/client/platforms/google.ts
@@ -122,16 +122,13 @@ export class GeminiProApi implements LLMApi {
     const controller = new AbortController();
     options.onController?.(controller);
     try {
-      // let baseUrl = accessStore.googleUrl;
-
-      if (!baseUrl) {
-        baseUrl = isApp
-          ? DEFAULT_API_HOST +
-            "/api/proxy/google/" +
-            Google.ChatPath(modelConfig.model)
-          : this.path(Google.ChatPath(modelConfig.model));
+      if (!baseUrl && isApp) {
+        baseUrl = DEFAULT_API_HOST + "/api/proxy/google/";
       }
-
+      baseUrl = `${baseUrl}/${Google.ChatPath(modelConfig.model)}`.replaceAll(
+        "//",
+        "/",
+      );
       if (isApp) {
         baseUrl += `?key=${accessStore.googleApiKey}`;
       }
diff --git a/app/constant.ts b/app/constant.ts
index c07adad25..a146200d6 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -35,6 +35,7 @@ export enum ApiPath {
   Azure = "/api/azure",
   OpenAI = "/api/openai",
   Anthropic = "/api/anthropic",
+  Google = "/api/google",
   Baidu = "/api/baidu",
   ByteDance = "/api/bytedance",
   Alibaba = "/api/alibaba",
diff --git a/app/store/access.ts b/app/store/access.ts
index c0fea9fea..5b07ff128 100644
--- a/app/store/access.ts
+++ b/app/store/access.ts
@@ -17,10 +17,10 @@ const DEFAULT_OPENAI_URL =
     ? DEFAULT_API_HOST + "/api/proxy/openai"
     : ApiPath.OpenAI;
 
-const DEFAULT_AZURE_URL =
+const DEFAULT_GOOGLE_URL =
   getClientConfig()?.buildMode === "export"
-    ? DEFAULT_API_HOST + "/api/proxy/azure/{resource_name}"
-    : ApiPath.Azure;
+    ? DEFAULT_API_HOST + "/api/proxy/google"
+    : ApiPath.Google;
 
 const DEFAULT_ACCESS_STATE = {
   accessCode: "",
@@ -33,12 +33,12 @@ const DEFAULT_ACCESS_STATE = {
   openaiApiKey: "",
 
   // azure
-  azureUrl: DEFAULT_AZURE_URL,
+  azureUrl: "",
   azureApiKey: "",
   azureApiVersion: "2023-08-01-preview",
 
   // google ai studio
-  googleUrl: "",
+  googleUrl: DEFAULT_GOOGLE_URL,
   googleApiKey: "",
   googleApiVersion: "v1",