From 37c9ef30871feaf114b5696325141d5904c1ddbe Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Thu, 19 Jun 2025 18:39:52 +0800
Subject: [PATCH] fix: document

---
 ui/src/layout/components/breadcrumb/index.vue |  2 +-
 .../modules-resource-management/knowledge.ts  |  2 +-
 .../stores/modules-shared-system/knowledge.ts |  2 +-
 ui/src/stores/modules/knowledge.ts            | 84 +++++++++----------
 .../views/application/ApplicationSetting.vue  |  2 +-
 .../component/AddKnowledgeDialog.vue          | 12 ++-
 .../component/SelectKnowledgeDialog.vue       | 58 +++++++------
 .../component/SelectDocumentDialog.vue        |  2 +-
 .../component/SelectKnowledgeDialog.vue       |  2 +-
 .../component/SelectDocumentDialog.vue        |  2 +-
 .../component/SelectDocumentDialog.vue        |  2 +-
 .../nodes/search-knowledge-node/index.vue     |  2 +-
 12 files changed, 91 insertions(+), 81 deletions(-)

diff --git a/ui/src/layout/components/breadcrumb/index.vue b/ui/src/layout/components/breadcrumb/index.vue
index baec9483d..e1f8cbfe8 100644
--- a/ui/src/layout/components/breadcrumb/index.vue
+++ b/ui/src/layout/components/breadcrumb/index.vue
@@ -165,7 +165,7 @@ function changeMenu(id: string) {
 function getKnowledge() {
   loading.value = true
   knowledge
-    .asyncGetRootKnowledge()
+    .asyncGetFolderKnowledge()
     .then((res: any) => {
       list.value = res.data
       common.saveBreadcrumb(list.value)
diff --git a/ui/src/stores/modules-resource-management/knowledge.ts b/ui/src/stores/modules-resource-management/knowledge.ts
index 17825f919..26c630643 100644
--- a/ui/src/stores/modules-resource-management/knowledge.ts
+++ b/ui/src/stores/modules-resource-management/knowledge.ts
@@ -31,7 +31,7 @@ const useKnowledgeStore = defineStore('knowled', {
     saveDocumentsFile(file: UploadUserFile[]) {
       this.documentsFiles = file
     },
-    async asyncGetRootKnowledge(loading?: Ref<boolean>) {
+    async asyncGetFolderKnowledge(loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         const params = {
           folder_id: localStorage.getItem('workspace_id'),
diff --git a/ui/src/stores/modules-shared-system/knowledge.ts b/ui/src/stores/modules-shared-system/knowledge.ts
index 312ae52a7..0fac3120a 100644
--- a/ui/src/stores/modules-shared-system/knowledge.ts
+++ b/ui/src/stores/modules-shared-system/knowledge.ts
@@ -31,7 +31,7 @@ const useKnowledgeStore = defineStore('knowledg', {
     saveDocumentsFile(file: UploadUserFile[]) {
       this.documentsFiles = file
     },
-    async asyncGetRootKnowledge(loading?: Ref<boolean>) {
+    async asyncGetFolderKnowledge(loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         const params = {
           folder_id: 'default',
diff --git a/ui/src/stores/modules/knowledge.ts b/ui/src/stores/modules/knowledge.ts
index a5129eb3f..c9bfd1b4b 100644
--- a/ui/src/stores/modules/knowledge.ts
+++ b/ui/src/stores/modules/knowledge.ts
@@ -32,10 +32,10 @@ const useKnowledgeStore = defineStore('knowledge', {
     saveDocumentsFile(file: UploadUserFile[]) {
       this.documentsFiles = file
     },
-    async asyncGetRootKnowledge(loading?: Ref<boolean>) {
+    async asyncGetFolderKnowledge(folder_id?: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         const params = {
-          folder_id: localStorage.getItem('workspace_id'),
+          folder_id: folder_id || localStorage.getItem('workspace_id'),
         }
         knowledgeApi
           .getKnowledgeList(params, loading)
@@ -47,46 +47,46 @@ const useKnowledgeStore = defineStore('knowledge', {
           })
       })
     },
-    async asyncGetTreeRootKnowledge(loading?: Ref<boolean>) {
-      const folder = useFolderStore()
-      return Promise.all([
-        folder.asyncGetFolder('KNOWLEDGE', {}, loading),
-        this.asyncGetRootKnowledge(loading),
-      ])
-        .then((res: any) => {
-          const folderList = res[0].data
-          const knowledgeList = res[1].data
-          const arrMap: any = {}
-          function buildIdMap(arr: any) {
-            arr.forEach((item: any) => {
-              arrMap[item.id] = item
-              // 递归处理子节点
-              if (item.children && item.children.length > 0) {
-                buildIdMap(item.children)
-              }
-            })
-          }
-          buildIdMap(folderList)
-          knowledgeList
-            .filter((v: any) => v.resource_type !== 'folder')
-            .forEach((item: any) => {
-              const targetFolder = arrMap[item.folder_id]
-              if (targetFolder) {
-                // 检查是否已有相同ID的子节点（避免重复插入）
-                const existingChild = targetFolder.children.find(
-                  (child: any) => child.id === item.id,
-                )
-                if (!existingChild) {
-                  targetFolder.children.push(item)
-                }
-              }
-            })
-          return Promise.resolve(folderList)
-        })
-        .catch((error) => {
-          return Promise.reject(error)
-        })
-    },
+    // async asyncGetTreeRootKnowledge(loading?: Ref<boolean>) {
+    //   const folder = useFolderStore()
+    //   return Promise.all([
+    //     folder.asyncGetFolder('KNOWLEDGE', {}, loading),
+    //     this.asyncGetFolderKnowledge(loading),
+    //   ])
+    //     .then((res: any) => {
+    //       const folderList = res[0].data
+    //       const knowledgeList = res[1].data
+    //       const arrMap: any = {}
+    //       function buildIdMap(arr: any) {
+    //         arr.forEach((item: any) => {
+    //           arrMap[item.id] = item
+    //           // 递归处理子节点
+    //           if (item.children && item.children.length > 0) {
+    //             buildIdMap(item.children)
+    //           }
+    //         })
+    //       }
+    //       buildIdMap(folderList)
+    //       knowledgeList
+    //         .filter((v: any) => v.resource_type !== 'folder')
+    //         .forEach((item: any) => {
+    //           const targetFolder = arrMap[item.folder_id]
+    //           if (targetFolder) {
+    //             // 检查是否已有相同ID的子节点（避免重复插入）
+    //             const existingChild = targetFolder.children.find(
+    //               (child: any) => child.id === item.id,
+    //             )
+    //             if (!existingChild) {
+    //               targetFolder.children.push(item)
+    //             }
+    //           }
+    //         })
+    //       return Promise.resolve(folderList)
+    //     })
+    //     .catch((error) => {
+    //       return Promise.reject(error)
+    //     })
+    // },
     async asyncGetKnowledgeDetail(knowledge_id: string, loading?: Ref<boolean>) {
       return new Promise((resolve, reject) => {
         knowledgeApi
diff --git a/ui/src/views/application/ApplicationSetting.vue b/ui/src/views/application/ApplicationSetting.vue
index cd4925f4e..8adc5291c 100644
--- a/ui/src/views/application/ApplicationSetting.vue
+++ b/ui/src/views/application/ApplicationSetting.vue
@@ -633,7 +633,7 @@ function getDetail() {
 }
 
 function getKnowledge() {
-  knowledge.asyncGetRootKnowledge(knowledgeLoading).then((res: any) => {
+  knowledge.asyncGetFolderKnowledge(knowledgeLoading).then((res: any) => {
     knowledgeList.value = res.data
   })
 }
diff --git a/ui/src/views/application/component/AddKnowledgeDialog.vue b/ui/src/views/application/component/AddKnowledgeDialog.vue
index 71a2e6319..3c319ef81 100644
--- a/ui/src/views/application/component/AddKnowledgeDialog.vue
+++ b/ui/src/views/application/component/AddKnowledgeDialog.vue
@@ -49,13 +49,17 @@
       <el-scrollbar>
         <div class="max-height layout-bg p-16-24">
           <el-row :gutter="12" v-loading="loading">
-            <el-col :span="12" v-for="(item, index) in filterData" :key="index" class="mb-16">
+            <el-col
+              :span="12"
+              v-for="(item, index) in filterData.filter((v:any) => v.resource_type !== 'folder')"
+              :key="index"
+              class="mb-16"
+            >
               <CardCheckbox
                 value-field="id"
                 :data="item"
                 v-model="checkList"
                 @change="changeHandle"
-                v-if="item.resource_type !== 'folder'"
               >
                 <span class="ellipsis cursor ml-12" :title="item.name"> {{ item.name }}</span>
               </CardCheckbox>
@@ -100,7 +104,7 @@ const props = defineProps({
 })
 
 const emit = defineEmits(['addData', 'refresh'])
-const { folder, user } = useStore()
+const { folder, user,knowledge } = useStore()
 
 const dialogVisible = ref<boolean>(false)
 const checkList = ref([])
@@ -191,7 +195,7 @@ function getList() {
   const params = {
     folder_id: currentFolder.value?.id || user.getWorkspaceId(),
   }
-  KnowledgeApi.getKnowledgeList(params, loading).then((res) => {
+  knowledge.asyncGetFolderKnowledge(params, loading).then((res) => {
     searchDate.value = res.data
   })
 }
diff --git a/ui/src/views/document/component/SelectKnowledgeDialog.vue b/ui/src/views/document/component/SelectKnowledgeDialog.vue
index e587a1f0c..f42d996a2 100644
--- a/ui/src/views/document/component/SelectKnowledgeDialog.vue
+++ b/ui/src/views/document/component/SelectKnowledgeDialog.vue
@@ -11,23 +11,28 @@
       <h4 :id="titleId" :class="titleClass">{{ '文档迁移到' }}</h4>
     </template>
     <el-form
-      class="p-24"
       ref="FormRef"
       :model="form"
       label-position="top"
       require-asterisk-position="right"
-      v-loading="loading"
     >
       <el-form-item :label="$t('views.chatLog.selectKnowledge')" required>
         <el-tree-select
           v-model="form.selectKnowledge"
-          :data="knowledgeList"
           :props="defaultProps"
           node-key="id"
+          :default-expanded-keys="['default']"
+          lazy
+          :load="loadTree"
         >
           <template #default="{ data }">
             <div class="flex align-center">
-              <KnowledgeIcon class="mr-12" :size="20" v-if="data.resource_type" :type="data.type" />
+              <KnowledgeIcon
+                class="mr-12"
+                :size="20"
+                v-if="data.resource_type !== 'folder'"
+                :type="data.type"
+              />
               <el-avatar v-else class="mr-12" shape="square" :size="20" style="background: none">
                 <img
                   src="@/assets/knowledge/icon_file-folder_colorful.svg"
@@ -80,9 +85,9 @@ const documentList = ref<any>([])
 const defaultProps = {
   children: 'children',
   label: 'name',
+  isLeaf: (data: any) => data.resource_type && data.resource_type !== 'folder',
   disabled: (data: any, node: any) => {
-    console.log(data, node)
-    return data.id === id || (node?.isLeaf && !data.resource_type)
+    return data.id === id
   },
 }
 
@@ -100,9 +105,17 @@ watch(dialogVisible, (bool) => {
 
 const open = (list: any) => {
   documentList.value = list
-  getKnowledge()
   dialogVisible.value = true
 }
+
+const loadTree = (node: any, resolve: any) => {
+  console.log(node)
+  if (node.isLeaf) return resolve([])
+  const folder_id = node.level === 0 ? '' : node.data.id
+  knowledge.asyncGetFolderKnowledge(folder_id, loading).then((res: any) => {
+    resolve(res.data)
+  })
+}
 const submitHandle = () => {
   documentApi
     .putMigrateMulDocument(id, form.value.selectKnowledge, documentList.value, loading)
@@ -112,26 +125,19 @@ const submitHandle = () => {
     })
 }
 
-function getKnowledge() {
-  knowledge.asyncGetTreeRootKnowledge(loading).then((res: any) => {
-    knowledgeList.value = res || []
-    console.log(knowledgeList.value)
-  })
-}
-
 defineExpose({ open })
 </script>
 <style lang="scss">
-.select-knowledge-dialog {
-  padding: 0;
-  .el-dialog__header {
-    padding: 24px 24px 0 24px;
-  }
-  .el-dialog__body {
-    padding: 8px !important;
-  }
-  .el-dialog__footer {
-    padding: 0 24px 24px;
-  }
-}
+// .select-knowledge-dialog {
+//   padding: 0;
+//   .el-dialog__header {
+//     padding: 24px 24px 0 24px;
+//   }
+//   .el-dialog__body {
+//     padding: 8px !important;
+//   }
+//   .el-dialog__footer {
+//     padding: 0 24px 24px;
+//   }
+// }
 </style>
diff --git a/ui/src/views/paragraph/component/SelectDocumentDialog.vue b/ui/src/views/paragraph/component/SelectDocumentDialog.vue
index 57a90876c..34e7d5b80 100644
--- a/ui/src/views/paragraph/component/SelectDocumentDialog.vue
+++ b/ui/src/views/paragraph/component/SelectDocumentDialog.vue
@@ -121,7 +121,7 @@ function getDocument(id: string) {
 }
 
 function getDataset() {
-  knowledge.asyncGetRootKnowledge(loading).then((res: any) => {
+  knowledge.asyncGetFolderKnowledge(loading).then((res: any) => {
     datasetList.value = res.data
   })
 }
diff --git a/ui/src/views/resource-management/document/component/SelectKnowledgeDialog.vue b/ui/src/views/resource-management/document/component/SelectKnowledgeDialog.vue
index 05eb55bfa..d8be64a44 100644
--- a/ui/src/views/resource-management/document/component/SelectKnowledgeDialog.vue
+++ b/ui/src/views/resource-management/document/component/SelectKnowledgeDialog.vue
@@ -92,7 +92,7 @@ const submitHandle = () => {
 }
 
 function getKnowledge() {
-  knowledge.asyncGetRootKnowledge(loading).then((res: any) => {
+  knowledge.asyncGetFolderKnowledge(loading).then((res: any) => {
     knowledgeList.value = res.data?.filter((v: any) => v.id !== id)
   })
 }
diff --git a/ui/src/views/resource-management/paragraph/component/SelectDocumentDialog.vue b/ui/src/views/resource-management/paragraph/component/SelectDocumentDialog.vue
index a1b8ba227..c1da37480 100644
--- a/ui/src/views/resource-management/paragraph/component/SelectDocumentDialog.vue
+++ b/ui/src/views/resource-management/paragraph/component/SelectDocumentDialog.vue
@@ -121,7 +121,7 @@ function getDocument(id: string) {
 }
 
 function getDataset() {
-  knowledge.asyncGetRootKnowledge(loading).then((res: any) => {
+  knowledge.asyncGetFolderKnowledge(loading).then((res: any) => {
     datasetList.value = res.data
   })
 }
diff --git a/ui/src/views/shared/paragraph-shared/component/SelectDocumentDialog.vue b/ui/src/views/shared/paragraph-shared/component/SelectDocumentDialog.vue
index 5997670e1..6d77304a6 100644
--- a/ui/src/views/shared/paragraph-shared/component/SelectDocumentDialog.vue
+++ b/ui/src/views/shared/paragraph-shared/component/SelectDocumentDialog.vue
@@ -144,7 +144,7 @@ function getDocument(id: string) {
 }
 
 function getknowledge() {
-  knowledge.asyncGetRootKnowledge(loading).then((res: any) => {
+  knowledge.asyncGetFolderKnowledge(loading).then((res: any) => {
     knowledgeList.value = res.data
   })
 }
diff --git a/ui/src/workflow/nodes/search-knowledge-node/index.vue b/ui/src/workflow/nodes/search-knowledge-node/index.vue
index 64285065a..3e47f7cee 100644
--- a/ui/src/workflow/nodes/search-knowledge-node/index.vue
+++ b/ui/src/workflow/nodes/search-knowledge-node/index.vue
@@ -190,7 +190,7 @@ function getknowledge() {
   //     knowledgeList.value = res.data
   //   })
   // } else {
-  knowledge.asyncGetRootKnowledge(knowledgeLoading).then((res: any) => {
+  knowledge.asyncGetFolderKnowledge(knowledgeLoading).then((res: any) => {
     knowledgeList.value = res.data?.filter((v: any) => v.user_id === user.userInfo?.id)
   })
   // }