Skip to content

Commit 365d190

Browse files
committed
add read timeout on tool call request
1 parent 8f423c5 commit 365d190

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

src/mcp/client/session.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,10 @@ async def unsubscribe_resource(self, uri: AnyUrl) -> types.EmptyResult:
254254
)
255255

256256
async def call_tool(
257-
self, name: str, arguments: dict[str, Any] | None = None
257+
self,
258+
name: str,
259+
arguments: dict[str, Any] | None = None,
260+
read_timeout_seconds: timedelta | None = None,
258261
) -> types.CallToolResult:
259262
"""Send a tools/call request."""
260263
return await self.send_request(
@@ -265,6 +268,7 @@ async def call_tool(
265268
)
266269
),
267270
types.CallToolResult,
271+
request_read_timeout_seconds=read_timeout_seconds,
268272
)
269273

270274
async def list_prompts(self) -> types.ListPromptsResult:

src/mcp/shared/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ async def send_request(
213213
self,
214214
request: SendRequestT,
215215
result_type: type[ReceiveResultT],
216-
request_read_timeout_seconds: timedelta | None = None
216+
request_read_timeout_seconds: timedelta | None = None,
217217
) -> ReceiveResultT:
218218
"""
219219
Sends a request and wait for a response. Raises an McpError if the

0 commit comments

Comments
 (0)