From 7b5efd9f3e2b642c648037b2090db737887bfe39 Mon Sep 17 00:00:00 2001
From: ckt1031 <65409152+ckt1031@users.noreply.github.com>
Date: Wed, 15 Nov 2023 20:01:14 +0800
Subject: [PATCH] Added error handling for unsupported image size in
 relayImageHelper function

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

diff --git a/controller/relay-image.go b/controller/relay-image.go
index 2f2e57cf..3ff7dc18 100644
--- a/controller/relay-image.go
+++ b/controller/relay-image.go
@@ -96,6 +96,8 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
 				sizeRatio *= 1.5
 			}
 		}
+	} else {
+		return errorWrapper(errors.New("size not supported"), "size_not_supported", http.StatusBadRequest)
 	}
 
 	quota := int(ratio*sizeRatio*1000) * imageRequest.N