diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetRespVO.java index 960b11f87..b8801e370 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/vo/DatasetRespVO.java @@ -66,4 +66,6 @@ public class DatasetRespVO { * 模型工厂文件id */ private String jobId; + + private String fileUrl; } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncDataSetService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncDataSetService.java index 6a224bc96..ca6c168bd 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncDataSetService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncDataSetService.java @@ -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][写入文件成功]"); } 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 777a1fcc7..892909dcf 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,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();