ChaosBot commited on
Commit
905d1ad
·
verified ·
1 Parent(s): c71e212

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -83,12 +83,14 @@ custom_role_conversions=None,
83
  # Import tool from Hub
84
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
85
 
 
 
86
  with open("prompts.yaml", 'r') as stream:
87
  prompt_templates = yaml.safe_load(stream)
88
 
89
  agent = CodeAgent(
90
  model=model,
91
- tools=[final_answer, image_generation_tool, get_current_time_in_timezone, get_weather], ## add your tools here (don't remove final answer)
92
  max_steps=6,
93
  verbosity_level=1,
94
  grammar=None,
 
83
  # Import tool from Hub
84
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
85
 
86
+ web_search = load_tool("agents-course/web_search", trust_remote_code=True)
87
+
88
  with open("prompts.yaml", 'r') as stream:
89
  prompt_templates = yaml.safe_load(stream)
90
 
91
  agent = CodeAgent(
92
  model=model,
93
+ tools=[final_answer, image_generation_tool, get_current_time_in_timezone, get_weather, web_search], ## add your tools here (don't remove final answer)
94
  max_steps=6,
95
  verbosity_level=1,
96
  grammar=None,