From 435bbee4dceee671cf18530121966a203278c141 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Tue, 12 Aug 2025 09:52:50 +0800 Subject: [PATCH] =?UTF-8?q?refactor(llm):=20=E4=BC=98=E5=8C=96=20FineTunin?= =?UTF-8?q?gTaskSyncService=20=E4=B8=AD=E5=A4=84=E7=90=86=E6=A3=80?= =?UTF-8?q?=E6=9F=A5=E7=82=B9=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在处理已成功但未获取到检查点的错误时,增加了对检查点文件路径的判断 - 仅当检查点文件路径为空时,才执行获取检查点的操作 - 优化了代码结构,提高了可读性和性能 --- .../FineTuningTaskSyncService.java | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java index 0bb593c60..035364e66 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java @@ -176,15 +176,17 @@ public class FineTuningTaskSyncService { fineTuningTaskMapper.updateById(updateObj); } else { // 处理已成功但是没有获取到检查点错误 - String jobModelName = fineTuningTaskDO.getJobModelName(); - String hostUrl = getHostUrl(fineTuningTaskDO); - FineTuningTaskDO updateObj = new FineTuningTaskDO(); - updateObj.setId(fineTuningTaskDO.getId()); - if (StringUtils.isBlank(hostUrl)){ - continue; + if (ObjectUtil.isAllEmpty(fineTuningTaskDO.getCheckPointFilePath()) && ObjectUtil.isAllEmpty(fineTuningTaskDO.getCheckPointFilePath())) { + String jobModelName = fineTuningTaskDO.getJobModelName(); + String hostUrl = getHostUrl(fineTuningTaskDO); + FineTuningTaskDO updateObj = new FineTuningTaskDO(); + updateObj.setId(fineTuningTaskDO.getId()); + if (StringUtils.isBlank(hostUrl)) { + continue; + } + getCheckPoint(fineTuningTaskDO, jobModelName, hostUrl, updateObj); + fineTuningTaskMapper.updateById(updateObj); } - getCheckPoint(fineTuningTaskDO, jobModelName, hostUrl, updateObj); - fineTuningTaskMapper.updateById(updateObj); } }