diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java index 71a58a432..eaa764765 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java @@ -24,6 +24,7 @@ import java.util.Map; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.BASE_MODEL_NOT_EXISTS; +import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.LABEL_NAME_EXISTS; /** * 基座模型 Service 实现类 @@ -55,6 +56,7 @@ public class BaseModelServiceImpl implements BaseModelService { @Override public Long createBaseModel (BaseModelSaveReqVO createReqVO) { + validateBaseModelNameExists(createReqVO); // 插入 BaseModelDO baseModel = BeanUtils.toBean(createReqVO, BaseModelDO.class); baseModelMapper.insert(baseModel); @@ -62,8 +64,27 @@ public class BaseModelServiceImpl implements BaseModelService { return baseModel.getId(); } + /** + * 校验模型名称是否重复 + * + * @param reqVO 模型信息 + */ + private void validateBaseModelNameExists (BaseModelSaveReqVO reqVO) { + LambdaQueryWrapper wrapper = new LambdaQueryWrapper() + .eq(BaseModelDO::getModelName, reqVO.getModelName()); + + if (reqVO.getId() != null) { + wrapper.ne(BaseModelDO::getId, reqVO.getId()); + } + List baseModel = baseModelMapper.selectList(wrapper); + if (CollectionUtils.isNotEmpty(baseModel)) { + throw exception(LABEL_NAME_EXISTS); + } + } + @Override public void updateBaseModel (BaseModelSaveReqVO updateReqVO) { + validateBaseModelNameExists(updateReqVO); // 校验存在 validateBaseModelExists(updateReqVO.getId()); // 更新