[update] 基座模型创建/修改 名称不能重复

This commit is contained in:
Liuyang 2025-01-23 18:01:59 +08:00
parent 4164cfd4d0
commit 2a119c13e3

View File

@ -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<BaseModelDO> wrapper = new LambdaQueryWrapper<BaseModelDO>()
.eq(BaseModelDO::getModelName, reqVO.getModelName());
if (reqVO.getId() != null) {
wrapper.ne(BaseModelDO::getId, reqVO.getId());
}
List<BaseModelDO> baseModel = baseModelMapper.selectList(wrapper);
if (CollectionUtils.isNotEmpty(baseModel)) {
throw exception(LABEL_NAME_EXISTS);
}
}
@Override
public void updateBaseModel (BaseModelSaveReqVO updateReqVO) {
validateBaseModelNameExists(updateReqVO);
// 校验存在
validateBaseModelExists(updateReqVO.getId());
// 更新