* fix ci
This commit is contained in:
parent
18050872fc
commit
68a5ca8928
1 changed files with 11 additions and 7 deletions
|
|
@ -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),
|
||||||
|
),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue