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

robust scrolling screenshot (#3050)

Browse Source
This commit is contained in:
LawyZheng
2025-07-29 14:56:13 +08:00
committed by GitHub
parent ecc0e2e17d
commit d9537327da
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
skyvern/webeye/utils/page.py
View File

@@ -253,8 +253,10 @@ class SkyvernFrame:
LOG.debug("Page is fully loaded, agent is about to generate the full page screenshot") LOG.debug("Page is fully loaded, agent is about to generate the full page screenshot")
start_time = time.time() start_time = time.time()
skyvern_frame = await SkyvernFrame.create_instance(frame=page) skyvern_frame = await SkyvernFrame.create_instance(frame=page)
x, y = await skyvern_frame.get_scroll_x_y() x: int | None = None
y: int | None = None
try: try:
x, y = await skyvern_frame.get_scroll_x_y()
async with asyncio.timeout(timeout): async with asyncio.timeout(timeout):
screenshots, positions = await _scrolling_screenshots_helper( screenshots, positions = await _scrolling_screenshots_helper(
skyvern_page=skyvern_frame, mode=mode, max_number=scrolling_number skyvern_page=skyvern_frame, mode=mode, max_number=scrolling_number
@@ -284,8 +286,17 @@ class SkyvernFrame:
file_path=file_path, file_path=file_path,
) )
return img_data return img_data
except Exception:
LOG.warning(
"Failed to take full page screenshot, fallback to use playwright full page screenshot",
exc_info=True,
)
return await _current_viewpoint_screenshot_helper(
page=page, file_path=file_path, timeout=timeout, full_page=True
)
finally: finally:
await skyvern_frame.scroll_to_x_y(x, y) if x is not None and y is not None:
await skyvern_frame.scroll_to_x_y(x, y)
@staticmethod @staticmethod
@TraceManager.traced_async(ignore_inputs=["page"]) @TraceManager.traced_async(ignore_inputs=["page"])