From b97120a635e67f05f1c455c2a9b8881021bd18d0 Mon Sep 17 00:00:00 2001 From: limin Date: Thu, 16 Jan 2025 15:01:43 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=AE=AD=E7=BB=83=E6=97=B6?= =?UTF-8?q?=E9=95=BF=E8=BF=94=E5=9B=9E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../admin/finetuningtask/vo/FineTuningTaskPageReqVO.java | 3 +++ .../admin/finetuningtask/vo/FineTuningTaskRespVO.java | 3 ++- .../admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java | 3 +++ .../llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java | 1 + .../llm/service/finetuningtask/FineTuningTaskSyncService.java | 2 ++ 5 files changed, 11 insertions(+), 1 deletion(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java index 5ba229950..370491db1 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java @@ -24,6 +24,9 @@ public class FineTuningTaskPageReqVO extends PageParam { @Schema(description = "训练阶段，使用字典（llm_training_stage）") private Integer trainStage; + @Schema(description = "训练时长") + private String trainDuration; + @Schema(description = "微调方法，使用字典（llm_fine_tuning_method）") private Integer method; diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java index 45ddeeae4..2124d9e62 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java @@ -134,7 +134,8 @@ public class FineTuningTaskRespVO { @Schema(description = "数据集名称") private String datasetName; - + @Schema(description = "训练时长") + private String trainDuration; @Schema(description = "模型广场信息") private AigcFineTuningDetailRespVO detailRespVO; diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java index c53129001..4451f79f8 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java @@ -28,6 +28,9 @@ public class FineTuningTaskSaveReqVO { @Schema(description = "描述") private String taskIntro; + @Schema(description = "训练时长") + private String trainDuration; + @Schema(description = "GPU类型，使用字典（llm_gpu_type）", example = "1") private Integer gpuType; diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java index 3f1bd8bbd..90af82c77 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java @@ -43,6 +43,7 @@ public class FineTuningTaskDO extends BaseDO { * 枚举 {@link TODO llm_training_stage 对应的类} */ private Integer trainStage; + private String trainDuration; /** * 微调方法，使用字典（llm_fine_tuning_method） * diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java index 9cf0a9730..879770a97 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java @@ -6,6 +6,7 @@ import cn.iocoder.yudao.module.llm.enums.FineTuningTaskStatusConstants; import cn.iocoder.yudao.module.llm.enums.FinetuningTaskStatusEnum; import cn.iocoder.yudao.module.llm.service.http.TrainHttpService; import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO; +import com.baomidou.mybatisplus.core.toolkit.StringUtils; import com.esotericsoftware.minlog.Log; import org.springframework.scheduling.annotation.Scheduled; import org.springframework.stereotype.Component; @@ -37,6 +38,7 @@ public class FineTuningTaskSyncService { FineTuningTaskDO updateObj = new FineTuningTaskDO(); updateObj.setId(fineTuningTaskDO.getId()); updateObj.setStatus(status); + updateObj.setTrainDuration(StringUtils.isNotBlank(resp.getTrainDuration())?resp.getTrainDuration():""); fineTuningTaskMapper.updateById(updateObj); } }