diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetMapper.java index b49a4ffa3..46caa6e36 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetMapper.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetMapper.java @@ -34,8 +34,8 @@ public interface DatasetMapper extends BaseMapperX { .orderByDesc(DatasetDO::getId)); } - @Update("update llm_dataset set annotate_progress = #{formattedRatio} where id = #{datasetId}") - void updateProcess(@Param("formattedRatio") Integer formattedRatio, @Param("datasetId") Long datasetId); + @Update("update llm_dataset set annotate_progress = #{formattedRatio},status=#{status} where id = #{datasetId}") + void updateProcess(@Param("formattedRatio") Integer formattedRatio, @Param("datasetId") Long datasetId,@Param("status") Integer status); @Delete("delete from llm_dataset where id = #{datasetPostId}") void deleteTrue(@Param("datasetPostId") Long datasetPostId); } 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 5e8bb246f..c705a207a 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 @@ -110,6 +110,9 @@ public class DatasetQuestionServiceImpl implements DatasetQuestionService { public void updateDatasetQuestionDataAnno(List updateReqVOS) { updateReqVOS.forEach(updateReqVO -> { 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); @@ -125,7 +128,8 @@ public class DatasetQuestionServiceImpl implements DatasetQuestionService { Long annoCount = datasetQuestionMapper.selectCount(wrapper); double ratio = sumCount == 0 ? 0 : ((double) annoCount / sumCount) *100; Integer formattedRatio = ratio == 0 ? 0 : (int) ratio; - datasetMapper.updateProcess(formattedRatio, updateReqVOS.get(0).getDatasetId()); + Integer status = formattedRatio == 100 ? 2 : 1; + datasetMapper.updateProcess(formattedRatio, updateReqVOS.get(0).getDatasetId(),status); } } \ No newline at end of file