From e381b5a5ae597867c9b02528d8393bcaca8b887a Mon Sep 17 00:00:00 2001 From: limin Date: Tue, 7 Jan 2025 15:40:53 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=95=B0=E6=8D=AE=E9=9B=86?= =?UTF-8?q?=E6=A0=87=E6=B3=A8=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/dataset/DatasetQuestionServiceImpl.java | 10 ++++++---- 1 file changed, 6 insertions(+), 4 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 499de9293..4690eeb2d 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 @@ -8,6 +8,7 @@ import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetMapper; import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetQuestionMapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; +import jodd.util.StringUtil; import org.springframework.stereotype.Service; import javax.annotation.Resource; import org.springframework.validation.annotation.Validated; @@ -115,14 +116,14 @@ public class DatasetQuestionServiceImpl implements DatasetQuestionService { datasetId = updateReqVO.getDatasetId(); } DatasetQuestionDO datasetQuestionDO = BeanUtils.toBean(updateReqVO, DatasetQuestionDO.class); - if (CollectionUtils.isNotEmpty(updateReqVO.getDatasetAnswerRespVO())){ - datasetQuestionDO.setStatus(2); - } - datasetQuestionMapper.updateById(datasetQuestionDO); + List datasetAnswerSaveReqVO = updateReqVO.getDatasetAnswerRespVO(); List datasetAnswerDOS = BeanUtils.toBean(datasetAnswerSaveReqVO, DatasetAnswerDO.class); if (CollectionUtils.isNotEmpty(datasetAnswerDOS)){ for (DatasetAnswerDO datasetAnswerDO : datasetAnswerDOS){ + if (StringUtil.isNotBlank(datasetAnswerDO.getAnswer())){ + datasetQuestionDO.setStatus(2); + } if (datasetAnswerDO.getId() == null){ datasetAnswerMapper.insert(datasetAnswerDO); ids.add(datasetAnswerDO.getId()); @@ -133,6 +134,7 @@ public class DatasetQuestionServiceImpl implements DatasetQuestionService { } // datasetAnswerMapper.insertOrUpdate(datasetAnswerDOS); } + datasetQuestionMapper.updateById(datasetQuestionDO); } LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); queryWrapper.eq(datasetId!= null,DatasetAnswerDO::getDatasetId, datasetId);