Client-Side Tools

fun tool(name: String, handler: suspend (input: Map<String, Any?>) -> Any)

client.tool("get_weather") { input ->
    val city = input["city"] as String

    // Call a weather API, read a sensor, etc.
    mapOf(
        "city" to city,
        "temperature" to "22°C",
        "condition" to "Sunny"
    )
}

fun removeTool(name: String)

client.sendMessage("What's the weather in Tokyo?") {
    onToolCall { tool ->
        println("Agent is calling: ${tool.toolName}")
        println("Input: ${tool.inputAsMap()}")
    }
    onToolResult { result ->
        println("Tool result: ${result.outputAsString()}")
    }
    onTextDelta { delta ->
        print(delta) // Agent's response after the tool result
    }
}

data class ToolCallInfo(
    val toolCallId: String,
    val toolName: String,
    val input: JsonElement
)

fun inputAsMap(): Map<String, Any?>

data class ToolResultInfo(
    val toolCallId: String,
    val toolName: String,
    val output: Any
)

fun outputAsString(): String

client.tool("pick_color") { _ ->
    val deferred = CompletableDeferred<String>()

    // Show a color picker dialog (UI-framework specific)
    colorPickerRequest.value = deferred  // Observed by the Composable

    // Suspend until the user picks a color
    val color = deferred.await()

    mapOf("color" to color)
}