fix timeout

Author: mayinghan

eval_protocol/mcp/client/connection.py

@@ -6,6 +6,7 @@
 """
 
 import asyncio
+from datetime import timedelta
 import hashlib
 import json
 import logging
@@ -69,7 +70,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, read_timeout_seconds=30.0)
+            ClientSession(read_stream, write_stream, client_info=client_info, read_timeout_seconds=timedelta(seconds=30))
         )
 
         await mcp_session.initialize()
@@ -414,7 +415,7 @@ async def call_tool(self, session: MCPSession, tool_name: str, arguments: Dict)
         while retry_count > 0:
             try:
                 logging.info(f"Session {session.session_id} Row {session.dataset_row.id}: Tool {tool_name} call started")
-                tool_result = await mcp_session.call_tool(tool_name, arguments, read_timeout_seconds=30.0)
+                tool_result = await mcp_session.call_tool(tool_name, arguments, read_timeout_seconds=timedelta(seconds=30))
                 logging.info(f"Session {session.session_id} Row {session.dataset_row.id}: Tool {tool_name} call completed")
                 break
             except httpx.TimeoutException: