use Vertex AI backend for imagen

thatfiredev · thatfiredev · commit a5cebcfe14d8 · 2025-05-20T20:36:41.000+01:00
diff --git a/core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt b/core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt
@@ -92,7 +92,7 @@ class FirebaseAiDataSourceImpl @Inject constructor(
     }
 
     private fun createGenerativeImageModel(): ImagenModel {
-        return Firebase.ai.imagenModel(
+        return Firebase.ai(backend = GenerativeBackend.vertexAI()).imagenModel(
             remoteConfigDataSource.imageModelName(),
             safetySettings =
             ImagenSafetySettings(

Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,7 @@ class FirebaseAiDataSourceImpl @Inject constructor(`
`92`	`92`	`}`
`93`	`93`
`94`	`94`	`private fun createGenerativeImageModel(): ImagenModel {`
`95`		`- return Firebase.ai.imagenModel(`
	`95`	`+ return Firebase.ai(backend = GenerativeBackend.vertexAI()).imagenModel(`
`96`	`96`	`remoteConfigDataSource.imageModelName(),`
`97`	`97`	`safetySettings =`
`98`	`98`	`ImagenSafetySettings(`