From d43c483cbdf610f4e1f47a9bbd755175d2294511 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Tue, 7 Jan 2025 16:32:02 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E9=9B=86=E7=AE=A1=E7=90=86?= =?UTF-8?q?=E5=88=A0=E9=99=A4=E6=95=B0=E6=8D=AE=E6=A0=87=E6=B3=A8=E5=A4=9A?= =?UTF-8?q?=E4=BD=99=E6=95=B0=E6=8D=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../llm/service/dataset/DatasetQuestionServiceImpl.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 collect1 = updateReqVOS.stream().map(DatasetQuestionSaveReqVO::getId).collect(Collectors.toList()); LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); - queryWrapper.eq(datasetId!= null,DatasetAnswerDO::getDatasetId, datasetId); + queryWrapper.in(collect1!= null,DatasetAnswerDO::getQuestionId, collect1); List datasetAnswerDOS = datasetAnswerMapper.selectList(queryWrapper); -// List collect = datasetAnswerDOS.stream().map(DatasetAnswerDO::getId).collect(Collectors.toList()); - List collect = updateReqVOS.stream().map(DatasetQuestionSaveReqVO::getId).collect(Collectors.toList()); + List collect = datasetAnswerDOS.stream().map(DatasetAnswerDO::getId).collect(Collectors.toList()); List diff1 = new ArrayList<>(); if (CollectionUtils.isNotEmpty(collect)){ HashSet set1 = new HashSet<>(ids);