agents-test/ai.koog.agents.testing.tools/AIAgentContextMockBuilderBase

AIAgentContextMockBuilderBase

@TestOnly

interface AIAgentContextMockBuilderBase : BaseBuilder<AIAgentContextBase> (source)

A base interface for building mock implementations of the AIAgentContextBase interface.

This interface provides configurable properties and methods for creating a mock AI agent context, enabling the customization of its environment, input, configuration, state management, and more. It is intended for use in testing scenarios and allows for the creation of testable mock instances of AI agent contexts.

Extends the BaseBuilder interface for constructing instances of type AIAgentContextBase.

Inheritors

NodeOutputAssertionsBuilder

EdgeAssertionsBuilder

AIAgentContextMockBuilder

Properties

agentInput

abstract 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

abstract var agentInputType: KType?

Represents the KType of the agentInput.

config

abstract var config: AIAgentConfigBase?

Specifies the configuration for the AI agent.

environment

abstract var environment: AIAgentEnvironment?

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

llm

abstract 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

abstract var runId: String?

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

stateManager

abstract 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

abstract 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

abstract var strategyId: String?

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

Functions

build

abstract 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

abstract fun copy(environment: AIAgentEnvironment? = this.environment, agentInput: Any? = this.agentInput, agentInputType: KType? = this.agentInputType, config: AIAgentConfigBase? = this.config, llm: AIAgentLLMContext? = this.llm, stateManager: AIAgentStateManager? = this.stateManager, storage: AIAgentStorage? = this.storage, runId: String? = this.runId, strategyId: String? = this.strategyId): AIAgentContextMockBuilderBase

Creates and returns a copy of the current instance of AIAgentContextMockBuilderBase.