@@ -22,19 +22,20 @@ import com.android.developers.androidify.model.ImageValidationError
22
22
import com.android.developers.androidify.model.ValidatedDescription
23
23
import com.android.developers.androidify.model.ValidatedImage
24
24
import com.google.firebase.Firebase
25
- import com.google.firebase.vertexai.GenerativeModel
26
- import com.google.firebase.vertexai.ImagenModel
27
- import com.google.firebase.vertexai.type.HarmBlockThreshold
28
- import com.google.firebase.vertexai.type.HarmCategory
29
- import com.google.firebase.vertexai.type.ImagenPersonFilterLevel
30
- import com.google.firebase.vertexai.type.ImagenSafetyFilterLevel
31
- import com.google.firebase.vertexai.type.ImagenSafetySettings
32
- import com.google.firebase.vertexai.type.PublicPreviewAPI
33
- import com.google.firebase.vertexai.type.SafetySetting
34
- import com.google.firebase.vertexai.type.Schema
35
- import com.google.firebase.vertexai.type.content
36
- import com.google.firebase.vertexai.type.generationConfig
37
- import com.google.firebase.vertexai.vertexAI
25
+ import com.google.firebase.ai.GenerativeModel
26
+ import com.google.firebase.ai.ImagenModel
27
+ import com.google.firebase.ai.type.HarmBlockThreshold
28
+ import com.google.firebase.ai.type.HarmCategory
29
+ import com.google.firebase.ai.type.ImagenPersonFilterLevel
30
+ import com.google.firebase.ai.type.ImagenSafetyFilterLevel
31
+ import com.google.firebase.ai.type.ImagenSafetySettings
32
+ import com.google.firebase.ai.type.PublicPreviewAPI
33
+ import com.google.firebase.ai.type.SafetySetting
34
+ import com.google.firebase.ai.type.Schema
35
+ import com.google.firebase.ai.type.content
36
+ import com.google.firebase.ai.type.generationConfig
37
+ import com.google.firebase.ai.ai
38
+ import com.google.firebase.ai.type.GenerativeBackend
38
39
import kotlinx.serialization.json.Json
39
40
import kotlinx.serialization.json.booleanOrNull
40
41
import kotlinx.serialization.json.decodeFromJsonElement
@@ -73,7 +74,7 @@ class FirebaseAiDataSourceImpl @Inject constructor(
73
74
private val remoteConfigDataSource : RemoteConfigDataSource ,
74
75
) : FirebaseAiDataSource {
75
76
private fun createGenerativeTextModel (jsonSchema : Schema , temperature : Float? = null): GenerativeModel {
76
- return Firebase .vertexAI.generativeModel(
77
+ return Firebase .ai(backend = GenerativeBackend . vertexAI()) .generativeModel(
77
78
modelName = remoteConfigDataSource.textModelName(),
78
79
generationConfig = generationConfig {
79
80
responseMimeType = " application/json"
@@ -91,7 +92,7 @@ class FirebaseAiDataSourceImpl @Inject constructor(
91
92
}
92
93
93
94
private fun createGenerativeImageModel (): ImagenModel {
94
- return Firebase .vertexAI.imagenModel(
95
+ return Firebase .ai(backend = GenerativeBackend . vertexAI()) .imagenModel(
95
96
remoteConfigDataSource.imageModelName(),
96
97
safetySettings =
97
98
ImagenSafetySettings (
0 commit comments