andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-20 05:18:16 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: allow custom client_timeout for MCPAdapter (#409)

Browse Source
This commit is contained in:
Lucas Gomide
2025-08-05 10:53:57 -03:00
committed by GitHub
parent 9220cfba28
commit 4daf18256d
2 changed files with 50 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/crewai_tools/adapters/mcp_adapter.py
View File

@@ -50,13 +50,17 @@ class MCPServerAdapter:
with MCPServerAdapter(..., "tool1", "tool2") as filtered_tools:
# only tool1 and tool2 are available
# context manager with custom connect timeout (60 seconds)
with MCPServerAdapter(..., connect_timeout=60) as tools:
# tools is now available with longer timeout
# manually stop mcp server
try:
mcp_server = MCPServerAdapter(...)
tools = mcp_server.tools # all tools
# or with filtered tools
mcp_server = MCPServerAdapter(..., "tool1", "tool2")
# or with filtered tools and custom timeout
mcp_server = MCPServerAdapter(..., "tool1", "tool2", connect_timeout=45)
filtered_tools = mcp_server.tools # only tool1 and tool2
...
finally:
@@ -70,6 +74,7 @@ class MCPServerAdapter:
self,
serverparams: StdioServerParameters | dict[str, Any],
*tool_names: str,
connect_timeout: int = 30,
):
"""Initialize the MCP Server
@@ -78,6 +83,7 @@ class MCPServerAdapter:
`StdioServerParameters` or a `dict` respectively for STDIO and SSE.
*tool_names: Optional names of tools to filter. If provided, only tools with
matching names will be available.
connect_timeout: Connection timeout in seconds to the MCP server (default is 30s).
"""
@@ -106,7 +112,7 @@ class MCPServerAdapter:
try:
self._serverparams = serverparams
self._adapter = MCPAdapt(self._serverparams, CrewAIAdapter())
self._adapter = MCPAdapt(self._serverparams, CrewAIAdapter(), connect_timeout)
self.start()
except Exception as e: