feature: support vertex open source models
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user