From 93760fe68cc1767dc619745140012fe212416191 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Fri, 3 Jan 2025 15:51:52 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=95=B0=E6=8D=AE=E5=9B=9E?= =?UTF-8?q?=E6=B5=81bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../llm/service/datarefluxdata/DataRefluxDataServiceImpl.java | 2 ++ 1 file changed, 2 insertions(+) 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 f2d8b7988..063396bfe 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 @@ -146,6 +146,8 @@ public class DataRefluxDataServiceImpl implements DataRefluxDataService { wrapper.in(pageReqVO.getDataids().size()>0,DataRefluxDataDO::getId, pageReqVO.getDataids()); List dataRefluxDataDOS = dataRefluxDataMapper.selectList(wrapper); DatasetDO dataRespVO = BeanUtils.toBean(pageReqVO, DatasetDO.class); + dataRespVO.setDataLength((long) pageReqVO.getDataids().size()); + dataRespVO.setDatasetCategory(Math.toIntExact(pageReqVO.getDatasetType())); dataRespVO.setId(null); List datasetDO = datasetMapper.selectList(new LambdaQueryWrapper().eq(DatasetDO::getDatasetName, dataRespVO.getDatasetName())); if (!CollectionUtils.isAnyEmpty(datasetDO)){