From 962f77a7f72aa2caf0d000ca421c41a44fa0448e Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Wed, 17 Jan 2024 18:06:24 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=9F=A5=E8=AF=86=E6=9D=A5=E6=BA=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/components/ai-chat/index.vue | 29 +++++++++++++++++++++++------
 ui/src/styles/element-plus.scss     |  2 +-
 ui/src/views/dataset/index.vue      |  4 ++--
 3 files changed, 26 insertions(+), 9 deletions(-)

diff --git a/ui/src/components/ai-chat/index.vue b/ui/src/components/ai-chat/index.vue
index b0a674062..2c39158b0 100644
--- a/ui/src/components/ai-chat/index.vue
+++ b/ui/src/components/ai-chat/index.vue
@@ -64,13 +64,30 @@
                 <MdRenderer :source="item.answer_text"></MdRenderer>
                 <div v-if="item.write_ed || log">
                   <el-divider> <el-text type="info">知识来源</el-text> </el-divider>
+                  <div class="mb-8">
+                    <el-space wrap>
+                      <el-button
+                        v-for="(dataset, index) in item.dataset_list"
+                        :key="index"
+                        type="primary"
+                        plain
+                        size="small"
+                        >{{ dataset.name }}</el-button
+                      >
+                    </el-space>
+                  </div>
 
-                  <el-tag type="info" effect="plain">
-                    消耗 tokens: {{ item?.message_tokens + item?.answer_tokens }}
-                  </el-tag>
-                  <el-tag class="ml-8" type="info" effect="plain">
-                    耗时: {{ item.run_time.toFixed(2) }} s
-                  </el-tag>
+                  <div>
+                    <el-button class="mr-8" type="primary" plain size="small"
+                      >引用分段：{{ item.paragraph_list.length }}</el-button
+                    >
+                    <el-tag type="info" effect="plain">
+                      消耗 tokens: {{ item?.message_tokens + item?.answer_tokens }}
+                    </el-tag>
+                    <el-tag class="ml-8" type="info" effect="plain">
+                      耗时: {{ item.run_time.toFixed(2) }} s
+                    </el-tag>
+                  </div>
                 </div>
               </el-card>
               <div class="flex-between mt-8" v-if="log">
diff --git a/ui/src/styles/element-plus.scss b/ui/src/styles/element-plus.scss
index ca5f60eb9..999f202c4 100644
--- a/ui/src/styles/element-plus.scss
+++ b/ui/src/styles/element-plus.scss
@@ -102,7 +102,7 @@
 
 .el-tag {
   --el-tag-border-radius: 2px;
-  height: 22px;
+  height: 24px;
   padding: 0 6px;
 }
 .el-table {
diff --git a/ui/src/views/dataset/index.vue b/ui/src/views/dataset/index.vue
index 23e557f11..8adaee0b1 100644
--- a/ui/src/views/dataset/index.vue
+++ b/ui/src/views/dataset/index.vue
@@ -48,9 +48,9 @@
                     </div>
                     <div @click.stop>
                       <el-dropdown trigger="click">
-                        <span class="el-dropdown-link cursor">
+                        <el-button text @click.stop>
                           <el-icon><MoreFilled /></el-icon>
-                        </span>
+                        </el-button>
                         <template #dropdown>
                           <el-dropdown-menu>
                             <el-dropdown-item