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

AIAgentPipelineImpl

class AIAgentPipelineImpl(val config: AIAgentConfig, val clock: Clock) : AIAgentPipelineAPI(source)

Default implementation of AIAgentPipelineAPI

Constructors

AIAgentPipelineImpl

constructor(config: AIAgentConfig, clock: Clock)

Properties

clock

open override val clock: Clock

config

open override val config: AIAgentConfig

Functions

closeAllFeaturesMessageProcessors

open suspend override fun closeAllFeaturesMessageProcessors()

feature

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

install

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

interceptAfterLLMCall

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

interceptAgentBeforeClose

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

interceptAgentClosing

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

interceptAgentCompleted

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

interceptAgentExecutionFailed

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

interceptAgentFinished

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

interceptAgentRunError

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

interceptAgentStarting

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

interceptBeforeAgentStarted

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

interceptBeforeLLMCall

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

interceptEnvironmentCreated

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

interceptLLMCallCompleted

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

interceptLLMCallStarting

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

interceptLLMStreamingCompleted

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

interceptLLMStreamingFailed

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

interceptLLMStreamingFrameReceived

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

interceptLLMStreamingStarting

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

interceptStrategyCompleted

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

interceptStrategyFinished

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

interceptStrategyStart

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

interceptStrategyStarting

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

interceptToolCall

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

interceptToolCallCompleted

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

interceptToolCallFailed

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

interceptToolCallFailure

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

interceptToolCallResult

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

interceptToolCallStarting

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

interceptToolValidationError

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

interceptToolValidationFailed

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

onAgentClosing

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

onAgentCompleted

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

onAgentEnvironmentTransforming

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

onAgentExecutionFailed

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

onAgentStarting

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

onLLMCallCompleted

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

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

onLLMStreamingCompleted

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

onLLMStreamingFailed

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

onLLMStreamingFrameReceived

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

onLLMStreamingStarting

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

onStrategyCompleted

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

onStrategyStarting

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

onToolCallCompleted

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

onToolCallFailed

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

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

onToolValidationFailed

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

open suspend override fun prepareFeatures()

uninstall

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