From dccdc7cca99e039cd36ee1ab01ce5b9090accc36 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 13 Apr 2024 21:26:47 -0700 Subject: [PATCH] screenshot missing error shows up when status is terminated or completed (#189) --- skyvern/forge/sdk/routes/agent_protocol.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 5065ef6b..577df4dd 100644 --- 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,