From 23b64f0e7f34cad3205f6e5598682b6c8f211a00 Mon Sep 17 00:00:00 2001 From: Tanzim Hossain Date: Mon, 27 May 2024 17:08:31 -0400 Subject: [PATCH 1/2] Update Vertex AI quickstart to use model 1.5-flash --- .../vertexai/GenerativeAiViewModelFactory.kt | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt b/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt index e01c9c5c8f..787f38fd98 100644 --- a/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt +++ b/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) @@ -87,9 +87,9 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory { ) ) - // Initialize a GenerativeModel with the `gemini-pro` AI model for function calling chat + // Initialize a GenerativeModel with the `gemini-flash` AI model for function calling chat val generativeModel = Firebase.vertexAI.generativeModel( - modelName = "gemini-1.5-pro-preview-0409", + modelName = "gemini-1.5-flash-preview-0514", generationConfig = config, tools = tools ) From c4ec0cd9b8bbc7c68993b94440066c23c4a7aa5a Mon Sep 17 00:00:00 2001 From: Tanzim Hossain Date: Mon, 27 May 2024 17:43:52 -0400 Subject: [PATCH 2/2] use pro model for function example --- .../quickstart/vertexai/GenerativeAiViewModelFactory.kt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt b/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt index 787f38fd98..af1b7e4946 100644 --- a/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt +++ b/vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt @@ -87,9 +87,9 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory { ) ) - // Initialize a GenerativeModel with the `gemini-flash` AI model for function calling chat + // Initialize a GenerativeModel with the `gemini-pro` AI model for function calling chat val generativeModel = Firebase.vertexAI.generativeModel( - modelName = "gemini-1.5-flash-preview-0514", + modelName = "gemini-1.5-pro-preview-0514", generationConfig = config, tools = tools )