idZddlmZmZddlmZmZddlmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z eGddZ!GddeZ"y)zkAbstract base classes for node executors. Defines the interfaces that every node executor must implement. )ABCabstractmethod) dataclassfield)AnyDictOptionalList)Node)MessageMessageContent MessageRoleserialize_messages) MemoryManager)ThinkingManagerBase) ToolManager)FunctionManager)HumanPromptService) LogManager) TokenTracker)WorkflowCancelledErrorcBeZdZUdZeed<eed<eed<ee Z e e e fed<ee Ze e efed<dZeeed <ee Ze e efed <dZeeed <dZeeed <dZeeed <de dee fdZde deefdZdeefdZdeefdZy)ExecutionContexta Node execution context that bundles every service and state the executor needs. Attributes: tool_manager: Tool manager shared by executors function_manager: Function manager registry log_manager: Structured log manager memory_managers: Mapping of node_id to ``MemoryManager`` instances thinking_managers: Mapping of node_id to ``ThinkingManagerBase`` instances token_tracker: Token tracker used for accounting global_state: Shared global state dictionary tool_managerfunction_manager log_manager)default_factorymemory_managersthinking_managersN token_tracker global_stateworkspace_hookhuman_prompt_service cancel_eventnode_idreturnc8|jj|S)z+Return the memory manager for a given node.)rgetselfr%s G/Users/bowang/.openclaw/workspace/ChatDev/runtime/node/executor/base.pyget_memory_managerz#ExecutionContext.get_memory_manager.s##''00c8|jj|S)z-Return the thinking manager for a given node.)rr(r)s r+get_thinking_managerz%ExecutionContext.get_thinking_manager2s%%))'22r-c|jS)z$Return the configured token tracker.)r r*s r+get_token_trackerz"ExecutionContext.get_token_tracker6s!!!r-c|jS)z,Return the interactive human prompt service.)r#r1s r+get_human_prompt_servicez)ExecutionContext.get_human_prompt_service:s(((r-)__name__ __module__ __qualname____doc__r__annotations__rrrdictrrstrrrrr r rr!rr"r#rr$r,r/r2r4r-r+rrs %%05d0KOT#},-K8=d8StC!445S,0M8L)0#(#>L$sCx.>$(NHSM(9=(#56="&L(3-&1#1(=2I13C3H=P4Q3"8L#9")(3E*F)r-rc eZdZdZdefdZededee dee fdZ e de fdZ e defd Ze defd Zdee defd Zdee ded zfd Zd d dddededed zdeeefd zdede f dZdee dee fdZddZy ) NodeExecutorzAbstract base class for node executors. Every concrete executor must inherit from this class and implement ``execute``. contextc||_y)zInitialize the executor with the shared execution context. Args: context: Execution context N)r?)r*r?s r+__init__zNodeExecutor.__init__Es  r-nodeinputsr&cy)aExecute the node logic. Args: node: Node definition to execute inputs: Input queue for the node Returns: List of payload messages produced by the node. Empty list when the node intentionally suppresses downstream propagation. Standard nodes return a single-element list. Raises: Exception: Raised when execution fails Nr<)r*rBrCs r+executezNodeExecutor.executeMs r-c.|jjS)zReturn the shared tool manager.)r?rr1s r+rzNodeExecutor.tool_manager_s||(((r-c.|jjS)z#Return the shared function manager.)r?rr1s r+rzNodeExecutor.function_managerds||,,,r-c.|jjS)z"Return the structured log manager.)r?rr1s r+rzNodeExecutor.log_manageris||'''r-c |syg}|D]Z}|jjdd}|jd|d|jjd|j \dj |S)NsourceUNKNOWNz=== INPUT FROM z (z) === z )metadatar(appendrolevalue text_contentjoin)r*rCpartsmessagerKs r+_inputs_to_textzNodeExecutor._inputs_to_textnswG%%))(I>F LL!&GLL,>,>+?yI]I]I_H`a  {{5!!r-Nc|syt|SN)r)r*rCs r+_inputs_to_message_jsonz$NodeExecutor._inputs_to_message_jsonys!&))r-F)rKrM preserve_rolerOcontentrKrMrYcft|xsi}|r|jd|t||||S)NrK)rOrZrMrY)r: setdefaultr )r*rOrZrKrMrYmetas r+_build_messagezNodeExecutor._build_message~s5HN#  OOHf -D'DP]^^r-messagescH|Dcgc]}|jc}Scc}wrW)clone)r*r_rTs r+_clone_messageszNodeExecutor._clone_messagess /78xG x888scnt|jdd}||jr tdyy)Nr$zWorkflow execution cancelled)getattrr?is_setr)r*events r+_ensure_not_cancelledz"NodeExecutor._ensure_not_cancelleds6 nd;  ()GH H"0 r-)r&N)r5r6r7r8rrArr r r rEpropertyrrrrrrr;rUrXrr rrboolr^rbrgr<r-r+r>r>?sK  0 D $w- DM  ")k))-/--(Z(( "d7m " "*d7m*d *"*.# _ _  _ d _ sCx.4' _ _  _9W 9$w-9Ir-r>N)#r8abcrr dataclassesrrtypingrrr r entity.configsr entity.messagesr r rrruntime.node.agentrrrutils.function_managerrutils.human_promptrutils.log_managerrutils.token_trackerrutils.exceptionsrrr>r<r-r+rusc $(,,TT,2*21(,3 %)%) %)PSI3SIr-