From 733e206f7c462dc8b4650069fe328e828429ac55 Mon Sep 17 00:00:00 2001 From: limin Date: Mon, 6 Jan 2025 16:38:26 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E9=9B=86=E5=AF=BC=E5=85=A5?= =?UTF-8?q?=20=E7=8A=B6=E6=80=81=E4=BF=AE=E6=94=B9=E5=88=A4=E6=96=AD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../module/llm/enums/ErrorCodeConstants.java | 1 + .../FineTuningTaskServiceImpl.java | 16 +++++++++++++++- 2 files changed, 16 insertions(+), 1 deletion(-) diff --git a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java index 18e4d6e71..bf816359b 100644 --- a/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java +++ b/yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java @@ -92,6 +92,7 @@ public interface ErrorCodeConstants { ErrorCode PROMPT_TEMPLATES_TAGS_BACKUP_NOT_EXISTS = new ErrorCode(10037, "模板信息不存在"); ErrorCode DATA_PROCESS_TASK_NAME_NOT_EXISTS = new ErrorCode(10038, "数据处理任务名称已存在"); + ErrorCode FINE_TUNING_TASK_NAME_NOT_EXISTS = new ErrorCode(10039, "模型调优任务名称已存在"); } 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 a8f7db577..0b4becb67 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 @@ -25,7 +25,7 @@ import java.util.Map; import java.util.stream.Collectors; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; -import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.FINE_TUNING_TASK_NOT_EXISTS; +import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.*; /** @@ -49,6 +49,7 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { @Override public Long createFineTuningTask(FineTuningTaskSaveReqVO createReqVO) { // 插入 + validateFineTuningNameTaskExists(createReqVO); FineTuningTaskDO fineTuningTask = BeanUtils.toBean(createReqVO, FineTuningTaskDO.class); fineTuningTaskMapper.insert(fineTuningTask); //todo 调用模型服务，创建调优任务 @@ -61,6 +62,7 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { public void updateFineTuningTask(FineTuningTaskSaveReqVO updateReqVO) { // 校验存在 validateFineTuningTaskExists(updateReqVO.getId()); + validateFineTuningNameTaskExists(updateReqVO); // 更新 FineTuningTaskDO updateObj = BeanUtils.toBean(updateReqVO, FineTuningTaskDO.class); fineTuningTaskMapper.updateById(updateObj); @@ -83,6 +85,18 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { throw exception(FINE_TUNING_TASK_NOT_EXISTS); } } + private void validateFineTuningNameTaskExists(FineTuningTaskSaveReqVO reqVO) { + LambdaQueryWrapper wrapper = new LambdaQueryWrapper() + .eq(FineTuningTaskDO::getTaskName, reqVO.getTaskName()); + + if (reqVO.getId() != null){ + wrapper.ne(FineTuningTaskDO::getId, reqVO.getId()); + } + List dineTuningTaskDos = fineTuningTaskMapper.selectList(wrapper); + if (CollectionUtils.isNotEmpty(dineTuningTaskDos)){ + throw exception(FINE_TUNING_TASK_NAME_NOT_EXISTS); + } + } @Override public FineTuningTaskRespVO getFineTuningTask(Long id) {