Updated

app.py

@@ -24,12 +24,27 @@ def suggest_menu(occasion: str) -> str:
     else:
         return "Custom menu for the butler."
 
-# Alfred, the butler, preparing the playlist for the party
-agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
-agent.run("Search for the best music recommendations for a party at the Wayne's mansion.")
+model = HfApiModel(
+max_tokens=2096,
+temperature=0.5,
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+custom_role_conversions=None,
+)
+
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+agent = CodeAgent(
+    model=model,
+    tools=[get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
+    max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
+    prompt_templates=prompt_templates
+)
 
-# Alfred, the butler, preparing the menu for the party
-agent = CodeAgent(tools=[suggest_menu], model=HfApiModel())
-agent.run("Prepare a formal menu for the party.")
 
 GradioUI(agent).launch()