JCJuice commited on
Commit
e2f3a46
·
verified ·
1 Parent(s): 2df94aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -64,12 +64,14 @@ custom_role_conversions=None,
64
  # Import tool from Hub
65
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
66
 
 
 
67
  with open("prompts.yaml", 'r') as stream:
68
  prompt_templates = yaml.safe_load(stream)
69
 
70
  agent = CodeAgent(
71
  model=model,
72
- tools=[final_answer, image_generation_tool, DuckDuckGoSearchTool(), search_duckduckgo, my_custom_tool], ## add your tools here (don't remove final answer)
73
  max_steps=6,
74
  verbosity_level=1,
75
  grammar=None,
 
64
  # Import tool from Hub
65
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
66
 
67
+ duckduckgo_tool = DuckDuckGoSearchTool()
68
+
69
  with open("prompts.yaml", 'r') as stream:
70
  prompt_templates = yaml.safe_load(stream)
71
 
72
  agent = CodeAgent(
73
  model=model,
74
+ tools=[final_answer, image_generation_tool, duckduckgo_tool, search_duckduckgo, my_custom_tool], ## add your tools here (don't remove final answer)
75
  max_steps=6,
76
  verbosity_level=1,
77
  grammar=None,