From 6e8698c5fcfef60b51bcdecb3afb8582915e7884 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Tue, 7 Jan 2025 15:20:34 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=9B=9E=E6=B5=81=E4=BF=9D?= =?UTF-8?q?=E5=AD=98=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../datarefluxdata/DataRefluxDataServiceImpl.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java index 23efd7243..6efbaff3b 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java @@ -149,8 +149,13 @@ public class DataRefluxDataServiceImpl implements DataRefluxDataService { DatasetDO dataRespVO = BeanUtils.toBean(pageReqVO, DatasetDO.class); dataRespVO.setDataLength((long) pageReqVO.getDataids().size()); dataRespVO.setDatasetCategory(Math.toIntExact(pageReqVO.getDatasetType())); - dataRespVO.setAnnotateProgress(100); - dataRespVO.setStatus(2); + if (pageReqVO.getSaveField()==0){ + dataRespVO.setAnnotateProgress(0); + dataRespVO.setStatus(0); + }else if (pageReqVO.getSaveField()==1){ + dataRespVO.setAnnotateProgress(100); + dataRespVO.setStatus(2); + } dataRespVO.setId(null); List datasetDO = datasetMapper.selectList(new LambdaQueryWrapper().eq(DatasetDO::getDatasetName, dataRespVO.getDatasetName())); if (!CollectionUtils.isAnyEmpty(datasetDO)){