petrov826 commited on
Commit
04fa3b7
·
verified ·
1 Parent(s): 528f6c4

add some tools

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -9,9 +9,13 @@ from smolagents import (
9
  LiteLLMModel,
10
  PythonInterpreterTool,
11
  DuckDuckGoSearchTool,
12
- VisitWebpageTool
 
 
13
  )
 
14
  from rate_limiter import RateLimiter
 
15
 
16
  # (Keep Constants as is)
17
  # --- Constants ---
@@ -32,8 +36,10 @@ class BasicAgent:
32
  self.agent = CodeAgent(
33
  tools=[
34
  PythonInterpreterTool(),
35
- DuckDuckGoSearchTool(),
36
- VisitWebpageTool()
 
 
37
  ],
38
  model=model,
39
  )
 
9
  LiteLLMModel,
10
  PythonInterpreterTool,
11
  DuckDuckGoSearchTool,
12
+ VisitWebpageTool,
13
+ GoogleSearchTool,
14
+ WikipediaSearchTool
15
  )
16
+
17
  from rate_limiter import RateLimiter
18
+ from youtube_transcript_tool import YoutubeTranscriptTool
19
 
20
  # (Keep Constants as is)
21
  # --- Constants ---
 
36
  self.agent = CodeAgent(
37
  tools=[
38
  PythonInterpreterTool(),
39
+ VisitWebpageTool(),
40
+ GoogleSearchTool(provider="serper"),
41
+ WikipediaSearchTool(),
42
+ YoutubeTranscriptTool()
43
  ],
44
  model=model,
45
  )