blocks experiment

app.py

@@ -21,6 +21,7 @@ from sqlalchemy import (
     inspect,
     text,
     select,
+    Engine,
 )
 import spaces
 
@@ -38,7 +39,7 @@ def dummy():
 
 
 @tool
-def sql_engine_tool(query: str, engine: any) -> str:
+def sql_engine_tool(query: str, engine: Engine) -> str:
     """
     Allows you to perform SQL queries on the table. Returns a string representation of the result.
     The table is named 'receipts'. Its description is as follows:
@@ -55,7 +56,7 @@ def sql_engine_tool(query: str, engine: any) -> str:
     output = ""
     print("debug sql_engine_tool")
     print(engine)
-    with engine.connect() as con:
+    with engine.begin() as con:
         print(con.connection)
         print(metadata_objects.tables.keys())
         result = con.execute(