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))