Update agent_node.py