Przeglądaj źródła

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

CaIon 6 miesięcy temu
rodzic
commit
9b0913343c
1 zmienionych plików z 2 dodań i 0 usunięć
  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
 }