feat: gemini Embeddings support

2025-03-10 23:32:06 +08:00
parent 5c794a362c
commit 9a878df8c0
4 changed files with 113 additions and 2 deletions
@@ -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{