From 1e24b7bc45689772c753d46663ffccee2ce8a3ad Mon Sep 17 00:00:00 2001 From: Konjac-XZ <1951801592@qq.com> Date: Tue, 4 Mar 2025 16:38:43 +0800 Subject: [PATCH] fix: Token counter display issues when translating assistant messages. --- src/renderer/src/pages/home/Messages/Message.tsx | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/renderer/src/pages/home/Messages/Message.tsx b/src/renderer/src/pages/home/Messages/Message.tsx index 30363452..ad56ca40 100644 --- a/src/renderer/src/pages/home/Messages/Message.tsx +++ b/src/renderer/src/pages/home/Messages/Message.tsx @@ -97,8 +97,10 @@ const MessageItem: FC = ({ const onEditMessage = useCallback( async (msg: Message) => { - const usage = await estimateMessageUsage(msg) - msg.usage = usage + if (msg.role === 'user') { + const usage = await estimateMessageUsage(msg) + msg.usage = usage + } setMessage(msg) const messages = onGetMessages?.()?.map((m) => (m.id === message.id ? msg : m))