fix: claude usage

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
+3
View File
@@ -337,6 +337,9 @@ func claudeStreamHandler(requestMode int, modelName string, promptTokens int, c
if requestMode == RequestModeCompletion {
usage, _ = service.ResponseText2Usage(responseText, modelName, promptTokens)
} else {
if usage.PromptTokens == 0 {
usage.PromptTokens = promptTokens
}
if usage.CompletionTokens == 0 {
usage, _ = service.ResponseText2Usage(responseText, modelName, usage.PromptTokens)
}