in task response, do not error when there's no screenshot if task is …

…still in progress (#188)
Skyvern-AI · Apr 14, 2024 · 1638e7d · 1638e7d
1 parent 935453c
commit 1638e7d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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):