try to fix sceenshot timeout (#272)

Skyvern-AI · May 8, 2024 · bce6326 · bce6326
1 parent 7b36e04
commit bce6326
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,