petrov826 commited on
Commit
57020e7
·
verified ·
1 Parent(s): 0716797

rollback changes because langchain's wikipedia tool returns 2 documents but I just want a single document

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -15,8 +15,8 @@ from smolagents import (
15
  )
16
 
17
  from rate_limiter import RateLimiter
18
-
19
  from assemblyai_audio__transcript_tool import AssemblyAIAudioTranscriptTool
 
20
  from youtube_transcript_tool import YoutubeTranscriptTool
21
 
22
  # (Keep Constants as is)
@@ -38,7 +38,6 @@ CUSTOM_PROMPT = """
38
  """
39
 
40
  rate_limiter = RateLimiter()
41
- wikipedia_tool = Tool.from_langchain(load_tools(["wikipedia"])[0])
42
 
43
  # --- Basic Agent Definition ---
44
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -53,7 +52,7 @@ class BasicAgent:
53
  PythonInterpreterTool(),
54
  VisitWebpageTool(),
55
  GoogleSearchTool(provider="serper"),
56
- wikipedia_tool,
57
  YoutubeTranscriptTool(),
58
  AssemblyAIAudioTranscriptTool(),
59
  ],
 
15
  )
16
 
17
  from rate_limiter import RateLimiter
 
18
  from assemblyai_audio__transcript_tool import AssemblyAIAudioTranscriptTool
19
+ from wikipedia_tool import WikipediaTool
20
  from youtube_transcript_tool import YoutubeTranscriptTool
21
 
22
  # (Keep Constants as is)
 
38
  """
39
 
40
  rate_limiter = RateLimiter()
 
41
 
42
  # --- Basic Agent Definition ---
43
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
52
  PythonInterpreterTool(),
53
  VisitWebpageTool(),
54
  GoogleSearchTool(provider="serper"),
55
+ WikipediaTool(),
56
  YoutubeTranscriptTool(),
57
  AssemblyAIAudioTranscriptTool(),
58
  ],