소스 검색

Merge pull request #5141 from ConnectAI-E/hotfix-google-api-in-app

hotfix for using google api in app
Dogtiti 1 년 전
부모
커밋
c9eb9f3eda
1개의 변경된 파일6개의 추가작업 그리고 4개의 파일을 삭제
  1. 6 4
      app/client/platforms/google.ts

+ 6 - 4
app/client/platforms/google.ts

@@ -25,11 +25,9 @@ export class GeminiProApi implements LLMApi {
       baseUrl = accessStore.googleUrl;
       baseUrl = accessStore.googleUrl;
     }
     }
 
 
+    const isApp = !!getClientConfig()?.isApp;
     if (baseUrl.length === 0) {
     if (baseUrl.length === 0) {
-      const isApp = !!getClientConfig()?.isApp;
-      baseUrl = isApp
-        ? DEFAULT_API_HOST + `/api/proxy/google?key=${accessStore.googleApiKey}`
-        : ApiPath.Google;
+      baseUrl = isApp ? DEFAULT_API_HOST + `/api/proxy/google` : ApiPath.Google;
     }
     }
     if (baseUrl.endsWith("/")) {
     if (baseUrl.endsWith("/")) {
       baseUrl = baseUrl.slice(0, baseUrl.length - 1);
       baseUrl = baseUrl.slice(0, baseUrl.length - 1);
@@ -43,6 +41,10 @@ export class GeminiProApi implements LLMApi {
     let chatPath = [baseUrl, path].join("/");
     let chatPath = [baseUrl, path].join("/");
 
 
     chatPath += chatPath.includes("?") ? "&alt=sse" : "?alt=sse";
     chatPath += chatPath.includes("?") ? "&alt=sse" : "?alt=sse";
+    // if chatPath.startsWith('http') then add key in query string
+    if (chatPath.startsWith("http") && accessStore.googleApiKey) {
+      chatPath += `&key=${accessStore.googleApiKey}`;
+    }
     return chatPath;
     return chatPath;
   }
   }
   extractMessage(res: any) {
   extractMessage(res: any) {