Skip to content

phsl / new-api

Go to a project
Commit 0106d270 by Calcium-Ion Committed by GitHub
Options

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

feat: support vertex open source models
parents 716e4107 f4c0972d
Showing with 3 additions and 1 deletions
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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment