diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java index 6efbaff3b..beedb3e40 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxdata/DataRefluxDataServiceImpl.java @@ -167,6 +167,11 @@ public class DataRefluxDataServiceImpl implements DataRefluxDataService { datasetQuestionDO.setDatasetId(dataRespVO.getId()); datasetQuestionDO.setQuestion(dataRefluxDataDO.getPrompt()); datasetQuestionDO.setSystem(dataRefluxDataDO.getSystem()); + if (pageReqVO.getSaveField()==0){ + datasetQuestionDO.setStatus(0); + }else if (pageReqVO.getSaveField()==1){ + datasetQuestionDO.setStatus(2); + } datasetQuestionMapper.insert(datasetQuestionDO); if (pageReqVO.getSaveField()==1) { DatasetAnswerDO datasetAnswerDO = new DatasetAnswerDO();