BCopeland64 commited on
Commit
56f1c1f
·
verified ·
1 Parent(s): ed22e6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -42,12 +42,16 @@ model = HfApiModel(
42
  # Import tool from Hub
43
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
44
 
 
 
 
 
45
  with open("prompts.yaml", 'r') as stream:
46
  prompt_templates = yaml.safe_load(stream)
47
 
48
  agent = CodeAgent(
49
  model=model,
50
- tools=[final_answer, get_ai_agent_info, get_ai_agent_roadmap], # add your tools here
51
  max_steps=6,
52
  verbosity_level=1,
53
  grammar=None,
 
42
  # Import tool from Hub
43
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
44
 
45
+ # Create instances of the DuckDuckGoSearchTool class
46
+ search_tool = DuckDuckGoSearchTool()
47
+ roadmap_tool = FinalAnswerTool()
48
+
49
  with open("prompts.yaml", 'r') as stream:
50
  prompt_templates = yaml.safe_load(stream)
51
 
52
  agent = CodeAgent(
53
  model=model,
54
+ tools=[final_answer, search_tool, roadmap_tool], # add your tools here
55
  max_steps=6,
56
  verbosity_level=1,
57
  grammar=None,