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

SimpleAIAgentNodeImpl

open class SimpleAIAgentNodeImpl<TInput, TOutput> : AIAgentNodeBase<TInput, TOutput> (source)

Represents a simple implementation of an AI agent node within a graph-based structure.

This class facilitates the execution of a computation node, receiving an input of type TInput and producing an output of type TOutput. It is designed to be used within an AI agent graph context.

Type Parameters

TInput

The type of the input data this node accepts.

TOutput

The type of the output data this node produces.

Inheritors

AIAgentNode

Properties

edges

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

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

execute

val execute: suspend AIAgentGraphContextBase.(input: TInput) -> TOutput

A suspendable lambda function that defines the execution logic of the node. It operates in the context of an AIAgentGraphContextBase, taking an input of type TInput and producing an output of type TOutput.

val id: String

Represents the unique identifier of the AI agent node.

inputType

open override val inputType: TypeToken

The TypeToken representing the expected type of input for the node.

name

open override val name: String

The name of the node, used for identification within the graph.

outputType

open override val outputType: TypeToken

The TypeToken representing the type of output produced by the node.

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.

execute

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

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

executeUnsafe

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

Executes the node operation using the provided execution context and input, bypassing static type checks. This method internally performs an unchecked cast of input to TInput and delegates to execute. The caller is responsible for ensuring that input is actually assignable to TInput; otherwise a ClassCastException may be thrown at the call site or later during execution.

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.

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.