From 44361d75e88d066a8554dfeff384606d19df2d29 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Wed, 13 Mar 2024 23:17:12 +0800
Subject: [PATCH] fix: "Inpaint" code error

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

diff --git a/relay/relay-mj.go b/relay/relay-mj.go
index 6cdd9e0..7342717 100644
--- a/relay/relay-mj.go
+++ b/relay/relay-mj.go
@@ -494,8 +494,10 @@ func RelayMidjourneySubmit(c *gin.Context, relayMode int) *dto.MidjourneyRespons
 			}
 		}
 		//修改返回值
-		newBody := strings.Replace(string(responseBody), `"code":21`, `"code":1`, -1)
-		responseBody = []byte(newBody)
+		if midjRequest.Action != constant.MjActionInPaintPre {
+			newBody := strings.Replace(string(responseBody), `"code":21`, `"code":1`, -1)
+			responseBody = []byte(newBody)
+		}
 	}
 
 	err = midjourneyTask.Insert()