data_juicer_agents.capabilities

Scenario capabilities for DJX orchestration.

class data_juicer_agents.capabilities.PlanOrchestrator(*, planner_model_name: str | None = None, llm_api_key: str | None = None, llm_base_url: str | None = None, llm_thinking: bool | None = None)

基类：object

Fixed orchestration for CLI plan generation.

__init__(*, planner_model_name: str | None = None, llm_api_key: str | None = None, llm_base_url: str | None = None, llm_thinking: bool | None = None)

generate_plan(*, user_intent: str, dataset_path: str, export_path: str, custom_operator_paths: Iterable[Any] | None = None, retrieved_candidates: Dict[str, Any] | None = None, retrieval_top_k: int = 5, retrieval_mode: str = 'auto') → Dict[str, Any]

class data_juicer_agents.capabilities.ApplyUseCase

基类：object

Execute validated plans and return execution summaries.

execute(plan_payload: Dict[str, Any], runtime_dir: Path, dry_run: bool = False, timeout_seconds: int = 300, command_override: str | Iterable[str] | None = None, cancel_check: Callable[[], bool] | None = None) → Tuple[ApplyResult, int, str, str]

class data_juicer_agents.capabilities.DevUseCase

基类：object

Generate and optionally smoke-check custom operator scaffolds.

static execute(*, intent: str, operator_name: str, output_dir: str, operator_type: str | None = None, from_retrieve: str | None = None, smoke_check: bool = False) → Dict[str, Any]

class data_juicer_agents.capabilities.DJSessionAgent(use_llm_router: bool = True, dataset_path: str | None = None, export_path: str | None = None, working_dir: str | None = None, verbose: bool = False, api_key: str | None = None, base_url: str | None = None, model_name: str | None = None, thinking: bool | None = None, event_callback: Callable[[Dict[str, Any]], None] | None = None, enable_streaming: bool = False)

基类：object

Session agent that orchestrates djx atomic commands via ReAct tools.

__init__(use_llm_router: bool = True, dataset_path: str | None = None, export_path: str | None = None, working_dir: str | None = None, verbose: bool = False, api_key: str | None = None, base_url: str | None = None, model_name: str | None = None, thinking: bool | None = None, event_callback: Callable[[Dict[str, Any]], None] | None = None, enable_streaming: bool = False) → None

request_interrupt() → bool

async handle_message_async(message: str) → SessionReply

async handle_as_studio_turn_async(inbound_msg: Any, emit_chunk: Callable[[Any, bool], Awaitable[None] | None]) → _StudioTurnResult

handle_message(message: str) → SessionReply

class data_juicer_agents.capabilities.SessionReply(text: 'str', thinking: 'str' = '', stop: 'bool' = False, interrupted: 'bool' = False)

基类：object

text: str

thinking: str = ''

stop: bool = False

interrupted: bool = False

__init__(text: str, thinking: str = '', stop: bool = False, interrupted: bool = False) → None