Kaynağa Gözat

Merge pull request #3617 from H0llyW00dzZ/fixfixbug

Fix [Bug] Issue ChatGPTNextWeb#3615
Fred Liang 1 yıl önce
ebeveyn
işleme
c0ce70ccc7
2 değiştirilmiş dosya ile 8 ekleme ve 3 silme
  1. 6 3
      app/api/auth.ts
  2. 2 0
      app/config/server.ts

+ 6 - 3
app/api/auth.ts

@@ -55,15 +55,18 @@ export function auth(req: NextRequest) {
 
   // if user does not provide an api key, inject system api key
   if (!apiKey) {
-    const serverApiKey = serverConfig.isAzure
+    const serverConfig = getServerSideConfig();
+    const systemApiKey = serverConfig.isAzure
       ? serverConfig.azureApiKey
+      : serverConfig.isGoogle
+      ? serverConfig.googleApiKey
       : serverConfig.apiKey;
 
-    if (serverApiKey) {
+    if (systemApiKey) {
       console.log("[Auth] use system api key");
       req.headers.set(
         "Authorization",
-        `${serverConfig.isAzure ? "" : "Bearer "}${serverApiKey}`,
+        `Bearer ${systemApiKey}`,
       );
     } else {
       console.log("[Auth] admin did not provide an api key");

+ 2 - 0
app/config/server.ts

@@ -65,6 +65,7 @@ export const getServerSideConfig = () => {
   }
 
   const isAzure = !!process.env.AZURE_URL;
+  const isGoogle = !!process.env.GOOGLE_API_KEY;
 
   const apiKeyEnvVar = process.env.OPENAI_API_KEY ?? "";
   const apiKeys = apiKeyEnvVar.split(",").map((v) => v.trim());
@@ -84,6 +85,7 @@ export const getServerSideConfig = () => {
     azureApiKey: process.env.AZURE_API_KEY,
     azureApiVersion: process.env.AZURE_API_VERSION,
 
+    isGoogle,
     googleApiKey: process.env.GOOGLE_API_KEY,
     googleUrl: process.env.GOOGLE_URL,