fix: Token counter display issues when translating assistant messages.

This commit is contained in:
Konjac-XZ 2025-03-04 16:38:43 +08:00 committed by 亢奋猫
parent dd92dca34b
commit 1e24b7bc45

View File

@ -97,8 +97,10 @@ const MessageItem: FC<Props> = ({
const onEditMessage = useCallback( const onEditMessage = useCallback(
async (msg: Message) => { async (msg: Message) => {
if (msg.role === 'user') {
const usage = await estimateMessageUsage(msg) const usage = await estimateMessageUsage(msg)
msg.usage = usage msg.usage = usage
}
setMessage(msg) setMessage(msg)
const messages = onGetMessages?.()?.map((m) => (m.id === message.id ? msg : m)) const messages = onGetMessages?.()?.map((m) => (m.id === message.id ? msg : m))