workflow run streaming (#791)

2024-09-09 08:29:01 -07:00
parent f940c71e87
commit eb659ec17f
1 changed files with 2 additions and 0 deletions
--- 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")