feat: add timeout seconds args option when calling tool (mcp-use#178)

* feat: add timeout seconds args option when calling tool

* fix: unit test fix

Author: qwtsc

mcp_use/connectors/base.py

@@ -6,6 +6,7 @@
 """
 
 from abc import ABC, abstractmethod
+from datetime import timedelta
 from typing import Any
 
 from mcp import ClientSession, Implementation
@@ -266,12 +267,15 @@ async def _ensure_connected(self) -> None:
                     "Connection to MCP server has been lost. Auto-reconnection is disabled. Please reconnect manually."
                 )
 
-    async def call_tool(self, name: str, arguments: dict[str, Any]) -> CallToolResult:
+    async def call_tool(
+        self, name: str, arguments: dict[str, Any], read_timeout_seconds: timedelta | None = None
+    ) -> CallToolResult:
         """Call an MCP tool with automatic reconnection handling.
 
         Args:
             name: The name of the tool to call.
             arguments: The arguments to pass to the tool.
+            read_timeout_seconds: timeout seconds when calling tool
 
         Returns:
             The result of the tool call.
@@ -285,7 +289,7 @@ async def call_tool(self, name: str, arguments: dict[str, Any]) -> CallToolResul
 
         logger.debug(f"Calling tool '{name}' with arguments: {arguments}")
         try:
-            result = await self.client_session.call_tool(name, arguments)
+            result = await self.client_session.call_tool(name, arguments, read_timeout_seconds)
             logger.debug(f"Tool '{name}' called with result: {result}")
             return result
         except Exception as e:

tests/unit/test_http_connector.py

@@ -316,7 +316,7 @@ async def test_call_tool(self, _):
 
         result = await self.connector.call_tool("test_tool", {"param": "value"})
 
-        self.connector.client_session.call_tool.assert_called_once_with("test_tool", {"param": "value"})
+        self.connector.client_session.call_tool.assert_called_once_with("test_tool", {"param": "value"}, None)
         self.assertEqual(result, {"result": "success"})
 
     async def test_call_tool_no_client(self, _):

tests/unit/test_stdio_connector.py

@@ -280,7 +280,7 @@ async def test_call_tool(self):
         result = await connector.call_tool(tool_name, arguments)
 
         # Verify
-        mock_client.call_tool.assert_called_once_with(tool_name, arguments)
+        mock_client.call_tool.assert_called_once_with(tool_name, arguments, None)
         assert result == mock_result
 
     @pytest.mark.asyncio