给 vertexai 增加 Gemini 的 embedding 支持
This commit is contained in:
parent
f9774698e9
commit
ac6cf4c841
@ -62,10 +62,15 @@ func (a *Adaptor) GetChannelName() string {
|
||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
|
||||
suffix := ""
|
||||
if strings.HasPrefix(meta.ActualModelName, "gemini") {
|
||||
if meta.IsStream {
|
||||
suffix = "streamGenerateContent?alt=sse"
|
||||
} else {
|
||||
suffix = "generateContent"
|
||||
switch meta.Mode {
|
||||
case relaymode.Embeddings:
|
||||
suffix = "batchEmbedContents"
|
||||
default:
|
||||
if meta.IsStream {
|
||||
suffix = "streamGenerateContent?alt=sse"
|
||||
} else {
|
||||
suffix = "generateContent"
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if meta.IsStream {
|
||||
|
@ -15,7 +15,7 @@ import (
|
||||
)
|
||||
|
||||
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 {
|
||||
|
Loading…
Reference in New Issue
Block a user