add call tool timeout

mayinghan · mayinghan · commit 0bf03cd6b438 · 2025-08-25T17:20:40.000-07:00
diff --git a/eval_protocol/mcp/client/connection.py b/eval_protocol/mcp/client/connection.py
@@ -69,7 +69,7 @@ async def initialize_session(self, session: MCPSession) -> None:
         )
 
         mcp_session = await exit_stack.enter_async_context(
-            ClientSession(read_stream, write_stream, client_info=client_info)
+            ClientSession(read_stream, write_stream, client_info=client_info, read_timeout_seconds=30.0)
         )
 
         await mcp_session.initialize()

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ async def initialize_session(self, session: MCPSession) -> None:`
`69`	`69`	`)`
`70`	`70`
`71`	`71`	`mcp_session = await exit_stack.enter_async_context(`
`72`		`- ClientSession(read_stream, write_stream, client_info=client_info)`
	`72`	`+ ClientSession(read_stream, write_stream, client_info=client_info, read_timeout_seconds=30.0)`
`73`	`73`	`)`
`74`	`74`
`75`	`75`	`await mcp_session.initialize()`