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") { 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) }