diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java index 5225eabce..a3cdf6cdc 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelTaskService.java @@ -330,10 +330,15 @@ public class BaseModelTaskService { localModel.setJobId((long) pedestalModelVo.getId()); modelServiceMapper.updateById(localModel); BaseModelDO baseModelDO = baseModelMapper.selectById(localModel.getFineTuningTask()); - baseModelDO.setModelId((long) pedestalModelVo.getId()); - baseModelDO.setChatUrl(string1); - baseModelDO.setApiUrl(string1); - baseModelMapper.updateById(baseModelDO); + if (baseModelDO != null) { + baseModelDO.setModelId((long) pedestalModelVo.getId()); + baseModelDO.setChatUrl(string1); + baseModelDO.setApiUrl(string1); + baseModelMapper.updateById(baseModelDO); + }else { + localModel.setStatus(3); + modelServiceMapper.updateById(localModel); + } log.info("模型 {} 状态为 running,无需更新", pedestalModelVo.getDeploymentName()); } }