agents-test/ai.koog.agents.testing.feature/NodeReference

NodeReference

@TestOnly

sealed class NodeReference<Input, Output>(source)

Represents a reference to a specific type of node within an AI agent subgraph. This sealed class provides a way to resolve and identify nodes for various processing operations in the context of a subgraph.

Parameters

Input

The input data type handled by the related node.

Output

The output data type produced by the related node.

Inheritors

Start

Finish

NamedNode

Types

Finish

class Finish<Output> : NodeReference<Output, Output>

Represents the final node in an AI agent subgraph workflow. The Finish node signifies the end of the processing within the subgraph and provides the result of the subgraph's operations.

NamedNode

open class NamedNode<Input, Output>(val name: String) : NodeReference<Input, Output>

NamedNode is a class that references a specific node within a subgraph by name. This class allows resolving a node instance by its unique name within a given subgraph.

Start

class Start<Input> : NodeReference<Input, Input>

The Start class is a specialized type of NodeReference that serves as a reference to the starting node of an AIAgentSubgraph. This node initiates the processing of the subgraph by resolving and returning the subgraph's starting node.

Strategy

class Strategy<Input, Output>(name: String) : NodeReference.SubgraphNode<Input, Output>

Represents a specific strategy within an AI agent subgraph.

SubgraphNode

open class SubgraphNode<Input, Output>(name: String) : NodeReference.NamedNode<Input, Output>

Represents a specialized type of node used in the context of AI agent subgraphs. This class extends the functionality of a named node by providing specific behavior to resolve itself as a valid subgraph within the overarching system.

Functions

resolve

abstract fun resolve(subgraph: AIAgentSubgraph<*, *>): AIAgentNodeBase<Input, Output>

Resolves the current node reference within the context of the provided AI agent subgraph.