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

DummyAIAgentContext

@TestOnly

class DummyAIAgentContext(builder: AIAgentContextMockBuilder, val id: String = "DummyAgentId") : AIAgentContextBase(source)

A mock implementation of the AIAgentContextBase interface, used for testing purposes.

Parameters

builder

A builder object used to initialize the mock properties of the context.

Constructors

DummyAIAgentContext

constructor(builder: AIAgentContextMockBuilder, id: String = "DummyAgentId")

Creates a new instance of DummyAIAgentContext using a predefined AIAgentContextMockBuilder.

Properties

agentInput

open override val agentInput: Any?

agentInputType

open override val agentInputType: KType

config

open override val config: AIAgentConfigBase

environment

open override val environment: AIAgentEnvironment

open override val id: String

isEnvironmentDefined

val isEnvironmentDefined: Boolean

Indicates whether the environment for the agent context is defined.

isLLMDefined

val isLLMDefined: Boolean

Indicates whether a Language Learning Model (LLM) is defined in the current context.

llm

open override val llm: AIAgentLLMContext

pipeline

open override val pipeline: AIAgentPipeline

runId

open override val runId: String

stateManager

open override val stateManager: AIAgentStateManager

storage

open override val storage: AIAgentStorage

strategyName

open override val strategyName: String

Functions

copy

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

copyWithTools

open fun copyWithTools(tools: List<ToolDescriptor>): AIAgentContextBase

feature

open override fun <Feature : Any> feature(key: AIAgentStorageKey<Feature>): Feature?

open override fun <Feature : Any> feature(feature: AIAgentFeature<*, Feature>): Feature?

featureOrThrow

open fun <Feature : Any> featureOrThrow(feature: AIAgentFeature<*, Feature>): Feature

fork

open suspend override fun fork(): AIAgentContextBase

get

open override fun <T> get(key: AIAgentStorageKey<*>): T?

getHistory

open suspend override fun getHistory(): List<Message>

remove

open override fun remove(key: AIAgentStorageKey<*>): Boolean

replace

open suspend override fun replace(context: AIAgentContextBase)

store

open override fun store(key: AIAgentStorageKey<*>, value: Any)