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

feat(llm): 为 baseModelDO 添加 aigcModelName属性

Browse Source 
- 在创建 BaseModelDO 对象时，为 remoteModelName 设置对应的 aigcModelName
- 此修改有助于完善模型信息，支持 AIGC（人工智能生成内容）功能
This commit is contained in:
sunxiqing 2025-03-17 10:01:49 +08:00
parent c3106010db
commit b29cd0976b
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java
View File

@ -8,8 +8,10 @@ import cn.iocoder.yudao.module.llm.controller.admin.basemodel.vo.BaseModelPageRe
import cn.iocoder.yudao.module.llm.controller.admin.basemodel.vo.BaseModelSaveReqVO;
import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask.FineTuningTaskDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.servername.ServerNameDO;
import cn.iocoder.yudao.module.llm.dal.mysql.basemodel.BaseModelMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelservice.ModelServiceMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.servername.ServerNameMapper;
import cn.iocoder.yudao.module.llm.enums.FinetuningTaskStatusEnum;
import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties;
@ -18,6 +20,7 @@ import cn.iocoder.yudao.module.llm.service.finetuningtask.FineTuningTaskService;
import cn.iocoder.yudao.module.llm.service.modelassesstaskauto.ModelAssessTaskAutoService;
import cn.iocoder.yudao.module.llm.service.modelassesstaskmanual.ModelAssessTaskManualService;
import cn.iocoder.yudao.module.llm.service.modelassesstaskmanualbackup.ModelAssessTaskManualBackupService;
import cn.iocoder.yudao.module.llm.service.modelservice.ModelServiceService;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
@ -52,8 +55,6 @@ import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.*;
@Validated
@Slf4j
public class BaseModelServiceImpl implements BaseModelService {
@Resource
private ServerNameMapper serverNameMapper;
@Resource
private BaseModelMapper baseModelMapper;
@ -76,6 +77,9 @@ public class BaseModelServiceImpl implements BaseModelService {
@Resource
private LLMBackendProperties llmBackendProperties;
@Resource
private ModelServiceMapper modelServiceMapper;
@Override
public Long createBaseModel (BaseModelSaveReqVO createReqVO) {
validateBaseModelNameExists(createReqVO);
@ -230,6 +234,13 @@ public class BaseModelServiceImpl implements BaseModelService {
@Override
public void deleteTheModel(Long id) {
LambdaQueryWrapper<ModelServiceDO> wrapper = new LambdaQueryWrapper<>();
wrapper.eq(ModelServiceDO::getFineTuningTask, id);
List<ModelServiceDO> modelServiceDOS = modelServiceMapper.selectList(wrapper);
if (modelServiceDOS.size()>0){
ModelServiceDO modelServiceDO = modelServiceDOS.get(0);
throw exception(new ErrorCode(10003_001, "该基座模型已生成模型服务，请删除"+modelServiceDO.getServiceName()+"模型服务后重试"));
}
try {
BaseModelDO pageReqVO = baseModelMapper.selectById(id);
// String resStr = HttpUtils.del(llmBackendProperties.getDeleteTheModel()+"/"+pageReqVO.getModelName(), null);