fix: 支持 /mj-{mode} 路径

This commit is contained in:
Xiangyuan Liu 2024-03-29 17:36:44 +08:00
parent 2e595bdafb
commit 3065bf92ae
No known key found for this signature in database
GPG Key ID: A2FCA5A37FC0F888

View File

@ -56,29 +56,29 @@ func Path2RelayMode(path string) int {
func Path2RelayModeMidjourney(path string) int { func Path2RelayModeMidjourney(path string) int {
relayMode := RelayModeUnknown relayMode := RelayModeUnknown
if strings.HasPrefix(path, "/mj/submit/action") { if strings.HasSuffix(path, "/mj/submit/action") {
// midjourney plus // midjourney plus
relayMode = RelayModeMidjourneyAction relayMode = RelayModeMidjourneyAction
} else if strings.HasPrefix(path, "/mj/submit/modal") { } else if strings.HasSuffix(path, "/mj/submit/modal") {
// midjourney plus // midjourney plus
relayMode = RelayModeMidjourneyModal relayMode = RelayModeMidjourneyModal
} else if strings.HasPrefix(path, "/mj/submit/shorten") { } else if strings.HasSuffix(path, "/mj/submit/shorten") {
// midjourney plus // midjourney plus
relayMode = RelayModeMidjourneyShorten relayMode = RelayModeMidjourneyShorten
} else if strings.HasPrefix(path, "/mj/insight-face/swap") { } else if strings.HasSuffix(path, "/mj/insight-face/swap") {
// midjourney plus // midjourney plus
relayMode = RelayModeSwapFace relayMode = RelayModeSwapFace
} else if strings.HasPrefix(path, "/mj/submit/imagine") { } else if strings.HasSuffix(path, "/mj/submit/imagine") {
relayMode = RelayModeMidjourneyImagine relayMode = RelayModeMidjourneyImagine
} else if strings.HasPrefix(path, "/mj/submit/blend") { } else if strings.HasSuffix(path, "/mj/submit/blend") {
relayMode = RelayModeMidjourneyBlend relayMode = RelayModeMidjourneyBlend
} else if strings.HasPrefix(path, "/mj/submit/describe") { } else if strings.HasSuffix(path, "/mj/submit/describe") {
relayMode = RelayModeMidjourneyDescribe relayMode = RelayModeMidjourneyDescribe
} else if strings.HasPrefix(path, "/mj/notify") { } else if strings.HasSuffix(path, "/mj/notify") {
relayMode = RelayModeMidjourneyNotify relayMode = RelayModeMidjourneyNotify
} else if strings.HasPrefix(path, "/mj/submit/change") { } else if strings.HasSuffix(path, "/mj/submit/change") {
relayMode = RelayModeMidjourneyChange relayMode = RelayModeMidjourneyChange
} else if strings.HasPrefix(path, "/mj/submit/simple-change") { } else if strings.HasSuffix(path, "/mj/submit/simple-change") {
relayMode = RelayModeMidjourneyChange relayMode = RelayModeMidjourneyChange
} else if strings.HasSuffix(path, "/fetch") { } else if strings.HasSuffix(path, "/fetch") {
relayMode = RelayModeMidjourneyTaskFetch relayMode = RelayModeMidjourneyTaskFetch