4.6.3-alpha1 (#529)

projects/app/src/pages/api/admin/initv463.ts

@@ -0,0 +1,55 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@fastgpt/service/common/response';
+import { connectToDatabase } from '@/service/mongo';
+import { authCert } from '@fastgpt/service/support/permission/auth/common';
+import { MongoDatasetData } from '@fastgpt/service/core/dataset/data/schema';
+import { MongoDatasetCollection } from '@fastgpt/service/core/dataset/collection/schema';
+import { TrainingModeEnum } from '@fastgpt/global/core/dataset/constant';
+
+let success = 0;
+/* pg 中的数据搬到 mongo dataset.datas 中，并做映射 */
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const { limit = 50 } = req.body as { limit: number };
+    await authCert({ req, authRoot: true });
+    await connectToDatabase();
+    success = 0;
+
+    await MongoDatasetCollection.updateMany({}, [
+      {
+        $set: {
+          createTime: '$updateTime',
+          trainingType: {
+            $cond: {
+              if: { $ifNull: ['$a', false] },
+              then: TrainingModeEnum.qa,
+              else: TrainingModeEnum.chunk
+            }
+          },
+          chunkSize: 0,
+          fileId: '$metadata.fileId',
+          rawLink: '$metadata.rawLink'
+        }
+      }
+    ]);
+
+    await MongoDatasetData.updateMany(
+      {},
+      {
+        chunkIndex: 0,
+        updateTime: new Date()
+      }
+    );
+
+    jsonRes(res, {
+      message: 'success'
+    });
+  } catch (error) {
+    console.log(error);
+
+    jsonRes(res, {
+      code: 500,
+      error
+    });
+  }
+}