try to fix sceenshot timeout (#272)

2024-05-08 08:59:34 +08:00
parent 7b36e04418
commit bce6326eef
1 changed files with 2 additions and 0 deletions
--- a/skyvern/webeye/browser_factory.py
+++ b/skyvern/webeye/browser_factory.py
@@ -226,6 +226,8 @@ class BrowserState:
            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,