DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

修复

Browse Source
This commit is contained in:
limin 2025-02-21 10:29:49 +08:00
parent b16c1c9957
commit 93e4f44bf5
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
View File

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