DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(yudao-module-llm): 优化异步微调任务服务

Browse Source 
- 异常处理机制改进，直接抛出异常而不是打印堆栈
This commit is contained in:
Liuyang 2025-02-26 14:00:09 +08:00
parent 58215b015e
commit 13e30d4249
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
View File

@ -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);