agents-test/ai.koog.agents.testing.feature/Testing/Config/SubgraphAssertionsBuilder

SubgraphAssertionsBuilder

class SubgraphAssertionsBuilder<Input, Output>(subgraphRef: NodeReference.SubgraphNode<Input, Output>, clock: Clock, tokenizer: Tokenizer?)(source)

Builder class for constructing subgraph-level assertions. This includes setup for nodes, edges, reachability assertions, and context-related mock setups.

Parameters

subgraphRef

: A NodeReference.SubgraphNode reference to a subgraph of the agent's graph strategy

clock

: A clock that is used for mock message timestamps

tokenizer

: Tokenizer that will be used to estimate token counts in mock messages

Constructors

SubgraphAssertionsBuilder

constructor(subgraphRef: NodeReference.SubgraphNode<Input, Output>, clock: Clock, tokenizer: Tokenizer?)

Types

EdgeAssertionsBuilder

class EdgeAssertionsBuilder(stageBuilder: Testing.Config.SubgraphAssertionsBuilder<*, *>, context: AIAgentContextMockBuilder = stageBuilder.context.copy()) : AIAgentContextMockBuilderBase

A builder class used to facilitate the creation and management of edge assertions in a stage context. Delegates functionality to a local agent stage context mock builder for shared behaviors.

NodeOutputAssertionsBuilder

class NodeOutputAssertionsBuilder(stageBuilder: Testing.Config.SubgraphAssertionsBuilder<*, *>, context: AIAgentContextMockBuilder = stageBuilder.context.copy()) : AIAgentContextMockBuilderBase

A builder class for constructing and managing assertions related to the outputs of nodes within a stage. This class provides functionality to define and evaluate assertions for node outputs in a local agent's stage.

Functions

assertEdges

fun assertEdges(block: Testing.Config.SubgraphAssertionsBuilder.EdgeAssertionsBuilder.() -> Unit)

Asserts the edges in the context of a graph by applying a set of edge assertions built using the provided block.

assertNodeByName

fun <I, O> assertNodeByName(name: String): NodeReference.NamedNode<I, O>

Asserts the existence of a node by its name within the stage structure and returns a reference to it. If the node with the given name has not been asserted previously, it creates a new NamedNode reference and records it in the nodes map for tracking.

assertNodes

fun assertNodes(block: Testing.Config.SubgraphAssertionsBuilder.NodeOutputAssertionsBuilder.() -> Unit)

Asserts the state of nodes in the stage using a provided block to define the desired assertions. The block operates on a NodeOutputAssertionsBuilder to specify expectations for node inputs and outputs.

assertReachable