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

simplify wikipedia tool related codes using Tool.from_langchain

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,7 +4,9 @@ import litellm
4
  import requests
5
  import inspect
6
  import pandas as pd
 
7
  from smolagents import (
 
8
  CodeAgent,
9
  LiteLLMModel,
10
  PythonInterpreterTool,
@@ -15,7 +17,6 @@ from smolagents import (
15
  from rate_limiter import RateLimiter
16
 
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)
@@ -37,6 +38,7 @@ CUSTOM_PROMPT = """
37
  """
38
 
39
  rate_limiter = RateLimiter()
 
40
 
41
  # --- Basic Agent Definition ---
42
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -51,7 +53,7 @@ class BasicAgent:
51
  PythonInterpreterTool(),
52
  VisitWebpageTool(),
53
  GoogleSearchTool(provider="serper"),
54
- WikipediaTool(),
55
  YoutubeTranscriptTool(),
56
  AssemblyAIAudioTranscriptTool(),
57
  ],
 
4
  import requests
5
  import inspect
6
  import pandas as pd
7
+ from langchain.agents import load_tools
8
  from smolagents import (
9
+ Tool,
10
  CodeAgent,
11
  LiteLLMModel,
12
  PythonInterpreterTool,
 
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
  """
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
  PythonInterpreterTool(),
54
  VisitWebpageTool(),
55
  GoogleSearchTool(provider="serper"),
56
+ wikipedia_tool,
57
  YoutubeTranscriptTool(),
58
  AssemblyAIAudioTranscriptTool(),
59
  ],