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

fix: claude usage

Browse Source
This commit is contained in:
1808837298@qq.com committed 2024-06-15 01:30:51 +08:00
1 parent 6b91f85400
commit 31f7946866
1 file changed
+3
Download Patch File Download Diff File Expand all files Collapse all files
relay/channel/claude/relay-claude.go
+3
View File
@@ -337,6 +337,9 @@ func claudeStreamHandler(requestMode int, modelName string, promptTokens int, c
if requestMode == RequestModeCompletion { if requestMode == RequestModeCompletion {
usage, _ = service.ResponseText2Usage(responseText, modelName, promptTokens) usage, _ = service.ResponseText2Usage(responseText, modelName, promptTokens)
} else { } else {
if usage.PromptTokens == 0 {
usage.PromptTokens = promptTokens
}
if usage.CompletionTokens == 0 { if usage.CompletionTokens == 0 {
usage, _ = service.ResponseText2Usage(responseText, modelName, usage.PromptTokens) usage, _ = service.ResponseText2Usage(responseText, modelName, usage.PromptTokens)
} }