Skip to content

Commit de675d9

Browse files
authored
Add Gemini 3.5 flash models to testing (#8213)
Gemini 3.5 was recently released and passes existing testing, should be tested for the future.
1 parent 83d0dda commit de675d9

1 file changed

Lines changed: 15 additions & 1 deletion

File tree

  • ai-logic/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai

ai-logic/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/AIModels.kt

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,10 @@ class AIModels {
2828
var app: FirebaseApp? = null
2929
lateinit var vertexAIFlashModel: GenerativeModel
3030
lateinit var vertexAIFlashLiteModel: GenerativeModel
31+
lateinit var vertexAI3_5FlashModel: GenerativeModel
3132
lateinit var googleAIFlashModel: GenerativeModel
3233
lateinit var googleAIFlashLiteModel: GenerativeModel
34+
lateinit var googleAI3_5FlashModel: GenerativeModel
3335
lateinit var vertexAITemplateModel: TemplateGenerativeModel
3436
lateinit var googleAITemplateModel: TemplateGenerativeModel
3537

@@ -41,8 +43,10 @@ class AIModels {
4143
return listOf(
4244
vertexAIFlashModel,
4345
vertexAIFlashLiteModel,
46+
vertexAI3_5FlashModel,
4447
googleAIFlashModel,
45-
googleAIFlashLiteModel
48+
googleAIFlashLiteModel,
49+
googleAI3_5FlashModel,
4650
)
4751
}
4852

@@ -66,6 +70,11 @@ class AIModels {
6670
.generativeModel(
6771
modelName = "gemini-2.5-flash-lite",
6872
)
73+
vertexAI3_5FlashModel =
74+
FirebaseAI.getInstance(app!!, GenerativeBackend.vertexAI("global"))
75+
.generativeModel(
76+
modelName = "gemini-3.5-flash",
77+
)
6978
googleAIFlashModel =
7079
FirebaseAI.getInstance(app!!, GenerativeBackend.googleAI())
7180
.generativeModel(
@@ -76,6 +85,11 @@ class AIModels {
7685
.generativeModel(
7786
modelName = "gemini-2.5-flash-lite",
7887
)
88+
googleAI3_5FlashModel =
89+
FirebaseAI.getInstance(app!!, GenerativeBackend.googleAI())
90+
.generativeModel(
91+
modelName = "gemini-3.5-flash",
92+
)
7993
vertexAITemplateModel =
8094
FirebaseAI.getInstance(app!!, GenerativeBackend.vertexAI()).templateGenerativeModel()
8195
googleAITemplateModel =

0 commit comments

Comments
 (0)