fix bugs

taais-common/taais-common-core/src/main/java/com/taais/common/core/utils/file/MimeTypeUtils.java

@@ -35,7 +35,10 @@ public class MimeTypeUtils {
         // 视频格式
         "mp4", "avi", "rmvb",
         // pdf
-        "pdf"};
+        "pdf",
+        // pt
+        "pt"
+    };
 
     public static String getExtension(String prefix) {
         switch (prefix) {

taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/AlgorithmTaskServiceImpl.java

@@ -6,6 +6,7 @@ import com.google.gson.reflect.TypeToken;
 import com.mybatisflex.core.paginate.Page;
 import com.mybatisflex.core.query.QueryWrapper;
 import com.taais.biz.constant.BizConstant;
+import com.taais.biz.domain.AlgorithmConfig;
 import com.taais.biz.domain.AlgorithmTask;
 import com.taais.biz.domain.Data;
 import com.taais.biz.domain.bo.*;
@@ -279,6 +280,8 @@ public class AlgorithmTaskServiceImpl extends BaseServiceImpl<AlgorithmTaskMappe
             for (TaskDto dto : taskDto.getTrain()) {
                 // 创建算法处理任务
                 AlgorithmBizProcessBo algorithmBizProcessBo = new AlgorithmBizProcessBo();
+                AlgorithmConfig algorithmConfig = algorithmConfigService.getById(dto.getAlgorithmId());
+                algorithmBizProcessBo.setName(algorithmConfig != null ? algorithmConfig.getAlgorithmName() : null);
                 algorithmBizProcessBo.setAlgorithmId(dto.getAlgorithmId());
                 algorithmBizProcessBo.setModelId(dto.getModelId());
                 algorithmBizProcessBo.setStatus(BizConstant.TASK_STATUS_PENDING);
@@ -396,6 +399,8 @@ public class AlgorithmTaskServiceImpl extends BaseServiceImpl<AlgorithmTaskMappe
             for (TaskDto dto : taskDto.getTrain()) {
                 // 创建算法处理任务
                 AlgorithmBizProcessBo algorithmBizProcessBo = new AlgorithmBizProcessBo();
+                AlgorithmConfig algorithmConfig = algorithmConfigService.getById(dto.getAlgorithmId());
+                algorithmBizProcessBo.setName(algorithmConfig != null ? algorithmConfig.getAlgorithmName() : null);
                 algorithmBizProcessBo.setAlgorithmId(dto.getAlgorithmId());
                 algorithmBizProcessBo.setModelId(dto.getModelId());
                 algorithmBizProcessBo.setStatus(BizConstant.TASK_STATUS_PENDING);
@@ -552,6 +557,8 @@ public class AlgorithmTaskServiceImpl extends BaseServiceImpl<AlgorithmTaskMappe
             for (TaskDto dto : taskDto.getTrain()) {
                 // 创建算法处理任务
                 AlgorithmBizProcessBo algorithmBizProcessBo = new AlgorithmBizProcessBo();
+                AlgorithmConfig algorithmConfig = algorithmConfigService.getById(dto.getAlgorithmId());
+                algorithmBizProcessBo.setName(algorithmConfig != null ? algorithmConfig.getAlgorithmName() : null);
                 algorithmBizProcessBo.setAlgorithmId(dto.getAlgorithmId());
                 algorithmBizProcessBo.setModelId(dto.getModelId());
                 algorithmBizProcessBo.setStatus(BizConstant.TASK_STATUS_PENDING);
@@ -704,9 +711,4 @@ public class AlgorithmTaskServiceImpl extends BaseServiceImpl<AlgorithmTaskMappe
             return MapstructUtils.convert(data, DataVo. class);
         }
     }
-
-    public static void main(String[] args) throws IOException {
-        String profile = "D:/taais/uploadPath";
-        FileUtils.copyFile(profile + StringUtils.substringAfter("/profile/upload\\2024\\07\\13\\101953322/3.jpg", Constants.RESOURCE_PREFIX), profile + "/task/169578890091683840/train/original/0/"+BizConstant.IMAGE);
-    }
 }