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 ed3f229ea..bb1e26fb4 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 @@ -21,7 +21,7 @@ import lombok.extern.slf4j.Slf4j; import org.springframework.context.annotation.Lazy; import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Service; - +import org.springframework.beans.factory.annotation.Value; import javax.annotation.Resource; import java.util.HashMap; import java.util.List; @@ -49,6 +49,9 @@ public class AsyncFineTuningTaskService { @Resource private AsyncDataSetService dataSetService; + @Value("${spring.profiles.active}") + private String active; + //大模型平台创建调优任务 @Async public void createTuning(FineTuningTaskDO fineTuningTask) { @@ -69,7 +72,7 @@ public class AsyncFineTuningTaskService { String fileUrl = dataSetService.JsonFileWriteFine(serverNameDO.getHost(), datasetDO, datasetQuestionList); req.setDataset(StringUtils.isNotBlank(dataset.getDatasetFileUrl())?dataset.getDatasetFileUrl():"storage/json/2025/01/_xVKpfDH8.json"); req.setDataset(fileUrl); - req.setSuffix("T-"+String.valueOf(fineTuningTask.getId())); + req.setSuffix(active+"-"+String.valueOf(fineTuningTask.getId())); FineTuningTaskDO updateObj = new FineTuningTaskDO(); if (serverNameDO == null){ updateObj.setStatus(FinetuningTaskStatusEnum.CANCELLED.getStatus());