Tavily Tools

"""
Tavily Tools
=============================

Demonstrates tavily tools.
"""

from agno.agent import Agent
from agno.tools.tavily import TavilyTools

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------


# Example 1: default TavilyTools
agent = Agent(tools=[TavilyTools()])

# Example 1a: TavilyTools with custom API base URL
# useful for self-hosted or alternative Tavily endpoints
agent_custom = Agent(tools=[TavilyTools(api_base_url="https://custom.tavily.com")])

# Example 2: Enable all Tavily functions (search + extract)
agent_all = Agent(tools=[TavilyTools(all=True)])

# Example 3: Use advanced search with context
context_agent = Agent(
    tools=[
        TavilyTools(
            enable_search=True,
        )
    ]
)

# ============================================================================
# EXTRACT EXAMPLES
# ============================================================================

# Example 4: URL content extraction with markdown format
extract_agent = Agent(
    tools=[
        TavilyTools(
            enable_search=False,  # Disable search for this example
            enable_extract=True,
            extract_depth="basic",  # basic = 1 credit/5 URLs
            extract_format="markdown",
        )
    ]
)

# Example 5: Advanced extraction with images in text format
advanced_extract_agent = Agent(
    tools=[
        TavilyTools(
            enable_search=False,
            enable_extract=True,
            extract_depth="advanced",  # advanced = 2 credits/5 URLs
            extract_format="text",
            include_images=True,
            include_favicon=True,
        )
    ]
)

# Example 6: Combined search and extract
combined_agent = Agent(
    tools=[
        TavilyTools(
            enable_search=True,
            enable_extract=True,
            search_depth="basic",
            extract_depth="basic",
            format="markdown",  # Format for search results
            extract_format="markdown",  # Format for extracted content
        )
    ]
)

# ============================================================================
# TEST THE AGENTS
# ============================================================================

# Test search agents

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    print("=" * 80)
    print("SEARCH EXAMPLES")
    print("=" * 80)

    agent.print_response(
        "Search for 'language models' and recent developments", markdown=True
    )

    context_agent.print_response(
        "Get detailed context about artificial intelligence trends", markdown=True
    )

    # Test extract agents
    print("\n" + "=" * 80)
    print("EXTRACT EXAMPLES")
    print("=" * 80)

    extract_agent.print_response(
        "Extract the main content from https://docs.tavily.com/documentation/api-reference/endpoint/extract",
        markdown=True,
    )

    advanced_extract_agent.print_response(
        "Extract content with images from https://github.com/anthropics/anthropic-sdk-python",
        markdown=True,
    )

    # Test combined agent
    print("\n" + "=" * 80)
    print("COMBINED SEARCH & EXTRACT")
    print("=" * 80)

    combined_agent.print_response(
        "Search for 'Tavily API documentation' and extract content from the most relevant result",
        markdown=True,
    )