From f84663e2a96ed3df88bba86398dc8f2843c7b479 Mon Sep 17 00:00:00 2001
From: ikechan8370 <geyinchibuaa@gmail.com>
Date: Thu, 2 Mar 2023 19:42:36 +0800
Subject: [PATCH] feat: support openai api baseurl

---
 apps/chat.js             | 1 +
 config/config.example.js | 2 ++
 guoba.support.js         | 6 ++++++
 utils/config.js          | 3 ++-
 4 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/apps/chat.js b/apps/chat.js
index 5b0c6677..cadd9a19 100644
--- a/apps/chat.js
+++ b/apps/chat.js
@@ -675,6 +675,7 @@ export class chatgpt extends plugin {
         let promptPrefix = `You are ${Config.assistantLabel}, a large language model trained by OpenAI. ${Config.promptPrefixOverride || defaultPropmtPrefix}
         Current date: ${currentDate}`
         this.chatGPTApi = new ChatGPTAPI({
+          apiBaseUrl: Config.openAiBaseUrl,
           apiKey: Config.apiKey,
           debug: false,
           upsertMessage,
diff --git a/config/config.example.js b/config/config.example.js
index 11944479..8e5e377d 100644
--- a/config/config.example.js
+++ b/config/config.example.js
@@ -36,6 +36,8 @@ export default {
   //                                                     以下为API方式(默认)的配置                                                          *
   // ***********************************************************************************************************************************
   apiKey: API_KEY,
+  // 如果有openai api的加速反代可以写
+  openAiBaseUrl: '',
   // 模型名称，选填。如无特殊需求保持默认即可，会使用chatgpt-api库提供的当前可用的最适合的默认值。保底可用的是 text-davinci-003。当发现新的可用的chatGPT模型会更新这里的值
   // 20230211： text-chat-davinci-002-sh-alpha-aoruigiofdj83 中午存活了几分钟
   // model: '',
diff --git a/guoba.support.js b/guoba.support.js
index 049acf68..f9dc2fe6 100644
--- a/guoba.support.js
+++ b/guoba.support.js
@@ -167,6 +167,12 @@ export function supportGuoba () {
           bottomHelpMessage: 'OpenAI的ApiKey，用于访问OpenAI的API接口。',
           component: 'InputPassword'
         },
+        {
+          field: 'openAiBaseUrl',
+          label: 'OpenAI API服务器地址',
+          bottomHelpMessage: 'OpenAI的API服务器地址。默认为https://api.openai.com',
+          component: 'Input'
+        },
         {
           field: 'model',
           label: '模型',
diff --git a/utils/config.js b/utils/config.js
index bdb78535..6a317617 100644
--- a/utils/config.js
+++ b/utils/config.js
@@ -14,6 +14,7 @@ const defaultConfig = {
   cacheUrl: 'https://content.alcedogroup.com',
   cacheEntry: false,
   apiKey: '',
+  openAiBaseUrl: '',
   drawCD: 30,
   model: '',
   temperature: 0.8,
@@ -33,7 +34,7 @@ const defaultConfig = {
   debug: true,
   defaultTimeoutMs: 120000,
   chromeTimeoutMS: 120000,
-  version: '2.0.6'
+  version: '2.0.7'
 }
 const _path = process.cwd()
 let config = {}