Function code_execution

mcts_math/agents/tree.py:132–160

(
    node: Type[BaseNode], 
    parser_result: Dict[str, str],
)

def code_execution(
    node: Type[BaseNode], 
    parser_result: Dict[str, str],
) -> str:

    @timeout(TIMEOUT_SECONDS, exception_message=TIMEOUT_MESSAGE)
    def _code_execution(node: Type[BaseNode], parser_result: Dict[str, str]) -> str:
        # Define tool
        action = parser_result["action"]
        tool_func = tools[action]

        # Preventing variable update between different children
        # For each child, we re-run the historical code snippets with the same action (tool).
        history_action_inputs = collect_action_inputs(node, action)
        for history_ai in history_action_inputs:
            _ = tool_func(history_ai)
        
        # then, we execute current code snippets
        action_input = parser_result["action_input"]
        observation = str(tool_func(action_input)).strip()
        del tool_func
        return observation
    
    try:
        observation = _code_execution(node, parser_result)
    except Exception as e:
        observation = "{}: {}".format(type(e).__name__, str(e))
    
    return observation


def collect_action_inputs(