petrov826 commited on
Commit
c46e32c
·
verified ·
1 Parent(s): 3434345

add PythonInterpreterTool, DuckDuckGoSearchTool and VisitWebpageTool

Browse files
Files changed (1) hide show
  1. basic_agent.py +9 -2
basic_agent.py CHANGED
@@ -1,7 +1,10 @@
1
  import os
2
  from smolagents import (
3
  CodeAgent,
4
- LiteLLMModel
 
 
 
5
  )
6
  from rate_limiter import RateLimiter
7
 
@@ -12,7 +15,11 @@ class BasicAgent:
12
  api_key=os.getenv("MODEL_API_KEY")
13
  )
14
  self.agent = CodeAgent(
15
- tools=[],
 
 
 
 
16
  model=model,
17
  step_callbacks=[RateLimiter().increment_and_sleep_if_needed]
18
  )
 
1
  import os
2
  from smolagents import (
3
  CodeAgent,
4
+ LiteLLMModel,
5
+ PythonInterpreterTool,
6
+ DuckDuckGoSearchTool,
7
+ VisitWebpageTool,
8
  )
9
  from rate_limiter import RateLimiter
10
 
 
15
  api_key=os.getenv("MODEL_API_KEY")
16
  )
17
  self.agent = CodeAgent(
18
+ tools=[
19
+ PythonInterpreterTool(),
20
+ DuckDuckGoSearchTool(),
21
+ VisitWebpageTool(),
22
+ ],
23
  model=model,
24
  step_callbacks=[RateLimiter().increment_and_sleep_if_needed]
25
  )