data_juicer_agents.tools.context.inspect_dataset.input module

Input models for inspect_dataset.

class data_juicer_agents.tools.context.inspect_dataset.input.InspectDatasetInput(*, dataset_path: str, sample_size: Annotated[int, Ge(ge=1)] = 20)

Bases: BaseModel

dataset_path: str

sample_size: int

model_config = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class data_juicer_agents.tools.context.inspect_dataset.input.GenericOutput(*, ok: bool = True)

Bases: BaseModel

ok: bool

model_config = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].