add tool config

Author: VinayGuthal

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/LiveGenerativeModel.kt

@@ -22,7 +22,6 @@ import com.google.firebase.auth.internal.InternalAuthProvider
 import com.google.firebase.vertexai.common.APIController
 import com.google.firebase.vertexai.common.AppCheckHeaderProvider
 import com.google.firebase.vertexai.type.BidiGenerateContentClientMessage
-import com.google.firebase.vertexai.type.BidiGenerateContentSetup
 import com.google.firebase.vertexai.type.Content
 import com.google.firebase.vertexai.type.LiveGenerationConfig
 import com.google.firebase.vertexai.type.LiveSession
@@ -91,14 +90,16 @@ internal constructor(
     val client = HttpClient(OkHttp) { install(WebSockets) }
 
     val bidiEndPoint = this.controller.getBidiEndpoint(location)
-    val setup =
-      BidiGenerateContentSetup(
-        this.modelName,
-        this.config?.toInternal(),
-        this.tools?.map { it.toInternal() },
-        this.systemInstruction?.toInternal()
-      )
-    val data: String = Json.encodeToString(BidiGenerateContentClientMessage(setup))
+    val clientMessage =
+      BidiGenerateContentClientMessage(
+          this.modelName,
+          this.config?.toInternal(),
+          this.tools?.map { it.toInternal() },
+          this.toolConfig?.toInternal(),
+          this.systemInstruction?.toInternal()
+        )
+        .toInternal()
+    val data: String = Json.encodeToString(clientMessage)
     val webSession = client.webSocketSession(bidiEndPoint)
     webSession.send(Frame.Text(data))
     var shouldReturn = false

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/BidiGenerateContentSetup.kt

@@ -2,13 +2,34 @@ package com.google.firebase.vertexai.type
 
 import kotlinx.serialization.Serializable
 
-@Serializable
-internal data class BidiGenerateContentSetup(
+internal class BidiGenerateContentClientMessage(
   val model: String,
   val generationConfig: LiveGenerationConfig.Internal?,
   val tools: List<Tool.Internal>?,
+  val toolConfig: ToolConfig.Internal?,
   val systemInstruction: Content.Internal?
-)
+) {
 
-@Serializable
-internal data class BidiGenerateContentClientMessage(val setup: BidiGenerateContentSetup)
+  @Serializable
+  internal class Internal(val setup: BidiGenerateContentSetup) {
+    @Serializable
+    internal data class BidiGenerateContentSetup(
+      val model: String,
+      val generationConfig: LiveGenerationConfig.Internal?,
+      val tools: List<Tool.Internal>?,
+      val toolConfig: ToolConfig.Internal?,
+      val systemInstruction: Content.Internal?
+    )
+  }
+
+  fun toInternal() =
+    Internal(
+      Internal.BidiGenerateContentSetup(
+        model,
+        generationConfig,
+        tools,
+        toolConfig,
+        systemInstruction
+      )
+    )
+}