From f1c7f0f3af88fd04b8f473d55e5984e6e122e2a0 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Mon, 18 Aug 2025 14:04:21 +0800
Subject: [PATCH] feat: add allow_download option to document form

---
 ui/src/locales/lang/en-US/views/document.ts              | 3 +++
 ui/src/locales/lang/zh-CN/views/document.ts              | 3 +++
 ui/src/locales/lang/zh-Hant/views/document.ts            | 3 +++
 ui/src/views/document/component/ImportDocumentDialog.vue | 9 +++++++++
 4 files changed, 18 insertions(+)

diff --git a/ui/src/locales/lang/en-US/views/document.ts b/ui/src/locales/lang/en-US/views/document.ts
index eeb779ca0..e6dcd2f12 100644
--- a/ui/src/locales/lang/en-US/views/document.ts
+++ b/ui/src/locales/lang/en-US/views/document.ts
@@ -158,6 +158,9 @@ export default {
       placeholder: 'Directly return segment content',
       requiredMessage: 'Please enter similarity value',
     },
+    allow_download: {
+     label: 'Allow download in knowledge base source',
+    }
   },
   hitHandlingMethod: {
     optimization: 'Model optimization',
diff --git a/ui/src/locales/lang/zh-CN/views/document.ts b/ui/src/locales/lang/zh-CN/views/document.ts
index aaf6c9b30..482c7e013 100644
--- a/ui/src/locales/lang/zh-CN/views/document.ts
+++ b/ui/src/locales/lang/zh-CN/views/document.ts
@@ -151,6 +151,9 @@ export default {
       placeholder: '直接返回分段内容',
       requiredMessage: '请输入相似度',
     },
+    allow_download: {
+      label: '允许在知识库来源中下载',
+    }
   },
   hitHandlingMethod: {
     optimization: '模型优化',
diff --git a/ui/src/locales/lang/zh-Hant/views/document.ts b/ui/src/locales/lang/zh-Hant/views/document.ts
index 1be1d0176..e0ec72d3e 100644
--- a/ui/src/locales/lang/zh-Hant/views/document.ts
+++ b/ui/src/locales/lang/zh-Hant/views/document.ts
@@ -154,6 +154,9 @@ export default {
       placeholder: '直接返回分段内容',
       requiredMessage: '请输入相似度',
     },
+    allow_download: {
+     label: '允許在知識庫來源下載',
+    }
   },
   hitHandlingMethod: {
     optimization: '模型優化',
diff --git a/ui/src/views/document/component/ImportDocumentDialog.vue b/ui/src/views/document/component/ImportDocumentDialog.vue
index 22b0b824d..c49e4d675 100644
--- a/ui/src/views/document/component/ImportDocumentDialog.vue
+++ b/ui/src/views/document/component/ImportDocumentDialog.vue
@@ -80,6 +80,11 @@
           /><span>{{ $t('views.document.form.similarity.placeholder') }}</span>
         </div>
       </el-form-item>
+      <el-form-item prop="allow_download">
+        <el-checkbox v-model="form.allow_download">
+          {{ $t('views.document.form.allow_download.label') }}
+        </el-checkbox>
+      </el-form-item>
     </el-form>
     <template #footer>
       <span class="dialog-footer">
@@ -127,6 +132,7 @@ const form = ref<any>({
   selector: '',
   hit_handling_method: 'optimization',
   directly_return_similarity: 0.9,
+  allow_download: false,
 })
 
 // 文档设置
@@ -162,6 +168,7 @@ watch(dialogVisible, (bool) => {
       selector: '',
       hit_handling_method: 'optimization',
       directly_return_similarity: 0.9,
+      allow_download: false,
     }
     isImport.value = false
     documentType.value = ''
@@ -200,6 +207,7 @@ const submit = async (formEl: FormInstance | undefined) => {
         const obj = {
           source_url_list: form.value.source_url.split('\n'),
           selector: form.value.selector,
+          allow_download: form.value.allow_download,
         }
         loadSharedApi({ type: 'document', systemType: apiType.value })
           .postWebDocument(id, obj, loading)
@@ -219,6 +227,7 @@ const submit = async (formEl: FormInstance | undefined) => {
               ...{
                 source_url: form.value.source_url,
                 selector: form.value.selector,
+                allow_download: form.value.allow_download,
               },
             },
           }