from typing import Literal

from src.settings.datasets.base import DatasetType, MultiDatasetSettings
from src.settings.datasets.chat import ChatDatasetSettings


class SamplingDatasetSettings(MultiDatasetSettings):
    dataset_type: Literal[DatasetType.SAMPLING] = DatasetType.SAMPLING
    chat_dataset: ChatDatasetSettings
