DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Csv 解析 是否保存问题标注

Browse Source
This commit is contained in:
Liuyang 2025-01-13 17:02:14 +08:00
parent 9839839037
commit 494f078c29
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java
View File

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