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);