tools/tavily_tools.py

from langchain_tavily import TavilySearch
from langchain_tavily import TavilyExtract

# Initialize Tavily Search Tool
tavily_search_tool = TavilySearch(
    max_results=10,
    topic="general",
    # Make sure to avoid retrieving the response from a dataset or a space
    exclude_domains=["https://huggingface.co/datasets", "https://huggingface.co/spaces"]
)

# Define the LangChain extract tool
tavily_extract_tool = TavilyExtract(extract_depth="basic")

web_search_tools = [
    tavily_search_tool, tavily_extract_tool
]