Error when trying to use ext in sqlglot library - ModuleNotFoundError: No module named 'sqlglot.exp'

[pavanmakineni]

`import sqlglot
from sqlglot.exp import Select, Procedure

def find_selects_in_procedure(sql):
tree = parse_one(sql)

for proc in tree.find_all(exp.Procedure):
    print(f"Found stored procedure: {proc.name}")

    for select in proc.find_all(exp.Select):
        print(f"  - Found SELECT: {select}")

sql = """
CREATE PROCEDURE GetEmployeeSales
AS
BEGIN
SELECT EmployeeID, SUM(Total) as TotalSales
FROM Orders
GROUP BY EmployeeID;
END
"""

find_selects_in_procedure(sql)`

The above snippet is failing with following: ModuleNotFoundError: No module named 'sqlglot.exp'
Can someone help where i am going wrong?

[georgesittas]

You need to do from sqlglot.expressions import ....