diff --git a/controller/relay.go b/controller/relay.go index ffd9de47..d48db1de 100644 --- a/controller/relay.go +++ b/controller/relay.go @@ -318,8 +318,6 @@ func RelayMidjourney(c *gin.Context) { return } - relayInfo.InitChannelMeta(c) - var mjErr *dto.MidjourneyResponse switch relayInfo.RelayMode { case relayconstant.RelayModeMidjourneyNotify: diff --git a/relay/mjproxy_handler.go b/relay/mjproxy_handler.go index 756ad450..7c52cb6b 100644 --- a/relay/mjproxy_handler.go +++ b/relay/mjproxy_handler.go @@ -176,6 +176,9 @@ func RelaySwapFace(c *gin.Context, info *relaycommon.RelayInfo) *dto.MidjourneyR if err != nil { return service.MidjourneyErrorWrapper(constant.MjRequestError, "bind_request_body_failed") } + + info.InitChannelMeta(c) + if swapFaceRequest.SourceBase64 == "" || swapFaceRequest.TargetBase64 == "" { return service.MidjourneyErrorWrapper(constant.MjRequestError, "sour_base64_and_target_base64_is_required") } @@ -370,6 +373,8 @@ func RelayMidjourneySubmit(c *gin.Context, relayInfo *relaycommon.RelayInfo) *dt return service.MidjourneyErrorWrapper(constant.MjRequestError, "bind_request_body_failed") } + relayInfo.InitChannelMeta(c) + if relayInfo.RelayMode == relayconstant.RelayModeMidjourneyAction { // midjourney plus,需要从customId中获取任务信息 mjErr := service.CoverPlusActionToNormalAction(&midjRequest) if mjErr != nil {