working

Author: VinayGuthal

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

@@ -148,9 +148,6 @@ internal constructor(
           //emit(LiveContentResponse(null, functionContent.toolCall.functionCalls.map { it.toPublic() as FunctionCallPart })))
           break
         } catch (_: Exception){ }
-        if (receivedJson.contains("turnComplete")) {
-          break
-        }
 
         try {
           if (receivedJson.contains("interrupted")) {
@@ -195,17 +192,16 @@ internal constructor(
       val jsonString =
         Json.encodeToString(
           ClientContentSetup(
-            ClientContent(listOf(content.toInternal()), false)
+            ClientContent(listOf(content.toInternal()), true)
           )
         )
-      println(jsonString)
       session?.send(Frame.Text(jsonString))
       while (true) {
         try {
           val message = session?.incoming?.receive() ?: continue
           val receivedBytes = (message as Frame.Binary).readBytes()
           val receivedJson = receivedBytes.toString(Charsets.UTF_8)
-          println(receivedJson)
+          println(receivedBytes)
           try {
             val functionContent = Json.decodeFromString<ToolCallSetup>(receivedJson)
             emit(LiveContentResponse(null, false, functionContent.toolCall.functionCalls.map { FunctionCallPart(it.name, it.args!!) }))

firebase-vertexai/src/test/java/com/google/firebase/vertexai/LiveModelTesting.kt

@@ -57,24 +57,28 @@ class LiveModelTesting {
             )
         runBlocking{
             val session =  generativeModel.connect()
-            session!!.send("Tell me the status of the light of type tubelight", listOf(ContentModality.TEXT)).collect {
-                println(it)
-                if(!it.functionCalls.isNullOrEmpty()) {
-                    println("Entered??")
-                    val fetchWeatherCall = it.functionCalls!!.find { it.name == "fetchLight" }
-
-                    val functionResponse = fetchWeatherCall?.let {
-                        // Alternatively, if your `Location` class is marked as @Serializable, you can use
-
-                        val type = it.args["type"]!!.jsonPrimitive.content
-                        fetchWeather(type)
-                    }
-                    session!!.send(Content("function", listOf(FunctionResponsePart("fetchLight", functionResponse!!))), listOf(ContentModality.TEXT)).collect {
-                        x-> println(x.text)
-                    }
-                }
+            session!!.send("Tell me a story", listOf(ContentModality.TEXT)).collect {
+                println(it.text)
             }
 
+//            session!!.send("Tell me the status of the light ", listOf(ContentModality.TEXT)).collect {
+//                println(it)
+//                if(!it.functionCalls.isNullOrEmpty()) {
+//                    println("Entered??")
+//                    val fetchWeatherCall = it.functionCalls!!.find { it.name == "fetchLight" }
+//
+//                    val functionResponse = fetchWeatherCall?.let {
+//                        // Alternatively, if your `Location` class is marked as @Serializable, you can use
+//
+//                        val type = it.args["type"]!!.jsonPrimitive.content
+//                        fetchWeather(type)
+//                    }
+//                    session!!.send(Content("function", listOf(FunctionResponsePart("fetchLight", functionResponse!!))), listOf(ContentModality.TEXT)).collect {
+//                        x-> println(x.text)
+//                    }
+//                }
+//            }
+
 
         }