mobrobro commited on
Commit
6e67fed
·
verified ·
1 Parent(s): 26001c8

update tools defined

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -76,7 +76,12 @@ with open("prompts.yaml", 'r') as stream:
76
 
77
  agent = CodeAgent(
78
  model=model,
79
- tools=[final_answer], ## add your tools here (don't remove final answer)
 
 
 
 
 
80
  max_steps=6,
81
  verbosity_level=1,
82
  grammar=None,
@@ -86,5 +91,4 @@ agent = CodeAgent(
86
  prompt_templates=prompt_templates
87
  )
88
 
89
-
90
  GradioUI(agent).launch()
 
76
 
77
  agent = CodeAgent(
78
  model=model,
79
+ tools=[
80
+ final_answer, # Keep this
81
+ image_generation_tool, # Image generation
82
+ tool(get_current_time_in_timezone), # Timezone checker
83
+ tool(animal_battle), # Your animal battle simulator
84
+ ],
85
  max_steps=6,
86
  verbosity_level=1,
87
  grammar=None,
 
91
  prompt_templates=prompt_templates
92
  )
93
 
 
94
  GradioUI(agent).launch()