This commit is contained in:
lxowalle 2026-05-15 11:54:46 +08:00
parent 18050872fc
commit 68a5ca8928

View file

@ -394,7 +394,11 @@ func parseStreamResponse(
usage = &UsageInfo{ usage = &UsageInfo{
PromptTokens: int(aws.ToInt32(e.Value.Usage.InputTokens)), PromptTokens: int(aws.ToInt32(e.Value.Usage.InputTokens)),
CompletionTokens: int(aws.ToInt32(e.Value.Usage.OutputTokens)), CompletionTokens: int(aws.ToInt32(e.Value.Usage.OutputTokens)),
TotalTokens: int(aws.ToInt32(e.Value.Usage.InputTokens)) + int(aws.ToInt32(e.Value.Usage.OutputTokens)), TotalTokens: int(
aws.ToInt32(e.Value.Usage.InputTokens),
) + int(
aws.ToInt32(e.Value.Usage.OutputTokens),
),
} }
} }
} }