Add totalBillableCharacters to CountTokensResponse

.changes/baseball-comfort-dock-appliance.json

@@ -0,0 +1 @@
+{"type":"MAJOR","changes":["Make totalTokens in CountTokensResponse nullable"]}

common/src/main/kotlin/com/google/ai/client/generativeai/common/Response.kt

@@ -30,7 +30,9 @@ data class GenerateContentResponse(
   val usageMetadata: UsageMetadata? = null
 ) : Response
 
-@Serializable data class CountTokensResponse(val totalTokens: Int) : Response
+@Serializable
+data class CountTokensResponse(val totalTokens: Int, val totalBillableCharacters: Int? = null) :
+  Response
 
 @Serializable data class GRpcErrorResponse(val error: GRpcError) : Response
 

common/src/test/java/com/google/ai/client/generativeai/common/APIControllerTests.kt

@@ -160,7 +160,8 @@ internal class RequestFormatTests {
 
   @Test
   fun `countTokenRequest doesn't include the model name`() = doBlocking {
-    val response = JSON.encodeToString(CountTokensResponse(totalTokens = 10))
+    val response =
+      JSON.encodeToString(CountTokensResponse(totalTokens = 10, totalBillableCharacters = 10))
     val mockEngine = MockEngine {
       respond(response, HttpStatusCode.OK, headersOf(HttpHeaders.ContentType, "application/json"))
     }