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 62bf7f1ed..68076a046 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 @@ -38,6 +38,7 @@ public class AsyncFineTuningTaskService { Long datasetId = fineTuningTask.getDataset(); DatasetRespVO dataset = datasetService.getDataset(datasetId); req.setFileId(dataset.getJobId()); + req.setSuffix(String.valueOf(fineTuningTask.getId())); AigcFineTuningCreateRespVO resp = trainHttpService.finetuningCreate(new HashMap<>(), req); FineTuningTaskDO updateObj = new FineTuningTaskDO(); updateObj.setId(fineTuningTask.getId()); diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetTaskSyncService.java index e8df69f26..57b08bebc 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetTaskSyncService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetTaskSyncService.java @@ -23,7 +23,7 @@ public class DatasetTaskSyncService { @Resource private AsyncDataSetService dataSetService; -// @Scheduled(cron ="0/15 * * * * ?") + @Scheduled(cron ="0/15 * * * * ?") public void syncDatasetAigcTask() { log.info("[syncDatasetAigcTask][开始同步任务]"); // 查询已经标注完成的数据集 diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java index a393aecd2..64a586fdb 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java @@ -83,11 +83,13 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { public void deleteFineTuningTask(Long id) { // 校验存在 validateFineTuningTaskExists(id); + FineTuningTaskDO fineTuningTaskDO = fineTuningTaskMapper.selectById(id); // 删除 LambdaUpdateWrapper wrapper = new LambdaUpdateWrapper<>(); wrapper.eq(FineTuningTaskDO::getId, id) .set(FineTuningTaskDO::getDeleted,true); fineTuningTaskMapper.update(null,wrapper); + asyncFineTuningTaskService.stopFineTuningTask(fineTuningTaskDO); // fineTuningTaskMapper.deleteById(id); }