11package dev.johnoreilly.vertexai
22
33import com.google.firebase.Firebase
4- import com.google.firebase.vertexai.type.PublicPreviewAPI
5- import com.google.firebase.vertexai.type.Schema
6- import com.google.firebase.vertexai.type.generationConfig
7- import com.google.firebase.vertexai.vertexAI
4+ import com.google.firebase.ai.type.PublicPreviewAPI
5+ import com.google.firebase.ai.type.Schema
6+ import com.google.firebase.ai.type.generationConfig
7+ import com.google.firebase.ai.ai
8+ import com.google.firebase.ai.type.GenerativeBackend
89
910
1011class GenerativeModelAndroid : GenerativeModel {
@@ -18,16 +19,16 @@ class GenerativeModelAndroid : GenerativeModel {
1819 )
1920
2021 override suspend fun generateTextContent (prompt : String ): String? {
21- val generativeModel = Firebase .vertexAI .generativeModel(
22- modelName = " gemini-1 .5-flash"
22+ val generativeModel = Firebase .ai(backend = GenerativeBackend .googleAI()) .generativeModel(
23+ modelName = " gemini-2 .5-flash"
2324 )
2425
2526 return generativeModel.generateContent(prompt).text
2627 }
2728
2829 override suspend fun generateJsonContent (prompt : String ): String? {
29- val generativeModel = Firebase .vertexAI .generativeModel(
30- modelName = " gemini-1 .5-flash" ,
30+ val generativeModel = Firebase .ai .generativeModel(
31+ modelName = " gemini-2 .5-flash" ,
3132 generationConfig = generationConfig {
3233 responseMimeType = " application/json"
3334 responseSchema = jsonSchema
@@ -39,7 +40,7 @@ class GenerativeModelAndroid : GenerativeModel {
3940
4041 @OptIn(PublicPreviewAPI ::class )
4142 override suspend fun generateImage (prompt : String ): ByteArray? {
42- val imageModel = Firebase .vertexAI .imagenModel(
43+ val imageModel = Firebase .ai .imagenModel(
4344 modelName = " imagen-3.0-generate-002"
4445 )
4546 val imageResponse = imageModel.generateImages(prompt)
0 commit comments