From 9b0913343cc1204a745b77f0248b94345c4d4ec8 Mon Sep 17 00:00:00 2001
From: CaIon <i@caion.me>
Date: Sat, 16 Aug 2025 18:26:26 +0800
Subject: [PATCH] feat: add support for Midjourney relay mode based on path
 prefix

---
 relay/constant/relay_mode.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/relay/constant/relay_mode.go b/relay/constant/relay_mode.go
index b1599fd0..85a1b9c5 100644
--- a/relay/constant/relay_mode.go
+++ b/relay/constant/relay_mode.go
@@ -84,6 +84,8 @@ func Path2RelayMode(path string) int {
 		relayMode = RelayModeRealtime
 	} else if strings.HasPrefix(path, "/v1beta/models") || strings.HasPrefix(path, "/v1/models") {
 		relayMode = RelayModeGemini
+	} else if strings.HasPrefix(path, "/mj") {
+		relayMode = Path2RelayModeMidjourney(path)
 	}
 	return relayMode
 }