From 466340ff206f9fea56d1a8c5cda437e76f590e9e Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Wed, 17 Jan 2024 17:42:59 +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/LogOperationButton.vue | 1 -
 ui/src/components/ai-chat/index.vue              | 7 +++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/ui/src/components/ai-chat/LogOperationButton.vue b/ui/src/components/ai-chat/LogOperationButton.vue
index cf6529e93..59cc48634 100644
--- a/ui/src/components/ai-chat/LogOperationButton.vue
+++ b/ui/src/components/ai-chat/LogOperationButton.vue
@@ -1,7 +1,6 @@
 <template>
   <div>
     <el-text type="info">
-      消耗: {{ data?.message_tokens + data?.answer_tokens }} tokens
       <span class="ml-4">{{ datetimeFormat(data.create_time) }}</span>
     </el-text>
   </div>
diff --git a/ui/src/components/ai-chat/index.vue b/ui/src/components/ai-chat/index.vue
index d422285c0..b0a674062 100644
--- a/ui/src/components/ai-chat/index.vue
+++ b/ui/src/components/ai-chat/index.vue
@@ -64,6 +64,13 @@
                 <MdRenderer :source="item.answer_text"></MdRenderer>
                 <div v-if="item.write_ed || log">
                   <el-divider> <el-text type="info">知识来源</el-text> </el-divider>
+
+                  <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-card>
               <div class="flex-between mt-8" v-if="log">