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

AIAgentNode

open class AIAgentNode<Input, Output> : AIAgentNodeBase<Input, Output> (source)

Represents a simple implementation of an AI agent node, encapsulating a specific execution logic that processes the input data and produces an output.

Parameters

Input

The type of input data this node processes.

Output

The type of output data this node produces.

Inheritors

StartNode

FinishNode

Properties

edges

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

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.

execute

val execute: suspend AIAgentContextBase.(input: Input) -> Output

A suspending function that defines the execution logic for the node. It processes the provided input within the given execution context and produces an output.

val id: String

Represents the unique identifier of the AI agent node.

inputType

open override val inputType: KType

The KType of the Input

name

open override val name: String

The name of the node, used for identification and debugging.

outputType

open override val outputType: KType

The KType of the Output

Functions

addEdge

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

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

execute

open suspend override fun execute(context: AIAgentContextBase, input: Input): Output

Executes a specific operation based on the given context and input.

executeUnsafe

suspend fun executeUnsafe(context: AIAgentContextBase, 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. The lifecycle hooks onBeforeNode and onAfterNode are invoked before and after the execution respectively.

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.

resolveEdge

suspend fun resolveEdge(context: AIAgentContextBase, nodeOutput: Output): 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.