training queue

src/service/models/model.ts

@@ -2,7 +2,7 @@ import { Schema, model, models, Model as MongoModel } from 'mongoose';
 import { ModelSchema as ModelType } from '@/types/mongoSchema';
 import {
   ModelVectorSearchModeMap,
-  ModelVectorSearchModeEnum,
+  appVectorSearchModeEnum,
   ChatModelMap,
   OpenAiChatEnum
 } from '@/constants/model';
@@ -40,7 +40,7 @@ const ModelSchema = new Schema({
       // knowledge base search mode
       type: String,
       enum: Object.keys(ModelVectorSearchModeMap),
-      default: ModelVectorSearchModeEnum.hightSimilarity
+      default: appVectorSearchModeEnum.hightSimilarity
     },
     systemPrompt: {
       // 系统提示词

src/service/models/splitData.ts

@@ -1,32 +0,0 @@
-/* 模型的知识库 */
-import { Schema, model, models, Model as MongoModel } from 'mongoose';
-import { SplitDataSchema as SplitDataType } from '@/types/mongoSchema';
-
-const SplitDataSchema = new Schema({
-  userId: {
-    type: Schema.Types.ObjectId,
-    ref: 'user',
-    required: true
-  },
-  prompt: {
-    // 拆分时的提示词
-    type: String,
-    required: true
-  },
-  kbId: {
-    type: Schema.Types.ObjectId,
-    ref: 'kb',
-    required: true
-  },
-  textList: {
-    type: [String],
-    default: []
-  },
-  errorText: {
-    type: String,
-    default: ''
-  }
-});
-
-export const SplitData: MongoModel<SplitDataType> =
-  models['splitData'] || model('splitData', SplitDataSchema);

src/service/models/trainingData.ts

@@ -0,0 +1,38 @@
+/* 模型的知识库 */
+import { Schema, model, models, Model as MongoModel } from 'mongoose';
+import { TrainingDataSchema as TrainingDateType } from '@/types/mongoSchema';
+
+// pgList and vectorList, Only one of them will work
+
+const TrainingDataSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  kbId: {
+    type: Schema.Types.ObjectId,
+    ref: 'kb',
+    required: true
+  },
+  lockTime: {
+    type: Date,
+    default: () => new Date('2000/1/1')
+  },
+  vectorList: {
+    type: [{ q: String, a: String }],
+    default: []
+  },
+  prompt: {
+    // 拆分时的提示词
+    type: String,
+    default: ''
+  },
+  qaList: {
+    type: [String],
+    default: []
+  }
+});
+
+export const TrainingData: MongoModel<TrainingDateType> =
+  models['trainingData'] || model('trainingData', TrainingDataSchema);