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) {