TavilySearchTool

def __init__(
    self, *, llm_config: Optional[Union[LLMConfig, dict[str, Any]]] = None, tavily_api_key: Optional[str] = None
):
    """
    Initializes the TavilySearchTool.

    Args:
        llm_config (Optional[Union[LLMConfig, dict[str, Any]]]): LLM configuration. (Currently unused but kept for potential future integration).
        tavily_api_key (Optional[str]): The API key for the Tavily Search API. If not provided,
            it attempts to read from the `TAVILY_API_KEY` environment variable.

    Raises:
        ValueError: If `tavily_api_key` is not provided either directly or via the environment variable.
    """
    self.tavily_api_key = tavily_api_key or os.getenv("TAVILY_API_KEY")

    if self.tavily_api_key is None:
        raise ValueError("tavily_api_key must be provided either as an argument or via TAVILY_API_KEY env var")

    def tavily_search(
        query: Annotated[str, "The search query."],
        tavily_api_key: Annotated[Optional[str], Depends(on(self.tavily_api_key))],
        search_depth: Annotated[Optional[str], "Either 'advanced' or 'basic'"] = "basic",
        include_answer: Annotated[Optional[str], "Either 'advanced' or 'basic'"] = "basic",
        include_raw_content: Annotated[Optional[bool], "Include the raw contents"] = False,
        include_domains: Annotated[Optional[list[str]], "Specific web domains to search"] = [],
        num_results: Annotated[int, "The number of results to return."] = 5,
    ) -> list[dict[str, Any]]:
        """
        Performs a search using the Tavily API and returns formatted results.

        Args:
            query: The search query string.
            tavily_api_key: The API key for Tavily (injected dependency).
            search_depth: The depth of the search ('basic' or 'advanced'). Defaults to "basic".
            include_answer: Whether to include an AI-generated answer ('basic' or 'advanced'). Defaults to "basic".
            include_raw_content: Whether to include raw content in the results. Defaults to False.
            include_domains: A list of domains to include in the search. Defaults to [].
            num_results: The maximum number of results to return. Defaults to 5.

        Returns:
            A list of dictionaries, each containing 'title', 'link', and 'snippet' of a search result.

        Raises:
            ValueError: If the Tavily API key is not available.
        """
        if tavily_api_key is None:
            raise ValueError("Tavily API key is missing.")
        return _tavily_search(
            query=query,
            tavily_api_key=tavily_api_key,
            search_depth=search_depth or "basic",
            include_answer=include_answer or "basic",
            include_raw_content=include_raw_content or False,
            include_domains=include_domains or [],
            num_results=num_results,
        )

    super().__init__(
        name="tavily_search",
        description="Use the Tavily Search API to perform a search.",
        func_or_tool=tavily_search,
    )