From e8a46d2a1a8a243d5c1808269336218c42af24a4 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 4 May 2024 18:27:42 -0400 Subject: [PATCH] catch screenshot error when taking the final screenshot (#256) --- skyvern/forge/agent.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index 468f12ea..d0bb90a0 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -767,12 +767,17 @@ class ForgeAgent: artifact_type=ArtifactType.SCREENSHOT_FINAL, data=screenshot, ) - except TargetClosedError as e: + except TargetClosedError: LOG.warning( "Failed to take screenshot before sending task response, page is closed", task_id=task.task_id, step_id=last_step.step_id, - error=e, + ) + except Exception: + LOG.exception( + "Failed to take screenshot before sending task response", + task_id=task.task_id, + step_id=last_step.step_id, ) if task.workflow_run_id: