mirror of
https://github.com/google/adk-docs.git
synced 2026-09-14 16:16:59 +08:00
Toni Klopfenstein
27e1303857
* Migrating from Map to typed data classes in FuncTool and LongRunningTool kotlin examples * lint cleanup of kotlin snippets * adding gradle wrapper for kotlin snippets * cleaning up build file for kotlin * fixing ktlint_code_style, added focus on included Gradle Wrapper --------- Co-authored-by: Kristopher Overholt <koverholt@google.com>
50 lines
1.5 KiB
Kotlin
50 lines
1.5 KiB
Kotlin
package com.google.adk.kt.examples.tools
|
|
|
|
import com.google.adk.kt.agents.Instruction
|
|
import com.google.adk.kt.agents.LlmAgent
|
|
import com.google.adk.kt.annotations.Param
|
|
import com.google.adk.kt.annotations.Tool
|
|
import com.google.adk.kt.models.Gemini
|
|
|
|
// --8<-- [start:long_running_tool]
|
|
data class ReimbursementApproval(
|
|
val status: String,
|
|
val approver: String,
|
|
val purpose: String,
|
|
val amount: Double,
|
|
val ticketId: String,
|
|
)
|
|
|
|
class ReimbursementService {
|
|
/**
|
|
* Asks for approval for the reimbursement.
|
|
*/
|
|
@Tool(isLongRunning = true)
|
|
fun askForApproval(
|
|
@Param("The purpose of the reimbursement.") purpose: String,
|
|
@Param("The amount to be reimbursed.") amount: Double,
|
|
): ReimbursementApproval {
|
|
// Simulate creating a ticket and sending a notification.
|
|
// This tool returns the initial result and then the agent pauses.
|
|
return ReimbursementApproval(
|
|
status = "pending",
|
|
approver = "Sean Zhou",
|
|
purpose = purpose,
|
|
amount = amount,
|
|
ticketId = "approval-ticket-1",
|
|
)
|
|
}
|
|
}
|
|
|
|
fun main() {
|
|
val service = ReimbursementService()
|
|
val agent =
|
|
LlmAgent(
|
|
name = "approver_agent",
|
|
model = Gemini(name = "gemini-flash-latest"),
|
|
instruction = Instruction("You are a helpful reimbursement assistant."),
|
|
tools = service.generatedTools(),
|
|
)
|
|
}
|
|
// --8<-- [end:long_running_tool]
|