perf: config fe

client/src/pages/api/system/getInitData.ts

@@ -6,7 +6,7 @@ import {
   type ChatModelItemType,
   type VectorModelItemType
 } from '@/types/model';
-import { readFileSync } from 'fs';
+import type { FeConfigsType } from '@/types';
 
 export type InitDateResponse = {
   beianText: string;
@@ -15,48 +15,7 @@ export type InitDateResponse = {
   chatModels: ChatModelItemType[];
   qaModels: QAModelItemType[];
   vectorModels: VectorModelItemType[];
-};
-
-const defaultmodels = {
-  'FastAI-4k': {
-    model: 'gpt-3.5-turbo',
-    name: 'FastAI-4k',
-    contextMaxToken: 4000,
-    systemMaxToken: 2400,
-    maxTemperature: 1.2,
-    price: 1.5
-  },
-  'FastAI-16k': {
-    model: 'gpt-3.5-turbo',
-    name: 'FastAI-16k',
-    contextMaxToken: 16000,
-    systemMaxToken: 8000,
-    maxTemperature: 1.2,
-    price: 3
-  },
-  'FastAI-Plus': {
-    model: 'gpt-4',
-    name: 'FastAI-Plus',
-    contextMaxToken: 8000,
-    systemMaxToken: 4000,
-    maxTemperature: 1.2,
-    price: 45
-  }
-};
-const defaultQaModels = {
-  'FastAI-16k': {
-    model: 'gpt-3.5-turbo',
-    name: 'FastAI-16k',
-    maxToken: 16000,
-    price: 3
-  }
-};
-const defaultVectorModels = {
-  'text-embedding-ada-002': {
-    model: 'text-embedding-ada-002',
-    name: 'Embedding-2',
-    price: 0.2
-  }
+  feConfigs: FeConfigsType;
 };
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
@@ -69,46 +28,10 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
   jsonRes<InitDateResponse>(res, {
     data: {
       ...envs,
-      ...initSystemModels()
+      chatModels: global.chatModels,
+      qaModels: global.qaModels,
+      vectorModels: global.vectorModels,
+      feConfigs: global.feConfigs
     }
   });
 }
-
-export function initSystemModels() {
-  const { chatModels, qaModels, vectorModels } = (() => {
-    try {
-      const chatModels = Object.values(JSON.parse(readFileSync('data/ChatModels.json', 'utf-8')));
-      const qaModels = Object.values(JSON.parse(readFileSync('data/QAModels.json', 'utf-8')));
-      const vectorModels = Object.values(
-        JSON.parse(readFileSync('data/VectorModels.json', 'utf-8'))
-      );
-
-      return {
-        chatModels,
-        qaModels,
-        vectorModels
-      };
-    } catch (error) {
-      console.log(error);
-
-      return {
-        chatModels: Object.values(defaultmodels),
-        qaModels: Object.values(defaultQaModels),
-        vectorModels: Object.values(defaultVectorModels)
-      };
-    }
-  })() as {
-    chatModels: ChatModelItemType[];
-    qaModels: QAModelItemType[];
-    vectorModels: VectorModelItemType[];
-  };
-  global.chatModels = chatModels;
-  global.qaModels = qaModels;
-  global.vectorModels = vectorModels;
-
-  return {
-    chatModels,
-    qaModels,
-    vectorModels
-  };
-}

client/src/pages/api/system/updateEnv.ts

@@ -2,6 +2,11 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { authUser } from '@/service/utils/auth';
 import { readFileSync } from 'fs';
+import {
+  type QAModelItemType,
+  type ChatModelItemType,
+  type VectorModelItemType
+} from '@/types/model';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   await authUser({ req, authRoot: true });
@@ -22,3 +27,97 @@ export async function updateSystemEnv() {
     console.log('update system env error');
   }
 }
+
+const defaultmodels = {
+  'FastAI-4k': {
+    model: 'gpt-3.5-turbo',
+    name: 'FastAI-4k',
+    contextMaxToken: 4000,
+    systemMaxToken: 2400,
+    maxTemperature: 1.2,
+    price: 1.5
+  },
+  'FastAI-16k': {
+    model: 'gpt-3.5-turbo',
+    name: 'FastAI-16k',
+    contextMaxToken: 16000,
+    systemMaxToken: 8000,
+    maxTemperature: 1.2,
+    price: 3
+  },
+  'FastAI-Plus': {
+    model: 'gpt-4',
+    name: 'FastAI-Plus',
+    contextMaxToken: 8000,
+    systemMaxToken: 4000,
+    maxTemperature: 1.2,
+    price: 45
+  }
+};
+const defaultQaModels = {
+  'FastAI-16k': {
+    model: 'gpt-3.5-turbo',
+    name: 'FastAI-16k',
+    maxToken: 16000,
+    price: 3
+  }
+};
+const defaultVectorModels = {
+  'text-embedding-ada-002': {
+    model: 'text-embedding-ada-002',
+    name: 'Embedding-2',
+    price: 0.2
+  }
+};
+export function initSystemModels() {
+  const { chatModels, qaModels, vectorModels } = (() => {
+    try {
+      const chatModels = Object.values(JSON.parse(readFileSync('data/ChatModels.json', 'utf-8')));
+      const qaModels = Object.values(JSON.parse(readFileSync('data/QAModels.json', 'utf-8')));
+      const vectorModels = Object.values(
+        JSON.parse(readFileSync('data/VectorModels.json', 'utf-8'))
+      );
+
+      return {
+        chatModels,
+        qaModels,
+        vectorModels
+      };
+    } catch (error) {
+      console.log(error);
+
+      return {
+        chatModels: Object.values(defaultmodels),
+        qaModels: Object.values(defaultQaModels),
+        vectorModels: Object.values(defaultVectorModels)
+      };
+    }
+  })() as {
+    chatModels: ChatModelItemType[];
+    qaModels: QAModelItemType[];
+    vectorModels: VectorModelItemType[];
+  };
+  global.chatModels = chatModels;
+  global.qaModels = qaModels;
+  global.vectorModels = vectorModels;
+  console.log({
+    chatModels,
+    qaModels,
+    vectorModels
+  });
+
+  return {
+    chatModels,
+    qaModels,
+    vectorModels
+  };
+}
+
+export function initFeConfig() {
+  const feConfig = JSON.parse(readFileSync('data/FeConfig.json', 'utf-8'));
+
+  global.feConfigs = feConfig;
+  console.log(feConfig);
+
+  return feConfig;
+}

client/src/pages/api/user/sendAuthCode.ts

@@ -32,6 +32,11 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     await connectToDatabase();
 
+    // register switch
+    if (type === UserAuthTypeEnum.register && !global.feConfigs.show_register) {
+      throw new Error('Register is closed');
+    }
+
     const code = nanoid();
 
     // 判断 1 分钟内是否有重复数据