Просмотр исходного кода

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

CaIon 6 месяцев назад
Родитель
Сommit
9b0913343c
1 измененных файлов с 2 добавлено и 0 удалено
  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
 }