Add GenerationConfig to CountTokenRequest's

firebase-vertexai/CHANGELOG.md

@@ -1,5 +1,5 @@
 # Unreleased
-
+* [feature] `CountTokenRequest` now includes `GenerationConfig` from the model.
 
 # 16.2.0
 * [fixed] Added support for new values sent by the server for `FinishReason` and `BlockReason`.
@@ -69,4 +69,3 @@
 * [feature] Added support for `responseMimeType` in `GenerationConfig`.
 * [changed] Renamed `GoogleGenerativeAIException` to `FirebaseVertexAIException`.
 * [changed] Updated the KDocs for various classes and functions.
-

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/Request.kt

@@ -49,6 +49,7 @@ internal data class CountTokensRequest(
   val contents: List<Content.Internal>? = null,
   val tools: List<Tool.Internal>? = null,
   @SerialName("system_instruction") val systemInstruction: Content.Internal? = null,
+  val generationConfig: GenerationConfig.Internal? = null
 ) : Request {
   companion object {
 
@@ -58,6 +59,7 @@ internal data class CountTokensRequest(
         contents = generateContentRequest.contents,
         tools = generateContentRequest.tools,
         systemInstruction = generateContentRequest.systemInstruction,
+        generationConfig = generateContentRequest.generationConfig,
       )
   }
 }