monorepo packages (#344)

projects/app/src/pages/api/system/getInitData.ts

@@ -0,0 +1,162 @@
+import type { FeConfigsType, SystemEnvType } from '@/types';
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import { readFileSync } from 'fs';
+import {
+  type QAModelItemType,
+  type ChatModelItemType,
+  type VectorModelItemType,
+  FunctionModelItemType
+} from '@/types/model';
+
+export type InitDateResponse = {
+  chatModels: ChatModelItemType[];
+  qaModel: QAModelItemType;
+  vectorModels: VectorModelItemType[];
+  feConfigs: FeConfigsType;
+  systemVersion: string;
+};
+
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  if (!global.feConfigs) {
+    await getInitConfig();
+  }
+  jsonRes<InitDateResponse>(res, {
+    data: {
+      feConfigs: global.feConfigs,
+      chatModels: global.chatModels,
+      qaModel: global.qaModel,
+      vectorModels: global.vectorModels,
+      systemVersion: global.systemVersion || '0.0.0'
+    }
+  });
+}
+
+const defaultSystemEnv: SystemEnvType = {
+  vectorMaxProcess: 15,
+  qaMaxProcess: 15,
+  pgIvfflatProbe: 20
+};
+const defaultFeConfigs: FeConfigsType = {
+  show_emptyChat: true,
+  show_contact: true,
+  show_git: true,
+  show_doc: true,
+  systemTitle: 'FastGPT',
+  authorText: 'Made by FastGPT Team.',
+  limit: {
+    exportLimitMinutes: 0
+  },
+  scripts: []
+};
+const defaultChatModels = [
+  {
+    model: 'gpt-3.5-turbo',
+    name: 'GPT35-4k',
+    contextMaxToken: 4000,
+    quoteMaxToken: 2400,
+    maxTemperature: 1.2,
+    price: 0
+  },
+  {
+    model: 'gpt-3.5-turbo-16k',
+    name: 'GPT35-16k',
+    contextMaxToken: 16000,
+    quoteMaxToken: 8000,
+    maxTemperature: 1.2,
+    price: 0
+  },
+  {
+    model: 'gpt-4',
+    name: 'GPT4-8k',
+    contextMaxToken: 8000,
+    quoteMaxToken: 4000,
+    maxTemperature: 1.2,
+    price: 0
+  }
+];
+const defaultQAModel = {
+  model: 'gpt-3.5-turbo-16k',
+  name: 'GPT35-16k',
+  maxToken: 16000,
+  price: 0
+};
+export const defaultExtractModel: FunctionModelItemType = {
+  model: 'gpt-3.5-turbo-16k',
+  name: 'GPT35-16k',
+  maxToken: 16000,
+  price: 0,
+  prompt: '',
+  functionCall: true
+};
+export const defaultCQModel: FunctionModelItemType = {
+  model: 'gpt-3.5-turbo-16k',
+  name: 'GPT35-16k',
+  maxToken: 16000,
+  price: 0,
+  prompt: '',
+  functionCall: true
+};
+
+const defaultVectorModels: VectorModelItemType[] = [
+  {
+    model: 'text-embedding-ada-002',
+    name: 'Embedding-2',
+    price: 0,
+    defaultToken: 500,
+    maxToken: 3000
+  }
+];
+
+export async function getInitConfig() {
+  try {
+    if (global.feConfigs) return;
+
+    getSystemVersion();
+
+    const filename =
+      process.env.NODE_ENV === 'development' ? 'data/config.local.json' : '/app/data/config.json';
+    const res = JSON.parse(readFileSync(filename, 'utf-8'));
+
+    console.log(`System Version: ${global.systemVersion}`);
+
+    console.log(res);
+
+    global.systemEnv = res.SystemParams
+      ? { ...defaultSystemEnv, ...res.SystemParams }
+      : defaultSystemEnv;
+    global.feConfigs = res.FeConfig ? { ...defaultFeConfigs, ...res.FeConfig } : defaultFeConfigs;
+    global.chatModels = res.ChatModels || defaultChatModels;
+    global.qaModel = res.QAModel || defaultQAModel;
+    global.extractModel = res.ExtractModel || defaultExtractModel;
+    global.cqModel = res.CQModel || defaultCQModel;
+    global.vectorModels = res.VectorModels || defaultVectorModels;
+  } catch (error) {
+    setDefaultData();
+    console.log('get init config error, set default', error);
+  }
+}
+
+export function setDefaultData() {
+  global.systemEnv = defaultSystemEnv;
+  global.feConfigs = defaultFeConfigs;
+  global.chatModels = defaultChatModels;
+  global.qaModel = defaultQAModel;
+  global.vectorModels = defaultVectorModels;
+}
+
+export function getSystemVersion() {
+  try {
+    if (process.env.NODE_ENV === 'development') {
+      global.systemVersion = process.env.npm_package_version || '0.0.0';
+      return;
+    }
+    const packageJson = JSON.parse(readFileSync('/app/package.json', 'utf-8'));
+
+    global.systemVersion = packageJson?.version;
+  } catch (error) {
+    console.log(error);
+
+    global.systemVersion = '0.0.0';
+  }
+}