vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

workflow run streaming (#791)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-09-09 08:29:01 -07:00
committed by GitHub
parent f940c71e87
commit eb659ec17f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/sdk/routes/streaming.py
View File

@@ -89,6 +89,8 @@ async def task_stream(
if task.status == TaskStatus.running: if task.status == TaskStatus.running:
file_name = f"{task_id}.png" 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) screenshot = await app.STORAGE.get_streaming_file(organization_id, file_name)
if screenshot: if screenshot:
encoded_screenshot = base64.b64encode(screenshot).decode("utf-8") encoded_screenshot = base64.b64encode(screenshot).decode("utf-8")