diff --git a/src/renderer/src/store/thunk/messageThunk.ts b/src/renderer/src/store/thunk/messageThunk.ts index 3b9485565f..3cd8be0b3c 100644 --- a/src/renderer/src/store/thunk/messageThunk.ts +++ b/src/renderer/src/store/thunk/messageThunk.ts @@ -603,7 +603,7 @@ const fetchAndProcessAssistantResponseImpl = async ( const finalStateOnComplete = getState() const finalAssistantMsg = finalStateOnComplete.messages.entities[assistantMsgId] - if (status === 'success' && finalAssistantMsg && response && !response?.usage) { + if (status === 'success' && finalAssistantMsg) { const userMsgId = finalAssistantMsg.askId const orderedMsgs = selectMessagesForTopic(finalStateOnComplete, topicId) const userMsgIndex = orderedMsgs.findIndex((m) => m.id === userMsgId) @@ -613,8 +613,10 @@ const fetchAndProcessAssistantResponseImpl = async ( // 更新topic的name autoRenameTopic(assistant, topicId) - const usage = await estimateMessagesUsage({ assistant, messages: finalContextWithAssistant }) - response.usage = usage + if (response && !response.usage) { + const usage = await estimateMessagesUsage({ assistant, messages: finalContextWithAssistant }) + response.usage = usage + } } if (response && response.metrics) { if (!response.metrics.completion_tokens && response.usage) {