agents-test/ai.koog.agents.testing.tools

Package-level declarations

Types

Link copied to clipboard
class DefaultResponseReceiver(val response: String)

Receiver class for configuring text responses from the LLM.

Link copied to clipboard
class DummyTool : SimpleTool<DummyTool.Args>

Simple tool implementation for testing purposes. This tool accepts a dummy parameter and returns a constant result.

Link copied to clipboard
class MockLLMBuilder

Builder class for creating mock LLM executors for testing.

Link copied to clipboard
class MockToolFromCallableReceiver<Result>(callable: KFunction<Result>, builder: MockLLMBuilder)

A class that facilitates mocking of callable tools by defining their behavior based on the received arguments.

Link copied to clipboard
class ToolCondition<Args : Tool.Args, Result : ToolResult>(val tool: Tool<Args, Result>, val argsCondition: suspend (Args) -> Boolean, val produceResult: suspend (Args) -> Result)

Represents a condition for a tool call and its corresponding result.

Functions

Link copied to clipboard
fun getMockExecutor(toolRegistry: ToolRegistry? = null, init: MockLLMBuilder.() -> Unit): PromptExecutor

Creates a mock LLM executor for testing.

Link copied to clipboard
fun mockLLMAnswer(response: String): DefaultResponseReceiver

Creates a mock LLM text response.

Link copied to clipboard
fun MockLLMBuilder.mockLLMToolCall(tool: KFunction<*>, vararg args: Any?): MockLLMBuilder.ToolCallReceiver<ToolFromCallable.VarArgs>

Mocks a tool call for the provided tool function and arguments.

Link copied to clipboard
infix fun <Result> MockLLMBuilder.mockTool(tool: KFunction<Result>): MockToolFromCallableReceiver<Result>

Associates a tool function with the MockLLMBuilder to create a MockToolFromCallableReceiver instance.