Fix ImagenIntegrationTests

andrewheard · andrewheard · commit 1b6b391f00e1 · 2025-04-24T09:47:14.000-04:00
diff --git a/FirebaseAI/Tests/TestApp/Tests/Integration/ImagenIntegrationTests.swift b/FirebaseAI/Tests/TestApp/Tests/Integration/ImagenIntegrationTests.swift
@@ -40,7 +40,7 @@ struct ImagenIntegrationTests {
 
   init() async throws {
     userID1 = try await TestHelpers.getUserID()
-    vertex = FirebaseAI.vertexAI()
+    vertex = FirebaseAI.firebaseAI(backend: .vertexAI())
     storage = Storage.storage()
   }
 

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@ struct ImagenIntegrationTests {`
`40`	`40`
`41`	`41`	`init() async throws {`
`42`	`42`	`userID1 = try await TestHelpers.getUserID()`
`43`		`- vertex = FirebaseAI.vertexAI()`
	`43`	`+ vertex = FirebaseAI.firebaseAI(backend: .vertexAI())`
`44`	`44`	`storage = Storage.storage()`
`45`	`45`	`}`
`46`	`46`