Embabel Agent Documentation/com.embabel.chat.support/InMemoryAssetTracker

InMemoryAssetTracker

class InMemoryAssetTracker(initialAssets: Collection<Asset> = emptyList()) : AssetTracker

Simple in-memory implementation of AssetTracker for testing and ephemeral use cases.

Constructors

constructor(initialAssets: Collection<Asset> = emptyList())

Properties

assets

open override val assets: List<Asset>

All tracked assets, in order in which they were added

Functions

addAnyDurablyReturnedAssets

open fun addAnyDurablyReturnedAssets(tools: List<Tool>, assetStore: AssetStore): List<Tool>

Make these tools materialize and track any returned MaterializableAsset instances.

addAnyReturnedAssets

open fun addAnyReturnedAssets(tools: List<Tool>): List<Tool>

Make these tools track any assets produced.

open fun addAnyReturnedAssets(tools: List<Tool>, filter: Predicate<Asset>): List<Tool>

Make these tools track any assets produced, with a filter. Only assets that pass the filter will be tracked.

addAsset

open override fun addAsset(asset: Asset)

Add an asset to be tracked. If the asset is already being tracked, do nothing.

addDurablyReturnedAssets

open fun addDurablyReturnedAssets(tool: Tool, assetStore: AssetStore): Tool

Wrap a tool so any returned MaterializableAsset instances are copied to assetStore and the resulting DurableAsset instances are tracked. Callers must supply a configured store; durable tracking is not enabled implicitly.

addReturnedAssets

open fun addReturnedAssets(tool: Tool): Tool

Wrap a tool so any outputs are tracked as assets.

open fun addReturnedAssets(tool: Tool, filter: Predicate<Asset>): Tool

Wrap a tool so any outputs are tracked as assets, with a filter (Java-friendly). Only assets that pass the filter will be tracked.

mostRecent

open fun mostRecent(n: Int): AssetView

The most recently timestamped assets

mostRecentlyAdded

open fun mostRecentlyAdded(n: Int): AssetView

The most recently added assets

references

open fun references(): List<LlmReference>

Convenience method to return references. References will be converted to matryoshka tools

since

open fun since(instant: Instant): AssetView

Assets timestamped since the given instant

withAsset

fun withAsset(asset: Asset): InMemoryAssetTracker

Convenience method to add an asset and return this tracker for chaining.