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="用户设置" >
@@ -53,7 +53,7 @@ {{ form['calls'] }} - + {{ form['tokens'] }} @@ -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 @@ - +