diff --git a/relay/constant/relay_mode.go b/relay/constant/relay_mode.go index 1790c57..2e94bc0 100644 --- a/relay/constant/relay_mode.go +++ b/relay/constant/relay_mode.go @@ -56,29 +56,29 @@ func Path2RelayMode(path string) int { func Path2RelayModeMidjourney(path string) int { relayMode := RelayModeUnknown - if strings.HasPrefix(path, "/mj/submit/action") { + if strings.HasSuffix(path, "/mj/submit/action") { // midjourney plus relayMode = RelayModeMidjourneyAction - } else if strings.HasPrefix(path, "/mj/submit/modal") { + } else if strings.HasSuffix(path, "/mj/submit/modal") { // midjourney plus relayMode = RelayModeMidjourneyModal - } else if strings.HasPrefix(path, "/mj/submit/shorten") { + } else if strings.HasSuffix(path, "/mj/submit/shorten") { // midjourney plus relayMode = RelayModeMidjourneyShorten - } else if strings.HasPrefix(path, "/mj/insight-face/swap") { + } else if strings.HasSuffix(path, "/mj/insight-face/swap") { // midjourney plus relayMode = RelayModeSwapFace - } else if strings.HasPrefix(path, "/mj/submit/imagine") { + } else if strings.HasSuffix(path, "/mj/submit/imagine") { relayMode = RelayModeMidjourneyImagine - } else if strings.HasPrefix(path, "/mj/submit/blend") { + } else if strings.HasSuffix(path, "/mj/submit/blend") { relayMode = RelayModeMidjourneyBlend - } else if strings.HasPrefix(path, "/mj/submit/describe") { + } else if strings.HasSuffix(path, "/mj/submit/describe") { relayMode = RelayModeMidjourneyDescribe - } else if strings.HasPrefix(path, "/mj/notify") { + } else if strings.HasSuffix(path, "/mj/notify") { relayMode = RelayModeMidjourneyNotify - } else if strings.HasPrefix(path, "/mj/submit/change") { + } else if strings.HasSuffix(path, "/mj/submit/change") { relayMode = RelayModeMidjourneyChange - } else if strings.HasPrefix(path, "/mj/submit/simple-change") { + } else if strings.HasSuffix(path, "/mj/submit/simple-change") { relayMode = RelayModeMidjourneyChange } else if strings.HasSuffix(path, "/fetch") { relayMode = RelayModeMidjourneyTaskFetch