feat(llm): 增加微调任务部署次数记录
- 在 FineTuningTaskDO 中添加 deployCount 字段,用于记录部署次数 - 在创建微调任务时,增加部署次数并更新任务信息
This commit is contained in:
parent
c0f597f81c
commit
91a3e03143
@ -159,5 +159,9 @@ public class FineTuningTaskDO extends BaseDO {
|
||||
//检查点数据内容
|
||||
private String checkPointData;
|
||||
|
||||
/**
|
||||
* 部署次数
|
||||
*/
|
||||
private Integer deployCount;
|
||||
|
||||
}
|
||||
|
@ -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());
|
||||
|
Loading…
x
Reference in New Issue
Block a user