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

修复数据集标注问题

Browse Source
This commit is contained in:
limin 2025-01-07 15:40:53 +08:00
parent 6e8698c5fc
commit e381b5a5ae
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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> datasetAnswerSaveReqVO = updateReqVO.getDatasetAnswerRespVO();
List<DatasetAnswerDO> 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<DatasetAnswerDO> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(datasetId!= null,DatasetAnswerDO::getDatasetId, datasetId);