| 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 |