From 29131f4504c287a188708df294dc541d6f50dfd1 Mon Sep 17 00:00:00 2001
From: sunxiqing <2240398334@qq.com>
Date: Tue, 7 Jan 2025 16:13:19 +0800
Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=9B=9E=E6=B5=81=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/datarefluxdata/DataRefluxDataServiceImpl.java    | 5 +++++
 1 file changed, 5 insertions(+)

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();