diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetQuestionServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetQuestionServiceImpl.java
index 27330a2fd..480da758d 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetQuestionServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetQuestionServiceImpl.java
@@ -136,11 +136,11 @@ public class DatasetQuestionServiceImpl implements DatasetQuestionService {
             }
             datasetQuestionMapper.updateById(datasetQuestionDO);
         }
+        List<Long> collect1 = updateReqVOS.stream().map(DatasetQuestionSaveReqVO::getId).collect(Collectors.toList());
         LambdaQueryWrapper<DatasetAnswerDO> queryWrapper = new LambdaQueryWrapper<>();
-        queryWrapper.eq(datasetId!= null,DatasetAnswerDO::getDatasetId, datasetId);
+        queryWrapper.in(collect1!= null,DatasetAnswerDO::getQuestionId, collect1);
         List<DatasetAnswerDO> datasetAnswerDOS = datasetAnswerMapper.selectList(queryWrapper);
-//        List<Long> collect = datasetAnswerDOS.stream().map(DatasetAnswerDO::getId).collect(Collectors.toList());
-        List<Long> collect = updateReqVOS.stream().map(DatasetQuestionSaveReqVO::getId).collect(Collectors.toList());
+        List<Long> collect = datasetAnswerDOS.stream().map(DatasetAnswerDO::getId).collect(Collectors.toList());
         List<Long> diff1 = new ArrayList<>();
         if (CollectionUtils.isNotEmpty(collect)){
             HashSet<Long> set1 = new HashSet<>(ids);