From ac99336e2b22f0aad13a4b9675af0915a61c8f12 Mon Sep 17 00:00:00 2001 From: Andrew Heard Date: Thu, 16 May 2024 15:46:04 -0400 Subject: [PATCH] [Vertex AI] Update samples/docs to use `gemini-1.5-flash-preview-0514` --- .../ChatSample/ViewModels/ConversationViewModel.swift | 2 +- .../ViewModels/FunctionCallingViewModel.swift | 2 +- .../ViewModels/PhotoReasoningViewModel.swift | 2 +- .../ViewModels/SummarizeViewModel.swift | 2 +- FirebaseVertexAI/Sources/GenerativeModel.swift | 3 +-- FirebaseVertexAI/Sources/VertexAI.swift | 10 +++++++--- 6 files changed, 12 insertions(+), 9 deletions(-) diff --git a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift index ea63f6af95b..d8020b3ac41 100644 --- a/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift +++ b/FirebaseVertexAI/Sample/ChatSample/ViewModels/ConversationViewModel.swift @@ -36,7 +36,7 @@ class ConversationViewModel: ObservableObject { private var chatTask: Task? init() { - model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.5-pro-preview-0409") + model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.5-flash-preview-0514") chat = model.startChat() } diff --git a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift index 43a267e0c09..3b6074b8cd5 100644 --- a/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift +++ b/FirebaseVertexAI/Sample/FunctionCallingSample/ViewModels/FunctionCallingViewModel.swift @@ -39,7 +39,7 @@ class FunctionCallingViewModel: ObservableObject { init() { model = VertexAI.vertexAI().generativeModel( - modelName: "gemini-1.5-pro-preview-0409", + modelName: "gemini-1.5-flash-preview-0514", tools: [Tool(functionDeclarations: [ FunctionDeclaration( name: "get_exchange_rate", diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift index 3568d6cb335..ec0f3a7c6b3 100644 --- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift +++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift @@ -44,7 +44,7 @@ class PhotoReasoningViewModel: ObservableObject { private var model: GenerativeModel? init() { - model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.5-pro-preview-0409") + model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.5-flash-preview-0514") } func reason() async { diff --git a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift index 95e18d72114..77e815ea77d 100644 --- a/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift +++ b/FirebaseVertexAI/Sample/GenerativeAITextSample/ViewModels/SummarizeViewModel.swift @@ -32,7 +32,7 @@ class SummarizeViewModel: ObservableObject { private var model: GenerativeModel? init() { - model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.5-pro-preview-0409") + model = VertexAI.vertexAI().generativeModel(modelName: "gemini-1.5-flash-preview-0514") } func summarize(inputText: String) async { diff --git a/FirebaseVertexAI/Sources/GenerativeModel.swift b/FirebaseVertexAI/Sources/GenerativeModel.swift index 1a5fc8c1533..e0778a4f228 100644 --- a/FirebaseVertexAI/Sources/GenerativeModel.swift +++ b/FirebaseVertexAI/Sources/GenerativeModel.swift @@ -50,8 +50,7 @@ public final class GenerativeModel { /// Initializes a new remote model with the given parameters. /// /// - Parameters: - /// - name: The name of the model to use, for example `"gemini-1.0-pro"`; see - /// [Gemini models](https://ai.google.dev/models/gemini) for a list of supported model names. + /// - name: The name of the model to use, for example `"gemini-1.0-pro"`. /// - apiKey: The API key for your project. /// - generationConfig: The content generation parameters your model should use. /// - safetySettings: A value describing what types of harmful content your model should allow. diff --git a/FirebaseVertexAI/Sources/VertexAI.swift b/FirebaseVertexAI/Sources/VertexAI.swift index 3be9043d07c..5069476b6ad 100644 --- a/FirebaseVertexAI/Sources/VertexAI.swift +++ b/FirebaseVertexAI/Sources/VertexAI.swift @@ -60,10 +60,14 @@ public class VertexAI: NSObject { /// Initializes a generative model with the given parameters. /// + /// - Note: Refer to [Gemini models](https://firebase.google.com/docs/vertex-ai/gemini-models) for + /// guidance on choosing an appropriate model for your use case. + /// /// - Parameters: - /// - modelName: The name of the model to use, for example `"gemini-1.0-pro"`; see - /// [Gemini models](https://firebase.google.com/docs/vertex-ai/gemini-model#available-models) - /// for a list of supported model names. + /// - modelName: The name of the model to use, for example `"gemini-1.5-flash-preview-0514"`; + /// see [available model names + /// ](https://firebase.google.com/docs/vertex-ai/gemini-models#available-model-names) for a + /// list of supported model names. /// - generationConfig: The content generation parameters your model should use. /// - safetySettings: A value describing what types of harmful content your model should allow. /// - tools: A list of ``Tool`` objects that the model may use to generate the next response.