From bf2310cc291716e4c3716e6213af29f135b6655a Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 26 Jul 2023 16:15:21 +0800
Subject: [PATCH] perf: chat source

---
 .../src/pages/api/chat/history/getHistory.ts  |   2 +
 client/src/pages/api/system/getInitData.ts    | 128 +++++++++---------
 2 files changed, 69 insertions(+), 61 deletions(-)

diff --git a/client/src/pages/api/chat/history/getHistory.ts b/client/src/pages/api/chat/history/getHistory.ts
index e2af65df3..e6fcb862a 100644
--- a/client/src/pages/api/chat/history/getHistory.ts
+++ b/client/src/pages/api/chat/history/getHistory.ts
@@ -3,6 +3,7 @@ import { jsonRes } from '@/service/response';
 import { connectToDatabase, Chat } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
 import type { ChatHistoryItemType } from '@/types/chat';
+import { ChatSourceEnum } from '@/constants/chat';
 
 /* 获取历史记录 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -15,6 +16,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const data = await Chat.find(
       {
         userId,
+        source: ChatSourceEnum.online,
         ...(appId && { appId })
       },
       'chatId title top customTitle appId updateTime'
diff --git a/client/src/pages/api/system/getInitData.ts b/client/src/pages/api/system/getInitData.ts
index 2202a9cdd..6fa33345e 100644
--- a/client/src/pages/api/system/getInitData.ts
+++ b/client/src/pages/api/system/getInitData.ts
@@ -31,16 +31,73 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
   });
 }
 
+const defaultSystemEnv = {
+  vectorMaxProcess: 15,
+  qaMaxProcess: 15,
+  pgIvfflatProbe: 20,
+  sensitiveCheck: false
+};
+const defaultFeConfigs = {
+  show_emptyChat: true,
+  show_register: true,
+  show_appStore: true,
+  show_userDetail: true,
+  show_git: true,
+  systemTitle: 'FastAI',
+  authorText: 'Made by FastAI Team.'
+};
+const defaultChatModels = [
+  {
+    model: 'gpt-3.5-turbo',
+    name: 'FastAI-4k',
+    contextMaxToken: 4000,
+    quoteMaxToken: 2400,
+    maxTemperature: 1.2,
+    price: 1.5
+  },
+  {
+    model: 'gpt-3.5-turbo-16k',
+    name: 'FastAI-16k',
+    contextMaxToken: 16000,
+    quoteMaxToken: 8000,
+    maxTemperature: 1.2,
+    price: 3
+  },
+  {
+    model: 'gpt-4',
+    name: 'FastAI-Plus',
+    contextMaxToken: 8000,
+    quoteMaxToken: 4000,
+    maxTemperature: 1.2,
+    price: 45
+  }
+];
+const defaultQAModels = [
+  {
+    model: 'gpt-3.5-turbo-16k',
+    name: 'FastAI-16k',
+    maxToken: 16000,
+    price: 3
+  }
+];
+const defaultVectorModels = [
+  {
+    model: 'text-embedding-ada-002',
+    name: 'Embedding-2',
+    price: 0.2
+  }
+];
+
 export async function getInitConfig() {
   try {
     const res = JSON.parse(readFileSync('data/config.json', 'utf-8'));
     console.log(res);
 
-    global.systemEnv = res.SystemParams;
-    global.feConfigs = res.FeConfig;
-    global.chatModels = res.ChatModels;
-    global.qaModels = res.QAModels;
-    global.vectorModels = res.VectorModels;
+    global.systemEnv = res.SystemParams || defaultSystemEnv;
+    global.feConfigs = res.FeConfig || defaultFeConfigs;
+    global.chatModels = res.ChatModels || defaultChatModels;
+    global.qaModels = res.QAModels || defaultQAModels;
+    global.vectorModels = res.VectorModels || defaultVectorModels;
   } catch (error) {
     setDefaultData();
     return Promise.reject('get init config error');
@@ -48,60 +105,9 @@ export async function getInitConfig() {
 }
 
 export function setDefaultData() {
-  global.systemEnv = {
-    vectorMaxProcess: 15,
-    qaMaxProcess: 15,
-    pgIvfflatProbe: 20,
-    sensitiveCheck: false
-  };
-  global.feConfigs = {
-    show_emptyChat: true,
-    show_register: true,
-    show_appStore: true,
-    show_userDetail: true,
-    show_git: true,
-    systemTitle: 'FastAI',
-    authorText: 'Made by FastAI Team.'
-  };
-  global.chatModels = [
-    {
-      model: 'gpt-3.5-turbo',
-      name: 'FastAI-4k',
-      contextMaxToken: 4000,
-      quoteMaxToken: 2400,
-      maxTemperature: 1.2,
-      price: 1.5
-    },
-    {
-      model: 'gpt-3.5-turbo-16k',
-      name: 'FastAI-16k',
-      contextMaxToken: 16000,
-      quoteMaxToken: 8000,
-      maxTemperature: 1.2,
-      price: 3
-    },
-    {
-      model: 'gpt-4',
-      name: 'FastAI-Plus',
-      contextMaxToken: 8000,
-      quoteMaxToken: 4000,
-      maxTemperature: 1.2,
-      price: 45
-    }
-  ];
-  global.qaModels = [
-    {
-      model: 'gpt-3.5-turbo-16k',
-      name: 'FastAI-16k',
-      maxToken: 16000,
-      price: 3
-    }
-  ];
-  global.vectorModels = [
-    {
-      model: 'text-embedding-ada-002',
-      name: 'Embedding-2',
-      price: 0.2
-    }
-  ];
+  global.systemEnv = defaultSystemEnv;
+  global.feConfigs = defaultFeConfigs;
+  global.chatModels = defaultChatModels;
+  global.qaModels = defaultQAModels;
+  global.vectorModels = defaultVectorModels;
 }