fix: Improve metrics completion tokens handling

This commit is contained in:
kangfenmao 2025-03-02 19:41:06 +08:00
parent 1c60375d71
commit ed96940e82

View File

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