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

dataset

Browse Source
This commit is contained in:
ire 2025-01-24 14:36:40 +08:00
parent 1880e5aa4a
commit 8b44edea3d
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetRespVO.java
View File

@ -66,4 +66,6 @@ public class DatasetRespVO {
* 模型工厂文件id
*/
private String jobId;
private String fileUrl;
}

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

@ -61,9 +61,9 @@ public class AsyncDataSetService {
datasetMapper.setJobid(datasetDO.getId(),aigcDatasetFileRespV0.getFileId());
String s3Url = aigcDatasetFileRespV0.getS3Url();
int lastIndex = s3Url.lastIndexOf("/");
int lastIndex = s3Url.lastIndexOf("/storage");
String url = s3Url.substring(lastIndex + 1);
datasetMapper.setUrl(datasetDO.getId(),"storage/" + url);
datasetMapper.setUrl(datasetDO.getId(),"storage" + url);
log.info("[JsonFileWrite][写入文件成功]");
}

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

@ -38,7 +38,7 @@ public class AsyncFineTuningTaskService {
Long datasetId = fineTuningTask.getDataset();
DatasetRespVO dataset = datasetService.getDataset(datasetId);
// req.setFileId(dataset.getJobId());
req.setDataset(dataset.getDatasetFileUrl());
req.setDataset(dataset.getFileUrl());
req.setSuffix(String.valueOf(fineTuningTask.getId()));
AigcFineTuningCreateRespVO resp = trainHttpService.finetuningCreate(new HashMap<>(), req);
FineTuningTaskDO updateObj = new FineTuningTaskDO();