feat: support openai api baseurl

ikechan8370 · Mar 2, 2023 · f84663e · f84663e
1 parent 17fb123
commit f84663e
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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
@@ -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 = {}