Function get_response_via_edit_button

browser_utils/operations.py:801–968

Get response via edit button

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

async def get_response_via_edit_button(
    page: AsyncPage, req_id: str, check_client_disconnected: Callable
) -> Optional[str]:
    """Get response via edit button"""
    logger.info(f"[{req_id}] (Helper) Attempting to get response via edit button...")
    last_message_container = page.locator("ms-chat-turn").last
    edit_button = last_message_container.get_by_label("Edit")
    finish_edit_button = last_message_container.get_by_label("Stop editing")
    autosize_textarea_locator = last_message_container.locator("ms-autosize-textarea")
    actual_textarea_locator = autosize_textarea_locator.locator("textarea")

    try:
        logger.info(
            f"[{req_id}]   - Attempting to hover last message to show 'Edit' button..."
        )
        try:
            # Perform hover on message container
            await last_message_container.hover(timeout=CLICK_TIMEOUT_MS / 2)
            await asyncio.sleep(0.3)  # Wait for hover effect
            check_client_disconnected("Edit Response - after hover: ")
        except (ClientDisconnectedError, asyncio.CancelledError):
            raise
        except Exception as hover_err:
            logger.warning(
                f"[{req_id}]   - (get_response_via_edit_button) Hover last message failed (ignoring): {type(hover_err).__name__}"
            )

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

            await expect_async(edit_button).to_be_visible(timeout=CLICK_TIMEOUT_MS)
            check_client_disconnected("Edit Response - 'Edit' button visible after: ")
            await edit_button.click(timeout=CLICK_TIMEOUT_MS)
            logger.info(f"[{req_id}]   - 'Edit' button clicked.")
        except (ClientDisconnectedError, asyncio.CancelledError):
            raise
        except Exception as edit_btn_err:
            logger.error(
                f"[{req_id}]   - 'Edit' button not visible or click failed: {edit_btn_err}"
            )
            await save_error_snapshot(f"edit_response_edit_button_failed_{req_id}")
            return None

        check_client_disconnected("Edit Response - after clicking 'Edit' button: ")
        await asyncio.sleep(0.3)
        check_client_disconnected(
            "Edit Response - after delay following 'Edit' button click: "
        )

        logger.info(f"[{req_id}]   - Getting content from textarea...")
        response_content = None
        textarea_failed = False

        try:
            await expect_async(autosize_textarea_locator).to_be_visible(
                timeout=CLICK_TIMEOUT_MS
            )