Update function calling kotlin sample

google-gemini · Jul 12, 2024 · ef62034 · ef62034
1 parent b04745e
commit ef62034
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/samples/src/main/java/com/google/ai/client/generative/samples/function_calling.kt b/samples/src/main/java/com/google/ai/client/generative/samples/function_calling.kt
@@ -63,17 +63,14 @@ suspend fun functionCalling() {
     var response = chat.sendMessage(prompt)
 
     // Check if the model responded with a function call
-    if (response.functionCalls.isNotEmpty() &&
-        response.functionCalls.first().name == "multiply") {
-        val functionCall = response.functionCalls.first()
-
-        val a = functionCall.args["a"]?.toDouble() ?: throw InvalidStateException("Missing param 'a'")
-        val b = functionCall.args["b"]?.toDouble() ?: throw InvalidStateException("Missing param 'b'")
-        val result = JSONObject().put("result", multiply(a, b))
+    response.functionCalls.first { it.name == "multiply" }.apply {
+        val a: String by args
+        val b: String by args
 
+        val result = JSONObject(mapOf("result" to multiply(a.toDouble(), b.toDouble())))
         response = chat.sendMessage(
             content(role = "function") {
-                part(FunctionResponsePart(functionCall.name, result))
+                part(FunctionResponsePart("multiply", result))
             }
         )
     }