petrov826 commited on
Commit
53dcff2
·
verified ·
1 Parent(s): 05e1f26

replace WikipediaSearchTool with WikipediaTool

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -10,11 +10,12 @@ from smolagents import (
10
  PythonInterpreterTool,
11
  VisitWebpageTool,
12
  GoogleSearchTool,
13
- WikipediaSearchTool,
14
  )
15
 
16
  from rate_limiter import RateLimiter
17
  from assemblyai_audio__transcript_tool import AssemblyAIAudioTranscriptTool
 
18
  from youtube_transcript_tool import YoutubeTranscriptTool
19
 
20
  # (Keep Constants as is)
@@ -50,7 +51,7 @@ class BasicAgent:
50
  PythonInterpreterTool(),
51
  VisitWebpageTool(),
52
  GoogleSearchTool(provider="serper"),
53
- WikipediaSearchTool(),
54
  YoutubeTranscriptTool(),
55
  AssemblyAIAudioTranscriptTool(),
56
  ],
 
10
  PythonInterpreterTool,
11
  VisitWebpageTool,
12
  GoogleSearchTool,
13
+ WikipediaTool,
14
  )
15
 
16
  from rate_limiter import RateLimiter
17
  from assemblyai_audio__transcript_tool import AssemblyAIAudioTranscriptTool
18
+ from wikipedia_tool import WikipediaTool
19
  from youtube_transcript_tool import YoutubeTranscriptTool
20
 
21
  # (Keep Constants as is)
 
51
  PythonInterpreterTool(),
52
  VisitWebpageTool(),
53
  GoogleSearchTool(provider="serper"),
54
+ WikipediaTool(),
55
  YoutubeTranscriptTool(),
56
  AssemblyAIAudioTranscriptTool(),
57
  ],