diff --git a/browser_utils/initialization/core.py b/browser_utils/initialization/core.py index 09680f21..4a220fdc 100644 --- a/browser_utils/initialization/core.py +++ b/browser_utils/initialization/core.py @@ -567,7 +567,8 @@ async def enable_temporary_chat_mode(page: AsyncPage) -> None: # pragma: no cov """ try: incognito_button_locator = page.locator( - 'button[aria-label="Temporary chat toggle"]' + 'button[aria-label="Temporary chat toggle"]', + 'button[aria-label="Toggle temporary chat"]' ) await incognito_button_locator.wait_for(state="visible", timeout=10000) diff --git a/browser_utils/models/switcher.py b/browser_utils/models/switcher.py index c2c22573..89446ee1 100644 --- a/browser_utils/models/switcher.py +++ b/browser_utils/models/switcher.py @@ -161,7 +161,8 @@ async def switch_ai_studio_model(page: AsyncPage, model_id: str, req_id: str) -> try: logger.debug("[Model] Re-enabling temporary chat mode...") incognito_button_locator = page.locator( - 'button[aria-label="Temporary chat toggle"]' + 'button[aria-label="Temporary chat toggle"]', + 'button[aria-label="Toggle temporary chat"]' ) await incognito_button_locator.wait_for(