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

Package-level declarations

Types

@TestOnly

Represents the result of an assertion used within the testing framework.

@TestOnly

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

Represents the assertion of an edge in a graph structure related to AI agent operations.

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>)

Represents a set of assertions for validating the structure and behavior of a graph-based system.

NodeOutputAssertion

@TestOnly

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

Represents an assertion for testing the output of a specific node within an AI agent's subgraph.

NodeReference

@TestOnly

sealed class NodeReference<Input, Output>

Represents a reference to a specific type of node within an AI agent subgraph. This sealed class provides a way to resolve and identify nodes for various processing operations in the context of a subgraph.

ReachabilityAssertion

@TestOnly

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

Represents an assertion for testing the reachability between two specific nodes in a graph.

SubGraphAssertions

@TestOnly

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

Encapsulates assertions for a subgraph within a graph testing framework.

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<*, *>)

Represents an assertion for an unconditional edge in a graph structure.

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 <Input, Output> AIAgent.FeatureContext.testGraph(name: String, test: Testing.Config.SubgraphAssertionsBuilder<Input, Output>.() -> Unit)

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

toolCallMessage

fun <Args : ToolArgs> 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.