feat: 模型数据管理

2023-03-29 00:22:48 +08:00
parent 713332522f
commit f32c557bdd
14 changed files with 366 additions and 95 deletions
--- a/src/constants/model.ts
+++ b/src/constants/model.ts
@@ -1,5 +1,4 @@
-import type { ServiceName } from '@/types/mongoSchema';
-import { ModelSchema } from '../types/mongoSchema';
+import type { ServiceName, ModelDataType, ModelSchema } from '@/types/mongoSchema';

 export enum ChatModelNameEnum {
  GPT35 = 'gpt-3.5-turbo',
@@ -76,6 +75,12 @@ export const formatModelStatus = {
  }
 };

+export const ModelDataStatusMap: Record<ModelDataType, string> = {
+  0: '训练完成',
+  1: '等待训练',
+  2: '训练中'
+};
+
 export const defaultModel: ModelSchema = {
  _id: '',
  userId: '',