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 0624630e66
commit ba7c3d8e37

View File

@ -97,8 +97,10 @@ const MessageItem: FC<Props> = ({
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))