Spaces:
Running
Running
File size: 396 Bytes
b0540b3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from langchain_community.tools.tavily_search import TavilySearchResults
from langgraph.prebuilt import ToolNode
def get_tools():
"""
Returns a list of tools to be used in the chatbot.
"""
tool = TavilySearchResults( max_results=2)
return [tool]
def create_tool_node(tools):
"""
Creates and returns a tool node for the graph.
"""
return ToolNode(tools=tools)
|