agents-features-longterm-memory/ai.koog.agents.longtermmemory.storage/InMemoryRecordStorage

InMemoryRecordStorage

open class InMemoryRecordStorage(defaultNamespace: String = "default") : SearchStorage<TextDocument, SearchRequest> , WriteStorage<TextDocument> , LookupStorage<TextDocument> , DeletionStorage(source)

In-memory implementation of SearchStorage and WriteStorage that stores records in a map.

This implementation is useful for testing, development, and scenarios where persistence is not required. All data is stored in memory and will be lost when the application stops.

Limitations:

Data is not persisted and will be lost on application restart
Both ai.koog.rag.base.storage.search.KeywordSearchRequest and ai.koog.rag.base.storage.search.SimilaritySearchRequest are accepted, but both are implemented as simple case-insensitive substring matching; no vector embeddings are used
Filter expressions are ignored

Parameters

defaultNamespace

The default namespace to use when none is specified in method calls. Defaults to "default".

Constructors

InMemoryRecordStorage

constructor(defaultNamespace: String = "default")

Functions

add

open suspend override fun add(documents: List<TextDocument>, namespace: String?): List<String>

delete

open suspend override fun delete(ids: List<String>, namespace: String?): List<String>

get

open suspend override fun get(ids: List<String>, namespace: String?): List<TextDocument>

rankDocuments

open fun ~~rankDocuments~~(query: String): Flow<SearchResult<TextDocument>>

open suspend override fun search(request: SearchRequest, namespace: String?): List<SearchResult<TextDocument>>

size

suspend fun size(namespace: String? = null): Int

Returns the number of records in the repository for the specified namespace.

update

open suspend override fun update(documents: Map<String, TextDocument>, namespace: String?): List<String>