data_juicer_agents.capabilities.plan.generator module

LLM generator for process operator lists used by CLI plan orchestration.

class data_juicer_agents.capabilities.plan.generator.ProcessOperatorGenerator(*, model_name: str, api_key: str | None = None, base_url: str | None = None, thinking: bool | None = None)

Bases: object

Generate an operator list for staged plan assembly.

__init__(*, model_name: str, api_key: str | None = None, base_url: str | None = None, thinking: bool | None = None)

generate(*, user_intent: str, retrieval_payload: Dict[str, Any], dataset_spec: Dict[str, Any], dataset_profile: Dict[str, Any] | None = None) → Dict[str, Any]