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

NodeOutputAssertionsBuilder

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

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.

Constructors

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

Types

Link copied to clipboard
inner class NodeOutputPair<I, O>(val node: NodeReference<I, O>, val input: I)

Represents a pairing of a specific node reference and its corresponding input. This is used to define the expected output for a given input within the context of a specific node.

Properties

Link copied to clipboard
open override var agentInput: String?
Link copied to clipboard
val assertions: MutableList<NodeOutputAssertion<*, *>>

A mutable list that stores assertions representing the expected behavior and output of nodes in the context of a specific staging environment for testing purposes.

Link copied to clipboard
open override var config: AIAgentConfigBase?
Link copied to clipboard
open override var environment: AIAgentEnvironment?
Link copied to clipboard
open override var llm: AIAgentLLMContext?
Link copied to clipboard
open override var sessionUuid: Uuid?
Link copied to clipboard
open override var stateManager: AIAgentStateManager?
Link copied to clipboard
open override var storage: AIAgentStorage?
Link copied to clipboard
open override var strategyId: String?

Functions

Link copied to clipboard
open override fun build(): AIAgentContextBase
Link copied to clipboard
open override fun copy(): Testing.Config.SubgraphAssertionsBuilder.NodeOutputAssertionsBuilder

Creates and returns a new copy of the NodeOutputAssertionsBuilder instance.

Link copied to clipboard
fun withContext(block: Testing.Config.SubgraphAssertionsBuilder.NodeOutputAssertionsBuilder.() -> Unit)

Executes the specified block of code within a duplicate context of the current NodeOutputAssertionsBuilder.

Link copied to clipboard
infix fun <I, O> NodeReference<I, O>.withInput(input: I): Testing.Config.SubgraphAssertionsBuilder.NodeOutputAssertionsBuilder.NodeOutputPair<I, O>

Associates the provided input with the current node reference, creating a pair that links the node to its corresponding input.