fix bug

taais-modules/taais-biz/src/main/java/com/taais/biz/constant/BizConstant.java

@@ -46,6 +46,15 @@ public class BizConstant {
     public static final String TASK_STATUS_PROCESSING = "1";
     public static final String TASK_STATUS_SUCCEED = "2";
     public static final String TASK_STATUS_FAILED = "3";
+    // 同步数据字典配置biz_ag_type
+    public static final String TASK_BIZ_AG_TYPE1_CODE = "1"; // 训练
+    public static final String TASK_BIZ_AG_TYPE1_CN = "训练"; // 训练
+    public static final String TASK_BIZ_AG_TYPE2_CODE = "2"; // 测试
+    public static final String TASK_BIZ_AG_TYPE2_CN = "验证"; // 测试
+    public static final String TASK_BIZ_AG_TYPE3_CODE = "3"; // 推理
+    public static final String TASK_BIZ_AG_TYPE3_CN = "测试"; // 推理
+    public static final String TASK_BIZ_AG_TYPE4_CODE = "4"; // 增广
+    public static final String TASK_BIZ_AG_TYPE4_CN = "增广"; // 增广
 
     public static final String TYPE_DATA_PROCESS = "dataProcess";
     public static final String TYPE_DATA_BIZ_PROCESS = "dataBizProcess";

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

@@ -609,10 +609,21 @@ public class AlgorithmTaskServiceImpl extends BaseServiceImpl<AlgorithmTaskMappe
         algorithmSubtask.setTaskId(algorithmTask.getId());
         algorithmSubtask.setIndex(Long.valueOf(step));
         // todo allen
-        algorithmSubtask.setName("todo allen");
-        algorithmSubtask.setType("todo allen");
-        algorithmSubtask.setParameters("null");
-
+        switch (step) {
+            case BizConstant.STEP5_TRAIN -> {
+                algorithmSubtask.setName(BizConstant.TASK_BIZ_AG_TYPE1_CN);
+                algorithmSubtask.setType(BizConstant.TASK_BIZ_AG_TYPE1_CODE);
+            }
+            case BizConstant.STEP9_TEST -> {
+                algorithmSubtask.setName(BizConstant.TASK_BIZ_AG_TYPE2_CN);
+                algorithmSubtask.setType(BizConstant.TASK_BIZ_AG_TYPE2_CODE);
+            }
+            case BizConstant.STEP13_REASONING -> {
+                algorithmSubtask.setName(BizConstant.TASK_BIZ_AG_TYPE3_CN);
+                algorithmSubtask.setType(BizConstant.TASK_BIZ_AG_TYPE3_CODE);
+            }
+            default -> log.warn("BuildAlgorithmSubtaskBo 这块出现了一些错误，step：{}", step);
+        }
         algorithmSubtask.setStatus(BizConstant.TASK_STATUS_PENDING);
         return algorithmSubtask;
     }