agents-core/ai.koog.agents.core.feature.pipeline/AIAgentGraphPipeline

AIAgentGraphPipeline

expect open class AIAgentGraphPipeline(agentConfig: AIAgentConfig, clock: Clock = Clock.System, basePipelineDelegate: AIAgentPipelineImpl = AIAgentPipelineImpl(agentConfig, clock)) : AIAgentPipeline, AIAgentGraphPipelineAPI(source)

Represents a pipeline for AI agent graph execution, extending the functionality of AIAgentPipeline. This class manages the execution of specific nodes in the pipeline using registered handlers.

actual open class AIAgentGraphPipeline(agentConfig: AIAgentConfig, clock: Clock, basePipelineDelegate: AIAgentPipelineImpl) : AIAgentPipeline, AIAgentGraphPipelineAPI(source)

Constructors

expect constructor(agentConfig: AIAgentConfig, clock: Clock = Clock.System, basePipelineDelegate: AIAgentPipelineImpl = AIAgentPipelineImpl(agentConfig, clock))

actual constructor(agentConfig: AIAgentConfig, clock: Clock, basePipelineDelegate: AIAgentPipelineImpl)

Properties

clock

expect open override val clock: Clock

Provides access to a Clock instance representing the current system time. The Clock can be used to retrieve the current time, create date-time instances, or perform operations based on a specific moment in time.

open override val clock: Clock

config

expect open override val config: AIAgentConfig

Represents the configuration settings for the AI agent.

open override val config: AIAgentConfig

Functions

closeAllFeaturesMessageProcessors

expect open suspend override fun closeAllFeaturesMessageProcessors()

open suspend override fun closeAllFeaturesMessageProcessors()

feature

expect open override fun <TFeature : Any> feature(featureClass: KClass<TFeature>, feature: AIAgentFeature<*, TFeature>): TFeature?

Retrieves a feature implementation from the current pipeline using the specified feature, if it is registered.

open override fun <TFeature : Any> feature(featureClass: KClass<TFeature>, feature: AIAgentFeature<*, TFeature>): TFeature?

install

expect fun <TConfig : FeatureConfig, TFeatureImpl : Any> install(feature: AIAgentGraphFeature<TConfig, TFeatureImpl>, configure: TConfig.() -> Unit)

Installs a feature into the pipeline with the provided configuration.

expect open override fun <TConfig : FeatureConfig, TFeatureImpl : Any> install(featureKey: AIAgentStorageKey<TFeatureImpl>, featureConfig: TConfig, featureImpl: TFeatureImpl)

Installs a feature into the AI agent storage using the provided feature key, configuration, and implementation.

actual fun <TConfig : FeatureConfig, TFeatureImpl : Any> install(feature: AIAgentGraphFeature<TConfig, TFeatureImpl>, configure: TConfig.() -> Unit)

open override fun <TConfig : FeatureConfig, TFeatureImpl : Any> install(featureKey: AIAgentStorageKey<TFeatureImpl>, featureConfig: TConfig, featureImpl: TFeatureImpl)

actual fun <TConfig : FeatureConfig, TFeatureImpl : Any> install(feature: AIAgentGraphFeature<TConfig, TFeatureImpl>, configure: TConfig.() -> Unit)

open override fun <TConfig : FeatureConfig, TFeatureImpl : Any> install(featureKey: AIAgentStorageKey<TFeatureImpl>, featureConfig: TConfig, featureImpl: TFeatureImpl)

interceptAfterLLMCall

expect open override fun ~~interceptAfterLLMCall~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallCompletedContext) -> Unit)

Intercepts LLM calls after they are made to process or log the response.

open override fun interceptAfterLLMCall(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallCompletedContext) -> Unit)

interceptAgentBeforeClose

expect open override fun ~~interceptAgentBeforeClose~~(feature: AIAgentFeature<*, *>, handle: suspend (AgentClosingContext) -> Unit)

Intercepts and sets a handler to be invoked before an agent is closed.

open override fun interceptAgentBeforeClose(feature: AIAgentFeature<*, *>, handle: suspend (AgentClosingContext) -> Unit)

interceptAgentClosing

expect open override fun interceptAgentClosing(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentClosingContext) -> Unit)

Intercepts and sets a handler to be invoked before an agent is closed.

open override fun interceptAgentClosing(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentClosingContext) -> Unit)

interceptAgentCompleted

expect open override fun interceptAgentCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentCompletedContext) -> Unit)

Intercepts the completion of an agent's operation and assigns a custom handler to process the result.

open override fun interceptAgentCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentCompletedContext) -> Unit)

interceptAgentExecutionFailed

expect open override fun interceptAgentExecutionFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentExecutionFailedContext) -> Unit)

Intercepts and handles errors occurring during the execution of an AI agent's strategy.

open override fun interceptAgentExecutionFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentExecutionFailedContext) -> Unit)

interceptAgentFinished

expect open override fun ~~interceptAgentFinished~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentCompletedContext) -> Unit)

Intercepts the completion of an agent's operation and assigns a custom handler to process the result.

open override fun interceptAgentFinished(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentCompletedContext) -> Unit)

interceptAgentRunError

expect open override fun ~~interceptAgentRunError~~(feature: AIAgentFeature<*, *>, handle: suspend (AgentExecutionFailedContext) -> Unit)

Intercepts and handles errors occurring during the execution of an AI agent's strategy.

open override fun interceptAgentRunError(feature: AIAgentFeature<*, *>, handle: suspend (AgentExecutionFailedContext) -> Unit)

interceptAgentStarting

expect open override fun interceptAgentStarting(feature: AIAgentFeature<*, *>, handle: suspend (AgentStartingContext) -> Unit)

Intercepts on before an agent started to modify or enhance the agent.

open override fun interceptAgentStarting(feature: AIAgentFeature<*, *>, handle: suspend (AgentStartingContext) -> Unit)

interceptBeforeAgentStarted

expect open override fun ~~interceptBeforeAgentStarted~~(feature: AIAgentFeature<*, *>, handle: suspend (AgentStartingContext) -> Unit)

Intercepts on before an agent started to modify or enhance the agent.

open override fun interceptBeforeAgentStarted(feature: AIAgentFeature<*, *>, handle: suspend (AgentStartingContext) -> Unit)

interceptBeforeLLMCall

expect open override fun ~~interceptBeforeLLMCall~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallStartingContext) -> Unit)

Intercepts LLM calls before they are made (deprecated name).

open override fun interceptBeforeLLMCall(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallStartingContext) -> Unit)

interceptEnvironmentCreated

expect open override fun interceptEnvironmentCreated(feature: AIAgentFeature<*, *>, handle: suspend (AgentEnvironmentTransformingContext, AIAgentEnvironment) -> AIAgentEnvironment)

Intercepts environment creation to allow features to modify or enhance the agent environment.

open override fun interceptEnvironmentCreated(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: AgentEnvironmentTransformingContext, environment: AIAgentEnvironment) -> AIAgentEnvironment)

interceptLLMCallCompleted

expect open override fun interceptLLMCallCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallCompletedContext) -> Unit)

Intercepts LLM calls after they are made to process or log the response.

open override fun interceptLLMCallCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallCompletedContext) -> Unit)

interceptLLMCallStarting

expect open override fun interceptLLMCallStarting(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallStartingContext) -> Unit)

Intercepts LLM calls before they are made to modify or log the prompt.

open override fun interceptLLMCallStarting(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMCallStartingContext) -> Unit)

interceptLLMStreamingCompleted

expect open override fun interceptLLMStreamingCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingCompletedContext) -> Unit)

Intercepts streaming operations after they complete to perform post-processing or cleanup.

open override fun interceptLLMStreamingCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingCompletedContext) -> Unit)

interceptLLMStreamingFailed

expect open override fun interceptLLMStreamingFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingFailedContext) -> Unit)

Intercepts errors during the streaming process.

open override fun interceptLLMStreamingFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingFailedContext) -> Unit)

interceptLLMStreamingFrameReceived

expect open override fun interceptLLMStreamingFrameReceived(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingFrameReceivedContext) -> Unit)

Intercepts stream frames as they are received during the streaming process.

open override fun interceptLLMStreamingFrameReceived(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingFrameReceivedContext) -> Unit)

interceptLLMStreamingStarting

expect open override fun interceptLLMStreamingStarting(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingStartingContext) -> Unit)

Intercepts streaming operations before they begin to modify or log the streaming request.

open override fun interceptLLMStreamingStarting(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: LLMStreamingStartingContext) -> Unit)

interceptNodeExecutionCompleted

expect open override fun interceptNodeExecutionCompleted(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: NodeExecutionCompletedContext) -> Unit)

Intercepts node execution after it completes.

open override fun interceptNodeExecutionCompleted(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: NodeExecutionCompletedContext) -> Unit)

interceptNodeExecutionFailed

expect open override fun interceptNodeExecutionFailed(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: NodeExecutionFailedContext) -> Unit)

Intercepts and handles node execution errors for a given feature.

open override fun interceptNodeExecutionFailed(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: NodeExecutionFailedContext) -> Unit)

interceptNodeExecutionStarting

expect open override fun interceptNodeExecutionStarting(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: NodeExecutionStartingContext) -> Unit)

Intercepts node execution before it starts.

open override fun interceptNodeExecutionStarting(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: NodeExecutionStartingContext) -> Unit)

interceptStrategyCompleted

expect open override fun interceptStrategyCompleted(feature: AIAgentFeature<*, *>, handle: suspend (StrategyCompletedContext) -> Unit)

Sets up an interceptor to handle the completion of a strategy for the given feature.

open override fun interceptStrategyCompleted(feature: AIAgentFeature<*, *>, handle: suspend (StrategyCompletedContext) -> Unit)

interceptStrategyFinished

expect open override fun ~~interceptStrategyFinished~~(feature: AIAgentFeature<*, *>, handle: suspend (StrategyCompletedContext) -> Unit)

Sets up an interceptor to handle the completion of a strategy for the given feature.

open override fun interceptStrategyFinished(feature: AIAgentFeature<*, *>, handle: suspend (StrategyCompletedContext) -> Unit)

interceptStrategyStart

expect open override fun ~~interceptStrategyStart~~(feature: AIAgentFeature<*, *>, handle: suspend (StrategyStartingContext) -> Unit)

Intercepts strategy started event to perform actions when an agent strategy begins execution.

open override fun interceptStrategyStart(feature: AIAgentFeature<*, *>, handle: suspend (StrategyStartingContext) -> Unit)

interceptStrategyStarting

expect open override fun interceptStrategyStarting(feature: AIAgentFeature<*, *>, handle: suspend (StrategyStartingContext) -> Unit)

Intercepts strategy started event to perform actions when an agent strategy begins execution.

open override fun interceptStrategyStarting(feature: AIAgentFeature<*, *>, handle: suspend (StrategyStartingContext) -> Unit)

interceptSubgraphExecutionCompleted

expect open override fun interceptSubgraphExecutionCompleted(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: SubgraphExecutionCompletedContext) -> Unit)

Intercepts the completion of a subgraph execution and allows handling of the event.

open override fun interceptSubgraphExecutionCompleted(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: SubgraphExecutionCompletedContext) -> Unit)

interceptSubgraphExecutionFailed

expect open override fun interceptSubgraphExecutionFailed(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: SubgraphExecutionFailedContext) -> Unit)

Intercepts and handles subgraph execution failures for a given feature.

open override fun interceptSubgraphExecutionFailed(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: SubgraphExecutionFailedContext) -> Unit)

interceptSubgraphExecutionStarting

expect open override fun interceptSubgraphExecutionStarting(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: SubgraphExecutionStartingContext) -> Unit)

Intercepts the execution of a subgraph when it starts.

open override fun interceptSubgraphExecutionStarting(feature: AIAgentGraphFeature<*, *>, handle: suspend (eventContext: SubgraphExecutionStartingContext) -> Unit)

interceptToolCall

expect open override fun ~~interceptToolCall~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallStartingContext) -> Unit)

Intercepts and handles tool calls for the specified feature and its implementation. Updates the tool call handler for the given feature key with a custom handler.

open override fun interceptToolCall(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallStartingContext) -> Unit)

interceptToolCallCompleted

expect open override fun interceptToolCallCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallCompletedContext) -> Unit)

Intercepts the result of a tool call with a custom handler for a specific feature.

open override fun interceptToolCallCompleted(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallCompletedContext) -> Unit)

interceptToolCallFailed

expect open override fun interceptToolCallFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallFailedContext) -> Unit)

Sets up an interception mechanism to handle tool call failures for a specific feature.

open override fun interceptToolCallFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallFailedContext) -> Unit)

interceptToolCallFailure

expect open override fun ~~interceptToolCallFailure~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallFailedContext) -> Unit)

Sets up an interception mechanism to handle tool call failures for a specific feature.

open override fun interceptToolCallFailure(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallFailedContext) -> Unit)

interceptToolCallResult

expect open override fun ~~interceptToolCallResult~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallCompletedContext) -> Unit)

Intercepts the result of a tool call with a custom handler for a specific feature.

open override fun interceptToolCallResult(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallCompletedContext) -> Unit)

interceptToolCallStarting

expect open override fun interceptToolCallStarting(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallStartingContext) -> Unit)

Intercepts and handles tool calls for the specified feature.

open override fun interceptToolCallStarting(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolCallStartingContext) -> Unit)

interceptToolValidationError

expect open override fun ~~interceptToolValidationError~~(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolValidationFailedContext) -> Unit)

Intercepts validation errors encountered during the execution of tools associated with the specified feature.

open override fun interceptToolValidationError(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolValidationFailedContext) -> Unit)

interceptToolValidationFailed

expect open override fun interceptToolValidationFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolValidationFailedContext) -> Unit)

Intercepts validation errors encountered during the execution of tools associated with the specified feature.

open override fun interceptToolValidationFailed(feature: AIAgentFeature<*, *>, handle: suspend (eventContext: ToolValidationFailedContext) -> Unit)

javaApiInterceptAgentClosing

fun javaApiInterceptAgentClosing(feature: AIAgentFeature<*, *>, handle: Interceptor<AgentClosingContext>)

Intercepts and sets a handler to be invoked before an agent is closed.

javaApiInterceptAgentCompleted

fun javaApiInterceptAgentCompleted(feature: AIAgentFeature<*, *>, handle: Interceptor<AgentCompletedContext>)

Intercepts the completion of an agent's operation and assigns a custom handler to process the result.

javaApiInterceptAgentExecutionFailed

fun javaApiInterceptAgentExecutionFailed(feature: AIAgentFeature<*, *>, handle: Interceptor<AgentExecutionFailedContext>)

Intercepts and handles errors occurring during the execution of an AI agent's strategy.

javaApiInterceptAgentStarting

fun javaApiInterceptAgentStarting(feature: AIAgentFeature<*, *>, handle: Interceptor<AgentStartingContext>)

Intercepts on before an agent started to modify or enhance the agent.

javaApiInterceptEnvironmentCreated

fun javaApiInterceptEnvironmentCreated(feature: AIAgentFeature<*, *>, transform: TransformInterceptor<AgentEnvironmentTransformingContext, AIAgentEnvironment>)

Intercepts environment creation to allow features to modify or enhance the agent environment.

javaApiInterceptLLMCallCompleted

fun javaApiInterceptLLMCallCompleted(feature: AIAgentFeature<*, *>, handle: Interceptor<LLMCallCompletedContext>)

Intercepts LLM calls after they are made to process or log the response.

javaApiInterceptLLMCallStarting

fun javaApiInterceptLLMCallStarting(feature: AIAgentFeature<*, *>, handle: Interceptor<LLMCallStartingContext>)

Intercepts LLM calls before they are made to modify or log the prompt.

javaApiInterceptLLMStreamingCompleted

fun javaApiInterceptLLMStreamingCompleted(feature: AIAgentFeature<*, *>, handle: Interceptor<LLMStreamingCompletedContext>)

Intercepts streaming operations after they complete to perform post-processing or cleanup.

javaApiInterceptLLMStreamingFailed

fun javaApiInterceptLLMStreamingFailed(feature: AIAgentFeature<*, *>, handle: Interceptor<LLMStreamingFailedContext>)

Intercepts errors during the streaming process.

javaApiInterceptLLMStreamingFrameReceived

fun javaApiInterceptLLMStreamingFrameReceived(feature: AIAgentFeature<*, *>, handle: Interceptor<LLMStreamingFrameReceivedContext>)

Intercepts stream frames as they are received during the streaming process.

javaApiInterceptLLMStreamingStarting

fun javaApiInterceptLLMStreamingStarting(feature: AIAgentFeature<*, *>, handle: Interceptor<LLMStreamingStartingContext>)

Intercepts streaming operations before they begin to modify or log the streaming request.

javaApiInterceptNodeExecutionCompleted

fun javaApiInterceptNodeExecutionCompleted(feature: AIAgentGraphFeature<*, *>, handle: Interceptor<NodeExecutionCompletedContext>)

Intercepts node execution after it is successfully completed.

javaApiInterceptNodeExecutionFailed

fun javaApiInterceptNodeExecutionFailed(feature: AIAgentGraphFeature<*, *>, handle: Interceptor<NodeExecutionFailedContext>)

Intercepts node execution when it fails.

javaApiInterceptNodeExecutionStarting

fun javaApiInterceptNodeExecutionStarting(feature: AIAgentGraphFeature<*, *>, handle: Interceptor<NodeExecutionStartingContext>)

Intercepts node execution before it starts.

javaApiInterceptStrategyCompleted

fun javaApiInterceptStrategyCompleted(feature: AIAgentFeature<*, *>, handle: Interceptor<StrategyCompletedContext>)

Sets up an interceptor to handle the completion of a strategy for the given feature.

javaApiInterceptStrategyStarting

fun javaApiInterceptStrategyStarting(feature: AIAgentFeature<*, *>, handle: Interceptor<StrategyStartingContext>)

Intercepts strategy started event to perform actions when an agent strategy begins execution.

javaApiInterceptSubgraphExecutionCompleted

fun javaApiInterceptSubgraphExecutionCompleted(feature: AIAgentGraphFeature<*, *>, handle: Interceptor<SubgraphExecutionCompletedContext>)

Intercepts subgraph execution after it is successfully completed.

javaApiInterceptSubgraphExecutionFailed

fun javaApiInterceptSubgraphExecutionFailed(feature: AIAgentGraphFeature<*, *>, handle: Interceptor<SubgraphExecutionFailedContext>)

Intercepts subgraph execution when it fails.

javaApiInterceptSubgraphExecutionStarting

fun javaApiInterceptSubgraphExecutionStarting(feature: AIAgentGraphFeature<*, *>, handle: Interceptor<SubgraphExecutionStartingContext>)

Intercepts subgraph execution before it starts.

javaApiInterceptToolCallCompleted

fun javaApiInterceptToolCallCompleted(feature: AIAgentFeature<*, *>, handle: Interceptor<ToolCallCompletedContext>)

Intercepts the result of a tool call with a custom handler for a specific feature.

javaApiInterceptToolCallFailed

fun javaApiInterceptToolCallFailed(feature: AIAgentFeature<*, *>, handle: Interceptor<ToolCallFailedContext>)

Sets up an interception mechanism to handle tool call failures for a specific feature.

javaApiInterceptToolCallStarting

fun javaApiInterceptToolCallStarting(feature: AIAgentFeature<*, *>, handle: Interceptor<ToolCallStartingContext>)

Intercepts and handles tool calls for the specified feature.

javaApiInterceptToolValidationFailed

fun javaApiInterceptToolValidationFailed(feature: AIAgentFeature<*, *>, handle: Interceptor<ToolValidationFailedContext>)

Intercepts validation errors encountered during the execution of tools associated with the specified feature.

onAgentClosing

expect open suspend override fun onAgentClosing(eventId: String, executionInfo: AgentExecutionInfo, agentId: String)

Invoked before an agent is closed to perform necessary pre-closure operations.

open suspend override fun onAgentClosing(eventId: String, executionInfo: AgentExecutionInfo, agentId: String)

onAgentCompleted

expect open suspend override fun onAgentCompleted(eventId: String, executionInfo: AgentExecutionInfo, agentId: String, runId: String, result: Any?, context: AIAgentContext)

Notifies all registered handlers that an agent has finished execution.

open suspend override fun onAgentCompleted(eventId: String, executionInfo: AgentExecutionInfo, agentId: String, runId: String, result: Any?, context: AIAgentContext)

onAgentEnvironmentTransforming

expect open suspend override fun onAgentEnvironmentTransforming(eventId: String, executionInfo: AgentExecutionInfo, agent: GraphAIAgent<*, *>, baseEnvironment: AIAgentEnvironment): AIAgentEnvironment

Transforms the agent environment by applying all registered environment transformers.

open suspend override fun onAgentEnvironmentTransforming(eventId: String, executionInfo: AgentExecutionInfo, agent: GraphAIAgent<*, *>, baseEnvironment: AIAgentEnvironment): AIAgentEnvironment

onAgentExecutionFailed

expect open suspend override fun onAgentExecutionFailed(eventId: String, executionInfo: AgentExecutionInfo, agentId: String, runId: String, throwable: Throwable, context: AIAgentContext)

Notifies all registered handlers about an error that occurred during agent execution.

open suspend override fun onAgentExecutionFailed(eventId: String, executionInfo: AgentExecutionInfo, agentId: String, runId: String, throwable: Throwable, context: AIAgentContext)

onAgentStarting

expect open suspend override fun <TInput, TOutput> onAgentStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, agent: AIAgent<*, *>, context: AIAgentContext)

Notifies all registered handlers that an agent has started execution.

open suspend override fun <TInput, TOutput> onAgentStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, agent: AIAgent<*, *>, context: AIAgentContext)

onLLMCallCompleted

expect open suspend override fun onLLMCallCompleted(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, responses: List<Message.Response>, moderationResponse: ModerationResult?, context: AIAgentContext)

Notifies all registered LLM handlers after a language model call has completed.

open suspend override fun onLLMCallCompleted(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, responses: List<Message.Response>, moderationResponse: ModerationResult?, context: AIAgentContext)

onLLMCallStarting

expect open suspend override fun onLLMCallStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, context: AIAgentContext)

Notifies all registered LLM handlers before a language model call is made.

open suspend override fun onLLMCallStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, context: AIAgentContext)

onLLMStreamingCompleted

expect open suspend override fun onLLMStreamingCompleted(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, context: AIAgentContext)

Invoked after streaming from a language model completes.

open suspend override fun onLLMStreamingCompleted(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, context: AIAgentContext)

onLLMStreamingFailed

expect open suspend override fun onLLMStreamingFailed(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, throwable: Throwable, context: AIAgentContext)

Invoked if an error occurs during the streaming process.

open suspend override fun onLLMStreamingFailed(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, throwable: Throwable, context: AIAgentContext)

onLLMStreamingFrameReceived

expect open suspend override fun onLLMStreamingFrameReceived(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, streamFrame: StreamFrame, context: AIAgentContext)

Invoked when a stream frame is received during the streaming process.

open suspend override fun onLLMStreamingFrameReceived(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, streamFrame: StreamFrame, context: AIAgentContext)

onLLMStreamingStarting

expect open suspend override fun onLLMStreamingStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, context: AIAgentContext)

Invoked before streaming from a language model begins.

open suspend override fun onLLMStreamingStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>, context: AIAgentContext)

onNodeExecutionCompleted

expect open suspend override fun onNodeExecutionCompleted(eventId: String, executionInfo: AgentExecutionInfo, node: AIAgentNodeBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, output: Any?, outputType: TypeToken)

Notifies all registered node handlers after a node has been executed.

open suspend override fun onNodeExecutionCompleted(eventId: String, executionInfo: AgentExecutionInfo, node: AIAgentNodeBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, output: Any?, outputType: TypeToken)

onNodeExecutionFailed

expect open suspend override fun onNodeExecutionFailed(eventId: String, executionInfo: AgentExecutionInfo, node: AIAgentNodeBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, throwable: Throwable)

Handles errors occurring during the execution of a node by invoking all registered node execution error handlers.

open suspend override fun onNodeExecutionFailed(eventId: String, executionInfo: AgentExecutionInfo, node: AIAgentNodeBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, throwable: Throwable)

onNodeExecutionStarting

expect open suspend override fun onNodeExecutionStarting(eventId: String, executionInfo: AgentExecutionInfo, node: AIAgentNodeBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken)

Notifies all registered node handlers before a node is executed.

open suspend override fun onNodeExecutionStarting(eventId: String, executionInfo: AgentExecutionInfo, node: AIAgentNodeBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken)

onStrategyCompleted

expect open suspend override fun onStrategyCompleted(eventId: String, executionInfo: AgentExecutionInfo, strategy: AIAgentStrategy<*, *, *>, context: AIAgentContext, result: Any?, resultType: TypeToken)

Notifies all registered strategy handlers that a strategy has finished execution.

open suspend override fun onStrategyCompleted(eventId: String, executionInfo: AgentExecutionInfo, strategy: AIAgentStrategy<*, *, *>, context: AIAgentContext, result: Any?, resultType: TypeToken)

onStrategyStarting

expect open suspend override fun onStrategyStarting(eventId: String, executionInfo: AgentExecutionInfo, strategy: AIAgentStrategy<*, *, *>, context: AIAgentContext)

Notifies all registered strategy handlers that a strategy has started execution.

open suspend override fun onStrategyStarting(eventId: String, executionInfo: AgentExecutionInfo, strategy: AIAgentStrategy<*, *, *>, context: AIAgentContext)

onSubgraphExecutionCompleted

expect open suspend override fun onSubgraphExecutionCompleted(eventId: String, executionInfo: AgentExecutionInfo, subgraph: AIAgentSubgraphBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, output: Any?, outputType: TypeToken)

Notifies all registered subgraph handlers after a subgraph has been executed.

open suspend override fun onSubgraphExecutionCompleted(eventId: String, executionInfo: AgentExecutionInfo, subgraph: AIAgentSubgraphBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, output: Any?, outputType: TypeToken)

onSubgraphExecutionFailed

expect open suspend override fun onSubgraphExecutionFailed(eventId: String, executionInfo: AgentExecutionInfo, subgraph: AIAgentSubgraphBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, throwable: Throwable)

Notifies all registered subgraph handlers when a subgraph execution fails.

open suspend override fun onSubgraphExecutionFailed(eventId: String, executionInfo: AgentExecutionInfo, subgraph: AIAgentSubgraphBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken, throwable: Throwable)

onSubgraphExecutionStarting

expect open suspend override fun onSubgraphExecutionStarting(eventId: String, executionInfo: AgentExecutionInfo, subgraph: AIAgentSubgraphBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken)

Notifies all registered subgraph handlers before a subgraph is executed.

open suspend override fun onSubgraphExecutionStarting(eventId: String, executionInfo: AgentExecutionInfo, subgraph: AIAgentSubgraphBase<*, *>, context: AIAgentGraphContextBase, input: Any?, inputType: TypeToken)

onToolCallCompleted

expect open suspend override fun onToolCallCompleted(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, toolResult: JSONElement?, context: AIAgentContext)

Notifies all registered tool handlers about the result of a tool call.

open suspend override fun onToolCallCompleted(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, toolResult: JSONElement?, context: AIAgentContext)

onToolCallFailed

expect open suspend override fun onToolCallFailed(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, message: String, error: AIAgentError?, context: AIAgentContext)

Notifies all registered tool handlers when a tool call fails with an exception.

open suspend override fun onToolCallFailed(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, message: String, error: AIAgentError?, context: AIAgentContext)

onToolCallStarting

expect open suspend override fun onToolCallStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, context: AIAgentContext)

Notifies all registered tool handlers when a tool is called.

open suspend override fun onToolCallStarting(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, context: AIAgentContext)

onToolValidationFailed

expect open suspend override fun onToolValidationFailed(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, message: String, error: AIAgentError, context: AIAgentContext)

Notifies all registered tool handlers when a validation error occurs during a tool call.

open suspend override fun onToolValidationFailed(eventId: String, executionInfo: AgentExecutionInfo, runId: String, toolCallId: String?, toolName: String, toolDescription: String?, toolArgs: JSONObject, message: String, error: AIAgentError, context: AIAgentContext)

prepareFeatures

expect open suspend override fun prepareFeatures()

open suspend override fun prepareFeatures()

uninstall

expect open suspend override fun uninstall(featureKey: AIAgentStorageKey<*>)

Uninstalls the specified feature associated with the given key from storage.

open suspend override fun uninstall(featureKey: AIAgentStorageKey<*>)