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

add screenshot for the first task block (#1471)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-01-02 16:55:55 -08:00
committed by GitHub
parent c85d5428f3
commit 241f648548
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -438,7 +438,7 @@ class BaseTaskBlock(Block):
task_order=task_order, task_order=task_order,
task_retry=task_retry, task_retry=task_retry,
) )
await app.DATABASE.update_workflow_run_block( workflow_run_block = await app.DATABASE.update_workflow_run_block(
workflow_run_block_id=workflow_run_block_id, workflow_run_block_id=workflow_run_block_id,
task_id=task.task_id, task_id=task.task_id,
organization_id=workflow.organization_id, organization_id=workflow.organization_id,
@@ -455,6 +455,14 @@ class BaseTaskBlock(Block):
browser_state = await app.BROWSER_MANAGER.get_or_create_for_workflow_run( browser_state = await app.BROWSER_MANAGER.get_or_create_for_workflow_run(
workflow_run=workflow_run, url=self.url workflow_run=workflow_run, url=self.url
) )
# add screenshot artifact for the first task
screenshot = await browser_state.take_screenshot(full_page=True)
if screenshot:
await app.ARTIFACT_MANAGER.create_workflow_run_block_artifact(
workflow_run_block=workflow_run_block,
artifact_type=ArtifactType.SCREENSHOT_LLM,
data=screenshot,
)
except Exception as e: except Exception as e:
LOG.exception( LOG.exception(
"Failed to get browser state for first task", "Failed to get browser state for first task",