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