screenshot missing error shows up when status is terminated or completed (#189)

2024-04-13 21:26:47 -07:00
parent 1638e7d073
commit dccdc7cca9
1 changed files with 1 additions and 1 deletions
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -236,7 +236,7 @@ async def get_task(
                    "Failed to get share link for action screenshot",
                    artifact_id=artifact.artifact_id,
                )
-    elif task_obj.status in [TaskStatus.failed, TaskStatus.completed]:
+    elif task_obj.status in [TaskStatus.terminated, TaskStatus.completed]:
        LOG.error(
            "Failed to get latest action screenshots in task response",
            task_id=task_id,