Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -55,7 +55,12 @@ with open("prompts.yaml", 'r') as stream:
55
 
56
  agent = CodeAgent(
57
  model=model,
58
- tools=[final_answer], ## add your tools here (don't remove final answer)
 
 
 
 
 
59
  max_steps=6,
60
  verbosity_level=1,
61
  grammar=None,
@@ -66,4 +71,5 @@ agent = CodeAgent(
66
  )
67
 
68
 
 
69
  GradioUI(agent).launch()
 
55
 
56
  agent = CodeAgent(
57
  model=model,
58
+ tools=[
59
+ final_answer,
60
+ my_custom_tool,
61
+ get_current_time_in_timezone,
62
+ image_generation_tool
63
+ ],
64
  max_steps=6,
65
  verbosity_level=1,
66
  grammar=None,
 
71
  )
72
 
73
 
74
+
75
  GradioUI(agent).launch()