DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

模型调优详情 获取训练日志

Browse Source
This commit is contained in:
limin 2025-01-09 17:24:14 +08:00
parent 795d7c81b3
commit 906dd7b3c4
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java
View File

@ -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;
}

7
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
View File

@ -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