Merge pull request #370 from Chanzhaoyu/feature

feat: 转义状态码
Chanzhaoyu · Mar 7, 2023 · 2ff82b1 · 2ff82b1
2 parents dea74e2 + 8e1b4ed
commit 2ff82b1
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/service/src/chatgpt/index.ts b/service/src/chatgpt/index.ts
@@ -7,6 +7,14 @@ import fetch from 'node-fetch'
 import { sendResponse } from '../utils'
 import type { ApiModel, ChatContext, ChatGPTUnofficialProxyAPIOptions, ModelConfig } from '../types'
 
+const ErrorCodeMessage: Record<string, string> = {
+  401: '提供错误的API密钥 | Incorrect API key provided',
+  429: '服务器限流，请稍后再试 | Server was limited, please try again later',
+  503: '服务器繁忙，请稍后再试 | Server is busy, please try again later',
+  500: '服务器繁忙，请稍后再试 | Server is busy, please try again later',
+  403: '服务器拒绝访问，请稍后再试 | Server refused to access, please try again later',
+}
+
 dotenv.config()
 
 const timeoutMs: number = !isNaN(+process.env.TIMEOUT_MS) ? +process.env.TIMEOUT_MS : 30 * 1000
@@ -98,8 +106,10 @@ async function chatReplyProcess(
     return sendResponse({ type: 'Success', data: response })
   }
   catch (error: any) {
-    global.console.error(error)
-    return sendResponse({ type: 'Fail', message: error.message })
+    const code = error.statusCode || 'unknown'
+    if (Reflect.has(ErrorCodeMessage, code))
+      return sendResponse({ type: 'Fail', message: ErrorCodeMessage[code] })
+    return sendResponse({ type: 'Fail', message: `${error.statusCode}-${error.statusText}` })
   }
 }