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

Package-level declarations

Types

@TestOnly

class AIAgentContextMockBuilder : AIAgentContextMockBuilderBase

AIAgentContextMockBuilder is a builder class for constructing a mock implementation of an AI agent context. It provides mechanisms to configure various components of the AI agent context before constructing it. This class is intended for use in testing scenarios and extends AIAgentContextMockBuilderBase.

AIAgentContextMockBuilderBase

@TestOnly

interface AIAgentContextMockBuilderBase : BaseBuilder<AIAgentContextBase>

A base interface for building mock implementations of the AIAgentContextBase interface.

DefaultResponseReceiver

class DefaultResponseReceiver(val response: String)

Receiver class for configuring text responses from the LLM.

DummyAIAgentContext

@TestOnly

class DummyAIAgentContext(builder: AIAgentContextMockBuilder, val agentId: String = "DummyAgentId") : AIAgentContextBase

A mock implementation of the AIAgentContextBase interface, used for testing purposes.

DummyTool

class DummyTool : SimpleTool<DummyTool.Args>

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

MockEnvironment

class MockEnvironment(toolRegistry: ToolRegistry, promptExecutor: PromptExecutor, baseEnvironment: AIAgentEnvironment? = null) : AIAgentEnvironment

A mock implementation of AIAgentEnvironment used for testing agent behavior.

MockLLMBuilder

class MockLLMBuilder(clock: Clock, tokenizer: Tokenizer? = null)

Builder class for creating mock LLM executors for testing.

MockToolFromCallableReceiver

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.

ToolCondition

class ToolCondition<Args : ToolArgs, 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

getMockExecutor

fun getMockExecutor(toolRegistry: ToolRegistry? = null, clock: Clock = Clock.System, tokenizer: Tokenizer? = null, handleLastAssistantMessage: Boolean = false, init: MockLLMBuilder.() -> Unit): PromptExecutor

Creates a mock LLM executor for testing.

mockLLMAnswer

fun mockLLMAnswer(response: String): DefaultResponseReceiver

Creates a mock LLM text response.

mockLLMToolCall

fun MockLLMBuilder.mockLLMToolCall(tool: KFunction<*>, vararg args: Any?, toolCallId: String? = null): MockLLMBuilder.ToolCallReceiver<ToolFromCallable.VarArgs>

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

mockTool

infix fun <Result> MockLLMBuilder.mockTool(tool: KFunction<Result>): MockToolFromCallableReceiver<Result>

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