From 494f078c2977219c69ba4718d99e19fe80b36e47 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Mon, 13 Jan 2025 17:02:14 +0800 Subject: [PATCH] =?UTF-8?q?Csv=20=E8=A7=A3=E6=9E=90=20=E6=98=AF=E5=90=A6?= =?UTF-8?q?=E4=BF=9D=E5=AD=98=E9=97=AE=E9=A2=98=E6=A0=87=E6=B3=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/dataset/DatasetServiceImpl.java | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) 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); + } + }); }