From 390e0f99ae04ad497ab5cbeff32d9ca92f494244 Mon Sep 17 00:00:00 2001 From: Derek Xu Date: Wed, 13 Aug 2025 13:55:08 -0700 Subject: [PATCH] better exception --- eval_protocol/mcp/execution/manager.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/eval_protocol/mcp/execution/manager.py b/eval_protocol/mcp/execution/manager.py index e0d101a7..e30f67d5 100644 --- a/eval_protocol/mcp/execution/manager.py +++ b/eval_protocol/mcp/execution/manager.py @@ -466,12 +466,13 @@ async def _execute_rollout( trajectory.control_plane_summary.update({"error_message": f"{failure_reason}"}) try: await envs.connection_manager.reset_session(session) - except: # noqa: E722 - logger.error(f"Error resetting session {session.session_id}") + except Exception as e: + logger.warning(f"Failed to reset session {session.session_id}: {type(e).__name__}: {e}", exc_info=True) + try: await envs.connection_manager.close_session(session) - except: # noqa: E722 - logger.error(f"Error closing session {session.session_id}") + except Exception as e: + logger.warning(f"Failed to close session {session.session_id}: {type(e).__name__}: {e}", exc_info=True) return trajectory async def _get_control_plane_status(self, session) -> Optional[Dict[str, Any]]: