diff --git a/packages/narada/src/narada/client.py b/packages/narada/src/narada/client.py index 354a690..166e429 100644 --- a/packages/narada/src/narada/client.py +++ b/packages/narada/src/narada/client.py @@ -283,11 +283,13 @@ async def _initialize_cloud_browser_window( # Wait for browser window ID. The extension can take a bit to be installed, so we retry a # few times. - max_attempts = 5 + max_attempts = 10 for attempt in range(max_attempts): try: browser_window_id = await self._wait_for_browser_window_id( - initialization_page, config + initialization_page, + config, + timeout=4_000, ) break except NaradaExtensionMissingError: @@ -295,6 +297,11 @@ async def _initialize_cloud_browser_window( raise logging.info("Waiting for Narada extension to be installed...") await asyncio.sleep(1) + except NaradaTimeoutError: + if attempt == max_attempts - 1: + raise + # If browser window ID is not found, reload the page and try again + await initialization_page.reload() cloud_window = CloudBrowserWindow( browser_window_id=browser_window_id,