Firebase AI: Unexpected model name format when calling countTokens

[SeanChinJunKai]

[REQUIRED] Step 2: Describe your environment

• Android Studio version: Android Studio Meerkat Feature Drop | 2024.3.2
• Firebase Component: FirebaseAI
• Component version: implementation(platform("com.google.firebase:firebase-bom:33.14.0"))
  implementation("com.google.firebase:firebase-ai")

[REQUIRED] Step 3: Describe the problem

I am getting the following exception when using googleAI()

com.google.firebase.ai.type.ServerException: * CountTokensRequest.generate_content_request.model: unexpected model name format

Steps to reproduce:

1. Create a android project with Empty Activity
2. Add the following code into the MainActivity

Relevant Code:

// This does not work
 lifecycleScope.launch {
        val generativeModel = Firebase.ai(backend = GenerativeBackend.googleAI())
            .generativeModel("gemini-2.0-flash")
        val response = generativeModel.countTokens("Write a story about a magic backpack.")
        Log.i("firebase-ai", "Total Tokens: ${response.totalTokens}")
        Log.i("firebase-ai", "Total Billable Characters: ${response.totalBillableCharacters}")
}


// This works
lifecycleScope.launch {
        val generativeModel = Firebase.ai(backend = GenerativeBackend.vertexAI())
            .generativeModel("gemini-2.0-flash")
        val response = generativeModel.countTokens("Write a story about a magic backpack.")
        Log.i("firebase-ai", "Total Tokens: ${response.totalTokens}")
        Log.i("firebase-ai", "Total Billable Characters: ${response.totalBillableCharacters}")

 }

[google-oss-bot]

I couldn't figure out how to label this issue, so I've labeled it for a human to triage. Hang tight.

[lehcar09]

Hi @SeanChinJunKai, thank you for reaching out and raising this. I was able to reproduce the issue. Per checking with our engineers, this was a confirmed bug. Though, I cannot provide any timeline when the fix will be released. We'll keep you posted for updates. Thanks!

[SeanChinJunKai]

Hi @lehcar09, you are most welcome. I am always happy to help