"""Bidirectional encoder rendering and multi-question packing for Jue-Flash."""

from jue_flash.encoding.pack import (
    MPS_BUCKETS,
    EncodedQuestionSpan,
    EncodedRequest,
    PackedFlashBatch,
    collate_encoded_requests,
    encode_group_request,
    select_mps_bucket,
)
from jue_flash.encoding.render import (
    ID_TO_PRIMITIVE,
    PRIMITIVE_TO_ID,
    option_text,
    render,
    render_question_parts,
)

__all__ = [
    "EncodedQuestionSpan",
    "EncodedRequest",
    "ID_TO_PRIMITIVE",
    "MPS_BUCKETS",
    "PRIMITIVE_TO_ID",
    "PackedFlashBatch",
    "collate_encoded_requests",
    "encode_group_request",
    "option_text",
    "render",
    "render_question_parts",
    "select_mps_bucket",
]