DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

修复模型服务名称 报错

Browse Source
This commit is contained in:
limin 2025-01-10 15:13:17 +08:00
parent c0edd77c05
commit 10efb123f4
2 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java
View File

@ -93,8 +93,9 @@ 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, "模型调优任务名称已存在");
ErrorCode LEARNING_RESOURCES_NAME_NOT_EXISTS = new ErrorCode(10040, "学习资源标题名称已存在");
ErrorCode KNOWLEDGE_BASE_NAME_NOT_EXISTS = new ErrorCode(10040, "知识库名称已存在");
ErrorCode LEARNING_RESOURCES_NAME_NOT_EXISTS = new ErrorCode(10041, "学习资源标题名称已存在");
ErrorCode KNOWLEDGE_BASE_NAME_NOT_EXISTS = new ErrorCode(10042, "知识库名称已存在");
ErrorCode MODEL_SERVIC_ENAME_NOT_EXISTS = new ErrorCode(10043, "模型名称已存在");
}

16
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelservice/ModelServiceServiceImpl.java
View File

@ -1,9 +1,11 @@
package cn.iocoder.yudao.module.llm.service.modelservice;
import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils;
import cn.iocoder.yudao.module.llm.dal.dataobject.dataset.DatasetDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask.FineTuningTaskDO;
import cn.iocoder.yudao.module.llm.dal.mysql.finetuningtask.FineTuningTaskMapper;
import cn.iocoder.yudao.module.llm.service.async.AsyncModelServiceService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
@ -43,6 +45,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
@Override
public Long createModelService(ModelServiceSaveReqVO createReqVO) {
// 插入
validateModelServiceNameExists(createReqVO);
ModelServiceDO modelService = BeanUtils.toBean(createReqVO, ModelServiceDO.class);
modelServiceMapper.insert(modelService);
asyncModelServiceService.createModelService(modelService);
@ -54,6 +57,7 @@ public class ModelServiceServiceImpl implements ModelServiceService {
public void updateModelService(ModelServiceSaveReqVO updateReqVO) {
// 校验存在
validateModelServiceExists(updateReqVO.getId());
validateModelServiceNameExists(updateReqVO);
// 更新
ModelServiceDO updateObj = BeanUtils.toBean(updateReqVO, ModelServiceDO.class);
modelServiceMapper.updateById(updateObj);
@ -72,6 +76,18 @@ public class ModelServiceServiceImpl implements ModelServiceService {
throw exception(MODEL_SERVICE_NOT_EXISTS);
}
}
private void validateModelServiceNameExists(ModelServiceSaveReqVO updateReqVO) {
LambdaQueryWrapper<ModelServiceDO> wrapper = new LambdaQueryWrapper<ModelServiceDO>()
.eq(ModelServiceDO::getServiceName, updateReqVO.getServiceName());
if (updateReqVO.getId() != null){
wrapper.ne(ModelServiceDO::getId, updateReqVO.getId());
}
List<ModelServiceDO> modelServiceDOS = modelServiceMapper.selectList(wrapper);
if (com.baomidou.mybatisplus.core.toolkit.CollectionUtils.isNotEmpty(modelServiceDOS)){
throw exception(MODEL_SERVIC_ENAME_NOT_EXISTS);
}
}
@Override
public ModelServiceDO getModelService(Long id) {