From 062586bb48998fe9f44607b128c67fa08563889c Mon Sep 17 00:00:00 2001
From: LawyZheng <lawyzheng1106@gmail.com>
Date: Wed, 8 May 2024 08:57:12 +0800
Subject: [PATCH] try to fix sceenshot timeout

---
 skyvern/webeye/browser_factory.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/skyvern/webeye/browser_factory.py b/skyvern/webeye/browser_factory.py
index 62fdc7f17..5a9ceaa23 100644
--- a/skyvern/webeye/browser_factory.py
+++ b/skyvern/webeye/browser_factory.py
@@ -226,6 +226,8 @@ async def take_screenshot(self, full_page: bool = False, file_path: str | None =
             LOG.error("BrowserState has no page")
             raise MissingBrowserStatePage()
         try:
+            await self.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")
             if file_path:
                 return await self.page.screenshot(
                     path=file_path,