Function _handle_playwright_response

api_utils/request_processor.py:482–640

Handle response using Playwright - Enhanced version with integrity verification

(
    req_id: str,
    request: ChatCompletionRequest,
    page: AsyncPage,
    context: dict,
    result_future: Future,
    submit_button_locator: Locator,
    check_client_disconnected: Callable,
    prompt_length: int,
    timeout: float,
)

async def _handle_playwright_response(
    req_id: str,
    request: ChatCompletionRequest,
    page: AsyncPage,
    context: dict,
    result_future: Future,
    submit_button_locator: Locator,
    check_client_disconnected: Callable,
    prompt_length: int,
    timeout: float,
) -> Optional[Tuple[Event, Locator, Callable]]:
    """Handle response using Playwright - Enhanced version with integrity verification"""
    from api_utils.server_state import state

    logger = state.logger

    is_streaming = request.stream
    current_ai_studio_model_id = context.get("current_ai_studio_model_id")

    await locate_response_elements(page, req_id, logger, check_client_disconnected)
    check_client_disconnected("After Response Element Located: ")

    if is_streaming:
        completion_event = Event()

        def playwright_stream_factory(event_to_signal: Event):
            return gen_sse_from_playwright(
                page,
                logger,
                req_id,
                current_ai_studio_model_id or MODEL_NAME,
                request,
                check_client_disconnected,
                event_to_signal,
                prompt_length=prompt_length,
                timeout=timeout,
            )

        resilient_gen = resilient_stream_generator(
            req_id,
            current_ai_studio_model_id or MODEL_NAME,
            playwright_stream_factory,
            completion_event,
        )

        if not result_future.done():
            result_future.set_result(
                StreamingResponse(resilient_gen, media_type="text/event-stream")
            )

        return completion_event, submit_button_locator, check_client_disconnected
    else:
        page_controller = PageController(page, logger, req_id)
        response_data = await page_controller.get_response_with_integrity_check(
            check_client_disconnected, prompt_length, timeout=timeout
        )

        final_content = response_data.get("content", "")