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 f859c34af..c4c0ce264 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 @@ -313,6 +313,11 @@ public class BaseModelTaskService { // localModel.setApiUrl(string1); 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); log.info("模型 {} 状态为 running,无需更新", pedestalModelVo.getDeploymentName()); } }