From a4acd34e5f277383eb6cedd79c9009e3c18b65ad Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Thu, 18 Jul 2024 15:46:28 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E7=9F=A5=E8=AF=86=E5=BA=93=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/layout/components/breadcrumb/index.vue         | 11 +++++++++--
 .../views/dataset/component/CreateDatasetDialog.vue   |  4 ++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/ui/src/layout/components/breadcrumb/index.vue b/ui/src/layout/components/breadcrumb/index.vue
index c884d5b09..2b4d77f7b 100644
--- a/ui/src/layout/components/breadcrumb/index.vue
+++ b/ui/src/layout/components/breadcrumb/index.vue
@@ -99,7 +99,7 @@
             </div>
           </template>
           <template v-else-if="isDataset">
-            <div class="w-full text-left cursor" @click="router.push({ path: '/dataset/create' })">
+            <div class="w-full text-left cursor" @click="openCreateDialog">
               <el-button link>
                 <el-icon class="mr-4"><Plus /></el-icon> 创建知识库
               </el-button>
@@ -110,12 +110,14 @@
     </el-dropdown>
   </div>
   <CreateApplicationDialog ref="CreateApplicationDialogRef" @refresh="refresh" />
+  <CreateDatasetDialog ref="CreateDatasetDialogRef" @refresh="refresh" />
 </template>
 
 <script setup lang="ts">
 import { ref, onMounted, computed } from 'vue'
 import { onBeforeRouteLeave, useRouter, useRoute } from 'vue-router'
 import CreateApplicationDialog from '@/views/application/component/CreateApplicationDialog.vue'
+import CreateDatasetDialog from '@/views/dataset/component/CreateDatasetDialog.vue'
 import { isAppIcon, isWorkFlow } from '@/utils/application'
 import useStore from '@/stores'
 const { common, dataset, application } = useStore()
@@ -130,6 +132,7 @@ onBeforeRouteLeave((to, from) => {
   common.saveBreadcrumb(null)
 })
 
+const CreateDatasetDialogRef = ref()
 const CreateApplicationDialogRef = ref()
 const list = ref<any[]>([])
 const loading = ref(false)
@@ -148,7 +151,11 @@ const isDataset = computed(() => {
 })
 
 function openCreateDialog() {
-  CreateApplicationDialogRef.value.open()
+  if (isDataset.value) {
+    CreateDatasetDialogRef.value.open()
+  } else if (isApplication.value) {
+    CreateApplicationDialogRef.value.open()
+  }
 }
 
 function changeMenu(id: string) {
diff --git a/ui/src/views/dataset/component/CreateDatasetDialog.vue b/ui/src/views/dataset/component/CreateDatasetDialog.vue
index df1363e59..df47c9ff7 100644
--- a/ui/src/views/dataset/component/CreateDatasetDialog.vue
+++ b/ui/src/views/dataset/component/CreateDatasetDialog.vue
@@ -89,6 +89,8 @@ import { MsgSuccess, MsgAlert } from '@/utils/message'
 import useStore from '@/stores'
 import { ValidType, ValidCount } from '@/enums/common'
 
+const emit = defineEmits(['refresh'])
+
 const { common, user } = useStore()
 const router = useRouter()
 const BaseFormRef = ref()
@@ -150,12 +152,14 @@ const submitHandle = async () => {
           datasetApi.postDataset(obj, loading).then((res) => {
             MsgSuccess('创建成功')
             router.push({ path: `/dataset/${res.data.id}/document` })
+            emit('refresh')
           })
         } else {
           const obj = { ...BaseFormRef.value.form, ...datasetForm.value }
           datasetApi.postWebDataset(obj, loading).then((res) => {
             MsgSuccess('创建成功')
             router.push({ path: `/dataset/${res.data.id}/document` })
+            emit('refresh')
           })
         }
       } else {