Sfoglia il codice sorgente

feat: add support for Midjourney relay mode based on path prefix

CaIon 7 mesi fa
parent
commit
9b0913343c
1 ha cambiato i file con 2 aggiunte e 0 eliminazioni
  1. 2 0
      relay/constant/relay_mode.go

+ 2 - 0
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
 }