Skip to content

Commit fcfe7b7

Browse files
authored
[Vertex AI] Update integration tests to Gemini 2.0 Flash (#14441)
1 parent 22dce81 commit fcfe7b7

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ final class IntegrationTests: XCTestCase {
5454

5555
vertex = VertexAI.vertexAI()
5656
model = vertex.generativeModel(
57-
modelName: "gemini-1.5-flash",
57+
modelName: "gemini-2.0-flash",
5858
generationConfig: generationConfig,
5959
safetySettings: safetySettings,
6060
tools: [],
@@ -80,7 +80,7 @@ final class IntegrationTests: XCTestCase {
8080
let app = try FirebaseApp.defaultNamedCopy(name: TestAppCheckProviderFactory.notConfiguredName)
8181
addTeardownBlock { await app.delete() }
8282
let vertex = VertexAI.vertexAI(app: app)
83-
let model = vertex.generativeModel(modelName: "gemini-1.5-flash")
83+
let model = vertex.generativeModel(modelName: "gemini-2.0-flash")
8484
let prompt = "Where is Google headquarters located? Answer with the city name only."
8585

8686
do {
@@ -183,7 +183,7 @@ final class IntegrationTests: XCTestCase {
183183
parameters: ["x": .integer(), "y": .integer()]
184184
)
185185
model = vertex.generativeModel(
186-
modelName: "gemini-1.5-flash",
186+
modelName: "gemini-2.0-flash",
187187
tools: [.functionDeclarations([sumDeclaration])],
188188
toolConfig: .init(functionCallingConfig: .any(allowedFunctionNames: ["sum"]))
189189
)
@@ -203,7 +203,7 @@ final class IntegrationTests: XCTestCase {
203203

204204
func testCountTokens_jsonSchema() async throws {
205205
model = vertex.generativeModel(
206-
modelName: "gemini-1.5-flash",
206+
modelName: "gemini-2.0-flash",
207207
generationConfig: GenerationConfig(
208208
responseMIMEType: "application/json",
209209
responseSchema: Schema.object(properties: [
@@ -226,7 +226,7 @@ final class IntegrationTests: XCTestCase {
226226
let app = try FirebaseApp.defaultNamedCopy(name: TestAppCheckProviderFactory.notConfiguredName)
227227
addTeardownBlock { await app.delete() }
228228
let vertex = VertexAI.vertexAI(app: app)
229-
let model = vertex.generativeModel(modelName: "gemini-1.5-flash")
229+
let model = vertex.generativeModel(modelName: "gemini-2.0-flash")
230230
let prompt = "Why is the sky blue?"
231231

232232
do {

0 commit comments

Comments
 (0)