agents-core/ai.koog.agents.core.dsl.builder/AIAgentParallelNodesMergeContext

AIAgentParallelNodesMergeContext

class AIAgentParallelNodesMergeContext<Input, Output>(underlyingContextBase: AIAgentGraphContextBase, val results: List<ParallelResult<Input, Output>>) : AIAgentGraphContextBase(source)

Context for merging parallel node execution results.

This class provides DSL methods for selecting and folding results from parallel node executions. It delegates all AIAgentContextBase methods and properties to the underlying context.

Parameters

Input

The input type of the parallel nodes

Output

The output type of the parallel nodes

Constructors

AIAgentParallelNodesMergeContext

constructor(underlyingContextBase: AIAgentGraphContextBase, results: List<ParallelResult<Input, Output>>)

Properties

agent

open override val agent: GraphAIAgent<*, *>

Represents the AIAgent holding the current AIAgentContext.

agentId

open val agentId: String

A unique identifier representing the current agent instance within the context. The value is derived from the underlying agent.id.

agentInput

open override val agentInput: Any?

Represents the input provided to the agent's execution.

agentInputType

open override val agentInputType: KType

KType representing the type of the agentInput

config

open override val config: AIAgentConfig

Represents the configuration for an AI agent.

environment

open override val environment: AIAgentEnvironment

Represents the environment in which the agent operates.

llm

open override val llm: AIAgentLLMContext

Represents the AI agent's LLM context, providing mechanisms for managing tools, prompts, and interaction with the execution environment. It ensures thread safety during concurrent read and write operations through the use of sessions.

parentContext

open override val parentContext: AIAgentGraphContextBase

Represents the parent context of the AI Agent.

pipeline

open override val pipeline: AIAgentGraphPipeline

Represents the execution pipeline associated with the AI agent operating in a graph-based context.

results

val results: List<ParallelResult<Input, Output>>

The results of the parallel node executions

runId

open override val runId: String

A unique identifier for the current session associated with the AI agent context. Used to track and differentiate sessions within the execution of the agent pipeline.

stateManager

open override val stateManager: AIAgentStateManager

Manages and tracks the state of aт AI agent within the context of its execution.

storage

open override val storage: AIAgentStorage

Concurrent-safe key-value storage for an agent, used to manage and persist data within the context of a the AI agent stage execution. The storage property provides a thread-safe mechanism for sharing and storing data specific to the agent's operation.

strategyName

open override val strategyName: String

Represents the name of the strategy being used in the current AI agent context.

Functions

agentInput

inline fun <T> AIAgentContext.agentInput(): T

Utility function to get AIAgentContext.agentInput and try to cast it to some expected type.

copy

open override fun copy(environment: AIAgentEnvironment, agentInput: Any?, agentInputType: KType, config: AIAgentConfig, llm: AIAgentLLMContext, stateManager: AIAgentStateManager, storage: AIAgentStorage, runId: String, strategyName: String, pipeline: AIAgentGraphPipeline): AIAgentGraphContextBase

Creates a copy of the current AIAgentContextBase object with the specified parameters.

feature

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

Retrieves a feature from the current context using the specified key.

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

Retrieves a feature of the specified type from the current context.

featureOrThrow

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

Retrieves a feature of the specified type from the context or throws an exception if it is not available.

fold

suspend fun <R> fold(initial: R, operation: suspend (acc: R, result: Output) -> R): ParallelNodeExecutionResult<R>

Folds the result output into a single value and leaves the base context.

fork

open suspend override fun fork(): AIAgentGraphContextBase

Creates a forked instance of the underlying agent context, resulting in a new independent copy of the AIAgentContextBase. This can be used to create isolated contexts for parallel or independent operations.

get

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

Retrieves data from the agent's storage using the specified key.

getAgentContextData

fun AIAgentContext.getAgentContextData(): AgentContextData?

Retrieves the agent-specific context data associated with the current instance.

getHistory

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

Retrieves the history of messages exchanged during the agent's execution.

remove

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

Removes a feature or data associated with the specified key from the agent's storage.

removeAgentContextData

fun AIAgentContext.removeAgentContextData(): Boolean

Removes the agent-specific context data associated with the current context.

replace

open suspend override fun replace(context: AIAgentContext)

Replaces the current context with the specified context in the underlying context base.

rootContext

open fun rootContext(): AIAgentContext

Provides the root context of the current agent. If the root context is not defined, this function defaults to returning the current instance.

selectBy

suspend fun selectBy(predicate: suspend (Output) -> Boolean): ParallelNodeExecutionResult<Output>

Selects a result based on a predicate.

selectByIndex

suspend fun selectByIndex(selectIndex: suspend (List<Output>) -> Int): ParallelNodeExecutionResult<Output>

Selects a result from a list of outputs based on a provided selection function.

selectByMax

suspend fun <T : Comparable<T>> selectByMax(function: suspend (Output) -> T): ParallelNodeExecutionResult<Output>

Selects the maximum result based on a given comparison function and returns the corresponding NodeExecutionResult containing the selected output and its associated context.

store

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

Stores a feature in the agent's storage using the specified key.

store

fun AIAgentContext.store(data: AgentContextData)

Stores the given agent context data within the current AI agent context.