mjschock commited on
Commit
0d02f24
·
unverified ·
1 Parent(s): ae7a494
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -52,10 +52,17 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(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,
 
52
 
53
  with open("prompts.yaml", 'r') as stream:
54
  prompt_templates = yaml.safe_load(stream)
55
+
56
+ tools = [
57
+ DuckDuckGoSearchTool(),
58
+ image_generation_tool,
59
+ get_current_time_in_timezone,
60
+ my_custom_tool
61
+ ]
62
+
63
  agent = CodeAgent(
64
  model=model,
65
+ tools=tools, ## add your tools here (don't remove final answer)
66
  max_steps=6,
67
  verbosity_level=1,
68
  grammar=None,