From 13e30d42490c4ed5e9ba564f4d68fca41994127a Mon Sep 17 00:00:00 2001
From: Liuyang <2746366019@qq.com>
Date: Wed, 26 Feb 2025 14:00:09 +0800
Subject: [PATCH] =?UTF-8?q?refactor(yudao-module-llm):=20=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E5=BC=82=E6=AD=A5=E5=BE=AE=E8=B0=83=E4=BB=BB=E5=8A=A1?=
 =?UTF-8?q?=E6=9C=8D=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 异常处理机制改进，直接抛出异常而不是打印堆栈
---
 .../llm/service/async/AsyncFineTuningTaskService.java     | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
index f620cc157..b9f14ad23 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
@@ -133,7 +133,7 @@ public class AsyncFineTuningTaskService {
                 updateObj.setStatus(FinetuningTaskStatusEnum.WAITING.getStatus());
                 updateObj.setJobModelName(resp.getFineTunedModel());
                 updateObj.setTrainLog(resp.getTrainLog());
-                log.info("微调任务创建成功。任务ID: {}, 任务模型名称: {} , 任务状态: {}", fineTuningTask.getId(), fineTuningTask.getJobModelName(),FinetuningTaskStatusEnum.WAITING.getStatus());
+                log.info("微调任务创建成功。任务ID: {}, 任务模型名称: {} , 任务状态: {}", fineTuningTask.getId(), fineTuningTask.getJobModelName(), FinetuningTaskStatusEnum.WAITING.getStatus());
             } else {
                 updateObj.setStatus(FinetuningTaskStatusEnum.FAILED.getStatus());
                 log.error("微调任务创建失败。任务ID: {}", fineTuningTask.getId());
@@ -146,7 +146,7 @@ public class AsyncFineTuningTaskService {
 
         } catch (Exception e) {
             log.error("创建微调任务时发生异常。任务ID: {}", fineTuningTask.getId(), e);
-            e.printStackTrace();
+            throw e;
         }
     }
 
@@ -178,7 +178,7 @@ public class AsyncFineTuningTaskService {
         req.setModel(fineTuningTask.getBaseModel());
         req.setTrainEpoch((fineTuningTask.getEpoch() == 0) ? 1 : fineTuningTask.getEpoch());
         int saveStep = Optional.ofNullable(fineTuningTask.getSaveSteps())
-                                     .orElse(0) + 1;
+                               .orElse(0) + 1;
         req.setTrainEpoch(saveStep);
         req.setSuffix(fineTuningTask.getTaskName());
         req.setRemark(fineTuningTask.getTaskIntro());
@@ -196,7 +196,7 @@ public class AsyncFineTuningTaskService {
     @Async
     public void startFineTuningTask (FineTuningTaskDO fineTuningTask) {
         log.info(" ===== 开始微调任务 ===== stopFinetuning");
-        String modelName = fineTuningTask.getJobModelName()+"-"+fineTuningTask.getDeployCount()+1;
+        String modelName = fineTuningTask.getJobModelName() + "-" + fineTuningTask.getDeployCount() + 1;
         try {
             log.info("开始调用HTTP服务开始微调任务，任务模型名称: {}", modelName);
             createTuning(fineTuningTask);