feat: qa生成

src/service/models/bill.ts

@@ -1,4 +1,5 @@
 import { Schema, model, models } from 'mongoose';
+import { ModelList } from '@/constants/model';
 
 const BillSchema = new Schema({
   userId: {
@@ -6,6 +7,16 @@ const BillSchema = new Schema({
     ref: 'user',
     required: true
   },
+  type: {
+    type: String,
+    enum: ['chat', 'generateData', 'return'],
+    required: true
+  },
+  modelName: {
+    type: String,
+    enum: ModelList.map((item) => item.model),
+    required: true
+  },
   chatId: {
     type: Schema.Types.ObjectId,
     ref: 'chat',

src/service/models/data.ts

@@ -6,21 +6,13 @@ const DataSchema = new Schema({
     ref: 'user',
     required: true
   },
-  docId: {
+  name: {
     type: String,
     required: true
   },
   createTime: {
     type: Date,
     default: () => new Date()
-  },
-  q: {
-    type: String,
-    required: true
-  },
-  a: {
-    type: String,
-    required: true
   }
 });
 

src/service/models/dataItem.ts

@@ -0,0 +1,48 @@
+import { Schema, model, models } from 'mongoose';
+
+const DataItemSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  dataId: {
+    type: Schema.Types.ObjectId,
+    ref: 'data',
+    required: true
+  },
+  times: {
+    type: Number,
+    default: 3
+  },
+  text: {
+    type: String,
+    required: true
+  },
+  temperature: {
+    type: Number,
+    required: true
+  },
+  result: {
+    type: [
+      {
+        q: {
+          type: String,
+          required: true
+        },
+        a: {
+          type: String,
+          required: true
+        }
+      }
+    ],
+    default: []
+  },
+  status: {
+    // 0-闲置，1-待生成，2-生成中
+    type: Number,
+    default: 1
+  }
+});
+
+export const DataItem = models['dataItem'] || model('dataItem', DataItemSchema);