feature: support vertex open source models

This commit is contained in:
Seefs
2025-11-16 14:23:11 +08:00
parent e07347ac53
commit ab36de3725

View File

@@ -76,7 +76,9 @@ func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInf
func (a *Adaptor) Init(info *relaycommon.RelayInfo) { func (a *Adaptor) Init(info *relaycommon.RelayInfo) {
if strings.HasPrefix(info.UpstreamModelName, "claude") { if strings.HasPrefix(info.UpstreamModelName, "claude") {
a.RequestMode = RequestModeClaude a.RequestMode = RequestModeClaude
} else if strings.Contains(info.UpstreamModelName, "llama") { } else if strings.Contains(info.UpstreamModelName, "llama") ||
// open source models
strings.Contains(info.UpstreamModelName, "-maas") {
a.RequestMode = RequestModeLlama a.RequestMode = RequestModeLlama
} else { } else {
a.RequestMode = RequestModeGemini a.RequestMode = RequestModeGemini