shreyassahu22's picture
Upload 10 files
4b58858 verified
Raw
History Blame Contribute Delete
861 Bytes
from typing import List
from langchain_core.tools import BaseTool
from .read_file_tool import read_file_tool
from .visit_webpage_tool import visit_webpage
from .web_search_tool import web_search_tool
from .youtube_transcription_tool import youtube_transcription_tool
from langchain_experimental.tools import PythonREPLTool
from langchain_community.tools import WikipediaQueryRun
from langchain_community.utilities import WikipediaAPIWrapper
python_tool = PythonREPLTool()
wiki_tool = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
def get_tools() -> List[BaseTool]:
"""Returns a list of available tools for the agent."""
tools = [
read_file_tool,
visit_webpage,
web_search_tool,
youtube_transcription_tool,
python_tool,
wiki_tool,
]
return tools