From 9f181c21f3bb627b581305a7703bd43bac438034 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Mon, 20 Jan 2025 14:09:29 +0800 Subject: [PATCH] =?UTF-8?q?[insert]=20=E8=8E=B7=E5=8F=96=E5=B7=B2=E9=83=A8?= =?UTF-8?q?=E7=BD=B2=E5=AE=8C=E6=88=90=E7=9A=84=E6=A8=A1=E5=9E=8B=E8=B0=83?= =?UTF-8?q?=E4=BC=98=E4=BB=BB=E5=8A=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../llm/enums/FinetuningTaskStatusEnum.java | 4 +- .../FineTuningTaskController.java | 10 ++++- .../finetuningtask/FineTuningTaskService.java | 40 ++++++++++++------- .../FineTuningTaskServiceImpl.java | 14 +++++++ 4 files changed, 50 insertions(+), 18 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/FinetuningTaskStatusEnum.java b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/FinetuningTaskStatusEnum.java index bfd03b305..eabf11fa5 100644 --- a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/FinetuningTaskStatusEnum.java +++ b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/FinetuningTaskStatusEnum.java @@ -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; 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 901cb9102..a53380669 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 @@ -101,6 +101,14 @@ public class FineTuningTaskController { return success(list); } + @GetMapping("/enable") + @Operation(summary = "获得模型调优 —— 获取已部署完成的调优任务") + // @PreAuthorize("@ss.hasPermission('llm:fine-tuning-task:query')") + public CommonResult> getEnableFineTuningTaskList() { + List 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)); } -} \ 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/FineTuningTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskService.java index f118f566d..c77a075ed 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 @@ -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 getFineTuningTaskPage(FineTuningTaskPageReqVO pageReqVO); + PageResult getFineTuningTaskPage (FineTuningTaskPageReqVO pageReqVO); /** * 获得模型调优 —— 微调任务分页 @@ -58,20 +60,28 @@ public interface FineTuningTaskService { * @param pageReqVO 分页查询 * @return 模型调优 —— 微调任务分页 */ - PageResult getFineTuningTaskPage1(FineTuningTaskPageReqVO pageReqVO); + PageResult getFineTuningTaskPage1 (FineTuningTaskPageReqVO pageReqVO); /** * 停止微调任务 + * * @param id */ - void stopFineTuningTask(Long id); + void stopFineTuningTask (Long id); /** * 部署微调任务 + * * @param id */ - void startFineTuningTask(Long id); + void startFineTuningTask (Long id); - List selectAll(); + List selectAll (); -} \ No newline at end of file + /** + * 获得模型调优 —— 获取已部署完成的调优任务 + * + * @return 已部署完成的调优任务 + */ + List selectEnable (); +} 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 44f553334..418280074 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 @@ -178,6 +178,20 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { return BeanUtils.toBean(fineTuningTaskDOS, FineTuningTaskRespVO.class); } + /** + * 获得模型调优 —— 获取已部署完成的调优任务 + * + * @return 已部署完成的调优任务 + */ + @Override + public List selectEnable () { + List 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);