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

AIAgentSubgraphBase

open class AIAgentSubgraphBase<TInput, TOutput>(val name: String, val start: StartNode<TInput>, val finish: FinishNode<TOutput>, toolSelectionStrategy: ToolSelectionStrategy, llmModel: LLModel? = null, llmParams: LLMParams? = null, responseProcessor: ResponseProcessor? = null) : AIAgentNodeBase<TInput, TOutput> , ExecutionPointNode(source)

Base class for AIAgentSubgraph.

Parameters

TInput

The type of input data accepted by the subgraph.

TOutput

The type of output data returned by the subgraph.

name

The name of the subgraph.

start

The starting node of the subgraph, which initiates the processing.

finish

The finishing node of the subgraph, which concludes the processing.

toolSelectionStrategy

Strategy determining which tools should be available during this subgraph's execution.

llmModel

Optional LLModel override for the subgraph execution.

llmParams

Optional LLMParams override for the prompt for the subgraph execution.

responseProcessor

Optional ResponseProcessor override for the subgraph execution.

Inheritors

AIAgentGraphStrategyBase

AIAgentSubgraph

Constructors

AIAgentSubgraphBase

constructor(name: String, start: StartNode<TInput>, finish: FinishNode<TOutput>, toolSelectionStrategy: ToolSelectionStrategy, llmModel: LLModel? = null, llmParams: LLMParams? = null, responseProcessor: ResponseProcessor? = null)

Types

Companion

object Companion

Companion object for the AIAgentSubgraphBase class.

Properties

edges

var edges: List<AIAgentEdge<TOutput, *>>

Represents the directed edges connecting the current node in the AI agent strategy graph to other nodes. Each edge defines the flow and transformation of output data from this node to another.

finish

val finish: FinishNode<TOutput>

val id: String

Represents the unique identifier of the AI agent node.

inputType

open override val inputType: TypeToken

The TypeToken of the TInput

name

open override val name: String

outputType

open override val outputType: TypeToken

The TypeToken of the TOutput

start

val start: StartNode<TInput>

Functions

addEdge

open fun addEdge(edge: AIAgentEdge<TOutput, *>)

Adds a directed edge from the current node, enabling connections between this node and other nodes in the AI agent strategy graph.

enforceExecutionPoint

open override fun enforceExecutionPoint(node: AIAgentNodeBase<*, *>, input: Any?)

Sets a forced node for the entity.

execute

open suspend override fun execute(context: AIAgentGraphContextBase, input: TInput): TOutput?

Executes the desired operation based on the input and the provided context. This function determines the execution strategy based on the tool selection strategy configured in the class.

executeUnsafe

suspend fun executeUnsafe(context: AIAgentGraphContextBase, input: Any?): Any?

Executes the node operation using the provided execution context and input, bypassing type safety checks. This method internally calls the type-safe execute method after casting the input.

forwardTo

infix fun <OutgoingInput> forwardTo(otherNode: AIAgentNodeBase<OutgoingInput, *>): AIAgentEdgeBuilderIntermediate<TOutput, TOutput, OutgoingInput>

Creates a directed edge from this AIAgentNodeBase to another AIAgentNodeBase, allowing data to flow from the output of the current node to the input of the specified node.

forwardTo

infix fun <IncomingOutput, OutgoingInput> AIAgentNodeBase<*, IncomingOutput>.~~forwardTo~~(otherNode: AIAgentNodeBase<OutgoingInput, *>): AIAgentEdgeBuilderIntermediate<IncomingOutput, IncomingOutput, OutgoingInput>

Creates a directed edge from this AIAgentNodeBase to another AIAgentNodeBase, allowing data to flow from the output of the current node to the input of the specified node.

getExecutionPoint

open override fun getExecutionPoint(): ExecutionPoint?

Retrieves the current execution point, which consists of a specific node in the execution graph and an optional input. If no forced node is defined, the method returns null.

resetExecutionPoint

open override fun resetExecutionPoint()

Resets the currently enforced execution point for the AI agent, including clearing any forced node and input data. Once the execution point is reset, the system will revert to its default execution behavior without targeting a specific node explicitly.

resolveEdge

suspend fun resolveEdge(context: AIAgentGraphContextBase, nodeOutput: TOutput): AIAgentNodeBase.ResolvedEdge?

Resolves the edge associated with the provided node output and execution context. Iterates through available edges and identifies the first edge that can successfully process the given node output within the provided context. If a resolvable edge is found, it returns a ResolvedEdge containing the edge and its output. Otherwise, returns null.