RemVdH commited on
Commit
152d857
·
verified ·
1 Parent(s): 6f8e73f

Update app.py

Browse files

Added base tools

Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,4 +1,3 @@
1
-
2
  from smolagents import CodeAgent, HfApiModel, load_tool, tool
3
  from tools.final_answer import FinalAnswerTool
4
  from typing import List
@@ -27,6 +26,7 @@ def search_wikipedia_pages(search_string: str) -> List[str]:
27
  print (f"An error occured {s_error}")
28
 
29
  @tool
 
30
  def get_wikipedia_page_content(page_name: str, p_summary: bool = True) -> str:
31
  """This tool helps to get the content of wikipedia pages by their name.
32
  If you use this tool based on the output of search_wikipedia_pages
@@ -71,7 +71,8 @@ agent = CodeAgent(
71
  planning_interval=None,
72
  name=None,
73
  description=None,
74
- prompt_templates=prompt_templates
 
75
  )
76
 
77
  GradioUI(agent).launch()
 
 
1
  from smolagents import CodeAgent, HfApiModel, load_tool, tool
2
  from tools.final_answer import FinalAnswerTool
3
  from typing import List
 
26
  print (f"An error occured {s_error}")
27
 
28
  @tool
29
+
30
  def get_wikipedia_page_content(page_name: str, p_summary: bool = True) -> str:
31
  """This tool helps to get the content of wikipedia pages by their name.
32
  If you use this tool based on the output of search_wikipedia_pages
 
71
  planning_interval=None,
72
  name=None,
73
  description=None,
74
+ prompt_templates=prompt_templates,
75
+ add_base_tools=True
76
  )
77
 
78
  GradioUI(agent).launch()