1
1
package ai.koog.agents.core.feature
2
2
3
- import ai.koog.agents.core.agent.GraphAIAgent
4
3
import ai.koog.agents.core.agent.context.AIAgentContextBase
5
4
import ai.koog.agents.core.agent.entity.AIAgentGraphStrategy
6
5
import ai.koog.agents.core.agent.entity.AIAgentStorageKey
@@ -18,9 +17,7 @@ import ai.koog.agents.core.feature.handler.AgentFinishedHandler
18
17
import ai.koog.agents.core.feature.handler.AgentHandler
19
18
import ai.koog.agents.core.feature.handler.AgentRunErrorContext
20
19
import ai.koog.agents.core.feature.handler.AgentRunErrorHandler
21
- import ai.koog.agents.core.feature.handler.AgentStartContext
22
20
import ai.koog.agents.core.feature.handler.AgentTransformEnvironmentContext
23
- import ai.koog.agents.core.feature.handler.BeforeAgentStartedHandler
24
21
import ai.koog.agents.core.feature.handler.BeforeLLMCallContext
25
22
import ai.koog.agents.core.feature.handler.BeforeLLMCallHandler
26
23
import ai.koog.agents.core.feature.handler.ExecuteLLMHandler
@@ -143,7 +140,6 @@ public abstract class AIAgentPipeline {
143
140
144
141
// region Trigger Agent Handlers
145
142
146
-
147
143
/* *
148
144
* Notifies all registered handlers that an agent has finished execution.
149
145
*
@@ -322,7 +318,8 @@ public abstract class AIAgentPipeline {
322
318
toolArgs : ToolArgs ,
323
319
error : String
324
320
) {
325
- val eventContext = ToolValidationErrorContext (runId, toolCallId, tool, toolArgs, error)
321
+ val eventContext =
322
+ ToolValidationErrorContext (runId, toolCallId, tool, toolArgs, error)
326
323
executeToolHandlers.values.forEach { handler -> handler.toolValidationErrorHandler.handle(eventContext) }
327
324
}
328
325
@@ -419,7 +416,6 @@ public abstract class AIAgentPipeline {
419
416
existingHandler.environmentTransformer = AgentEnvironmentTransformer { context, env -> context.transform(env) }
420
417
}
421
418
422
-
423
419
/* *
424
420
* Intercepts the completion of an agent's operation and assigns a custom handler to process the result.
425
421
*
0 commit comments