flash upgrade:

nohe427 · Jun 14, 2024 · 5e98ae3 · 5e98ae3
1 parent 3dca4c6
commit 5e98ae3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/genai/vertex.go b/genai/vertex.go
@@ -35,7 +35,8 @@ type ClientOptions struct {
 }
 
 // var MODEL_NAME = "gemini-pro"
-var MODEL_NAME = "gemini-1.5-pro-preview-0409"
+// var MODEL_NAME = "gemini-1.5-pro-preview-0409"
+var MODEL_NAME = "gemini-1.5-flash-001"
 
 func NewVertexClient(opts ClientOptions) *VertexClient {
 	return &VertexClient{Opts: opts}