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

fix(llm): 修复模型训练任务完成后的状态更新问题

Browse Source 
- 在更新模型信息时增加空值检查，避免空指针异常- 如果基础模型信息不存在，将本地模型状态标记为失败并更新
- 优化了日志输出，仅在模型状态为 running时打印相关信息
This commit is contained in:
sunxiqing 2025-04-21 18:45:47 +08:00
parent 062bed0584
commit 2a3bc9c0a9
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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());
}
}