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

AIAgentEdgeBuilderIntermediate

class AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>(source)

Represents an intermediate stage in the construction of a directed edge between two nodes in an AI agent strategy graph. This class provides mechanisms to define conditions and transformations that dictate how data flows and is processed between the nodes.

Parameters

IncomingOutput

The type of the output data produced by the originating node.

IntermediateOutput

The type of intermediate data produced after transformation or filtering.

OutgoingInput

The type of input data that the destination node expects.

Functions

onAssistantMessage

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onAssistantMessage(block: suspend (Message.Assistant) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, String, OutgoingInput>

Creates an edge that filters assistant messages based on a custom condition and extracts their content.

onAssistantMessageWithMedia

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onAssistantMessageWithMedia(block: suspend (Message.Assistant) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, Attachment, OutgoingInput>

Creates an edge that filters assistant messages based on a custom condition and provides access to media content.

onCondition

infix fun onCondition(block: suspend AIAgentContextBase.(output: IntermediateOutput) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>

Filters the intermediate outputs of the ai.koog.agents.core.agent.entity.AIAgentNode based on a specified condition.

onFailure

infix inline fun <IncomingOutput, OutgoingInput, TResult : ToolResult> AIAgentEdgeBuilderIntermediate<IncomingOutput, SafeTool.Result<TResult>, OutgoingInput>.onFailure(crossinline condition: suspend (error: String) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, SafeTool.Result.Failure<TResult>, OutgoingInput>

Defines a handler to process failure cases in a directed edge strategy by applying a condition to filter intermediate results of type SafeTool.Result.Failure. This method is used to specialize processing for failure results and to propagate or transform them based on the provided condition.

onIsInstance

infix inline fun <IncomingOutput, IntermediateOutput, OutgoingInput, T : Any> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onIsInstance(klass: KClass<T>): AIAgentEdgeBuilderIntermediate<IncomingOutput, T, OutgoingInput>

Creates an edge that filters outputs based on their type.

onMultipleAssistantMessages

infix fun <IncomingOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, List<Message.Response>, OutgoingInput>.onMultipleAssistantMessages(block: suspend (List<Message.Assistant>) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, List<Message.Assistant>, OutgoingInput>

Creates an edge that filters assistant messages based on a custom condition and extracts their content.

onMultipleToolCalls

infix fun <IncomingOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, List<Message.Response>, OutgoingInput>.onMultipleToolCalls(block: suspend (List<Message.Tool.Call>) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, List<Message.Tool.Call>, OutgoingInput>

Creates an edge that filters lists of tool call messages based on a custom condition.

onMultipleToolResults

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onMultipleToolResults(block: suspend (List<ReceivedToolResult>) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, List<ReceivedToolResult>, OutgoingInput>

Creates an edge that filters lists of tool result messages based on a custom condition.

onSuccessful

infix inline fun <IncomingOutput, OutgoingInput, TResult : ToolResult> AIAgentEdgeBuilderIntermediate<IncomingOutput, SafeTool.Result<TResult>, OutgoingInput>.onSuccessful(crossinline condition: suspend (TResult) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, SafeTool.Result.Success<TResult>, OutgoingInput>

Filters and transforms the intermediate outputs of the AI agent node based on the success results of a tool operation.

onToolCall

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onToolCall(tool: Tool<*, *>): AIAgentEdgeBuilderIntermediate<IncomingOutput, Message.Tool.Call, OutgoingInput>

Creates an edge that filters tool call messages for a specific tool.

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onToolCall(block: suspend (Message.Tool.Call) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, Message.Tool.Call, OutgoingInput>

Creates an edge that filters tool call messages based on a custom condition.

inline fun <IncomingOutput, IntermediateOutput, OutgoingInput, Args : ToolArgs> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onToolCall(tool: Tool<Args, *>, crossinline block: suspend (Args) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, Message.Tool.Call, OutgoingInput>

Creates an edge that filters tool call messages for a specific tool and arguments condition.

onToolNotCalled

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onToolNotCalled(tool: Tool<*, *>): AIAgentEdgeBuilderIntermediate<IncomingOutput, Message.Tool.Call, OutgoingInput>

Creates an edge that filters tool call messages to NOT be a specific tool

onToolResult

inline fun <IncomingOutput, IntermediateOutput, OutgoingInput, Result : ToolResult> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onToolResult(tool: Tool<*, Result>, crossinline block: suspend (SafeTool.Result<Result>) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, ReceivedToolResult, OutgoingInput>

Creates an edge that filters tool result messages for a specific tool and result condition.

transformed

infix fun <NewIntermediateOutput> transformed(block: suspend AIAgentContextBase.(IntermediateOutput) -> NewIntermediateOutput): AIAgentEdgeBuilderIntermediate<IncomingOutput, NewIntermediateOutput, OutgoingInput>

Transforms the intermediate output of the ai.koog.agents.core.agent.entity.AIAgentNode by applying a given transformation block.