whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: Improve metrics completion tokens handling

Browse Source
This commit is contained in:
kangfenmao 2025-03-02 19:41:06 +08:00
parent 1c60375d71
commit ed96940e82
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/renderer/src/services/ApiService.ts
View File

@ -114,11 +114,13 @@ export async function fetchChatCompletion({
assistant, assistant,
messages: [..._messages, message] messages: [..._messages, message]
}) })
// 设置 metrics.completion_tokens // Set metrics.completion_tokens
if (message.metrics && message.usage.completion_tokens && !message.metrics.completion_tokens) { if (message.metrics && message?.usage?.completion_tokens) {
if (!message.metrics?.completion_tokens) {
message.metrics.completion_tokens = message.usage.completion_tokens message.metrics.completion_tokens = message.usage.completion_tokens
} }
} }
}
} catch (error: any) { } catch (error: any) {
message.status = 'error' message.status = 'error'
message.error = formatMessageError(error) message.error = formatMessageError(error)