from states.state import State def node_2(state: State) -> State: # --- Node 2: Initialize State --- state.set_graph_state("node_2", "Node 2: State initialized.") state.log_node_end("node_2") return state