Function _force_ui_state_settings

browser_utils/models/ui_state.py:97–147

Forcefully set the UI state. Args: page: Playwright page object. req_id: Request ID for logging. Returns: bool: Whether the setting was successful.

(page: AsyncPage, req_id: str = "unknown")

async def _force_ui_state_settings(page: AsyncPage, req_id: str = "unknown") -> bool:
    """
    Forcefully set the UI state.

    Args:
        page: Playwright page object.
        req_id: Request ID for logging.

    Returns:
        bool: Whether the setting was successful.
    """
    try:
        logger.debug("[State] Forcefully setting UI state...")

        # First verify current state
        current_state = await _verify_ui_state_settings(page, req_id)

        if not current_state["needsUpdate"]:
            logger.debug("[State] State is already correct, no update needed")
            return True

        # Get existing preferences or create new ones
        prefs = current_state.get("prefs", {})

        # Force key configurations
        prefs["isAdvancedOpen"] = True
        prefs["areToolsOpen"] = True

        # Save to localStorage
        prefs_str = json.dumps(prefs)
        await page.evaluate(
            "(prefsStr) => localStorage.setItem('aiStudioUserPreference', prefsStr)",
            prefs_str,
        )

        logger.debug("[State] Set: isAdvancedOpen=true, areToolsOpen=true")

        # Verify if setting was successful
        verify_state = await _verify_ui_state_settings(page, req_id)
        if not verify_state["needsUpdate"]:
            logger.debug("[State] Setting verification successful")
            return True
        else:
            logger.warning("UI state setting verification failed, may need to retry")
            return False

    except asyncio.CancelledError:
        raise
    except Exception as e:
        logger.error(f"Error forcefully setting UI state: {e}")
        return False


async def _force_ui_state_with_retry(