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

[insert] 获取已部署完成的模型调优任务

Browse Source
This commit is contained in:
Liuyang 2025-01-20 14:09:29 +08:00
parent befe0aa672
commit 9f181c21f3
4 changed files with 50 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/FinetuningTaskStatusEnum.java
View File

@ -7,7 +7,7 @@ import lombok.Getter;
import java.util.Arrays;
/**
* 用户类型的枚举值
* 微调任务状态的枚举值
*
* @author 张陶
*/
@ -29,7 +29,7 @@ public enum FinetuningTaskStatusEnum implements IntArrayValuable {
public static final int[] ARRAYS = Arrays.stream(values()).mapToInt(FinetuningTaskStatusEnum::getStatus).toArray();
/**
* 用户类型
* 微调任务类型
*/
private final Integer status;

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

@ -101,6 +101,14 @@ public class FineTuningTaskController {
return success(list);
}
@GetMapping("/enable")
@Operation(summary = "获得模型调优 —— 获取已部署完成的调优任务")
// @PreAuthorize("@ss.hasPermission('llm:fine-tuning-task:query')")
public CommonResult<List<FineTuningTaskRespVO>> getEnableFineTuningTaskList() {
List<FineTuningTaskRespVO> list = fineTuningTaskService.selectEnable();
return success(list);
}
@GetMapping("/export-excel")
@Operation(summary = "导出模型调优 —— 微调任务 Excel")
// @PreAuthorize("@ss.hasPermission('llm:fine-tuning-task:export')")
@ -114,4 +122,4 @@ public class FineTuningTaskController {
BeanUtils.toBean(list, FineTuningTaskRespVO.class));
}
}
}

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

@ -1,11 +1,13 @@
package cn.iocoder.yudao.module.llm.service.finetuningtask;
import java.util.*;
import javax.validation.*;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask.FineTuningTaskDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.FineTuningTaskPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.FineTuningTaskRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.FineTuningTaskSaveReqVO;
import cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask.FineTuningTaskDO;
import javax.validation.Valid;
import java.util.List;
/**
* 模型调优 —— 微调任务 Service 接口
@ -20,21 +22,21 @@ public interface FineTuningTaskService {
* @param createReqVO 创建信息
* @return 编号
*/
Long createFineTuningTask(@Valid FineTuningTaskSaveReqVO createReqVO);
Long createFineTuningTask (@Valid FineTuningTaskSaveReqVO createReqVO);
/**
* 更新模型调优 —— 微调任务
*
* @param updateReqVO 更新信息
*/
void updateFineTuningTask(@Valid FineTuningTaskSaveReqVO updateReqVO);
void updateFineTuningTask (@Valid FineTuningTaskSaveReqVO updateReqVO);
/**
* 删除模型调优 —— 微调任务
*
* @param id 编号
*/
void deleteFineTuningTask(Long id);
void deleteFineTuningTask (Long id);
/**
* 获得模型调优 —— 微调任务
@ -42,7 +44,7 @@ public interface FineTuningTaskService {
* @param id 编号
* @return 模型调优 —— 微调任务
*/
FineTuningTaskRespVO getFineTuningTask(Long id);
FineTuningTaskRespVO getFineTuningTask (Long id);
/**
* 获得模型调优 —— 微调任务分页
@ -50,7 +52,7 @@ public interface FineTuningTaskService {
* @param pageReqVO 分页查询
* @return 模型调优 —— 微调任务分页
*/
PageResult<FineTuningTaskDO> getFineTuningTaskPage(FineTuningTaskPageReqVO pageReqVO);
PageResult<FineTuningTaskDO> getFineTuningTaskPage (FineTuningTaskPageReqVO pageReqVO);
/**
* 获得模型调优 —— 微调任务分页
@ -58,20 +60,28 @@ public interface FineTuningTaskService {
* @param pageReqVO 分页查询
* @return 模型调优 —— 微调任务分页
*/
PageResult<FineTuningTaskRespVO> getFineTuningTaskPage1(FineTuningTaskPageReqVO pageReqVO);
PageResult<FineTuningTaskRespVO> getFineTuningTaskPage1 (FineTuningTaskPageReqVO pageReqVO);
/**
* 停止微调任务
*
* @param id
*/
void stopFineTuningTask(Long id);
void stopFineTuningTask (Long id);
/**
* 部署微调任务
*
* @param id
*/
void startFineTuningTask(Long id);
void startFineTuningTask (Long id);
List<FineTuningTaskRespVO> selectAll();
List<FineTuningTaskRespVO> selectAll ();
}
/**
* 获得模型调优 —— 获取已部署完成的调优任务
*
* @return 已部署完成的调优任务
*/
List<FineTuningTaskRespVO> selectEnable ();
}

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

@ -178,6 +178,20 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
return BeanUtils.toBean(fineTuningTaskDOS, FineTuningTaskRespVO.class);
}
/**
* 获得模型调优 —— 获取已部署完成的调优任务
*
* @return 已部署完成的调优任务
*/
@Override
public List<FineTuningTaskRespVO> selectEnable () {
List<FineTuningTaskDO> fineTuningTaskDOS = fineTuningTaskMapper.selectList();
fineTuningTaskDOS = fineTuningTaskDOS.stream()
.filter(item->item.getStatus().equals(FinetuningTaskStatusEnum.FINISHED.getStatus()))
.collect(Collectors.toList());
return BeanUtils.toBean(fineTuningTaskDOS, FineTuningTaskRespVO.class);
}
@Override
public void stopFineTuningTask(Long id) {
FineTuningTaskDO fineTuningTaskDO = validateFineTuningTaskExists(id);