catch screenshot error when taking the final screenshot (#256)

Skyvern-AI · May 4, 2024 · e8a46d2 · e8a46d2
1 parent e75b86c
commit e8a46d2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py
@@ -767,12 +767,17 @@ async def send_task_response(
                 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: