feat: gemini Embeddings support

2025-03-10 23:32:06 +08:00
parent 69db1f1465
commit e1b9f164f9
4 changed files with 113 additions and 2 deletions
--- a/relay/channel/gemini/constant.go
+++ b/relay/channel/gemini/constant.go
@@ -18,6 +18,10 @@ var ModelList = []string{
 	"gemini-2.0-flash-thinking-exp",
 	// imagen models
 	"imagen-3.0-generate-002",
+	// embedding models
+	"gemini-embedding-exp-03-07",
+	"text-embedding-004",
+	"embedding-001",
 }

 var SafetySettingList = []string{