data_juicer_agents.capabilities.session

Session capability.

class data_juicer_agents.capabilities.session.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)

Bases: 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.session.SessionReply(text: 'str', thinking: 'str' = '', stop: 'bool' = False, interrupted: 'bool' = False)

Bases: object

text: str

thinking: str = ''

stop: bool = False

interrupted: bool = False

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