chore(vertexai): Add supported models in tokenizer (#10939)

googleapis · Oct 1, 2024 · 3d9ef48 · 3d9ef48
1 parent a38650e
commit 3d9ef48
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vertexai/genai/tokenizer/tokenizer.go b/vertexai/genai/tokenizer/tokenizer.go
@@ -38,7 +38,9 @@ var supportedModels = map[string]bool{
 	"gemini-1.0-pro-001":   true,
 	"gemini-1.0-pro-002":   true,
 	"gemini-1.5-pro-001":   true,
+	"gemini-1.5-pro-002":   true,
 	"gemini-1.5-flash-001": true,
+	"gemini-1.5-flash-002": true,
 }
 
 // Tokenizer is a local tokenizer for text.