Merge pull request #1626 from firebase/tanzim.vertex

Update Vertex AI quickstart to use model 1.5-flash

Author: rlazo

vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt

@@ -43,29 +43,29 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory {
         return with(viewModelClass) {
             when {
                 isAssignableFrom(SummarizeViewModel::class.java) -> {
-                    // Initialize a GenerativeModel with the `gemini-pro` AI model
+                    // Initialize a GenerativeModel with the `gemini-flash` AI model
                     // for text generation
                     val generativeModel = Firebase.vertexAI.generativeModel(
-                        modelName = "gemini-1.5-pro-preview-0409",
+                        modelName = "gemini-1.5-flash-preview-0514",
                         generationConfig = config
                     )
                     SummarizeViewModel(generativeModel)
                 }
 
                 isAssignableFrom(PhotoReasoningViewModel::class.java) -> {
-                    // Initialize a GenerativeModel with the `gemini-pro` AI model
+                    // Initialize a GenerativeModel with the `gemini-flash` AI model
                     // for multimodal text generation
                     val generativeModel = Firebase.vertexAI.generativeModel(
-                        modelName = "gemini-1.5-pro-preview-0409",
+                        modelName = "gemini-1.5-flash-preview-0514",
                         generationConfig = config
                     )
                     PhotoReasoningViewModel(generativeModel)
                 }
 
                 isAssignableFrom(ChatViewModel::class.java) -> {
-                    // Initialize a GenerativeModel with the `gemini-pro` AI model for chat
+                    // Initialize a GenerativeModel with the `gemini-flash` AI model for chat
                     val generativeModel = Firebase.vertexAI.generativeModel(
-                        modelName = "gemini-1.5-pro-preview-0409",
+                        modelName = "gemini-1.5-flash-preview-0514",
                         generationConfig = config
                     )
                     ChatViewModel(generativeModel)
@@ -89,7 +89,7 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory {
 
                     // Initialize a GenerativeModel with the `gemini-pro` AI model for function calling chat
                     val generativeModel = Firebase.vertexAI.generativeModel(
-                        modelName = "gemini-1.5-pro-preview-0409",
+                        modelName = "gemini-1.5-pro-preview-0514",
                         generationConfig = config,
                         tools = tools
                     )