chore: 单独设置API地址 (Close #369)

chatgpt-web-dev · Dec 30, 2023 · c275b1e · c275b1e
1 parent d5a1a09
commit c275b1e
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 9 deletions.
diff --git a/service/src/chatgpt/index.ts b/service/src/chatgpt/index.ts
@@ -41,7 +41,7 @@ export async function initApi(key: KeyConfig, chatModel: string, maxContextCount
   const model = chatModel as string
 
   if (key.keyModel === 'ChatGPTAPI') {
-    const OPENAI_API_BASE_URL = config.apiBaseUrl
+    const OPENAI_API_BASE_URL = isNotEmptyString(key.baseUrl) ? key.baseUrl : config.apiBaseUrl
 
     let contextCount = 0
     const options: ChatGPTAPIOptions = {
@@ -97,7 +97,11 @@ export async function initApi(key: KeyConfig, chatModel: string, maxContextCount
   else {
     const options: ChatGPTUnofficialProxyAPIOptions = {
       accessToken: key.key,
-      apiReverseProxyUrl: isNotEmptyString(config.reverseProxy) ? config.reverseProxy : 'https://ai.fakeopen.com/api/conversation',
+      apiReverseProxyUrl: isNotEmptyString(key.baseUrl)
+        ? key.baseUrl
+        : isNotEmptyString(config.reverseProxy)
+          ? config.reverseProxy
+          : 'https://ai.fakeopen.com/api/conversation',
       model,
       debug: !config.apiDisableDebug,
     }

diff --git a/service/src/storage/model.ts b/service/src/storage/model.ts
@@ -228,6 +228,7 @@ export class KeyConfig {
   userRoles: UserRole[]
   status: Status
   remark: string
+  baseUrl?: string
   constructor(key: string, keyModel: APIMODEL, chatModels: string[], userRoles: UserRole[], remark: string) {
     this.key = key
     this.keyModel = keyModel

diff --git a/src/components/common/Setting/Keys.vue b/src/components/common/Setting/Keys.vue
@@ -23,20 +23,25 @@ const columns = [
     title: 'Key',
     key: 'key',
     resizable: true,
-    width: 200,
-    minWidth: 100,
-    maxWidth: 200,
+    width: 120,
+    minWidth: 50,
+    maxWidth: 120,
     ellipsis: true,
   },
   {
     title: 'Api Model',
     key: 'keyModel',
-    width: 190,
+    width: 150,
+  },
+  {
+    title: 'Base url',
+    key: 'baseUrl',
+    width: 150,
   },
   {
     title: 'Chat Model',
     key: 'chatModels',
-    width: 320,
+    width: 300,
     render(row: any) {
       const tags = row.chatModels.map((chatModel: string) => {
         return h(
@@ -59,7 +64,7 @@ const columns = [
   {
     title: 'User Roles',
     key: 'userRoles',
-    width: 200,
+    width: 180,
     render(row: any) {
       const tags = row.userRoles.map((userRole: UserRole) => {
         return h(
@@ -82,7 +87,7 @@ const columns = [
   {
     title: 'Remark',
     key: 'remark',
-    width: 220,
+    width: 150,
   },
   {
     title: 'Action',
@@ -256,6 +261,15 @@ onMounted(async () => {
             />
           </div>
         </div>
+        <div class="flex items-center space-x-4">
+          <span class="flex-shrink-0 w-[100px]">{{ $t('setting.apiBaseUrl') }}</span>
+          <div class="flex-1">
+            <NInput
+              v-model:value="keyConfig.baseUrl"
+              style="width: 100%" placeholder=""
+            />
+          </div>
+        </div>
         <div class="flex items-center space-x-4">
           <span class="flex-shrink-0 w-[100px]">{{ $t('setting.chatModels') }}</span>
           <div class="flex-1">

diff --git a/src/components/common/Setting/model.ts b/src/components/common/Setting/model.ts
@@ -91,6 +91,7 @@ export class KeyConfig {
   userRoles: UserRole[]
   status: Status
   remark: string
+  baseUrl?: string
   constructor(key: string, keyModel: APIMODEL, chatModels: string[], userRoles: UserRole[], remark: string) {
     this.key = key
     this.keyModel = keyModel