diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java index afab18209..ebc6211e8 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java @@ -159,5 +159,9 @@ public class FineTuningTaskDO extends BaseDO { //检查点数据内容 private String checkPointData; + /** + * 部署次数 + */ + private Integer deployCount; } 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 926a4f30f..3b5fa122c 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 @@ -27,6 +27,7 @@ import org.springframework.stereotype.Service; import javax.annotation.Resource; import java.util.HashMap; import java.util.List; +import java.util.Optional; @Service @Slf4j @@ -110,9 +111,18 @@ public class AsyncFineTuningTaskService { req.setDataset(fileUrl); log.info("JSON 文件生成成功。文件 URL: {}", fileUrl); - // 设置请求参数 - req.setSuffix(active + "-" + fineTuningTask.getId()); - log.debug("请求参数设置完成。后缀: {}", req.getSuffix()); + // 设置部署次数 + int newDeployCount = Optional.ofNullable(fineTuningTask.getDeployCount()) + .orElse(0) + 1; + fineTuningTask.setDeployCount(newDeployCount); + + // 设置后缀 + req.setSuffix(active + "-" + fineTuningTask.getId() + "-" + newDeployCount); + fineTuningTask.setDeployCount(newDeployCount); + + // 设置后缀 + req.setSuffix(active + "-" + fineTuningTask.getId() + "-" + newDeployCount); + log.info("请求参数设置完成。后缀: {}", req.getSuffix()); // 调用模型服务创建微调任务 log.info("正在调用模型服务创建微调任务..."); @@ -121,6 +131,7 @@ public class AsyncFineTuningTaskService { // 更新任务状态 FineTuningTaskDO updateObj = new FineTuningTaskDO(); updateObj.setId(fineTuningTask.getId()); + updateObj.setDeployCount(newDeployCount); if (resp != null && resp.getId() != 0) { updateObj.setJobId(resp.getJobId()); updateObj.setStatus(FinetuningTaskStatusEnum.WAITING.getStatus());