fix serialization

Author: David Motsonashvili

firebase-ai/src/main/kotlin/com/google/firebase/ai/TemplateGenerativeModel.kt

@@ -33,6 +33,9 @@ import com.google.firebase.auth.internal.InternalAuthProvider
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.catch
 import kotlinx.coroutines.flow.map
+import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.jsonObject
+import org.json.JSONObject
 
 /**
  * Represents a multimodal model (like Gemini), capable of generating content based on various
@@ -111,7 +114,9 @@ internal constructor(
       .map { it.toPublic().validate() }
 
   internal fun constructRequest(inputs: Map<String, Any>): TemplateGenerateContentRequest {
-    return TemplateGenerateContentRequest(inputs)
+    return TemplateGenerateContentRequest(
+      Json.parseToJsonElement(JSONObject(inputs).toString()).jsonObject
+    )
   }
 
   private fun GenerateContentResponse.validate() = apply {

firebase-ai/src/main/kotlin/com/google/firebase/ai/TemplateImagenModel.kt

@@ -27,6 +27,9 @@ import com.google.firebase.ai.type.PublicPreviewAPI
 import com.google.firebase.ai.type.RequestOptions
 import com.google.firebase.appcheck.interop.InteropAppCheckTokenProvider
 import com.google.firebase.auth.internal.InternalAuthProvider
+import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.jsonObject
+import org.json.JSONObject
 
 /**
  * Represents a generative model (like Imagen), capable of generating images based a template.
@@ -92,7 +95,9 @@ internal constructor(
   private fun constructTemplateGenerateImageRequest(
     inputs: Map<String, Any>
   ): TemplateGenerateImageRequest {
-    return TemplateGenerateImageRequest(inputs)
+    return TemplateGenerateImageRequest(
+      Json.parseToJsonElement(JSONObject(inputs).toString()).jsonObject
+    )
   }
 
   internal companion object {

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

@@ -28,10 +28,10 @@ import com.google.firebase.ai.type.PublicPreviewAPI
 import com.google.firebase.ai.type.SafetySetting
 import com.google.firebase.ai.type.Tool
 import com.google.firebase.ai.type.ToolConfig
-import kotlinx.serialization.Contextual
 import kotlinx.serialization.ExperimentalSerializationApi
 import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
+import kotlinx.serialization.json.JsonObject
 
 internal interface Request
 
@@ -46,13 +46,9 @@ internal data class GenerateContentRequest(
   @SerialName("system_instruction") val systemInstruction: Content.Internal? = null,
 ) : Request
 
-@Serializable
-internal data class TemplateGenerateContentRequest(val inputs: Map<String, @Contextual Any>) :
-  Request
+@Serializable internal data class TemplateGenerateContentRequest(val inputs: JsonObject) : Request
 
-@Serializable
-internal data class TemplateGenerateImageRequest(val inputs: Map<String, @Contextual Any>) :
-  Request
+@Serializable internal data class TemplateGenerateImageRequest(val inputs: JsonObject) : Request
 
 @Serializable
 internal data class CountTokensRequest(