From cd1290d71e94cd26b701f7e3fb0df0d8f33fd25d Mon Sep 17 00:00:00 2001 From: luxl-1379 <726857632@qq.com> Date: Tue, 14 Nov 2023 08:55:10 +0800 Subject: [PATCH 1/2] Update relay.go --- controller/relay.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/controller/relay.go b/controller/relay.go index c505c22..351ed3e 100644 --- a/controller/relay.go +++ b/controller/relay.go @@ -255,6 +255,10 @@ func RelayMidjourney(c *gin.Context) { relayMode := RelayModeUnknown if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/imagine") { relayMode = RelayModeMidjourneyImagine + } else if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/blend") { + relayMode = RelayModeMidjourneyBlend + } else if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/describe") { + relayMode = RelayModeMidjourneyDescribe } else if strings.HasPrefix(c.Request.URL.Path, "/mj/notify") { relayMode = RelayModeMidjourneyNotify } else if strings.HasPrefix(c.Request.URL.Path, "/mj/submit/change") { From d0e2cd9f22728a75f3adea6bc716b83037945385 Mon Sep 17 00:00:00 2001 From: luxl-1379 <726857632@qq.com> Date: Tue, 14 Nov 2023 08:55:45 +0800 Subject: [PATCH 2/2] Update relay-router.go --- router/relay-router.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/router/relay-router.go b/router/relay-router.go index c97ea31..2b9b9c3 100644 --- a/router/relay-router.go +++ b/router/relay-router.go @@ -48,6 +48,8 @@ func SetRelayRouter(router *gin.Engine) { { relayMjRouter.POST("/submit/imagine", controller.RelayMidjourney) relayMjRouter.POST("/submit/change", controller.RelayMidjourney) + relayMjRouter.POST("/submit/describe", controller.RelayMidjourney) + relayMjRouter.POST("/submit/blend", controller.RelayMidjourney) relayMjRouter.POST("/notify", controller.RelayMidjourney) relayMjRouter.GET("/task/:id/fetch", controller.RelayMidjourney) }