From 2a3bc9c0a9d5dfa551bcb5285a1e28a4aee5a039 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Mon, 21 Apr 2025 18:45:47 +0800 Subject: [PATCH] =?UTF-8?q?fix(llm):=20=E4=BF=AE=E5=A4=8D=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E8=AE=AD=E7=BB=83=E4=BB=BB=E5=8A=A1=E5=AE=8C=E6=88=90?= =?UTF-8?q?=E5=90=8E=E7=9A=84=E7=8A=B6=E6=80=81=E6=9B=B4=E6=96=B0=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在更新模型信息时增加空值检查，避免空指针异常- 如果基础模型信息不存在，将本地模型状态标记为失败并更新 - 优化了日志输出，仅在模型状态为 running时打印相关信息 --- .../llm/service/basemodel/BaseModelTaskService.java | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) 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()); } }