feat: sync api collection will refresh title;perf: invite link ux (#4237)

* update queue

* feat: sync api collection will refresh title

* sync collection

* remove lock

* perf: invite link ux

projects/app/src/pages/api/core/dataset/collection/create/apiCollection.ts

@@ -41,7 +41,7 @@ async function handler(req: NextApiRequest): CreateCollectionResponse {
     return Promise.reject(DatasetErrEnum.sameApiCollection);
   }
 
-  const content = await readApiServerFileContent({
+  const { title, rawText } = await readApiServerFileContent({
     apiServer,
     feishuServer,
     yuqueServer,
@@ -53,14 +53,14 @@ async function handler(req: NextApiRequest): CreateCollectionResponse {
 
   const { collectionId, insertResults } = await createCollectionAndInsertData({
     dataset,
-    rawText: content,
+    rawText,
     relatedId: apiFileId,
     createCollectionParams: {
       ...body,
       teamId,
       tmbId,
       type: DatasetCollectionTypeEnum.apiFile,
-      name,
+      name: title || name,
       apiFileId,
       metadata: {
         relatedImgId: apiFileId

projects/app/src/pages/api/core/dataset/collection/create/reTrainingCollection.ts

@@ -77,7 +77,7 @@ async function handler(
     return Promise.reject(i18nT('dataset:collection_not_support_retraining'));
   })();
 
-  const rawText = await readDatasetSourceRawText({
+  const { title, rawText } = await readDatasetSourceRawText({
     teamId,
     tmbId,
     customPdfParse,
@@ -100,7 +100,7 @@ async function handler(
         teamId: collection.teamId,
         tmbId: collection.tmbId,
         datasetId: collection.dataset._id,
-        name: collection.name,
+        name: title || collection.name,
         type: collection.type,
 
         customPdfParse,

projects/app/src/pages/api/core/dataset/file/getPreviewChunks.ts

@@ -82,7 +82,7 @@ async function handler(
     };
   })();
 
-  const rawText = await readDatasetSourceRawText({
+  const { rawText } = await readDatasetSourceRawText({
     teamId,
     tmbId,
     type,

projects/app/src/pages/api/core/dataset/training/getQueueLen.ts

@@ -1,27 +1,31 @@
 import type { NextApiRequest } from 'next';
 import { MongoDatasetTraining } from '@fastgpt/service/core/dataset/training/schema';
 import { authCert } from '@fastgpt/service/support/permission/auth/common';
-import { GetTrainingQueueProps } from '@/global/core/dataset/api';
 import { NextAPI } from '@/service/middleware/entry';
 import { readFromSecondary } from '@fastgpt/service/common/mongo/utils';
+import { TrainingModeEnum } from '@fastgpt/global/core/dataset/constants';
+
+export type GetQueueLenResponse = {
+  vectorTrainingCount: number;
+  qaTrainingCount: number;
+  autoTrainingCount: number;
+  imageTrainingCount: number;
+};
 
 async function handler(req: NextApiRequest) {
   await authCert({ req, authToken: true });
-  const { vectorModel, agentModel } = req.query as GetTrainingQueueProps;
 
   // get queue data
-  // 分别统计 model = vectorModel和agentModel的数量
   const data = await MongoDatasetTraining.aggregate(
     [
       {
         $match: {
-          lockTime: { $lt: new Date('2040/1/1') },
-          $or: [{ model: { $eq: vectorModel } }, { model: { $eq: agentModel } }]
+          lockTime: { $lt: new Date('2040/1/1') }
         }
       },
       {
         $group: {
-          _id: '$model',
+          _id: '$mode',
           count: { $sum: 1 }
         }
       }
@@ -31,12 +35,16 @@ async function handler(req: NextApiRequest) {
     }
   );
 
-  const vectorTrainingCount = data.find((item) => item._id === vectorModel)?.count || 0;
-  const agentTrainingCount = data.find((item) => item._id === agentModel)?.count || 0;
+  const vectorTrainingCount = data.find((item) => item._id === TrainingModeEnum.chunk)?.count || 0;
+  const qaTrainingCount = data.find((item) => item._id === TrainingModeEnum.qa)?.count || 0;
+  const autoTrainingCount = data.find((item) => item._id === TrainingModeEnum.auto)?.count || 0;
+  const imageTrainingCount = data.find((item) => item._id === TrainingModeEnum.image)?.count || 0;
 
   return {
     vectorTrainingCount,
-    agentTrainingCount
+    qaTrainingCount,
+    autoTrainingCount,
+    imageTrainingCount
   };
 }