agents-core/ai.koog.agents.core.agent.entity/AIAgentStorage

AIAgentStorage

Concurrent-safe key-value storage for an agent. You can create typed keys for your data using the createStorageKey function and set and retrieve data using it by calling set and get.

Functions

clear

suspend fun clear()

Clears all data from the storage.

get

suspend fun <T : Any> get(key: AIAgentStorageKey<T>): T?

Retrieves the value associated with the given key from the storage.

getValue

suspend fun <T : Any> getValue(key: AIAgentStorageKey<T>): T

Retrieves the non-null value associated with the given key from the storage. If the key does not exist in the storage, a NoSuchElementException is thrown.

putAll

suspend fun putAll(map: Map<AIAgentStorageKey<*>, Any>)

Adds all key-value pairs from the given map to the storage.

remove

suspend fun <T : Any> remove(key: AIAgentStorageKey<T>): T?

Removes the value associated with the given key from the storage.

set

suspend fun <T : Any> set(key: AIAgentStorageKey<T>, value: T)

Sets the value associated with the given key in the storage.

toMap

suspend fun toMap(): Map<AIAgentStorageKey<*>, Any>

Converts the storage to a map representation.