Update app.py

app.py

@@ -52,6 +52,17 @@ def get_db_chain():
 
     return SQLDatabaseChain.from_llm(llm, db, prompt=PROMPT, verbose=True)
 
+def get_sql_agent():
+    toolkit = SQLDatabaseToolkit(llm=llm,db=db)
+
+    return create_sql_agent(
+        llm=llm,
+        toolkit=toolkit,
+        verbose=True,
+        use_query_checker=True, 
+        return_intermediate_steps=True
+    )
+
 
 def get_json_chain():
     prompt_template = "Reformat this {result} in JSON format"