From e154d2e340179a574d0920c6f0521675c5c1dc40 Mon Sep 17 00:00:00 2001 From: Rodrigo Lazo Paz Date: Wed, 24 Sep 2025 11:17:33 -0400 Subject: [PATCH] [AI] Update model name's in tests Update all model used in test to point to the 2.5 family Internal b/438111636 --- .../kotlin/com/google/firebase/ai/AIModels.kt | 4 ++-- .../kotlin/com/google/firebase/ai/ToolTests.kt | 2 +- .../google/firebase/ai/GenerativeModelTesting.kt | 8 ++++---- .../firebase/ai/common/APIControllerTests.kt | 16 ++++++++-------- 4 files changed, 15 insertions(+), 15 deletions(-) diff --git a/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/AIModels.kt b/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/AIModels.kt index d544f8df4e0..5b51b47fc0c 100644 --- a/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/AIModels.kt +++ b/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/AIModels.kt @@ -51,12 +51,12 @@ class AIModels { flash2Model = FirebaseAI.getInstance(app!!, GenerativeBackend.vertexAI()) .generativeModel( - modelName = "gemini-2.0-flash", + modelName = "gemini-2.5-flash", ) flash2LiteModel = FirebaseAI.getInstance(app!!, GenerativeBackend.vertexAI()) .generativeModel( - modelName = "gemini-2.0-flash-lite", + modelName = "gemini-2.5-flash-lite", ) } } diff --git a/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt b/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt index 5aef0f429c6..856154c161f 100644 --- a/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt +++ b/firebase-ai/src/androidTest/kotlin/com/google/firebase/ai/ToolTests.kt @@ -299,7 +299,7 @@ class ToolTests { val model = FirebaseAI.getInstance(app(), GenerativeBackend.vertexAI()) .generativeModel( - modelName = "gemini-2.0-flash", + modelName = "gemini-2.5-flash", toolConfig = ToolConfig( functionCallingConfig = FunctionCallingConfig(FunctionCallingConfig.Mode.ANY) diff --git a/firebase-ai/src/test/java/com/google/firebase/ai/GenerativeModelTesting.kt b/firebase-ai/src/test/java/com/google/firebase/ai/GenerativeModelTesting.kt index 8301f48d968..3a8b6473612 100644 --- a/firebase-ai/src/test/java/com/google/firebase/ai/GenerativeModelTesting.kt +++ b/firebase-ai/src/test/java/com/google/firebase/ai/GenerativeModelTesting.kt @@ -72,7 +72,7 @@ internal class GenerativeModelTesting { val apiController = APIController( "super_cool_test_key", - "gemini-1.5-flash", + "gemini-2.5-flash", RequestOptions(timeout = 5.seconds, endpoint = "https://my.custom.endpoint"), mockEngine, TEST_CLIENT_ID, @@ -84,7 +84,7 @@ internal class GenerativeModelTesting { val generativeModel = GenerativeModel( - "gemini-1.5-flash", + "gemini-2.5-flash", systemInstruction = content { text("system instruction") }, controller = apiController ) @@ -120,7 +120,7 @@ internal class GenerativeModelTesting { val apiController = APIController( "super_cool_test_key", - "gemini-1.5-flash", + "gemini-2.5-flash", RequestOptions(), mockEngine, TEST_CLIENT_ID, @@ -133,7 +133,7 @@ internal class GenerativeModelTesting { // Creating the val generativeModel = GenerativeModel( - "projects/PROJECTID/locations/INVALID_LOCATION/publishers/google/models/gemini-1.5-flash", + "projects/PROJECTID/locations/INVALID_LOCATION/publishers/google/models/gemini-2.5-flash", controller = apiController ) diff --git a/firebase-ai/src/test/java/com/google/firebase/ai/common/APIControllerTests.kt b/firebase-ai/src/test/java/com/google/firebase/ai/common/APIControllerTests.kt index c0e866e794a..c2e5dd591c7 100644 --- a/firebase-ai/src/test/java/com/google/firebase/ai/common/APIControllerTests.kt +++ b/firebase-ai/src/test/java/com/google/firebase/ai/common/APIControllerTests.kt @@ -112,7 +112,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, "genai-android/${BuildConfig.VERSION_NAME}", @@ -142,7 +142,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(timeout = 5.seconds, endpoint = "https://my.custom.endpoint"), mockEngine, TEST_CLIENT_ID, @@ -172,7 +172,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, TEST_CLIENT_ID, @@ -199,7 +199,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, TEST_CLIENT_ID, @@ -227,7 +227,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, TEST_CLIENT_ID, @@ -319,7 +319,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, TEST_CLIENT_ID, @@ -356,7 +356,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, TEST_CLIENT_ID, @@ -382,7 +382,7 @@ internal class RequestFormatTests { val controller = APIController( "super_cool_test_key", - "gemini-pro-1.5", + "gemini-pro-2.5", RequestOptions(), mockEngine, TEST_CLIENT_ID,