fix: record total tokens

labring · Feb 10, 2025 · 8dfcf08 · 8dfcf08
1 parent ce497ed
commit 8dfcf08
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/service/aiproxy/model/log.go b/service/aiproxy/model/log.go
@@ -171,6 +171,7 @@ func RecordConsumeLog(
 		Code:             code,
 		PromptTokens:     promptTokens,
 		CompletionTokens: completionTokens,
+		TotalTokens:      promptTokens + completionTokens,
 		TokenID:          tokenID,
 		TokenName:        tokenName,
 		Model:            modelName,