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

EdgeAssertionsBuilder

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

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.

Constructors

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

Types

EdgeOutputPair

inner class EdgeOutputPair<I, O>(val node: NodeReference<I, O>, val output: O)

Represents a pair consisting of a node and its corresponding output within an edge assertion context. This is used to define expected edge behavior in a graph of nodes.

Properties

agentInput

open override var agentInput: String?

assertions

val assertions: MutableList<EdgeAssertion<*, *>>

A mutable list that holds all the defined EdgeAssertion instances for the current context.

config

open override var config: AIAgentConfigBase?

environment

open override var environment: AIAgentEnvironment?

llm

open override var llm: AIAgentLLMContext?

sessionUuid

open override var sessionUuid: Uuid?

stateManager

open override var stateManager: AIAgentStateManager?

storage

open override var storage: AIAgentStorage?

strategyId

open override var strategyId: String?

unconditionalEdgeAssertions

val unconditionalEdgeAssertions: MutableList<UnconditionalEdgeAssertion>

A collection that stores assertions ensuring an unconditional connection between nodes in a graph testing context. Each assertion represents a defined relationship where a node always leads to a specified target node.

Functions

alwaysGoesTo

infix fun NodeReference<*, *>.alwaysGoesTo(targetNode: NodeReference<*, *>)

Creates an assertion to verify that the current node always leads to the given target node.

build

open override fun build(): AIAgentContextBase

copy

open override fun copy(): Testing.Config.SubgraphAssertionsBuilder.EdgeAssertionsBuilder

Creates a deep copy of the current EdgeAssertionsBuilder instance, duplicating its state and context.

withContext

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

Executes a given block of logic within the context of a copied instance of the current EdgeAssertionsBuilder.

withOutput

infix fun <I, O> NodeReference<I, O>.withOutput(output: O): Testing.Config.SubgraphAssertionsBuilder.EdgeAssertionsBuilder.EdgeOutputPair<I, O>

Associates the given output with the current node reference, creating a pair that represents the node and its corresponding output.