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

clean up fullpage screenshot exp (#4102)

Browse Source
This commit is contained in:
LawyZheng
2025-11-26 14:55:02 +08:00
committed by GitHub
parent 1fff8a2ac6
commit 8d09d9822a
5 changed files with 4 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
skyvern/forge/agent.py
View File

@@ -2198,11 +2198,6 @@ class ForgeAgent:
try:
screenshot = await browser_state.take_post_action_screenshot(
scrolling_number=scrolling_number,
use_playwright_fullpage=await app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
"ENABLE_PLAYWRIGHT_FULLPAGE",
task.workflow_run_id or task.task_id,
properties={"organization_id": task.organization_id},
),
)
await app.ARTIFACT_MANAGER.create_artifact(
step=step,
@@ -3069,13 +3064,7 @@ class ForgeAgent:
browser_state = app.BROWSER_MANAGER.get_for_task(task.task_id)
if browser_state is not None and await browser_state.get_working_page() is not None:
try:
screenshot = await browser_state.take_fullpage_screenshot(
use_playwright_fullpage=await app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
"ENABLE_PLAYWRIGHT_FULLPAGE",
task.workflow_run_id or task.task_id,
properties={"organization_id": task.organization_id},
)
)
screenshot = await browser_state.take_fullpage_screenshot()
await app.ARTIFACT_MANAGER.create_artifact(
step=last_step,
artifact_type=ArtifactType.SCREENSHOT_FINAL,

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

@@ -383,13 +383,7 @@ class Block(BaseModel, abc.ABC):
)
else:
try:
screenshot = await browser_state.take_fullpage_screenshot(
use_playwright_fullpage=await app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
"ENABLE_PLAYWRIGHT_FULLPAGE",
workflow_run_id,
properties={"organization_id": str(organization_id)},
)
)
screenshot = await browser_state.take_fullpage_screenshot()
except Exception:
LOG.warning(
"Failed to take screenshot before executing the block, ignoring the exception",
@@ -693,13 +687,7 @@ class BaseTaskBlock(Block):
try:
# add screenshot artifact for the first task
screenshot = await browser_state.take_fullpage_screenshot(
use_playwright_fullpage=await app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
"ENABLE_PLAYWRIGHT_FULLPAGE",
workflow_run_id,
properties={"organization_id": str(organization_id)},
)
)
screenshot = await browser_state.take_fullpage_screenshot()
if screenshot:
await app.ARTIFACT_MANAGER.create_workflow_run_block_artifact(
workflow_run_block=workflow_run_block,