[docs]classTavilySearchInput(BaseModel):"""Input for [TavilySearch]"""query:str=Field(description=("Search query to look up"))include_domains:Optional[List[str]]=Field(default=[],description="""A list of domains to restrict search results to. Use this parameter when: 1. The user explicitly requests information from specific websites (e.g., "Find climate data from nasa.gov") 2. The user mentions an organization or company without specifying the domain (e.g., "Find information about iPhones from Apple") In both cases, you should determine the appropriate domains (e.g., ["nasa.gov"] or ["apple.com"]) and set this parameter. Results will ONLY come from the specified domains - no other sources will be included. Default is None (no domain restriction). """,# noqa: E501)exclude_domains:Optional[List[str]]=Field(default=[],description="""A list of domains to exclude from search results. Use this parameter when: 1. The user explicitly requests to avoid certain websites (e.g., "Find information about climate change but not from twitter.com") 2. The user mentions not wanting results from specific organizations without naming the domain (e.g., "Find phone reviews but nothing from Apple") In both cases, you should determine the appropriate domains to exclude (e.g., ["twitter.com"] or ["apple.com"]) and set this parameter. Results will filter out all content from the specified domains. Default is None (no domain exclusion). """,# noqa: E501)search_depth:Optional[Literal["basic","advanced"]]=Field(default="basic",description="""Controls search thoroughness and result comprehensiveness. Use "basic" for simple queries requiring quick, straightforward answers. Use "advanced" (default) for complex queries, specialized topics, rare information, or when in-depth analysis is needed. """,# noqa: E501)include_images:Optional[bool]=Field(default=False,description="""Determines if the search returns relevant images along with text results. Set to True when the user explicitly requests visuals or when images would significantly enhance understanding (e.g., "Show me what black holes look like," "Find pictures of Renaissance art"). Leave as False (default) for most informational queries where text is sufficient. """,# noqa: E501)time_range:Optional[Literal["day","week","month","year"]]=Field(default=None,description="""Limits results to content published within a specific timeframe. ONLY set this when the user explicitly mentions a time period (e.g., "latest AI news," "articles from last week"). For less popular or niche topics, use broader time ranges ("month" or "year") to ensure sufficient relevant results. Options: "day" (24h), "week" (7d), "month" (30d), "year" (365d). Default is None. """,# noqa: E501)topic:Optional[Literal["general","news","finance"]]=Field(default="general",description="""Specifies search category for optimized results. Use "general" (default) for most queries, INCLUDING those with terms like "latest," "newest," or "recent" when referring to general information. Use "finance" for markets, investments, economic data, or financial news. Use "news" ONLY for politics, sports, or major current events covered by mainstream media - NOT simply because a query asks for "new" information. """,# noqa: E501)
def_generate_suggestions(params:dict)->list:"""Generate helpful suggestions based on the failed search parameters."""suggestions=[]search_depth=params.get("search_depth")exclude_domains=params.get("exclude_domains")include_domains=params.get("include_domains")time_range=params.get("time_range")topic=params.get("topic")iftime_range:suggestions.append("Remove time_range argument")ifinclude_domains:suggestions.append("Remove include_domains argument")ifexclude_domains:suggestions.append("Remove exclude_domains argument")ifsearch_depth=="basic":suggestions.append("Try a more detailed search using 'advanced' search_depth")iftopic!="general":suggestions.append("Try a general search using 'general' topic")returnsuggestions
[docs]classTavilySearch(BaseTool):# type: ignore[override]"""Tool that queries the Tavily Search API and gets back json. Setup: Install ``langchain-tavily`` and set environment variable ``TAVILY_API_KEY``. .. code-block:: bash pip install -U langchain-tavily export TAVILY_API_KEY="your-api-key" Instantiate: .. code-block:: python from langchain_tavily import TavilySearch tool = TavilySearch( max_results=1, topic="general", # include_answer=False, # include_raw_content=False, # include_images=False, # include_image_descriptions=False, # search_depth="basic", # time_range="day", # include_domains=None, # exclude_domains=None, # country=None ) Invoke directly with args: .. code-block:: python tool.invoke({"query": "What happened at the last wimbledon"}) .. code-block:: json { 'query': 'What happened at the last wimbledon', 'follow_up_questions': None, 'answer': None, 'images': [], 'results': [{'title': "Andy Murray pulls out of the men's singles draw at his last Wimbledon", 'url': 'https://www.nbcnews.com/news/sports/andy-murray-wimbledon-tennis-singles-draw-rcna159912', 'content': "NBC News Now LONDON — Andy Murray, one of the last decade's most successful ..." 'score': 0.6755297, 'raw_content': None }], 'response_time': 1.31 } """# noqa: E501name:str="tavily_search"description:str=("A search engine optimized for comprehensive, accurate, and trusted results. ""Useful for when you need to answer questions about current events. ""It not only retrieves URLs and snippets, but offers advanced search depths, ""domain management, time range filters, and image search, this tool delivers ""real-time, accurate, and citation-backed results.""Input should be a search query.")args_schema:Type[BaseModel]=TavilySearchInputhandle_tool_error:bool=Trueinclude_domains:Optional[List[str]]=None"""A list of domains to specifically include in the search results default is None """exclude_domains:Optional[List[str]]=None"""A list of domains to specifically exclude from the search results default is None """search_depth:Optional[Literal["basic","advanced"]]="basic""""The depth of the search. It can be 'basic' or 'advanced' default is "basic" """include_images:Optional[bool]=False"""Include a list of query related images in the response default is False """time_range:Optional[Literal["day","week","month","year"]]=None"""The time range back from the current date to filter results default is None """max_results:Optional[int]=5"""Max search results to return, default is 5 """topic:Optional[Literal["general","news","finance"]]="general""""The category of the search. Can be "general", "news", or "finance". Default is "general". """include_answer:Optional[Union[bool,Literal["basic","advanced"]]]=False"""Include a short answer to original query in the search results. Default is False. """include_raw_content:Optional[Union[bool,Literal["markdown","text"]]]=False"""Include an LLM-generated answer to the provided query. basic or true returns a quick answer. advanced returns a more detailed answer. Default is False. """include_image_descriptions:Optional[bool]=False"""Include a descriptive text for each image in the search results. Default is False. """country:Optional[str]=None"""Boost search results from a specific country. This will prioritize content from the selected country in the search results. Available only if topic is general. To see the countries supported visit our docs https://docs.tavily.com/documentation/api-reference/endpoint/search Default is None. """api_wrapper:TavilySearchAPIWrapper=Field(default_factory=TavilySearchAPIWrapper)# type: ignore[arg-type]def__init__(self,**kwargs:Any)->None:# Create api_wrapper with tavily_api_key if providedif"tavily_api_key"inkwargs:kwargs["api_wrapper"]=TavilySearchAPIWrapper(tavily_api_key=kwargs["tavily_api_key"])super().__init__(**kwargs)def_run(self,query:str,include_domains:Optional[List[str]]=None,exclude_domains:Optional[List[str]]=None,search_depth:Optional[Literal["basic","advanced"]]="basic",include_images:Optional[bool]=False,time_range:Optional[Literal["day","week","month","year"]]=None,topic:Optional[Literal["general","news","finance"]]="general",run_manager:Optional[CallbackManagerForToolRun]=None,)->Dict[str,Any]:"""Execute a search query using the Tavily Search API. Returns: Dict[str, Any]: Search results containing: - query: Original search query - results: List of search results, each with: - title: Title of the page - url: URL of the page - content: Relevant content snippet - score: Relevance score - images: List of relevant images (if include_images=True) - response_time: Time taken for the search """try:# Execute search with parameters directlyraw_results=self.api_wrapper.raw_results(query=query,include_domains=include_domainsifinclude_domainselseself.include_domains,exclude_domains=exclude_domainsifexclude_domainselseself.exclude_domains,search_depth=search_depthifsearch_depthelseself.search_depth,include_images=include_imagesifinclude_imageselseself.include_images,time_range=time_rangeiftime_rangeelseself.time_range,topic=topiciftopicelseself.topic,country=self.country,max_results=self.max_results,include_answer=self.include_answer,include_raw_content=self.include_raw_content,include_image_descriptions=self.include_image_descriptions,)# Check if results are empty and raise a specific exceptionifnotraw_results.get("results",[]):search_params={"time_range":time_range,"include_domains":include_domains,"search_depth":search_depth,"exclude_domains":exclude_domains,"topic":topic,}suggestions=_generate_suggestions(search_params)# Construct a detailed message for the agenterror_message=(f"No search results found for '{query}'. "f"Suggestions: {', '.join(suggestions)}. "f"Try modifying your search parameters with one of these approaches."# noqa: E501)raiseToolException(error_message)returnraw_resultsexceptToolException:# Re-raise tool exceptionsraiseexceptExceptionase:return{"error":e}asyncdef_arun(self,query:str,include_domains:Optional[List[str]]=None,exclude_domains:Optional[List[str]]=None,search_depth:Optional[Literal["basic","advanced"]]="basic",include_images:Optional[bool]=False,time_range:Optional[Literal["day","week","month","year"]]=None,topic:Optional[Literal["general","news","finance"]]="general",run_manager:Optional[AsyncCallbackManagerForToolRun]=None,)->Dict[str,Any]:"""Use the tool asynchronously."""try:raw_results=awaitself.api_wrapper.raw_results_async(query=query,include_domains=include_domainsifinclude_domainselseself.include_domains,exclude_domains=exclude_domainsifexclude_domainselseself.exclude_domains,search_depth=search_depthifsearch_depthelseself.search_depth,include_images=include_imagesifinclude_imageselseself.include_images,time_range=time_rangeiftime_rangeelseself.time_range,topic=topiciftopicelseself.topic,country=self.country,max_results=self.max_results,include_answer=self.include_answer,include_raw_content=self.include_raw_content,include_image_descriptions=self.include_image_descriptions,)# Check if results are empty and raise a specific exceptionifnotraw_results.get("results",[]):search_params={"time_range":time_range,"include_domains":include_domains,"search_depth":search_depth,"exclude_domains":exclude_domains,"topic":topic,}suggestions=_generate_suggestions(search_params)# Construct a detailed message for the agenterror_message=(f"No search results found for '{query}'. "f"Suggestions: {', '.join(suggestions)}. "f"Try modifying your search parameters with one of these approaches."# noqa: E501)raiseToolException(error_message)returnraw_resultsexceptToolException:# Re-raise tool exceptionsraiseexceptExceptionase:return{"error":e}