Building the stategraph

def create_text2sql_graph():
    """Create the LangGraph state graph for Text2SQL with visualization"""

    workflow = StateGraph(AgentState)

    # Add all agent nodes
    workflow.add_node("guardrails_agent", guardrails_agent)
    workflow.add_node("sql_agent", sql_agent)
    workflow.add_node("execute_sql", execute_sql)
    workflow.add_node("analysis_agent", analysis_agent)
    workflow.add_node("error_agent", error_agent)
    workflow.add_node("decide_graph_need", decide_graph_need)
    workflow.add_node("viz_agent", viz_agent)

    # Set the entry point - always start with guardrails
    workflow.set_entry_point("guardrails_agent")

    # Add conditional and regular edges (see next lesson)
    # ...

    return workflow.compile()

# Create the compiled graph at module load
text2sql_graph = create_text2sql_graph()