chaos/chaos-api
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: Update token usage calculation in FormatClaudeResponseInfo #865

Browse Source
This commit is contained in:
CalciumIon committed 2025-03-14 17:00:39 +08:00
1 parent 25aaefc6b1
commit 15c5fbd3c8
2 files changed
+4 -791
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

relay/channel/claude/relay-claude.go
+4 -1
View File
@@ -457,7 +457,10 @@ func FormatClaudeResponseInfo(requestMode int, claudeResponse *dto.ClaudeRespons
}
} else if claudeResponse.Type == "message_delta" {
claudeInfo.Usage.CompletionTokens = claudeResponse.Usage.OutputTokens
claudeInfo.Usage.TotalTokens = claudeResponse.Usage.InputTokens + claudeResponse.Usage.OutputTokens
if claudeResponse.Usage.InputTokens > 0 {
claudeInfo.Usage.PromptTokens = claudeResponse.Usage.InputTokens
}
claudeInfo.Usage.TotalTokens = claudeInfo.Usage.PromptTokens + claudeResponse.Usage.OutputTokens
} else if claudeResponse.Type == "content_block_start" {
} else {
return false