From deb80bce9c643953119297931cf2c9f99a5a2a3e Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Mon, 9 Sep 2024 23:07:13 -0700 Subject: [PATCH] use workflow_run_id screenshot when task has the workflow_run_id (#798) --- skyvern/forge/sdk/routes/streaming.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/skyvern/forge/sdk/routes/streaming.py b/skyvern/forge/sdk/routes/streaming.py index 8bbc3692..a5df5e99 100644 --- a/skyvern/forge/sdk/routes/streaming.py +++ b/skyvern/forge/sdk/routes/streaming.py @@ -89,6 +89,8 @@ async def task_stream( if task.status == TaskStatus.running: file_name = f"{task_id}.png" + if task.workflow_run_id: + file_name = f"{task.workflow_run_id}.png" screenshot = await app.STORAGE.get_streaming_file(organization_id, file_name) if screenshot: encoded_screenshot = base64.b64encode(screenshot).decode("utf-8")