diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java index dccd3fdfe..882fb9bdf 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java @@ -312,22 +312,26 @@ public class DatasetServiceImpl implements DatasetService { Long fileId = datasetFilesDO.getId(); dataSetVos.forEach(dataSetVO -> { + boolean answerBlank = StringUtils.isNotBlank(dataSetVO.getAnswer()); // 保存到 数据集数据问题 DatasetQuestionDO datasetQuestionDO = new DatasetQuestionDO();// 检查是否为空行 datasetQuestionDO.setDatasetId(datasetId); datasetQuestionDO.setDatasetFilesId(fileId); datasetQuestionDO.setSystem(dataSetVO.getSystem()); datasetQuestionDO.setQuestion(dataSetVO.getQuestion()); - datasetQuestionDO.setStatus(StringUtils.isNotBlank(dataSetVO.getAnswer()) ? 2 : 0); + datasetQuestionDO.setStatus(answerBlank ? 2 : 0); datasetQuestionMapper.insert(datasetQuestionDO); // 保存到 数据集数据问题标注 - DatasetAnswerDO datasetAnswerDO = new DatasetAnswerDO(); - datasetAnswerDO.setDatasetId(datasetId); - datasetAnswerDO.setDatasetFilesId(fileId); - datasetAnswerDO.setQuestionId(datasetQuestionDO.getId()); - datasetAnswerDO.setAnswer(dataSetVO.getAnswer()); - datasetAnswerMapper.insert(datasetAnswerDO); + if (answerBlank){ + DatasetAnswerDO datasetAnswerDO = new DatasetAnswerDO(); + datasetAnswerDO.setDatasetId(datasetId); + datasetAnswerDO.setDatasetFilesId(fileId); + datasetAnswerDO.setQuestionId(datasetQuestionDO.getId()); + datasetAnswerDO.setAnswer(dataSetVO.getAnswer()); + datasetAnswerMapper.insert(datasetAnswerDO); + } + }); }