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 d16389bb6..cfcb7451a 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 @@ -82,5 +82,6 @@ public interface ErrorCodeConstants { ErrorCode MODEL_ASSESS_TASK_MANUAL_BACKUP_NOT_EXISTS = new ErrorCode(10033, "人工评估备份不存在"); ErrorCode MODEL_ASSESS_TASK_MANUAL_BACKUP_EXISTS = new ErrorCode(100_1002, "人工评估已备份"); + ErrorCode DATASET_NAME_EXISTS = new ErrorCode(10034, "数据集名称重复"); } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java index 04204abbc..90e43e4da 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java @@ -35,6 +35,7 @@ import java.util.List; 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.DATASET_NAME_EXISTS; import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.DATASET_NOT_EXISTS; /** @@ -57,6 +58,8 @@ public class DatasetServiceImpl implements DatasetService { @Override public Long createDataset(DatasetSaveReqVO createReqVO) { + // 校验 + validateDatasetNameExists(createReqVO); if(createReqVO.getType() == null){ createReqVO.setType(0); } @@ -95,6 +98,7 @@ public class DatasetServiceImpl implements DatasetService { public void updateDataset(DatasetSaveReqVO updateReqVO) { // 校验存在 validateDatasetExists(updateReqVO.getId()); + validateDatasetNameExists(updateReqVO); // 更新 DatasetDO updateObj = BeanUtils.toBean(updateReqVO, DatasetDO.class); List datasetFiles = updateReqVO.getDatasetFiles(); @@ -133,6 +137,13 @@ public class DatasetServiceImpl implements DatasetService { throw exception(DATASET_NOT_EXISTS); } } + private void validateDatasetNameExists(DatasetSaveReqVO dateReqVO){ + List datasetDOS = datasetMapper.selectList(new LambdaQueryWrapper() + .eq(DatasetDO::getDatasetName, dateReqVO.getDatasetName())); + if (CollectionUtils.isNotEmpty(datasetDOS)){ + throw exception(DATASET_NAME_EXISTS); + } + } @Override public DatasetRespVO getDataset(Long id) {