Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-19 14:41:24 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix: add token usage to agent session message

Browse Source
This commit is contained in:
defi-failure 2025-11-06 17:59:38 +08:00
parent ff8e5ddd27
commit 7c8bf8b591
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/renderer/src/pages/home/Inputbar/AgentSessionInputbar.tsx
View File

@ -10,6 +10,7 @@ import { useShortcutDisplay } from '@renderer/hooks/useShortcuts'
import { useTimer } from '@renderer/hooks/useTimer' import { useTimer } from '@renderer/hooks/useTimer'
import PasteService from '@renderer/services/PasteService' import PasteService from '@renderer/services/PasteService'
import { pauseTrace } from '@renderer/services/SpanManagerService' import { pauseTrace } from '@renderer/services/SpanManagerService'
import { estimateUserPromptUsage } from '@renderer/services/TokenService'
import { useAppDispatch, useAppSelector } from '@renderer/store' import { useAppDispatch, useAppSelector } from '@renderer/store'
import { newMessagesActions, selectMessagesForTopic } from '@renderer/store/newMessage' import { newMessagesActions, selectMessagesForTopic } from '@renderer/store/newMessage'
import { sendMessage as dispatchSendMessage } from '@renderer/store/thunk/messageThunk' import { sendMessage as dispatchSendMessage } from '@renderer/store/thunk/messageThunk'
@ -199,11 +200,15 @@ const AgentSessionInputbar: FC<Props> = ({ agentId, sessionId }) => {
} }
: undefined : undefined
// Calculate token usage for the user message
const usage = await estimateUserPromptUsage({ content: text })
const userMessage: Message = createMessage('user', sessionTopicId, agentId, { const userMessage: Message = createMessage('user', sessionTopicId, agentId, {
id: userMessageId, id: userMessageId,
blocks: userMessageBlocks.map((block) => block?.id), blocks: userMessageBlocks.map((block) => block?.id),
model, model,
modelId: model?.id modelId: model?.id,
usage
}) })
const assistantStub: Assistant = { const assistantStub: Assistant = {