From 2ad5ec200d53c6fd0586d92bb64981faa8c263cb Mon Sep 17 00:00:00 2001
From: Yidadaa <yidadaa@qq.com>
Date: Sun, 19 Nov 2023 18:42:30 +0800
Subject: [PATCH] feat: close #3300 support multiple api keys

---
 README.md            |  2 +-
 README_CN.md         |  2 +-
 app/config/server.ts | 10 +++++++++-
 3 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index ef7f2e1d9fb..3050fcc95e1 100644
--- a/README.md
+++ b/README.md
@@ -161,7 +161,7 @@ Access password, separated by comma.
 
 ### `OPENAI_API_KEY` (required)
 
-Your openai api key.
+Your openai api key, join multiple api keys with comma.
 
 ### `BASE_URL` (optional)
 
diff --git a/README_CN.md b/README_CN.md
index 3b713255a47..0ef508f61ac 100644
--- a/README_CN.md
+++ b/README_CN.md
@@ -68,7 +68,7 @@ code1,code2,code3
 
 ### `OPENAI_API_KEY` （必填项）
 
-OpanAI 密钥，你在 openai 账户页面申请的 api key。
+OpanAI 密钥，你在 openai 账户页面申请的 api key，使用英文逗号隔开多个 key，这样可以随机轮询这些 key。
 
 ### `CODE` （可选）
 
diff --git a/app/config/server.ts b/app/config/server.ts
index 2f2e7d7fd8a..2398805a264 100644
--- a/app/config/server.ts
+++ b/app/config/server.ts
@@ -62,9 +62,17 @@ export const getServerSideConfig = () => {
 
   const isAzure = !!process.env.AZURE_URL;
 
+  const apiKeyEnvVar = process.env.OPENAI_API_KEY ?? "";
+  const apiKeys = apiKeyEnvVar.split(",").map((v) => v.trim());
+  const randomIndex = Math.floor(Math.random() * apiKeys.length);
+  const apiKey = apiKeys[randomIndex];
+  console.log(
+    `[Server Config] using ${randomIndex + 1} of ${apiKeys.length} api key`,
+  );
+
   return {
     baseUrl: process.env.BASE_URL,
-    apiKey: process.env.OPENAI_API_KEY,
+    apiKey,
     openaiOrgId: process.env.OPENAI_ORG_ID,
 
     isAzure,