HJeon commited on
Commit
efef216
·
verified ·
1 Parent(s): 6de3e62

Update Agent's tool list

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,7 +36,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
36
  import wikipedia
37
 
38
  @tool
39
- def wikipedia_summary(query: str) -> str:
40
  """
41
  A tool that fetches a Wikipedia summary for a given topic.
42
 
@@ -79,7 +79,7 @@ with open("prompts.yaml", 'r') as stream:
79
 
80
  agent = CodeAgent(
81
  model=model,
82
- tools=[final_answer], ## add your tools here (don't remove final answer)
83
  max_steps=6,
84
  verbosity_level=1,
85
  grammar=None,
 
36
  import wikipedia
37
 
38
  @tool
39
+ def get_wikipedia_summary(query: str) -> str:
40
  """
41
  A tool that fetches a Wikipedia summary for a given topic.
42
 
 
79
 
80
  agent = CodeAgent(
81
  model=model,
82
+ tools=[final_answer, get_wikipedia_summary], ## add your tools here (don't remove final answer)
83
  max_steps=6,
84
  verbosity_level=1,
85
  grammar=None,