Parcourir la source

Merge pull request #2236 from seefs001/feature/vertex-k2

feat: support vertex open source models
Calcium-Ion il y a 3 mois
Parent
commit
ff3225ab44
1 fichiers modifiés avec 3 ajouts et 1 suppressions
  1. 3 1
      relay/channel/vertex/adaptor.go

+ 3 - 1
relay/channel/vertex/adaptor.go

@@ -76,7 +76,9 @@ func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInf
 func (a *Adaptor) Init(info *relaycommon.RelayInfo) {
 	if strings.HasPrefix(info.UpstreamModelName, "claude") {
 		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
 	} else {
 		a.RequestMode = RequestModeGemini