From 70491ea1bb93fdd617d45f33880b8f6375e5893d Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Thu, 18 Jul 2024 17:12:28 +0800
Subject: [PATCH] fix: image relay quota

---
 relay/relay-image.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/relay/relay-image.go b/relay/relay-image.go
index 21e0582..f6a2641 100644
--- 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"