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
-  };
-}