refactor: use Firebase AI Logic SDK

Author: thatfiredev

app/build.gradle.kts

@@ -121,7 +121,7 @@ dependencies {
 
     implementation(platform(libs.firebase.bom))
     implementation(libs.firebase.crashlytics)
-    implementation(libs.firebase.vertexai)
+    implementation(libs.firebase.ai)
     implementation(libs.firebase.app.check)
     implementation(libs.firebase.config)
     implementation(libs.firebase.appcheck.debug)

core/network/build.gradle.kts

@@ -71,7 +71,7 @@ dependencies {
     implementation(libs.coil.compose.http)
     implementation(libs.coil.gif)
     implementation(platform(libs.firebase.bom))
-    implementation(libs.firebase.vertexai)
+    implementation(libs.firebase.ai)
     implementation(libs.firebase.analytics) {
         exclude(group = "com.google.guava")
     }

core/network/src/main/java/com/android/developers/androidify/vertexai/FirebaseAiDataSource.kt

@@ -22,19 +22,20 @@ import com.android.developers.androidify.model.ImageValidationError
 import com.android.developers.androidify.model.ValidatedDescription
 import com.android.developers.androidify.model.ValidatedImage
 import com.google.firebase.Firebase
-import com.google.firebase.vertexai.GenerativeModel
-import com.google.firebase.vertexai.ImagenModel
-import com.google.firebase.vertexai.type.HarmBlockThreshold
-import com.google.firebase.vertexai.type.HarmCategory
-import com.google.firebase.vertexai.type.ImagenPersonFilterLevel
-import com.google.firebase.vertexai.type.ImagenSafetyFilterLevel
-import com.google.firebase.vertexai.type.ImagenSafetySettings
-import com.google.firebase.vertexai.type.PublicPreviewAPI
-import com.google.firebase.vertexai.type.SafetySetting
-import com.google.firebase.vertexai.type.Schema
-import com.google.firebase.vertexai.type.content
-import com.google.firebase.vertexai.type.generationConfig
-import com.google.firebase.vertexai.vertexAI
+import com.google.firebase.ai.GenerativeModel
+import com.google.firebase.ai.ImagenModel
+import com.google.firebase.ai.type.HarmBlockThreshold
+import com.google.firebase.ai.type.HarmCategory
+import com.google.firebase.ai.type.ImagenPersonFilterLevel
+import com.google.firebase.ai.type.ImagenSafetyFilterLevel
+import com.google.firebase.ai.type.ImagenSafetySettings
+import com.google.firebase.ai.type.PublicPreviewAPI
+import com.google.firebase.ai.type.SafetySetting
+import com.google.firebase.ai.type.Schema
+import com.google.firebase.ai.type.content
+import com.google.firebase.ai.type.generationConfig
+import com.google.firebase.ai.ai
+import com.google.firebase.ai.type.GenerativeBackend
 import kotlinx.serialization.json.Json
 import kotlinx.serialization.json.booleanOrNull
 import kotlinx.serialization.json.decodeFromJsonElement
@@ -73,7 +74,7 @@ class FirebaseAiDataSourceImpl @Inject constructor(
     private val remoteConfigDataSource: RemoteConfigDataSource,
 ) : FirebaseAiDataSource {
     private fun createGenerativeTextModel(jsonSchema: Schema, temperature: Float? = null): GenerativeModel {
-        return Firebase.vertexAI.generativeModel(
+        return Firebase.ai(backend = GenerativeBackend.vertexAI()).generativeModel(
             modelName = remoteConfigDataSource.textModelName(),
             generationConfig = generationConfig {
                 responseMimeType = "application/json"
@@ -91,7 +92,7 @@ class FirebaseAiDataSourceImpl @Inject constructor(
     }
 
     private fun createGenerativeImageModel(): ImagenModel {
-        return Firebase.vertexAI.imagenModel(
+        return Firebase.ai.imagenModel(
             remoteConfigDataSource.imageModelName(),
             safetySettings =
             ImagenSafetySettings(

gradle/libs.versions.toml

@@ -25,7 +25,7 @@ coreKtx = "1.15.0"
 coreSplashscreen = "1.0.1"
 crashlytics = "3.0.3"
 espressoCore = "3.6.1"
-firebaseBom = "33.13.0"
+firebaseBom = "33.14.0"
 firebaseConfigKtx = "22.1.1"
 googleServices = "4.4.2"
 hiltAndroid = "2.56.2"
@@ -107,7 +107,7 @@ firebase-bom = { module = "com.google.firebase:firebase-bom", version.ref = "fir
 firebase-config = { module = "com.google.firebase:firebase-config" }
 firebase-config-ktx = { group = "com.google.firebase", name = "firebase-config-ktx", version.ref = "firebaseConfigKtx" }
 firebase-crashlytics = { module = "com.google.firebase:firebase-crashlytics" }
-firebase-vertexai = { module = "com.google.firebase:firebase-vertexai" }
+firebase-ai = { module = "com.google.firebase:firebase-ai" }
 hilt-android = { group = "com.google.dagger", name = "hilt-android", version.ref = "hiltAndroid" }
 hilt-android-testing = { group = "com.google.dagger", name = "hilt-android-testing", version.ref = "hiltAndroid" }
 hilt-compiler = { group = "com.google.dagger", name = "hilt-compiler", version.ref = "hiltAndroid" }