feat(llm): 增加微调任务部署次数记录

- 在 FineTuningTaskDO 中添加 deployCount 字段,用于记录部署次数
- 在创建微调任务时,增加部署次数并更新任务信息
This commit is contained in:
Liuyang 2025-02-26 10:01:57 +08:00
parent c0f597f81c
commit 91a3e03143
2 changed files with 18 additions and 3 deletions

View File

@ -159,5 +159,9 @@ public class FineTuningTaskDO extends BaseDO {
//检查点数据内容
private String checkPointData;
/**
* 部署次数
*/
private Integer deployCount;
}

View File

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