From 1b8eeb0c86ca80f48653b7d1f4f4b8da8e489bfa Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Sat, 8 Jul 2023 14:03:45 +0800
Subject: [PATCH] fix calculate user token in hide state

---
 src/chatbox.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/chatbox.tsx b/src/chatbox.tsx
index 1c9298d..746139f 100644
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -169,7 +169,8 @@ export default function ChatBOX(props: {
     if (data.usage.prompt_tokens) {
       const userMessageToken = data.usage.prompt_tokens - aboveToken;
       console.log("set user message token");
-      chatStore.history.slice(-1)[0].token = userMessageToken;
+      chatStore.history.filter((msg) => !msg.hide).slice(-1)[0].token =
+        userMessageToken;
     }
 
     chatStore.history.push({