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

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

Types

NodeOutputPair

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

agentInput

open override var agentInput: Any?

Represents the input to be used by the AI agent during its execution. This variable can be set to define specific data or context relevant to the agent's task. It is nullable, indicating that the agent may operate without an explicitly defined input.

agentInputType

open override var agentInputType: KType?

Represents the KType of the agentInput.

assertions

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.

config

open override var config: AIAgentConfigBase?

Specifies the configuration for the AI agent.

environment

open override var environment: AIAgentEnvironment?

Represents the environment used by the AI agent to interact with external systems.

llm

open override var llm: AIAgentLLMContext?

Represents the LLM context associated with an AI agent during testing scenarios. This variable is used to configure and manage the context for an AI agent's large language model (LLM) interactions, including tools, prompt handling, and model-specific attributes.

runId

open override var runId: String?

Represents the unique identifier associated with the session in the mock builder context.

stateManager

open override var stateManager: AIAgentStateManager?

Represents an optional state manager for an AI agent in the context of building its mock environment. The stateManager is responsible for maintaining and managing the internal state of the agent in a thread-safe manner, ensuring consistency during state modifications.

storage

open override var storage: AIAgentStorage?

Represents a concurrent-safe key-value storage used for managing data within the context of mock AI agent construction. This property typically holds an optional instance of AIAgentStorage, which provides methods to store, retrieve, and manage typed key-value pairs in a thread-safe manner.

strategyId

open override var strategyId: String?

Represents the identifier of a strategy to be used within the context of an AI agent.

Functions

build

open override fun build(): AIAgentContextBase

Builds and returns an instance of AIAgentContextBase based on the current properties of the builder. This method creates a finalized AI agent context, integrating all the specified configurations, environment settings, and components into a coherent context object ready for use.

copy

open override fun copy(environment: AIAgentEnvironment?, agentInput: Any?, agentInputType: KType?, config: AIAgentConfigBase?, llm: AIAgentLLMContext?, stateManager: AIAgentStateManager?, storage: AIAgentStorage?, runId: String?, strategyId: String?): Testing.Config.SubgraphAssertionsBuilder.NodeOutputAssertionsBuilder

Creates and returns a new copy of the NodeOutputAssertionsBuilder instance.

withContext

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

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

withInput

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.