From 3432d9e0f67a8abc42a2153094f7af152372f37d Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Wed, 10 Jan 2024 13:56:29 +0800
Subject: [PATCH] fix: do not consume user quota if failed

---
 controller/relay-image.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/controller/relay-image.go b/controller/relay-image.go
index a215b579..d7361cbf 100644
--- a/controller/relay-image.go
+++ b/controller/relay-image.go
@@ -170,6 +170,9 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
 	var textResponse ImageResponse
 	defer func(ctx context.Context) {
 		if consumeQuota {
+			if resp.StatusCode != http.StatusOK {
+				return
+			}
 			err := model.PostConsumeTokenQuota(tokenId, userQuota, quota, 0, true)
 			if err != nil {
 				common.SysError("error consuming token remain quota: " + err.Error())