feat: 训练数据管理

src/pages/api/data/splitData.ts

@@ -4,11 +4,10 @@ import { connectToDatabase, Data, DataItem } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
 import { generateQA } from '@/service/events/generateQA';
 
-/* 定时删除那些不活跃的内容 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
-    let { text, name } = req.body as { text: string; name: string };
-    if (!text || !name) {
+    let { text, dataId } = req.body as { text: string; dataId: string };
+    if (!text || !dataId) {
       throw new Error('参数错误');
     }
     text = text.replace(/\n+/g, '\n');
@@ -18,28 +17,16 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
     const userId = await authToken(authorization);
 
-    // 生成 data 父级
-    const data = await Data.create({
-      userId,
-      name
-    });
-
     const dataItems: any[] = [];
 
     // 格式化文本长度
     for (let i = 0; i <= text.length / 1000; i++) {
-      const dataItem = {
+      dataItems.push({
+        temperature: 0,
         userId,
-        dataId: data._id,
+        dataId,
         text: text.slice(i * 1000, (i + 1) * 1000),
         status: 1
-      };
-
-      [0, 0.2, 0.4, 0.6, 0.8, 1.0].forEach((temperature) => {
-        dataItems.push({
-          temperature,
-          ...dataItem
-        });
       });
     }
 
@@ -58,8 +45,3 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     });
   }
 }
-
-/**
- * 检查文本是否按格式返回
- */
-function splitText(text: string) {}