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

Package-level declarations

Types

AIAgentContextMockBuilder

@TestOnly

class AIAgentContextMockBuilder : AIAgentContextMockBuilderBase

AIAgentContextMockBuilderBase

@TestOnly

interface AIAgentContextMockBuilderBase : BaseBuilder<AIAgentContextBase>

AssertionResult

@TestOnly

sealed interface AssertionResult

DummyAgentContext

class DummyAgentContext(builder: AIAgentContextMockBuilder) : AIAgentContextBase

EdgeAssertion

@TestOnly

data class EdgeAssertion<Input, Output>(val node: NodeReference<Input, Output>, val context: AIAgentContextBase, val output: Output, val expectedNode: NodeReference<*, *>)

GraphAssertions

@TestOnly

data class GraphAssertions(val name: String, val start: NodeReference.Start<*>, val finish: NodeReference.Finish<*>, val nodes: Map<String, NodeReference<*, *>>, val nodeOutputAssertions: List<NodeOutputAssertion<*, *>>, val edgeAssertions: List<EdgeAssertion<*, *>>, val unconditionalEdgeAssertions: List<UnconditionalEdgeAssertion>, val reachabilityAssertions: List<ReachabilityAssertion>, val subgraphAssertions: MutableList<SubGraphAssertions>)

NodeOutputAssertion

@TestOnly

data class NodeOutputAssertion<Input, Output>(val node: NodeReference<Input, Output>, val context: DummyAgentContext, val input: Input, val expectedOutput: Output)

NodeReference

@TestOnly

sealed class NodeReference<Input, Output>

ReachabilityAssertion

@TestOnly

data class ReachabilityAssertion(val from: NodeReference<*, *>, val to: NodeReference<*, *>)

SubGraphAssertions

@TestOnly

data class SubGraphAssertions(val subgraphRef: NodeReference.SubgraphNode<*, *>, val graphAssertions: GraphAssertions)

Testing

@TestOnly

class Testing

Provides functionality for testing graph-related stages in an AI agent pipeline.

UnconditionalEdgeAssertion

@TestOnly

data class UnconditionalEdgeAssertion(val node: NodeReference<*, *>, val expectedNode: NodeReference<*, *>)

Functions

assistantMessage

fun Testing.Config.SubgraphAssertionsBuilder<*, *>.assistantMessage(text: String, finishReason: String? = null): Message.Assistant

Creates an assistant message with the provided text and finish reason.

graph

fun Testing.Config.graph(test: Testing.Config.() -> Unit)

Configures the Testing feature for graph testing.

testGraph

fun AIAgent.FeatureContext.testGraph(name: String, test: Testing.Config.SubgraphAssertionsBuilder<*, *>.() -> Unit)

Installs the Testing feature with graph testing enabled and executes the provided test block.

toolCallMessage

fun <Args : Tool.Args> Testing.Config.SubgraphAssertionsBuilder<*, *>.toolCallMessage(tool: Tool<Args, *>, args: Args): Message.Tool.Call

Creates a Message.Tool.Call instance for the given tool and its arguments.

toolResult

fun toolResult(tool: SimpleTool<*>, result: String): ReceivedToolResult

Constructs a ReceivedToolResult object using the provided tool and result string.

fun <Result : ToolResult> toolResult(tool: Tool<*, Result>, result: Result): ReceivedToolResult

Converts a tool and its corresponding result into a ReceivedToolResult object.

withTesting

fun AIAgent.FeatureContext.withTesting(config: Testing.Config.() -> Unit = {})

Enables and configures the Testing feature for a Kotlin AI Agent instance.