From 3455a371384d58d402d727a7f952d221c808fe4e Mon Sep 17 00:00:00 2001
From: dataeaseShu <tong.shu@fit2cloud.com>
Date: Mon, 22 Jul 2024 10:30:08 +0800
Subject: [PATCH] =?UTF-8?q?fix(=E6=9F=A5=E8=AF=A2=E7=BB=84=E4=BB=B6):=20?=
 =?UTF-8?q?=E8=BF=87=E6=BB=A4=E7=BB=84=E4=BB=B6=E6=B2=A1=E6=9C=89=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E6=8C=89=E9=92=AE=E6=83=85=E5=86=B5=E4=B8=8B=E6=97=A0?=
 =?UTF-8?q?=E6=B3=95=E8=A7=A6=E5=8F=91=E6=B8=85=E7=A9=BA=E5=92=8C=E9=87=8D?=
 =?UTF-8?q?=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../custom-component/v-query/Component.vue    |  2 +
 .../core-frontend/src/views/copilot/index.vue | 44 ++++++++++++++-----
 2 files changed, 35 insertions(+), 11 deletions(-)

diff --git a/core/core-frontend/src/custom-component/v-query/Component.vue b/core/core-frontend/src/custom-component/v-query/Component.vue
index e805fc5dfb..b0c059b69f 100644
--- a/core/core-frontend/src/custom-component/v-query/Component.vue
+++ b/core/core-frontend/src/custom-component/v-query/Component.vue
@@ -410,6 +410,7 @@ const resetData = () => {
     pre = [...new Set([...keyList, ...pre])]
     return pre
   }, [])
+  !componentWithSure.value && queryData()
 }
 
 const clearData = () => {
@@ -427,6 +428,7 @@ const clearData = () => {
     pre = [...new Set([...keyList, ...pre])]
     return pre
   }, [])
+  !componentWithSure.value && queryData()
 }
 const listVisible = computed(() => {
   return list.value.filter(itx => itx.visible)
diff --git a/core/core-frontend/src/views/copilot/index.vue b/core/core-frontend/src/views/copilot/index.vue
index 2479922add..6535d1628c 100644
--- a/core/core-frontend/src/views/copilot/index.vue
+++ b/core/core-frontend/src/views/copilot/index.vue
@@ -1,5 +1,5 @@
 <script lang="ts" setup>
-import { ref, shallowRef, computed, watch, onBeforeUnmount } from 'vue'
+import { ref, shallowRef, computed, watch, onBeforeUnmount, nextTick } from 'vue'
 import { ElMessageBox } from 'element-plus-secondary'
 import {
   getDatasetTree,
@@ -42,6 +42,7 @@ const dfs = arr => {
     return ele.leaf
   })
 }
+
 const computedTree = computed(() => {
   if (datasetTree.value[0]?.id === '0') {
     return dfs(datasetTree.value[0].children)
@@ -52,16 +53,44 @@ const computedTree = computed(() => {
 const isActive = computed(() => {
   return questionInput.value.trim().length && !!datasetId.value
 })
-const initDataset = () => {
-  getDatasetTree({}).then(res => {
+const initDataset = async () => {
+  await getDatasetTree({}).then(res => {
     datasetTree.value = (res as unknown as Tree[]) || []
   })
+  getListCopilot().then(res => {
+    const allList = (res as unknown as { history: object }[]) || []
+    historyBack = allList[allList.length - 1]?.history || []
+    historyArr.value = cloneDeep(allList).map(ele => ({ ...ele, loading: false }))
+    if (!!allList.length) {
+      datasetId.value = allList[0].datasetGroupId
+      oldId = datasetId.value
+      datasetId.value && getOptions(datasetId.value)
+      if (oldId && !oldName) {
+        nextTick(() => {
+          dfsName(computedTree.value)
+        })
+      }
+    }
+  })
 }
 
 const treeSelectRef = ref()
 let oldId = ''
 let currentId = ''
 let oldName = ''
+
+const dfsName = arr => {
+  return arr.filter(ele => {
+    if (ele.id === oldId) {
+      oldName = ele.name
+    }
+    if (!!ele.children?.length && !ele.leaf) {
+      ele.children = dfsName(ele.children)
+      return !!ele.children?.length
+    }
+    return ele.leaf
+  })
+}
 const handleDatasetChange = () => {
   if (!!oldId && !!historyArr.value.length) {
     currentId = datasetId.value
@@ -97,14 +126,7 @@ const getOptions = id => {
 }
 initDataset()
 let historyBack = []
-getListCopilot().then(res => {
-  historyBack = (res as unknown as string[]) || []
-  historyArr.value = cloneDeep(historyBack).map(ele => ({ ...ele, loading: false }))
-  if (!!historyBack.length) {
-    datasetId.value = historyBack[0].datasetGroupId
-    datasetId.value && getOptions(datasetId.value)
-  }
-})
+
 const questionInputRef = ref()
 const overHeight = ref(false)
 const { height } = useElementSize(questionInputRef)