tool

app.py

@@ -28,7 +28,7 @@ def dummy():
 
 
 @tool
-def sql_engine(query: str) -> str:
+def sql_engine_tool(query: str) -> 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:
@@ -97,13 +97,13 @@ if __name__ == "__main__":
     )
 
     agent = CodeAgent(
-        tools=[sql_engine],
+        tools=[sql_engine_tool],
         #         system_prompt="""
         # You are a text to sql converter
         # """,
         model=model,
-        max_steps=1,
-        verbosity_level=1,
+        max_steps=3,
+        verbosity_level=2,
     )
     GradioUI(agent).launch()
-# What is the average each customer paid?
+# What is the average each customer paid? Create a sql statement and invoke your sql_engine tool