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 8c1b0a17a..00ff06d4d 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 @@ -1,5 +1,6 @@ package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo; +import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO; import io.swagger.v3.oas.annotations.media.Schema; import lombok.*; import java.util.*; @@ -134,4 +135,7 @@ public class FineTuningTaskRespVO { @Schema(description = "数据集名称") private String datasetName; + @Schema(description = "模型广场信息") + private AigcFineTuningDetailRespVO detailRespVO; + } \ No newline at end of file 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 eceb561e6..aaea244c3 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 @@ -13,6 +13,8 @@ import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetMapper; import cn.iocoder.yudao.module.llm.dal.mysql.finetuningtask.FineTuningTaskMapper; import cn.iocoder.yudao.module.llm.enums.FinetuningTaskStatusEnum; import cn.iocoder.yudao.module.llm.service.async.AsyncFineTuningTaskService; +import cn.iocoder.yudao.module.llm.service.http.TrainHttpService; +import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; @@ -21,6 +23,7 @@ import org.springframework.validation.annotation.Validated; import javax.annotation.Resource; import java.util.Collections; +import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.stream.Collectors; @@ -46,6 +49,8 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { private DatasetMapper datasetMapper; @Resource private AsyncFineTuningTaskService asyncFineTuningTaskService; + @Resource + TrainHttpService trainHttpService; @Override public Long createFineTuningTask(FineTuningTaskSaveReqVO createReqVO) { @@ -116,6 +121,8 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { BaseModelDO baseModelDO = baseModelMapper.selectById(baseModelId); result.setBaseModelName(baseModelDO.getModelName()); } + AigcFineTuningDetailRespVO aigcFineTuningDetailRespVO = trainHttpService.finetuningDetail(new HashMap<>(), fineTuningTaskDO.getJobId()); + result.setDetailRespVO(aigcFineTuningDetailRespVO); return result; } @Override