From 954fa879dc7a342197e9951680b545a789680ffa Mon Sep 17 00:00:00 2001 From: "1808837298@qq.com" <1808837298@qq.com> Date: Sat, 15 Jun 2024 01:30:51 +0800 Subject: [PATCH] fix: claude usage --- relay/channel/claude/relay-claude.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/relay/channel/claude/relay-claude.go b/relay/channel/claude/relay-claude.go index f9c6ab3..ca53433 100644 --- a/relay/channel/claude/relay-claude.go +++ b/relay/channel/claude/relay-claude.go @@ -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) }