Jelajahi Sumber

fix: claude usage

1808837298@qq.com 1 tahun lalu
induk
melakukan
954fa879dc
1 mengubah file dengan 3 tambahan dan 0 penghapusan
  1. 3 0
      relay/channel/claude/relay-claude.go

+ 3 - 0
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)
 		}