JBAI-13757. Update the tracing feature with handling new events in an agent from the EventHandler class

- Update logic to set JSON config for remote writers in agent features to correctly process with dependencies;
- Add new events from the EventHandler class;
- Update all tests.

Author: sdubov

agents/agents-core/src/commonMain/kotlin/ai/grazie/code/agents/core/agent/AIAgentBase.kt

@@ -8,7 +8,7 @@ import ai.grazie.code.agents.core.environment.AgentEnvironmentUtils.mapToToolRes
 import ai.grazie.code.agents.core.environment.ReceivedToolResult
 import ai.grazie.code.agents.core.environment.TerminationTool
 import ai.grazie.code.agents.core.exception.AgentEngineException
-import ai.grazie.code.agents.core.feature.AIAgentPipeline
+import ai.grazie.code.agents.core.feature.AgentPipeline
 import ai.grazie.code.agents.core.feature.KotlinAIAgentFeature
 import ai.grazie.code.agents.core.model.AgentServiceError
 import ai.grazie.code.agents.core.model.AgentServiceErrorType
@@ -81,7 +81,7 @@ open class AIAgentBase(
 
     private val agentResultDeferred: CompletableDeferred<String?> = CompletableDeferred()
 
-    private val pipeline = AIAgentPipeline()
+    private val pipeline = AgentPipeline()
 
     init {
         cs.launch(context = Dispatchers.SuitableForIO, start = CoroutineStart.UNDISPATCHED) {
@@ -138,7 +138,7 @@ open class AIAgentBase(
 
     override suspend fun executeTools(toolCalls: List<Message.Tool.Call>): List<ReceivedToolResult> {
         logger.info { formatLog("Executing tools '$toolCalls'") }
-        pipeline.onBeforeToolCalls(toolCalls)
+        pipeline.onBeforeToolCalls(tools = toolCalls)
 
         val message = AgentToolCallsToEnvironmentMessage(
             sessionUuid = sessionUuid ?: throw IllegalStateException("Session UUID is null"),
@@ -153,7 +153,7 @@ open class AIAgentBase(
         )
 
         val results = processToolCallMultiple(message).mapToToolResult()
-        pipeline.onAfterToolCalls(results)
+        pipeline.onAfterToolCalls(tools = toolCalls, results)
         return results
     }
 
@@ -239,7 +239,7 @@ open class AIAgentBase(
             }
             catch (e: ToolException) {
 
-                pipeline.onToolValidationError(stage = stage, tool = tool, toolArgs = toolArgs, value = e.message)
+                pipeline.onToolValidationError(stage = stage, tool = tool, toolArgs = toolArgs, error = e.message)
 
                 return toolResult(
                     message = e.message,

agents/agents-core/src/commonMain/kotlin/ai/grazie/code/agents/core/agent/entity/LocalAgentStrategy.kt

@@ -15,7 +15,7 @@ import ai.grazie.code.agents.core.dsl.builder.forwardTo
 import ai.grazie.code.agents.core.dsl.extension.clearHistory
 import ai.grazie.code.agents.core.dsl.extension.nodeLLMCompressHistory
 import ai.grazie.code.agents.core.environment.AgentEnvironment
-import ai.grazie.code.agents.core.feature.AIAgentPipeline
+import ai.grazie.code.agents.core.feature.AgentPipeline
 import ai.grazie.code.agents.core.feature.PromptExecutorProxy
 import ai.grazie.utils.mpp.UUID
 import ai.jetbrains.code.prompt.executor.model.PromptExecutor
@@ -166,7 +166,7 @@ class LocalAgentStrategy(
         promptExecutor: PromptExecutor,
         environment: AgentEnvironment,
         config: LocalAgentConfig,
-        pipeline: AIAgentPipeline,
+        pipeline: AgentPipeline,
     ) {
         val stateManager = LocalAgentStateManager()
         val storage = LocalAgentStorage()

agents/agents-core/src/commonMain/kotlin/ai/grazie/code/agents/core/agent/entity/stage/LocalAgentStageContext.kt

@@ -9,7 +9,7 @@ import ai.grazie.code.agents.core.agent.entity.LocalAgentStorageKey
 import ai.grazie.code.agents.core.annotation.InternalAgentsApi
 import ai.grazie.code.agents.core.environment.AgentEnvironment
 import ai.grazie.code.agents.core.environment.SafeTool
-import ai.grazie.code.agents.core.feature.AIAgentPipeline
+import ai.grazie.code.agents.core.feature.AgentPipeline
 import ai.grazie.code.agents.core.feature.KotlinAIAgentFeature
 import ai.grazie.code.agents.core.tools.Tool
 import ai.grazie.code.agents.core.tools.ToolDescriptor
@@ -134,7 +134,7 @@ interface LocalAgentStageContext {
      * it is subject to changes or alterations in future releases.
      */
     @InternalAgentsApi
-    val pipeline: AIAgentPipeline
+    val pipeline: AgentPipeline
 
     /**
      * Retrieves a feature from the local agent's storage using the specified key.
@@ -191,7 +191,7 @@ interface LocalAgentStageContext {
         sessionUuid: UUID? = null,
         strategyId: String? = null,
         stageName: String? = null,
-        pipeline: AIAgentPipeline? = null,
+        pipeline: AgentPipeline? = null,
     ): LocalAgentStageContext
 }
 
@@ -223,7 +223,7 @@ class LocalAgentStageContextImpl constructor(
     override val sessionUuid: UUID,
     override val strategyId: String,
     override val stageName: String,
-    override val pipeline: AIAgentPipeline,
+    override val pipeline: AgentPipeline,
 ) : LocalAgentStageContext {
     /**
      * A map storing features associated with the current stage context.
@@ -291,7 +291,7 @@ class LocalAgentStageContextImpl constructor(
         sessionUuid: UUID?,
         strategyId: String?,
         stageName: String?,
-        pipeline: AIAgentPipeline?,
+        pipeline: AgentPipeline?,
     ) = LocalAgentStageContextImpl(
         environment = environment ?: this.environment,
         stageInput = stageInput ?: this.stageInput,