vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-16 18:29:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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_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,
task_id=task.task_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(
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:
LOG.exception(
"Failed to get browser state for first task",