Spaces:
Sleeping
Sleeping
| from langchain_community.tools.tavily_search import TavilySearchResults | |
| from langgraph.prebuilt import ToolNode | |
| def get_tools(): | |
| """ | |
| Return the list of tools to be used in the chatbot | |
| """ | |
| tools=[TavilySearchResults(max_results=2)] | |
| return tools | |
| def create_tool_node(tools): | |
| """ | |
| creates and returns a tool node for the graph | |
| """ | |
| return ToolNode(tools=tools) | |
| def get_tools_by_usecase(usecase): | |
| """ | |
| Return tools based on the specific use case | |
| """ | |
| if usecase == "Chatbot with Web Search": | |
| return get_tools() | |
| elif usecase == "Research Assistant": | |
| from .arxiv_tool import get_research_assistant_tools | |
| return get_research_assistant_tools() | |
| else: | |
| # Default to web search tools | |
| return get_tools() |