From 1638e7d07317da64e69800e3ea7ac23c5610836e Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 13 Apr 2024 21:20:16 -0700 Subject: [PATCH] in task response, do not error when there's no screenshot if task is still in progress (#188) --- skyvern/forge/sdk/routes/agent_protocol.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 0d1e28468..5065ef6b9 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -236,8 +236,12 @@ async def get_task( "Failed to get share link for action screenshot", artifact_id=artifact.artifact_id, ) - else: - LOG.error("Failed to get latest action screenshots") + elif task_obj.status in [TaskStatus.failed, TaskStatus.completed]: + LOG.error( + "Failed to get latest action screenshots in task response", + task_id=task_id, + task_status=task_obj.status, + ) failure_reason = None if task_obj.status == TaskStatus.failed and (latest_step.output or task_obj.failure_reason):