Skip to content

內建工具

Koog 框架提供內建工具,用於處理代理程式與使用者互動的常見情境。

以下是可用的內建工具:

工具
名稱
說明
SayToUser__say_to_user__讓代理程式傳送訊息給使用者。它會將代理程式訊息以 Agent says: 前綴印出至控制台。
AskUser__ask_user__讓代理程式向使用者請求輸入。它會將代理程式訊息印出至控制台並等待使用者回應。
ExitTool__exit__讓代理程式結束對話並終止會話。
ReadFileTool__read_file__讀取文字檔,可選行數範圍。傳回格式化內容,其中包含使用 0 基底行索引的後設資料。

註冊內建工具

如同任何其他工具,內建工具必須加入到工具註冊表才能供代理程式使用。以下是一個範例:

kotlin
// 建立一個包含所有內建工具的工具註冊表
val toolRegistry = ToolRegistry {
    tool(SayToUser)
    tool(AskUser)
    tool(ExitTool)
    tool(ReadFileTool(JVMFileSystemProvider.ReadOnly))
}

// 在建立代理程式時傳入註冊表
val agent = AIAgent(
    executor = simpleOpenAIExecutor(apiToken),
    systemPrompt = "You are a helpful assistant.",
    llmModel = OpenAIModels.Chat.GPT4o,
    toolRegistry = toolRegistry
)

您可以透過在同一個註冊表中結合內建工具和自訂工具,為您的代理程式建立一套全面的功能。 若要深入了解自訂工具,請參閱 基於註解的工具基於類別的工具