diff --git a/ui/src/views/dataset/UploadDocumentDataset.vue b/ui/src/views/dataset/UploadDocumentDataset.vue
index 6c13325c..498199fd 100644
--- a/ui/src/views/dataset/UploadDocumentDataset.vue
+++ b/ui/src/views/dataset/UploadDocumentDataset.vue
@@ -22,12 +22,17 @@
       </div>
     </div>
     <div class="create-dataset__footer text-right border-t" v-if="active !== 2">
-      <el-button @click="router.go(-1)" :disabled="loading">取消</el-button>
-      <el-button @click="prev" v-if="active === 1" :disabled="loading">上一步</el-button>
-      <el-button @click="next" type="primary" v-if="active === 0" :disabled="loading">
+      <el-button @click="router.go(-1)" :disabled="SetRulesRef?.loading || loading">取消</el-button>
+      <el-button @click="prev" v-if="active === 1" :disabled="SetRulesRef?.loading || loading">上一步</el-button>
+      <el-button @click="next" type="primary" v-if="active === 0" :disabled="SetRulesRef?.loading || loading">
         {{ documentsType === 'txt' ? '下一步' : '开始导入' }}
       </el-button>
-      <el-button @click="submit" type="primary" v-if="active === 1" :disabled="loading">
+      <el-button
+        @click="submit"
+        type="primary"
+        v-if="active === 1"
+        :disabled="SetRulesRef?.loading || loading"
+      >
         开始导入
       </el-button>
     </div>
diff --git a/ui/src/views/dataset/component/SetRules.vue b/ui/src/views/dataset/component/SetRules.vue
index 06654961..b523aa75 100644
--- a/ui/src/views/dataset/component/SetRules.vue
+++ b/ui/src/views/dataset/component/SetRules.vue
@@ -209,7 +209,8 @@ onMounted(() => {
 
 defineExpose({
   paragraphList,
-  checkedConnect
+  checkedConnect,
+  loading
 })
 </script>
 <style scoped lang="scss">