Function _wait_for_response_completion

browser_utils/operations.py:1078–1273

Wait for response completion

(
    page: AsyncPage,
    prompt_textarea_locator: Locator,
    submit_button_locator: Locator,
    edit_button_locator: Locator,
    req_id: str,
    check_client_disconnected_func: Callable,
    current_chat_id: Optional[str],
    prompt_length: int,
    initial_wait_ms=INITIAL_WAIT_MS_BEFORE_POLLING,
    timeout: Optional[float] = None,
)

async def _wait_for_response_completion(
    page: AsyncPage,
    prompt_textarea_locator: Locator,
    submit_button_locator: Locator,
    edit_button_locator: Locator,
    req_id: str,
    check_client_disconnected_func: Callable,
    current_chat_id: Optional[str],
    prompt_length: int,
    initial_wait_ms=INITIAL_WAIT_MS_BEFORE_POLLING,
    timeout: Optional[float] = None,
) -> bool:
    """Wait for response completion"""
    from playwright.async_api import TimeoutError

    # [FIX-03] Dynamic TTFB Timeout - Rotation Aware
    if timeout is None:
        base_timeout_seconds = 5 + (prompt_length / 1000.0)

        # Rotation-aware adjustments
        if GlobalState.IS_QUOTA_EXCEEDED:
            rotation_overhead = 20  # 20 second overhead for rotation
            timeout_seconds = max(
                base_timeout_seconds + rotation_overhead, 30
            )  # Minimum 30s during rotation
            logger.info(
                f"[{req_id}] (WaitV3) Rotation detected - applying extended timeout: {timeout_seconds:.2f}s"
            )
        else:
            timeout_seconds = max(base_timeout_seconds, 10)
            timeout_seconds = min(timeout_seconds, 120)

            if prompt_length > 5000:
                timeout_seconds *= 1.5
                logger.info(
                    f"[{req_id}] (WaitV3) Large prompt detected - extending timeout by 50%: {timeout_seconds:.2f}s"
                )
    else:
        timeout_seconds = timeout
        if GlobalState.IS_QUOTA_EXCEEDED and timeout_seconds < 30:
            timeout_seconds = 30
            logger.info(
                f"[{req_id}] (WaitV3) Rotation detected - enforcing minimum timeout: {timeout_seconds:.2f}s"
            )

    _timeout_ms = timeout_seconds * 1000

    logger.info(
        f"[{req_id}] (WaitV3) Waiting for response completion... (Dynamic Timeout: {timeout_seconds:.2f}s)"
    )
    await asyncio.sleep(initial_wait_ms / 1000)

    start_time = time.time()

    from config import UI_GENERATION_WAIT_TIMEOUT_MS

    wait_timeout_ms_short = UI_GENERATION_WAIT_TIMEOUT_MS