Function responses_websocket

examples/server/server.py:16174–16313

(  # pyright: ignore[reportUnusedFunction]
        websocket: WebSocket,
    )

    @app.websocket("/v1/responses")
    async def responses_websocket(  # pyright: ignore[reportUnusedFunction]
        websocket: WebSocket,
    ):
        await websocket.accept()
        service: CompletionService = app.state.service
        formatter = service.formatter
        # HTTP /v1/responses remains stateless. The websocket transport keeps
        # per-connection response history so Codex can use previous_response_id
        # within a single live session.
        websocket_response_history: Dict[str, ResponsesWebSocketState] = {}

        def websocket_request_with_ephemeral_history(
            ws_body: ResponseCreateWebSocketRequest,
        ) -> CreateResponseRequest:
            body = ws_body.to_create_response_request()
            previous_response_id = body.previous_response_id
            if previous_response_id is None:
                return body
            prior_state = websocket_response_history.get(previous_response_id)
            if prior_state is None:
                raise CompletionRequestValidationError(
                    f"unknown previous_response_id: {previous_response_id}"
                )
            current_items = formatter._clone_response_input_items(body.input)
            replay_items = formatter._clone_response_input_items(
                prior_state.input_items
            )
            for item in prior_state.output_items:
                normalized = formatter._normalize_response_output_item_for_input(item)
                if normalized is not None:
                    replay_items.append(normalized)
            replay_items.extend(current_items)
            return body.model_copy(
                update={
                    "input": replay_items,
                    "previous_response_id": None,
                }
            )

        async def send_error(message: str) -> None:
            await websocket.send_json(
                {
                    "type": "error",
                    "error": {
                        "message": message,
                    },
                }
            )

        try:
            while True:
                try:
                    payload = await websocket.receive_json()
                except WebSocketDisconnect:
                    break
                except Exception:
                    await send_error("invalid websocket request payload")
                    continue