From bd152b3f6045543a21144202dfdfff25b22ba199 Mon Sep 17 00:00:00 2001 From: limin Date: Tue, 11 Feb 2025 11:25:58 +0800 Subject: [PATCH] =?UTF-8?q?=E5=BE=AE=E8=B0=83=E8=AE=AD=E7=BB=83=E5=AE=8C?= =?UTF-8?q?=E6=88=90=20=E6=97=B6=E9=95=BF=E8=8E=B7=E5=8F=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../finetuningtask/FineTuningTaskSyncService.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) 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 83f58d734..617189e3e 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 @@ -22,6 +22,7 @@ import java.io.InputStreamReader; import java.net.MalformedURLException; import java.net.URL; import java.net.URLConnection; +import java.time.Duration; import java.util.*; @Component @@ -85,6 +86,12 @@ public class FineTuningTaskSyncService { String resModels = trainHttpService.modelTableQuery(new HashMap<>(), "models",querModels); log.info("获取 aigc models 表数据 info {}",resModels); JSONArray jsonArrayModels = JSONArray.parseArray(resModels); + + // 时长获取 + Duration duration = Duration.between(fineTuningTaskDO.getUpdateTime(), fineTuningTaskDO.getCreateTime()); + long minutes = duration.toMinutes(); + updateObj.setTrainDuration(String.valueOf(minutes)); + if(jsonArrayModels.size() > 0){ JSONObject jsonObjectModels = jsonArrayModels.getJSONObject(0); updateObj.setJobModelListId(jsonObjectModels.getLong("id")); @@ -139,12 +146,13 @@ public class FineTuningTaskSyncService { }catch (Exception e){ log.error(" error {}",e.getMessage()); } + fineTuningTaskMapper.updateById(updateObj); } else { updateObj.setStatus(status); - fineTuningTaskMapper.updateById(updateObj); } + fineTuningTaskMapper.updateById(updateObj); }