testing llm again

app.py

@@ -31,7 +31,7 @@ def dummy():
 
 
 @tool
-def sql_engine_tool(query: str) -> 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:
@@ -97,7 +97,7 @@ def init_db():
 if __name__ == "__main__":
     init_db()
     model = HfApiModel(
-        model_id="HuggingFaceH4/zephyr-7b-beta",
+        model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
         token=os.getenv("my_first_agents_hf_tokens"),
     )
 
@@ -110,4 +110,5 @@ if __name__ == "__main__":
         max_steps=1,
         verbosity_level=1,
     )
+    agent.run("What is the average each customer paid?")
     GradioUI(agent).launch()