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

log screenshot taking time (#300)

Browse Source
This commit is contained in:
LawyZheng
2024-05-13 10:41:23 +08:00
committed by GitHub
parent 25311dee86
commit ab88f24df6
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
skyvern/webeye/browser_factory.py
View File

@@ -224,17 +224,28 @@ class BrowserState:
try: try:
await page.wait_for_load_state(timeout=SettingsManager.get_settings().BROWSER_LOADING_TIMEOUT_MS) await page.wait_for_load_state(timeout=SettingsManager.get_settings().BROWSER_LOADING_TIMEOUT_MS)
LOG.info("Page is fully loaded, agent is about to take screenshots") LOG.info("Page is fully loaded, agent is about to take screenshots")
start_time = time.time()
screenshot: bytes = bytes()
if file_path: if file_path:
return await page.screenshot( screenshot = await page.screenshot(
path=file_path, path=file_path,
full_page=full_page, full_page=full_page,
timeout=SettingsManager.get_settings().BROWSER_SCREENSHOT_TIMEOUT_MS, timeout=SettingsManager.get_settings().BROWSER_SCREENSHOT_TIMEOUT_MS,
) )
return await page.screenshot( else:
screenshot = await page.screenshot(
full_page=full_page,
timeout=SettingsManager.get_settings().BROWSER_SCREENSHOT_TIMEOUT_MS,
animations="disabled",
)
end_time = time.time()
LOG.info(
f"Screenshot taking time",
screenshot_time=end_time - start_time,
full_page=full_page, full_page=full_page,
timeout=SettingsManager.get_settings().BROWSER_SCREENSHOT_TIMEOUT_MS, file_path=file_path,
animations="disabled",
) )
return screenshot
except TimeoutError as e: except TimeoutError as e:
LOG.exception(f"Timeout error while taking screenshot: {str(e)}") LOG.exception(f"Timeout error while taking screenshot: {str(e)}")
raise FailedToTakeScreenshot(error_message=str(e)) from e raise FailedToTakeScreenshot(error_message=str(e)) from e