Banana-Leopard commited on
Commit
51fcb56
·
verified ·
1 Parent(s): c510830

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -57,11 +57,12 @@ custom_role_conversions=None,
57
 
58
  # Import tool from Hub
59
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
60
 
61
  with open("prompts.yaml", 'r') as stream:
62
  prompt_templates = yaml.safe_load(stream)
63
 
64
- for t in [final_answer, rick_roll_tool, my_custom_tool, image_generation_tool, get_current_time_in_timezone, DuckDuckGoSearchTool]:
65
  print(t, type(t))
66
 
67
  agent = CodeAgent(
@@ -72,6 +73,7 @@ agent = CodeAgent(
72
  my_custom_tool,
73
  image_generation_tool,
74
  get_current_time_in_timezone,
 
75
  # DuckDuckGoSearchTool,
76
  ], ## add your tools here (don't remove final answer)
77
  max_steps=6,
 
57
 
58
  # Import tool from Hub
59
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
60
+ browser_tool = load_tool(DuckDuckGoSearchTool)
61
 
62
  with open("prompts.yaml", 'r') as stream:
63
  prompt_templates = yaml.safe_load(stream)
64
 
65
+ for t in [final_answer, rick_roll_tool, my_custom_tool, image_generation_tool, get_current_time_in_timezone, DuckDuckGoSearchTool, browser_tool]:
66
  print(t, type(t))
67
 
68
  agent = CodeAgent(
 
73
  my_custom_tool,
74
  image_generation_tool,
75
  get_current_time_in_timezone,
76
+ browser_tool,
77
  # DuckDuckGoSearchTool,
78
  ], ## add your tools here (don't remove final answer)
79
  max_steps=6,