Skip to content

Commit 9b9b8f8

Browse files
authored
[AI] Update model name's in tests (#7409)
Update all model used in test to point to the 2.5 family Internal b/438111636
1 parent 482ac0e commit 9b9b8f8

File tree

4 files changed

+15
-15
lines changed

4 files changed

+15
-15
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,12 +51,12 @@ class AIModels {
5151
flash2Model =
5252
FirebaseAI.getInstance(app!!, GenerativeBackend.vertexAI())
5353
.generativeModel(
54-
modelName = "gemini-2.0-flash",
54+
modelName = "gemini-2.5-flash",
5555
)
5656
flash2LiteModel =
5757
FirebaseAI.getInstance(app!!, GenerativeBackend.vertexAI())
5858
.generativeModel(
59-
modelName = "gemini-2.0-flash-lite",
59+
modelName = "gemini-2.5-flash-lite",
6060
)
6161
}
6262
}

firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,7 @@ class ToolTests {
299299
val model =
300300
FirebaseAI.getInstance(app(), GenerativeBackend.vertexAI())
301301
.generativeModel(
302-
modelName = "gemini-2.0-flash",
302+
modelName = "gemini-2.5-flash",
303303
toolConfig =
304304
ToolConfig(
305305
functionCallingConfig = FunctionCallingConfig(FunctionCallingConfig.Mode.ANY)

firebase-ai/src/test/java/com/google/firebase/ai/GenerativeModelTesting.kt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ internal class GenerativeModelTesting {
7272
val apiController =
7373
APIController(
7474
"super_cool_test_key",
75-
"gemini-1.5-flash",
75+
"gemini-2.5-flash",
7676
RequestOptions(timeout = 5.seconds, endpoint = "https://my.custom.endpoint"),
7777
mockEngine,
7878
TEST_CLIENT_ID,
@@ -84,7 +84,7 @@ internal class GenerativeModelTesting {
8484

8585
val generativeModel =
8686
GenerativeModel(
87-
"gemini-1.5-flash",
87+
"gemini-2.5-flash",
8888
systemInstruction = content { text("system instruction") },
8989
controller = apiController
9090
)
@@ -120,7 +120,7 @@ internal class GenerativeModelTesting {
120120
val apiController =
121121
APIController(
122122
"super_cool_test_key",
123-
"gemini-1.5-flash",
123+
"gemini-2.5-flash",
124124
RequestOptions(),
125125
mockEngine,
126126
TEST_CLIENT_ID,
@@ -133,7 +133,7 @@ internal class GenerativeModelTesting {
133133
// Creating the
134134
val generativeModel =
135135
GenerativeModel(
136-
"projects/PROJECTID/locations/INVALID_LOCATION/publishers/google/models/gemini-1.5-flash",
136+
"projects/PROJECTID/locations/INVALID_LOCATION/publishers/google/models/gemini-2.5-flash",
137137
controller = apiController
138138
)
139139

firebase-ai/src/test/java/com/google/firebase/ai/common/APIControllerTests.kt

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ internal class RequestFormatTests {
112112
val controller =
113113
APIController(
114114
"super_cool_test_key",
115-
"gemini-pro-1.5",
115+
"gemini-pro-2.5",
116116
RequestOptions(),
117117
mockEngine,
118118
"genai-android/${BuildConfig.VERSION_NAME}",
@@ -142,7 +142,7 @@ internal class RequestFormatTests {
142142
val controller =
143143
APIController(
144144
"super_cool_test_key",
145-
"gemini-pro-1.5",
145+
"gemini-pro-2.5",
146146
RequestOptions(timeout = 5.seconds, endpoint = "https://my.custom.endpoint"),
147147
mockEngine,
148148
TEST_CLIENT_ID,
@@ -172,7 +172,7 @@ internal class RequestFormatTests {
172172
val controller =
173173
APIController(
174174
"super_cool_test_key",
175-
"gemini-pro-1.5",
175+
"gemini-pro-2.5",
176176
RequestOptions(),
177177
mockEngine,
178178
TEST_CLIENT_ID,
@@ -199,7 +199,7 @@ internal class RequestFormatTests {
199199
val controller =
200200
APIController(
201201
"super_cool_test_key",
202-
"gemini-pro-1.5",
202+
"gemini-pro-2.5",
203203
RequestOptions(),
204204
mockEngine,
205205
TEST_CLIENT_ID,
@@ -227,7 +227,7 @@ internal class RequestFormatTests {
227227
val controller =
228228
APIController(
229229
"super_cool_test_key",
230-
"gemini-pro-1.5",
230+
"gemini-pro-2.5",
231231
RequestOptions(),
232232
mockEngine,
233233
TEST_CLIENT_ID,
@@ -319,7 +319,7 @@ internal class RequestFormatTests {
319319
val controller =
320320
APIController(
321321
"super_cool_test_key",
322-
"gemini-pro-1.5",
322+
"gemini-pro-2.5",
323323
RequestOptions(),
324324
mockEngine,
325325
TEST_CLIENT_ID,
@@ -356,7 +356,7 @@ internal class RequestFormatTests {
356356
val controller =
357357
APIController(
358358
"super_cool_test_key",
359-
"gemini-pro-1.5",
359+
"gemini-pro-2.5",
360360
RequestOptions(),
361361
mockEngine,
362362
TEST_CLIENT_ID,
@@ -382,7 +382,7 @@ internal class RequestFormatTests {
382382
val controller =
383383
APIController(
384384
"super_cool_test_key",
385-
"gemini-pro-1.5",
385+
"gemini-pro-2.5",
386386
RequestOptions(),
387387
mockEngine,
388388
TEST_CLIENT_ID,

0 commit comments

Comments
 (0)