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 2024-12-31 13:07:55 +08:00
parent e3b700a049
commit 1229366d67
3 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/FineTuningTaskController.java
View File

@ -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<FineTuningTaskRespVO> getFineTuningTask(@RequestParam("id") Long id) {
FineTuningTaskDO fineTuningTask = fineTuningTaskService.getFineTuningTask(id);
return success(BeanUtils.toBean(fineTuningTask, FineTuningTaskRespVO.class));
return success(fineTuningTaskService.getFineTuningTask(id));
}
@PutMapping("/stop")

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

@ -42,7 +42,7 @@ public interface FineTuningTaskService {
* @param id 编号
* @return 模型调优 —— 微调任务
*/
FineTuningTaskDO getFineTuningTask(Long id);
FineTuningTaskRespVO getFineTuningTask(Long id);
/**
* 获得模型调优 —— 微调任务分页

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

@ -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<FineTuningTaskRespVO> getFineTuningTaskPage1(FineTuningTaskPageReqVO pageReqVO) {