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, 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) {
message.metrics.completion_tokens = message.usage.completion_tokens if (!message.metrics?.completion_tokens) {
message.metrics.completion_tokens = message.usage.completion_tokens
}
} }
} }
} catch (error: any) { } catch (error: any) {