feat: add handle_tool_error and handle_validation_error to load_mcp_tools

Author: Chaitanya-Keyal

langchain_mcp_adapters/client.py

@@ -5,15 +5,16 @@
 """
 
 import asyncio
-from collections.abc import AsyncIterator
+from collections.abc import AsyncIterator, Callable
 from contextlib import asynccontextmanager
 from types import TracebackType
 from typing import Any
 
 from langchain_core.documents.base import Blob
 from langchain_core.messages import AIMessage, HumanMessage
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseTool, ToolException
 from mcp import ClientSession
+from pydantic import ValidationError
 
 from langchain_mcp_adapters.callbacks import CallbackContext, Callbacks
 from langchain_mcp_adapters.prompts import load_mcp_prompt
@@ -138,12 +139,22 @@ async def session(
                 await session.initialize()
             yield session
 
-    async def get_tools(self, *, server_name: str | None = None) -> list[BaseTool]:
+    async def get_tools(
+        self,
+        *,
+        server_name: str | None = None,
+        handle_tool_error: bool | str | Callable[[ToolException], str] | None = False,
+        handle_validation_error: (
+            bool | str | Callable[[ValidationError], str] | None
+        ) = False,
+    ) -> list[BaseTool]:
         """Get a list of all tools from all connected servers.
 
         Args:
             server_name: Optional name of the server to get tools from.
                 If None, all tools from all servers will be returned (default).
+            handle_tool_error: Optional error handler for tool execution errors.
+            handle_validation_error: Optional error handler for validation errors.
 
         NOTE: a new session will be created for each tool call
 
@@ -163,6 +174,8 @@ async def get_tools(self, *, server_name: str | None = None) -> list[BaseTool]:
                 connection=self.connections[server_name],
                 callbacks=self.callbacks,
                 server_name=server_name,
+                handle_tool_error=handle_tool_error,
+                handle_validation_error=handle_validation_error,
             )
 
         all_tools: list[BaseTool] = []
@@ -174,6 +187,8 @@ async def get_tools(self, *, server_name: str | None = None) -> list[BaseTool]:
                     connection=connection,
                     callbacks=self.callbacks,
                     server_name=name,
+                    handle_tool_error=handle_tool_error,
+                    handle_validation_error=handle_validation_error,
                 )
             )
             load_mcp_tool_tasks.append(load_mcp_tool_task)

langchain_mcp_adapters/tools.py

@@ -4,6 +4,7 @@
 tools, handle tool execution, and manage tool conversion between the two formats.
 """
 
+from collections.abc import Callable
 from typing import Any, cast, get_args
 
 from langchain_core.tools import (
@@ -25,7 +26,7 @@
     TextContent,
 )
 from mcp.types import Tool as MCPTool
-from pydantic import BaseModel, create_model
+from pydantic import BaseModel, ValidationError, create_model
 
 from langchain_mcp_adapters.callbacks import CallbackContext, Callbacks, _MCPCallbacks
 from langchain_mcp_adapters.sessions import Connection, create_session
@@ -112,6 +113,10 @@ def convert_mcp_tool_to_langchain_tool(
     connection: Connection | None = None,
     callbacks: Callbacks | None = None,
     server_name: str | None = None,
+    handle_tool_error: bool | str | Callable[[ToolException], str] | None = False,
+    handle_validation_error: (
+        bool | str | Callable[[ValidationError], str] | None
+    ) = False,
 ) -> BaseTool:
     """Convert an MCP tool to a LangChain tool.
 
@@ -124,6 +129,8 @@ def convert_mcp_tool_to_langchain_tool(
                     if a `session` is not provided
         callbacks: Optional callbacks for handling notifications and events
         server_name: Name of the server this tool belongs to
+        handle_tool_error: Optional error handler for tool execution errors.
+        handle_validation_error: Optional error handler for validation errors.
 
     Returns:
         a LangChain tool
@@ -192,6 +199,8 @@ async def call_tool(
         coroutine=call_tool,
         response_format="content_and_artifact",
         metadata=metadata,
+        handle_tool_error=handle_tool_error,
+        handle_validation_error=handle_validation_error,
     )
 
 
@@ -201,6 +210,10 @@ async def load_mcp_tools(
     connection: Connection | None = None,
     callbacks: Callbacks | None = None,
     server_name: str | None = None,
+    handle_tool_error: bool | str | Callable[[ToolException], str] | None = False,
+    handle_validation_error: (
+        bool | str | Callable[[ValidationError], str] | None
+    ) = False,
 ) -> list[BaseTool]:
     """Load all available MCP tools and convert them to LangChain tools.
 
@@ -209,6 +222,8 @@ async def load_mcp_tools(
         connection: Connection config to create a new session if session is None.
         callbacks: Optional callbacks for handling notifications and events.
         server_name: Name of the server these tools belong to.
+        handle_tool_error: Optional error handler for tool execution errors.
+        handle_validation_error: Optional error handler for validation errors.
 
     Returns:
         List of LangChain tools. Tool annotations are returned as part
@@ -247,6 +262,8 @@ async def load_mcp_tools(
             connection=connection,
             callbacks=callbacks,
             server_name=server_name,
+            handle_tool_error=handle_tool_error,
+            handle_validation_error=handle_validation_error,
         )
         for tool in tools
     ]