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

AIAgentContextMockBuilder

@TestOnly

class AIAgentContextMockBuilder : AIAgentContextMockBuilderBase(source)

AIAgentContextMockBuilder is a builder class for constructing a mock implementation of an AI agent context. It provides mechanisms to configure various components of the AI agent context before constructing it. This class is intended for use in testing scenarios and extends AIAgentContextMockBuilderBase.

Constructors

constructor()

Types

ProxyHandler

class ProxyHandler<T : Any>(name: String)

ProxyHandler is a utility class that dynamically creates a proxy object of a specified type. The proxy implements default behavior for methods such as toString, equals, property access, and method invocation, providing a placeholder implementation.

Properties

agentInput

open override var agentInput: Any?

Represents the agent's input data used in constructing or testing the agent's context.

agentInputType

open override var agentInputType: KType?

Represents the KType of the agentInput.

config

open override var config: AIAgentConfigBase?

Represents the AI agent configuration used in the mock builder.

environment

open override var environment: AIAgentEnvironment?

Represents the AI agent's environment in which the context is being executed.

llm

open override var llm: AIAgentLLMContext?

Represents the context for accessing and managing an AI agent's LLM (Large Language Model) configuration and behavior. The llm property allows you to define or override the LLM context for the agent, including tools, prompt handling, and interaction with external dependencies.

runId

open override var runId: String?

Defines the unique identifier for the session context within the agent's lifecycle. This property can be used to correlate and differentiate multiple sessions for the same agent or across different agents.

stateManager

open override var stateManager: AIAgentStateManager?

An overrideable property for managing the agent's state using an instance of AIAgentStateManager.

storage

open override var storage: AIAgentStorage?

Represents a concurrent-safe key-value storage instance for an AI agent.

strategyId

open override var strategyId: String?

Represents the identifier for the strategy to be used in the agent context.

Functions

build

open override fun build(): DummyAIAgentContext

Builds and returns an instance of DummyAgentContext.

copy

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

Creates and returns a new copy of the current AIAgentContextMockBuilder instance. The copied instance contains the same state and configuration as the original.