From ff3abf4363cb53aa529eb81b7150b3ce0ad37ac6 Mon Sep 17 00:00:00 2001 From: LawyZheng Date: Mon, 13 May 2024 10:34:31 +0800 Subject: [PATCH] log screenshot taking time --- skyvern/webeye/browser_factory.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/skyvern/webeye/browser_factory.py b/skyvern/webeye/browser_factory.py index 90de055e5..85edb750b 100644 --- a/skyvern/webeye/browser_factory.py +++ b/skyvern/webeye/browser_factory.py @@ -224,17 +224,28 @@ async def take_screenshot_from_page(page: Page, full_page: bool = False, file_pa try: 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") + start_time = time.time() + screenshot: bytes = bytes() if file_path: - return await page.screenshot( + screenshot = await page.screenshot( path=file_path, full_page=full_page, 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, - timeout=SettingsManager.get_settings().BROWSER_SCREENSHOT_TIMEOUT_MS, - animations="disabled", + file_path=file_path, ) + return screenshot except TimeoutError as e: LOG.exception(f"Timeout error while taking screenshot: {str(e)}") raise FailedToTakeScreenshot(error_message=str(e)) from e