aligned callbacks between Py and TS

Author: brnaba-aws

typescript/src/agents/agent.ts

@@ -41,15 +41,6 @@ export class AgentCallbacks {
    * Provides default implementations that can be overridden by subclasses.
    */
 
-  async onLlmNewToken(_token: string): Promise<void> {
-      /**
-       * Called when a new token is generated by the LLM.
-       *
-       * @param token The new token generated
-       */
-      // Default implementation does nothing
-  }
-
   async onAgentStart(
       _agentName: string,
       _input: any,
@@ -77,7 +68,7 @@ export class AgentCallbacks {
       // Default implementation does nothing
   }
 
-  async onAgentStop(
+  async onAgentEnd(
       _agentName: string,
       _response: any,
       _messages: any[],
@@ -129,7 +120,17 @@ export class AgentCallbacks {
       // Default implementation does nothing
   }
 
-  async onLlmStop(
+  async onLlmNewToken(_token: string, ..._kwargs: any[]): Promise<void> {
+    /**
+     * Called when a new token is generated by the LLM.
+     *
+     * @param token The new token generated
+     * @param kwargs Additional keyword arguments that might be passed to the callback
+     */
+    // Default implementation does nothing
+}
+
+  async onLlmEnd(
       _name: string,
       _output: any,
       _runId?: string,

typescript/src/utils/tool.ts

@@ -61,6 +61,7 @@ export class AgentToolCallbacks {
 
   onToolError(
       _toolName: string,
+      _input: any,
       _error: Error,
       _runId?: string,
       _tags?: string[],
@@ -71,6 +72,7 @@ export class AgentToolCallbacks {
        * Callback method that runs when a tool encounters an error.
        *
        * @param toolName Name of the tool that encountered an error
+       * @param input Object containing the tool's input
        * @param error The error that occurred
        * @param runId Unique identifier for this specific tool run
        * @param tags Optional list of string tags associated with this tool run