from diffulex.config import Config
from diffulex.engine.request import AutoReq
from diffulex.sampling_params import SamplingParams
from diffulex.strategy_template.multi_block.engine.request import MultiBlockReqTemplate


@AutoReq.register("d2f", is_default=True)
class D2fReq(MultiBlockReqTemplate):
    """Req for D2F strategy. Accepts config for AutoReq.create()."""

    def __init__(
        self,
        token_ids: list[int],
        sampling_params: SamplingParams = SamplingParams(),
        config: Config | None = None,
    ):
        super().__init__(token_ids, sampling_params)
        # config is passed by AutoReq.create(); req attributes are set in init_multi_block()