nodes/node2.py

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