kirbah commited on
Commit
c18904f
·
1 Parent(s): 8cfc5ec

add default tools

Browse files
Files changed (2) hide show
  1. basic_agent.py +5 -2
  2. requirements.txt +2 -1
basic_agent.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel, LiteLLMModel
3
 
4
 
5
  class BasicAgent:
@@ -13,11 +13,14 @@ class BasicAgent:
13
  api_key=os.getenv("GROQ_API_KEY"))
14
 
15
  agent = ToolCallingAgent(
16
- tools=[DuckDuckGoSearchTool()],
 
17
  model=model,
18
  max_steps=5,
19
  name="web_agent",
20
  description="Web Search Agent",
 
 
21
  verbosity_level=2
22
  )
23
  answer = agent.run(
 
1
  import os
2
+ from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, VisitWebpageTool, WikipediaSearchTool, PythonInterpreterTool, InferenceClientModel, LiteLLMModel
3
 
4
 
5
  class BasicAgent:
 
13
  api_key=os.getenv("GROQ_API_KEY"))
14
 
15
  agent = ToolCallingAgent(
16
+ tools=[DuckDuckGoSearchTool(), VisitWebpageTool(),
17
+ WikipediaSearchTool(), PythonInterpreterTool()],
18
  model=model,
19
  max_steps=5,
20
  name="web_agent",
21
  description="Web Search Agent",
22
+ authorized_imports=['statistics', 'unicodedata', 'collections', 'queue',
23
+ 'time', 'pandas', 'stat', 'random', 'datetime', 're', 'math', 'itertools'],
24
  verbosity_level=2
25
  )
26
  answer = agent.run(
requirements.txt CHANGED
@@ -2,4 +2,5 @@ gradio
2
  requests
3
  smolagents
4
  litellm
5
- duckduckgo-search
 
 
2
  requests
3
  smolagents
4
  litellm
5
+ duckduckgo-search
6
+ wikipedia-api