Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

给 vertexai 增加 Gemini 的 embedding 支持

Browse Source
This commit is contained in:
xinghejd 2024-08-27 15:49:30 +08:00 committed by GitHub
parent f9774698e9
commit ac6cf4c841
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
relay/adaptor/vertexai/adaptor.go
View File

@ -62,11 +62,16 @@ func (a *Adaptor) GetChannelName() string {
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) { func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
suffix := "" suffix := ""
if strings.HasPrefix(meta.ActualModelName, "gemini") { if strings.HasPrefix(meta.ActualModelName, "gemini") {
switch meta.Mode {
case relaymode.Embeddings:
suffix = "batchEmbedContents"
default:
if meta.IsStream { if meta.IsStream {
suffix = "streamGenerateContent?alt=sse" suffix = "streamGenerateContent?alt=sse"
} else { } else {
suffix = "generateContent" suffix = "generateContent"
} }
}
} else { } else {
if meta.IsStream { if meta.IsStream {
suffix = "streamRawPredict?alt=sse" suffix = "streamRawPredict?alt=sse"

2
relay/adaptor/vertexai/gemini/adapter.go
View File

@ -15,7 +15,7 @@ import (
) )
var ModelList = []string{ var ModelList = []string{
"gemini-1.5-pro-001", "gemini-1.5-flash-001", "gemini-pro", "gemini-pro-vision", "gemini-1.5-pro-001", "gemini-1.5-flash-001", "gemini-pro", "gemini-pro-vision","text-embedding-004"
} }
type Adaptor struct { type Adaptor struct {