From 32484bd32ee757d2e8a8fb79885d8b967b98b61e Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Sat, 1 Jul 2023 23:29:24 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E7=94=A8=E6=88=B7=20?=
 =?UTF-8?q?token=20=E6=B6=88=E8=80=97=E7=BB=9F=E8=AE=A1=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/go/handler/chat_handler.go      |  4 ++++
 api/go/handler/user_handler.go      |  2 +-
 web/src/components/ConfigDialog.vue | 11 ++---------
 web/src/views/mobile/Profile.vue    |  2 +-
 4 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/api/go/handler/chat_handler.go b/api/go/handler/chat_handler.go
index 177e5296..3707d80c 100644
--- a/api/go/handler/chat_handler.go
+++ b/api/go/handler/chat_handler.go
@@ -319,6 +319,10 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 				if res.Error != nil {
 					logger.Error("failed to save reply history message: ", res.Error)
 				}
+
+				// 统计用户 token 数量
+				h.db.Model(&user).UpdateColumn("tokens", gorm.Expr("tokens + ?",
+					historyUserMsg.Tokens+historyReplyMsg.Tokens))
 			}
 
 			// 保存当前会话
diff --git a/api/go/handler/user_handler.go b/api/go/handler/user_handler.go
index c329957c..8a2e662a 100644
--- a/api/go/handler/user_handler.go
+++ b/api/go/handler/user_handler.go
@@ -240,7 +240,7 @@ type userProfile struct {
 	Avatar     string           `json:"avatar"`
 	ChatConfig types.ChatConfig `json:"chat_config"`
 	Calls      int              `json:"calls"`
-	Tokens     int              `json:"tokens"`
+	Tokens     int64            `json:"tokens"`
 }
 
 func (h *UserHandler) Profile(c *gin.Context) {
diff --git a/web/src/components/ConfigDialog.vue b/web/src/components/ConfigDialog.vue
index 10931e10..8de7c8e3 100644
--- a/web/src/components/ConfigDialog.vue
+++ b/web/src/components/ConfigDialog.vue
@@ -3,7 +3,7 @@
       v-model="showDialog"
       :close-on-click-modal="false"
       :before-close="close"
-      :top="top"
+      :top="50+'px'"
       title="用户设置"
   >
     <div class="user-info" id="user-info">
@@ -53,7 +53,7 @@
         <el-form-item label="剩余调用次数">
           <el-tag>{{ form['calls'] }}</el-tag>
         </el-form-item>
-        <el-form-item label="剩余 Tokens">
+        <el-form-item label="消耗 Tokens">
           <el-tag type="info">{{ form['tokens'] }}</el-tag>
         </el-form-item>
         <el-form-item label="API KEY">
@@ -100,13 +100,6 @@ const form = ref({
   tokens: 0,
   chat_configs: {}
 })
-const top = computed(() => {
-  if (window.innerHeight < 1024) {
-    return '5vh';
-  } else {
-    return '15vh';
-  }
-})
 
 onMounted(() => {
   // 获取最新用户信息
diff --git a/web/src/views/mobile/Profile.vue b/web/src/views/mobile/Profile.vue
index bb6fea5a..800ec24d 100644
--- a/web/src/views/mobile/Profile.vue
+++ b/web/src/views/mobile/Profile.vue
@@ -35,7 +35,7 @@
             </template>
           </van-field>
 
-          <van-field label="剩余 Tokens">
+          <van-field label="消耗 Tokens">
             <template #input>
               <van-tag type="primary">{{ form.tokens }}</van-tag>
             </template>