From ac6cf4c841dd2eefa1662cd2f06697b9559d3134 Mon Sep 17 00:00:00 2001
From: xinghejd <31512683+xinghejd@users.noreply.github.com>
Date: Tue, 27 Aug 2024 15:49:30 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=99=20vertexai=20=E5=A2=9E=E5=8A=A0=20Gem?=
 =?UTF-8?q?ini=20=E7=9A=84=20embedding=20=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 relay/adaptor/vertexai/adaptor.go        | 13 +++++++++----
 relay/adaptor/vertexai/gemini/adapter.go |  2 +-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/relay/adaptor/vertexai/adaptor.go b/relay/adaptor/vertexai/adaptor.go
index 3fab4a45..6187a29a 100644
--- a/relay/adaptor/vertexai/adaptor.go
+++ b/relay/adaptor/vertexai/adaptor.go
@@ -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 {
diff --git a/relay/adaptor/vertexai/gemini/adapter.go b/relay/adaptor/vertexai/gemini/adapter.go
index 43e6cbcd..0ba74668 100644
--- a/relay/adaptor/vertexai/gemini/adapter.go
+++ b/relay/adaptor/vertexai/gemini/adapter.go
@@ -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 {