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<DataRefluxDataDO> 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> datasetDO = datasetMapper.selectList(new LambdaQueryWrapper<DatasetDO>().eq(DatasetDO::getDatasetName, dataRespVO.getDatasetName()));
         if (!CollectionUtils.isAnyEmpty(datasetDO)){