Function get_response_via_copy_button

browser_utils/operations.py:971–1075

Get response via copy button

(
    page: AsyncPage, req_id: str, check_client_disconnected: Callable
)

async def get_response_via_copy_button(
    page: AsyncPage, req_id: str, check_client_disconnected: Callable
) -> Optional[str]:
    """Get response via copy button"""
    logger.info(f"[{req_id}] (Helper) Attempting to get response via copy button...")
    last_message_container = page.locator("ms-chat-turn").last
    more_options_button = last_message_container.get_by_label("Open options")
    copy_markdown_button = page.get_by_role("menuitem", name="Copy markdown")

    try:
        logger.info(
            f"[{req_id}]   - Attempting to hover last message to show options..."
        )
        await last_message_container.hover(timeout=CLICK_TIMEOUT_MS)
        check_client_disconnected("Copy Response - after hover: ")
        await asyncio.sleep(0.5)
        check_client_disconnected("Copy Response - after hover delay: ")
        logger.info(f"[{req_id}]   - Hovered.")

        logger.info(f"[{req_id}]   - Locating and clicking 'More options' button...")
        try:
            from playwright.async_api import expect as expect_async

            await expect_async(more_options_button).to_be_visible(
                timeout=CLICK_TIMEOUT_MS
            )
            check_client_disconnected(
                "Copy Response - after more options button visible: "
            )
            await more_options_button.click(timeout=CLICK_TIMEOUT_MS)
            logger.info(f"[{req_id}]   - 'More options' clicked (via get_by_label).")
        except (ClientDisconnectedError, asyncio.CancelledError):
            raise
        except Exception as more_opts_err:
            logger.error(
                f"[{req_id}]   - 'More options' button (via get_by_label) not visible or click failed: {more_opts_err}"
            )
            await save_error_snapshot(f"copy_response_more_options_failed_{req_id}")
            return None

        check_client_disconnected("Copy Response - after clicking more options: ")
        await asyncio.sleep(0.5)
        check_client_disconnected(
            "Copy Response - after delay following more options click: "
        )

        logger.info(f"[{req_id}]   - Locating and clicking 'Copy markdown' button...")
        copy_success = False
        try:
            await expect_async(copy_markdown_button).to_be_visible(
                timeout=CLICK_TIMEOUT_MS
            )
            check_client_disconnected("Copy Response - after copy button visible: ")
            await copy_markdown_button.click(timeout=CLICK_TIMEOUT_MS, force=True)
            copy_success = True
            logger.info(f"[{req_id}]   - 'Copy markdown' clicked (via get_by_role).")
        except (ClientDisconnectedError, asyncio.CancelledError):
            raise