diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/FineTuningTaskController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/FineTuningTaskController.java index e13e6357b..5ca6bcfa2 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/FineTuningTaskController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/FineTuningTaskController.java @@ -67,8 +67,7 @@ public class FineTuningTaskController { @Parameter(name = "id", description = "编号", required = true, example = "1024") @PreAuthorize("@ss.hasPermission('llm:fine-tuning-task:query')") public CommonResult getFineTuningTask(@RequestParam("id") Long id) { - FineTuningTaskDO fineTuningTask = fineTuningTaskService.getFineTuningTask(id); - return success(BeanUtils.toBean(fineTuningTask, FineTuningTaskRespVO.class)); + return success(fineTuningTaskService.getFineTuningTask(id)); } @PutMapping("/stop") diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskService.java index d94bb516f..87c3d2ee1 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskService.java @@ -42,7 +42,7 @@ public interface FineTuningTaskService { * @param id 编号 * @return 模型调优 —— 微调任务 */ - FineTuningTaskDO getFineTuningTask(Long id); + FineTuningTaskRespVO getFineTuningTask(Long id); /** * 获得模型调优 —— 微调任务分页 diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java index f19020f76..e4ffc1047 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java @@ -81,8 +81,20 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { } @Override - public FineTuningTaskDO getFineTuningTask(Long id) { - return fineTuningTaskMapper.selectById(id); + public FineTuningTaskRespVO getFineTuningTask(Long id) { + FineTuningTaskDO fineTuningTaskDO = fineTuningTaskMapper.selectById(id); + FineTuningTaskRespVO result = BeanUtils.toBean(fineTuningTaskDO, FineTuningTaskRespVO.class); + Long dataset = fineTuningTaskDO.getDataset(); + if (dataset != null) { + DatasetDO datasetDO = datasetMapper.selectById(dataset); + result.setDatasetName(datasetDO.getDatasetName()); + } + Long baseModelId = fineTuningTaskDO.getBaseModelId(); + if (baseModelId != null) { + BaseModelDO baseModelDO = baseModelMapper.selectById(baseModelId); + result.setBaseModelName(baseModelDO.getModelName()); + } + return result; } @Override public PageResult getFineTuningTaskPage1(FineTuningTaskPageReqVO pageReqVO) {