get_a2ui_datapart

def get_a2ui_datapart(part: Part) -> DataPart | None:
    """Extract the A2UI DataPart from an A2A Part, if present.

    Args:
        part: The A2A Part to extract from.

    Returns:
        The ``DataPart`` if the part contains A2UI data, None otherwise.
    """
    if is_a2ui_part(part):
        return part.root  # type: ignore[return-value]
    return None