tool-error-callback

Author: tripolskypetr

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-swarm-kit",
-  "version": "1.1.92",
+  "version": "1.1.93",
   "description": "A TypeScript library for building orchestrated framework-agnostic multi-agent AI systems",
   "author": {
     "name": "Petr Tripolsky",

package.json

@@ -248,6 +248,8 @@ export class MergeMCP implements IMCP {
           );
           {
             const { callbacks = {} } = swarm.agentSchemaService.get(agentName);
+            callbacks.onToolError &&
+              callbacks.onToolError(dto.clientId, dto.agentName, error);
             callbacks.onResurrect &&
               callbacks.onResurrect(
                 dto.clientId,

src/classes/MCP.ts

@@ -153,6 +153,12 @@ const createToolCall = async (
         tool.function.arguments,
         error
       );
+    self.params.callbacks?.onToolError &&
+      self.params.callbacks.onToolError(
+        self.params.clientId,
+        self.params.agentName,
+        error
+      );
     self._toolErrorSubject.next(TOOL_ERROR_SYMBOL);
   }
 };

src/client/ClientAgent.ts

@@ -283,6 +283,20 @@ export interface IAgentSchemaInternalCallbacks {
     request: IToolRequest[]
   ) => void;
 
+  /**
+   * Optional callback triggered when a tool throw an error
+   * This callback is used to log the error before resurrect
+   * 
+   * @param {string} clientId - The ID of the client interacting with the agent.
+   * @param {AgentName} agentName - The name of the agent making the tool request.
+   * @param {IToolRequest} request - The content of the tool request.
+   */
+  onToolError?: (
+    clientId: string,
+    agentName: AgentName,
+    error: Error,
+  ) => void;
+
   /**
    * Optional callback triggered when an assistant message is committed.
    * @param {string} clientId - The ID of the client interacting with the agent.

src/interfaces/Agent.interface.ts

@@ -3490,6 +3490,15 @@ interface IAgentSchemaInternalCallbacks {
      * @param {IToolRequest} request - The content of the tool request.
      */
     onToolRequest?: (clientId: string, agentName: AgentName, request: IToolRequest[]) => void;
+    /**
+     * Optional callback triggered when a tool throw an error
+     * This callback is used to log the error before resurrect
+     *
+     * @param {string} clientId - The ID of the client interacting with the agent.
+     * @param {AgentName} agentName - The name of the agent making the tool request.
+     * @param {IToolRequest} request - The content of the tool request.
+     */
+    onToolError?: (clientId: string, agentName: AgentName, error: Error) => void;
     /**
      * Optional callback triggered when an assistant message is committed.
      * @param {string} clientId - The ID of the client interacting with the agent.