EdgeAssertionsBuilder

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

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

Types

Link copied to clipboard
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

Link copied to clipboard
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.

Link copied to clipboard
open override var agentInputType: KType?

Represents the KType of the agentInput.

Link copied to clipboard

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

Link copied to clipboard
open override var config: AIAgentConfigBase?

Specifies the configuration for the AI agent.

Link copied to clipboard
open override var environment: AIAgentEnvironment?

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

Link copied to clipboard
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.

Link copied to clipboard
open override var runId: String?

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

Link copied to clipboard
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.

Link copied to clipboard
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.

Link copied to clipboard
open override var strategyId: String?

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

Link copied to clipboard

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

Link copied to clipboard
infix fun NodeReference<*, *>.alwaysGoesTo(targetNode: NodeReference<*, *>)

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

Link copied to clipboard
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.

Link copied to clipboard
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.EdgeAssertionsBuilder

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

Link copied to clipboard

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

Link copied to clipboard

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