fix: image relay quota

9def2c6c · CalciumIon · 34c356b0 · 9def2c6c
Commit 9def2c6c authored Jul 18, 2024 by CalciumIon
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

relay/relay-image.go
+2 -2

No files found.
--- a/relay/relay-image.go
+++ b/relay/relay-image.go
@@ -170,8 +170,8 @@ func ImageHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusCode {
 	}

 	usage := &dto.Usage{
-		PromptTokens: relayInfo.PromptTokens,
-		TotalTokens:  relayInfo.PromptTokens,
+		PromptTokens: imageRequest.N,
+		TotalTokens:  imageRequest.N,
 	}

 	quality := "standard"