From d8029550f758fed626caa1615a8f6d66efa02db4 Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Mon, 1 Jan 2024 16:18:50 +0800
Subject: [PATCH] fix: do not consume user quota if failed (close #881)

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

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