Add new models to ModelList

2024-05-28 19:23:12 +08:00 · 2024-05-28 19:23:12 +08:00 · 3c535f5a6e
commit 3c535f5a6e
parent 5dee10cacf
1 changed files with 1 additions and 1 deletions
--- a/relay/adaptor/gemini/constants.go
+++ b/relay/adaptor/gemini/constants.go
@ -4,5 +4,5 @@ package gemini

 var ModelList = []string{
 	"gemini-pro", "gemini-1.0-pro-001", "gemini-1.5-pro",
-	"gemini-pro-vision", "gemini-1.0-pro-vision-001",
+	"gemini-pro-vision", "gemini-1.0-pro-vision-001", "embedding-001", "text-embedding-004",
 }