agno 1.8.1__py3-none-any.whl → 2.0.0__py3-none-any.whl

agno/tools/agentql.py

@@ -12,7 +12,15 @@ except ImportError:
 
 
 class AgentQLTools(Toolkit):
-    def __init__(self, api_key: Optional[str] = None, scrape: bool = True, agentql_query: str = "", **kwargs):
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        enable_scrape_website: bool = True,
+        enable_custom_scrape_website: bool = False,
+        all: bool = False,
+        agentql_query: str = "",
+        **kwargs,
+    ):
         self.api_key = api_key or getenv("AGENTQL_API_KEY")
         if not self.api_key:
             raise ValueError("AGENTQL_API_KEY not set. Please set the AGENTQL_API_KEY environment variable.")
@@ -20,11 +28,12 @@ class AgentQLTools(Toolkit):
         self.agentql_query = agentql_query
 
         tools: List[Any] = []
-        if scrape:
+        if all or enable_scrape_website:
             tools.append(self.scrape_website)
-        if agentql_query:
-            log_info("Custom AgentQL query provided. Registering custom scrape function.")
-            tools.append(self.custom_scrape_website)
+        if all or enable_custom_scrape_website or (agentql_query and not all and not enable_custom_scrape_website):
+            if agentql_query:
+                log_info("Custom AgentQL query provided. Registering custom scrape function.")
+                tools.append(self.custom_scrape_website)
 
         super().__init__(name="agentql_tools", tools=tools, **kwargs)
 

agno/tools/airflow.py

@@ -7,7 +7,12 @@ from agno.utils.log import log_debug, log_info, logger
 
 class AirflowTools(Toolkit):
     def __init__(
-        self, dags_dir: Optional[Union[Path, str]] = None, save_dag: bool = True, read_dag: bool = True, **kwargs
+        self,
+        dags_dir: Optional[Union[Path, str]] = None,
+        enable_save_dag_file: bool = True,
+        enable_read_dag_file: bool = True,
+        all: bool = False,
+        **kwargs,
     ):
         """
         quick start to work with airflow : https://airflow.apache.org/docs/apache-airflow/stable/start.html
@@ -22,9 +27,9 @@ class AirflowTools(Toolkit):
         self.dags_dir: Path = _dags_dir or Path.cwd()
 
         tools: List[Any] = []
-        if save_dag:
+        if all or enable_save_dag_file:
             tools.append(self.save_dag_file)
-        if read_dag:
+        if all or enable_read_dag_file:
             tools.append(self.read_dag_file)
 
         super().__init__(name="AirflowTools", tools=tools, **kwargs)
@@ -43,7 +48,7 @@ class AirflowTools(Toolkit):
                 file_path.parent.mkdir(parents=True, exist_ok=True)
             file_path.write_text(contents)
             log_info(f"Saved: {file_path}")
-            return str(str(file_path))
+            return str(file_path)
         except Exception as e:
             logger.error(f"Error saving to file: {e}")
             return f"Error saving to file: {e}"

agno/tools/api.py

@@ -21,7 +21,8 @@ class CustomApiTools(Toolkit):
         headers: Optional[Dict[str, str]] = None,
         verify_ssl: bool = True,
         timeout: int = 30,
-        make_request: bool = True,
+        enable_make_request: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.base_url = base_url
@@ -33,7 +34,7 @@ class CustomApiTools(Toolkit):
         self.timeout = timeout
 
         tools: List[Any] = []
-        if make_request:
+        if all or enable_make_request:
             tools.append(self.make_request)
 
         super().__init__(name="api_tools", tools=tools, **kwargs)
@@ -76,7 +77,10 @@ class CustomApiTools(Toolkit):
             str: JSON string containing response data or error message
         """
         try:
-            url = f"{self.base_url.rstrip('/')}/{endpoint.lstrip('/')}" if self.base_url else endpoint
+            if self.base_url:
+                url = f"{self.base_url.rstrip('/')}/{endpoint.lstrip('/')}"
+            else:
+                url = endpoint
             log_debug(f"Making {method} request to {url}")
 
             response = requests.request(

agno/tools/apify.py

@@ -1,6 +1,6 @@
 import json
-import os
 import string
+from os import getenv
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 import requests
@@ -21,53 +21,9 @@ class ApifyTools(Toolkit):
         Args:
             actors (Optional[Union[str, List[str]]]): Single Actor ID as string or list of Actor IDs to register as individual tools
             apify_api_token (Optional[str]): Apify API token (defaults to APIFY_API_TOKEN env variable)
-
-        Examples:
-            Configuration Instructions:
-            1. Install required dependencies:
-            pip install agno apify-client
-
-            2. Set the APIFY_API_TOKEN environment variable:
-            Add a .env file with APIFY_API_TOKEN=your_apify_api_key
-
-            Import necessary components:
-
-            from agno.agent import Agent
-            from agno.tools.apify import ApifyTools
-
-            # Create an agent with ApifyTools
-            agent = Agent(
-                tools=[
-                    ApifyTools(actors=["apify/rag-web-browser"])
-                ],
-                markdown=True
-            )
-
-            # Ask the agent to process web content
-            agent.print_response("Summarize the content from https://docs.agno.com/introduction", markdown=True)
-
-            # Using multiple actors with the agent
-            agent = Agent(
-                tools=[
-                    ApifyTools(actors=[
-                        "apify/rag-web-browser",
-                        "compass/crawler-google-places"
-                    ])
-                ],
-                show_tool_calls=True
-            )
-            agent.print_response(
-                '''
-                I'm traveling to Tokyo next month.
-                1. Research the best time to visit and major attractions
-                2. Find one good rated sushi restaurant near Shinjuku
-                Compile a comprehensive travel guide with this information.
-                ''',
-                markdown=True
-            )
         """
         # Get API token from args or environment
-        self.apify_api_token = apify_api_token or os.getenv("APIFY_API_TOKEN")
+        self.apify_api_token = apify_api_token or getenv("APIFY_API_TOKEN")
         if not self.apify_api_token:
             raise ValueError("APIFY_API_TOKEN environment variable or apify_api_token parameter must be set")
 

agno/tools/arxiv.py

@@ -18,15 +18,20 @@ except ImportError:
 
 class ArxivTools(Toolkit):
     def __init__(
-        self, search_arxiv: bool = True, read_arxiv_papers: bool = True, download_dir: Optional[Path] = None, **kwargs
+        self,
+        enable_search_arxiv: bool = True,
+        enable_read_arxiv_papers: bool = True,
+        all: bool = False,
+        download_dir: Optional[Path] = None,
+        **kwargs,
     ):
         self.client: arxiv.Client = arxiv.Client()
         self.download_dir: Path = download_dir or Path(__file__).parent.joinpath("arxiv_pdfs")
 
         tools: List[Any] = []
-        if search_arxiv:
+        if all or enable_search_arxiv:
             tools.append(self.search_arxiv_and_return_articles)
-        if read_arxiv_papers:
+        if all or enable_read_arxiv_papers:
             tools.append(self.read_arxiv_papers)
 
         super().__init__(name="arxiv_tools", tools=tools, **kwargs)

agno/tools/aws_lambda.py

@@ -9,19 +9,21 @@ except ImportError:
 
 
 class AWSLambdaTools(Toolkit):
-    name: str = "AWSLambdaTool"
-    description: str = "A tool for interacting with AWS Lambda functions"
-
     def __init__(
         self,
         region_name: str = "us-east-1",
+        enable_list_functions: bool = True,
+        enable_invoke_function: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.client = boto3.client("lambda", region_name=region_name)
 
         tools: List[Any] = []
-        tools.append(self.list_functions)
-        tools.append(self.invoke_function)
+        if all or enable_list_functions:
+            tools.append(self.list_functions)
+        if all or enable_invoke_function:
+            tools.append(self.invoke_function)
 
         super().__init__(name="aws-lambda", tools=tools, **kwargs)
 

agno/tools/aws_ses.py

@@ -15,8 +15,14 @@ class AWSSESTool(Toolkit):
         sender_email: Optional[str] = None,
         sender_name: Optional[str] = None,
         region_name: str = "us-east-1",
+        enable_send_email: bool = True,
+        all: bool = False,
+        **kwargs,
     ):
-        super().__init__(name="aws_ses_tool", tools=[self.send_email])
+        tools = []
+        if all or enable_send_email:
+            tools.append(self.send_email)
+        super().__init__(name="aws_ses_tool", tools=tools, **kwargs)
         self.client = boto3.client("ses", region_name=region_name)
         self.sender_email = sender_email
         self.sender_name = sender_name

agno/tools/baidusearch.py

@@ -35,6 +35,8 @@ class BaiduSearchTools(Toolkit):
         proxy: Optional[str] = None,
         timeout: Optional[int] = 10,
         debug: Optional[bool] = False,
+        enable_baidu_search: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.fixed_max_results = fixed_max_results
@@ -45,7 +47,8 @@ class BaiduSearchTools(Toolkit):
         self.debug = debug
 
         tools = []
-        tools.append(self.baidu_search)
+        if all or enable_baidu_search:
+            tools.append(self.baidu_search)
 
         super().__init__(name="baidusearch", tools=tools, **kwargs)
 

agno/tools/bitbucket.py

@@ -1,6 +1,6 @@
 import base64
 import json
-import os
+from os import getenv
 from typing import Any, Dict, Optional, Union
 
 import requests
@@ -21,9 +21,9 @@ class BitbucketTools(Toolkit):
         api_version: str = "2.0",
         **kwargs,
     ):
-        self.username = username or os.getenv("BITBUCKET_USERNAME")
-        self.password = password or os.getenv("BITBUCKET_PASSWORD")
-        self.token = token or os.getenv("BITBUCKET_TOKEN")
+        self.username = username or getenv("BITBUCKET_USERNAME")
+        self.password = password or getenv("BITBUCKET_PASSWORD")
+        self.token = token or getenv("BITBUCKET_TOKEN")
         self.auth_password = self.token or self.password
         self.server_url = server_url or "api.bitbucket.org"
         self.api_version = api_version or "2.0"

agno/tools/brandfetch.py

@@ -28,9 +28,11 @@ class BrandfetchTools(Toolkit):
 
     client_id: str - your Brandfetch Client ID
 
-    async_tools: bool = True - if True, will use async tools, if False, will use sync tools
-    brand: bool = False - if True, will use brand api, if False, will not use brand api
-    search: bool = False - if True, will use brand search api, if False, will not use brand search api
+    all: bool - if True, will use all tools
+    enable_search_by_identifier: bool - if True, will use search by identifier
+    enable_search_by_brand: bool - if True, will use search by brand
+    enable_asearch_by_identifier: bool - if True, will use async search by identifier
+    enable_asearch_by_brand: bool - if True, will use async search by brand
     """
 
     def __init__(
@@ -39,29 +41,30 @@ class BrandfetchTools(Toolkit):
         client_id: Optional[str] = None,
         base_url: str = "https://api.brandfetch.io/v2",
         timeout: Optional[float] = 20.0,
+        enable_search_by_identifier: bool = True,
+        enable_search_by_brand: bool = False,
+        all: bool = False,
         async_tools: bool = False,
-        brand: bool = True,
-        search: bool = False,
         **kwargs,
     ):
         self.api_key = api_key or getenv("BRANDFETCH_API_KEY")
         self.client_id = client_id or getenv("BRANDFETCH_CLIENT_ID")
         self.base_url = base_url
         self.timeout = httpx.Timeout(timeout)
-        self.async_tools = async_tools
         self.search_url = f"{self.base_url}/search"
         self.brand_url = f"{self.base_url}/brands"
 
         tools: list[Any] = []
-        if self.async_tools:
-            if brand:
+        # Backward-compat mapping: prefer new enable_* flags, but honor legacy toggles
+        if async_tools:
+            if all or enable_search_by_identifier:
                 tools.append(self.asearch_by_identifier)
-            if search:
+            if all or enable_search_by_brand:
                 tools.append(self.asearch_by_brand)
         else:
-            if brand:
+            if all or enable_search_by_identifier:
                 tools.append(self.search_by_identifier)
-            if search:
+            if all or enable_search_by_brand:
                 tools.append(self.search_by_brand)
         name = kwargs.pop("name", "brandfetch_tools")
         super().__init__(name=name, tools=tools, **kwargs)

agno/tools/bravesearch.py

@@ -26,6 +26,8 @@ class BraveSearchTools(Toolkit):
         api_key: Optional[str] = None,
         fixed_max_results: Optional[int] = None,
         fixed_language: Optional[str] = None,
+        enable_brave_search: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.api_key = api_key or getenv("BRAVE_API_KEY")
@@ -38,7 +40,8 @@ class BraveSearchTools(Toolkit):
         self.brave_client = Brave(api_key=self.api_key)
 
         tools = []
-        tools.append(self.brave_search)
+        if all or enable_brave_search:
+            tools.append(self.brave_search)
 
         super().__init__(
             name="brave_search",

agno/tools/brightdata.py

@@ -5,8 +5,9 @@ from typing import Any, Dict, List, Optional
 from uuid import uuid4
 
 from agno.agent import Agent
-from agno.media import ImageArtifact
+from agno.media import Image
 from agno.tools import Toolkit
+from agno.tools.function import ToolResult
 from agno.utils.log import log_debug, log_error, log_info
 
 try:
@@ -16,15 +17,32 @@ except ImportError:
 
 
 class BrightDataTools(Toolkit):
+    """
+    BrightData is a toolkit for web scraping, screenshots, search engines, and web data feeds.
+
+    Args:
+        api_key (Optional[str]): Bright Data API key. Retrieved from BRIGHT_DATA_API_KEY env variable if not provided.
+        enable_scrape_markdown (bool): Enable webpage scraping as Markdown. Default is True.
+        enable_screenshot (bool): Enable website screenshot capture. Default is True.
+        enable_search_engine (bool): Enable search engine functionality. Default is True.
+        enable_web_data_feed (bool): Enable web data feed retrieval. Default is True.
+        all (bool): Enable all tools. Overrides individual flags when True. Default is False.
+        serp_zone (str): SERP zone for search operations. Default is "serp_api".
+        web_unlocker_zone (str): Web unlocker zone for scraping operations. Default is "web_unlocker1".
+        verbose (bool): Enable verbose logging. Default is False.
+        timeout (int): Timeout in seconds for operations. Default is 600.
+    """
+
     def __init__(
         self,
         api_key: Optional[str] = None,
+        enable_scrape_markdown: bool = True,
+        enable_screenshot: bool = True,
+        enable_search_engine: bool = True,
+        enable_web_data_feed: bool = True,
+        all: bool = False,
         serp_zone: str = "serp_api",
         web_unlocker_zone: str = "web_unlocker1",
-        scrape_as_markdown: bool = True,
-        get_screenshot: bool = False,
-        search_engine: bool = True,
-        web_data_feed: bool = True,
         verbose: bool = False,
         timeout: int = 600,
         **kwargs,
@@ -47,14 +65,13 @@ class BrightDataTools(Toolkit):
         self.timeout = timeout
 
         tools: List[Any] = []
-
-        if scrape_as_markdown:
+        if all or enable_scrape_markdown:
             tools.append(self.scrape_as_markdown)
-        if get_screenshot:
+        if all or enable_screenshot:
             tools.append(self.get_screenshot)
-        if search_engine:
+        if all or enable_search_engine:
             tools.append(self.search_engine)
-        if web_data_feed:
+        if all or enable_web_data_feed:
             tools.append(self.web_data_feed)
 
         super().__init__(name="brightdata_tools", tools=tools, **kwargs)
@@ -104,21 +121,22 @@ class BrightDataTools(Toolkit):
         except Exception as e:
             return f"Error scraping URL {url}: {e}"
 
-    def get_screenshot(self, agent: Agent, url: str, output_path: str = "screenshot.png") -> str:
+    def get_screenshot(self, agent: Agent, url: str, output_path: str = "screenshot.png") -> ToolResult:
         """
         Capture a screenshot of a webpage
 
         Args:
             url (str): URL to screenshot
+            output_path (str): Output path for the screenshot (not used, kept for compatibility)
 
         Returns:
-            str: A message indicating success (including media ID) or failure.
+            ToolResult: Contains the screenshot image or error message.
         """
         try:
             if not self.api_key:
-                return "Please provide a Bright Data API key"
+                return ToolResult(content="Please provide a Bright Data API key")
             if not url:
-                return "Please provide a URL to screenshot"
+                return ToolResult(content="Please provide a URL to screenshot")
 
             log_info(f"Taking screenshot of: {url}")
 
@@ -140,18 +158,20 @@ class BrightDataTools(Toolkit):
 
             media_id = str(uuid4())
 
-            agent.add_image(
-                ImageArtifact(
-                    id=media_id,
-                    content=base64_encoded_image.encode("utf-8"),
-                    mime_type="image/png",
-                    original_prompt=f"Screenshot of {url}",
-                )
+            # Create Image for the screenshot
+            image_artifact = Image(
+                id=media_id,
+                content=base64_encoded_image.encode("utf-8"),
+                mime_type="image/png",
+                original_prompt=f"Screenshot of {url}",
             )
+
             log_debug(f"Screenshot captured and added as artifact with ID: {media_id}")
-            return f"Screenshot captured and added as artifact with ID: {media_id}"
+            return ToolResult(
+                content=f"Screenshot captured and added as artifact with ID: {media_id}", images=[image_artifact]
+            )
         except Exception as e:
-            return f"Error taking screenshot of {url}: {e}"
+            return ToolResult(content=f"Error taking screenshot of {url}: {e}")
 
     def search_engine(
         self,

agno/tools/browserbase.py

@@ -24,6 +24,11 @@ class BrowserbaseTools(Toolkit):
         api_key: Optional[str] = None,
         project_id: Optional[str] = None,
         base_url: Optional[str] = None,
+        enable_navigate_to: bool = True,
+        enable_screenshot: bool = True,
+        enable_get_page_content: bool = True,
+        enable_close_session: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         """Initialize BrowserbaseTools.
@@ -61,10 +66,14 @@ class BrowserbaseTools(Toolkit):
         self._connect_url = None
 
         tools: List[Any] = []
-        tools.append(self.navigate_to)
-        tools.append(self.screenshot)
-        tools.append(self.get_page_content)
-        tools.append(self.close_session)
+        if all or enable_navigate_to:
+            tools.append(self.navigate_to)
+        if all or enable_screenshot:
+            tools.append(self.screenshot)
+        if all or enable_get_page_content:
+            tools.append(self.get_page_content)
+        if all or enable_close_session:
+            tools.append(self.close_session)
 
         super().__init__(name="browserbase_tools", tools=tools, **kwargs)
 

agno/tools/calcom.py

@@ -18,11 +18,13 @@ class CalComTools(Toolkit):
         api_key: Optional[str] = None,
         event_type_id: Optional[int] = None,
         user_timezone: Optional[str] = None,
-        get_available_slots: bool = True,
-        create_booking: bool = True,
-        get_upcoming_bookings: bool = True,
-        reschedule_booking: bool = True,
-        cancel_booking: bool = True,
+        # Enable flags for <6 functions
+        enable_get_available_slots: bool = True,
+        enable_create_booking: bool = True,
+        enable_get_upcoming_bookings: bool = True,
+        enable_reschedule_booking: bool = True,
+        enable_cancel_booking: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         """Initialize the Cal.com toolkit.
@@ -49,15 +51,15 @@ class CalComTools(Toolkit):
         self.user_timezone = user_timezone or "America/New_York"
 
         tools: List[Any] = []
-        if get_available_slots:
+        if all or enable_get_available_slots:
             tools.append(self.get_available_slots)
-        if create_booking:
+        if all or enable_create_booking:
             tools.append(self.create_booking)
-        if get_upcoming_bookings:
+        if all or enable_get_upcoming_bookings:
             tools.append(self.get_upcoming_bookings)
-        if reschedule_booking:
+        if all or enable_reschedule_booking:
             tools.append(self.reschedule_booking)
-        if cancel_booking:
+        if all or enable_cancel_booking:
             tools.append(self.cancel_booking)
 
         super().__init__(name="calcom", tools=tools, **kwargs)

agno/tools/calculator.py

@@ -9,35 +9,18 @@ from agno.utils.log import log_debug, logger
 class CalculatorTools(Toolkit):
     def __init__(
         self,
-        add: bool = True,
-        subtract: bool = True,
-        multiply: bool = True,
-        divide: bool = True,
-        exponentiate: bool = False,
-        factorial: bool = False,
-        is_prime: bool = False,
-        square_root: bool = False,
-        enable_all: bool = False,
         **kwargs,
     ):
-        # Build the include_tools list based on enabled functions
-        tools: List[Callable] = []
-        if add or enable_all:
-            tools.append(self.add)
-        if subtract or enable_all:
-            tools.append(self.subtract)
-        if multiply or enable_all:
-            tools.append(self.multiply)
-        if divide or enable_all:
-            tools.append(self.divide)
-        if exponentiate or enable_all:
-            tools.append(self.exponentiate)
-        if factorial or enable_all:
-            tools.append(self.factorial)
-        if is_prime or enable_all:
-            tools.append(self.is_prime)
-        if square_root or enable_all:
-            tools.append(self.square_root)
+        tools: List[Callable] = [
+            self.add,
+            self.subtract,
+            self.multiply,
+            self.divide,
+            self.exponentiate,
+            self.factorial,
+            self.is_prime,
+            self.square_root,
+        ]
 
         # Initialize the toolkit with auto-registration enabled
         super().__init__(name="calculator", tools=tools, **kwargs)