data_juicer_agents.tools.plan.assemble_plan#

assemble_plan tool package.

class data_juicer_agents.tools.plan.assemble_plan.AssemblePlanInput(*, intent: str, dataset_spec: Dict[str, Any], process_spec: Dict[str, Any], system_spec: Dict[str, Any], approval_required: bool = True)[source]#

Bases: BaseModel

intent: str#
dataset_spec: Dict[str, Any]#
process_spec: Dict[str, Any]#
system_spec: Dict[str, Any]#
approval_required: bool#
model_config = {}#

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

exception data_juicer_agents.tools.plan.assemble_plan.PlannerBuildError[source]#

Bases: ValueError

Raised when planner core cannot build a valid plan.

class data_juicer_agents.tools.plan.assemble_plan.PlannerCore[source]#

Bases: object

Pure deterministic planner builder.

classmethod normalize_context(*, user_intent: str, dataset_path: str, export_path: str, custom_operator_paths: Iterable[Any] | None = None) → PlanContext[source]#
classmethod normalize_dataset_spec(dataset_spec: DatasetSpec | Dict[str, Any]) → DatasetSpec[source]#
classmethod build_plan_from_specs(*, user_intent: str, dataset_spec: DatasetSpec | Dict[str, Any], process_spec: Dict[str, Any], system_spec: Dict[str, Any] | None = None, risk_notes: Iterable[Any] | None = None, estimation: Dict[str, Any] | None = None, approval_required: bool = True) → PlanModel[source]#
data_juicer_agents.tools.plan.assemble_plan.assemble_plan(*, user_intent: str, dataset_spec: Dict[str, Any], process_spec: Dict[str, Any], system_spec: Dict[str, Any] | None = None, approval_required: bool = True) → Dict[str, Any][source]#