agno 1.8.0__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

agno/tools/slack.py

@@ -1,8 +1,8 @@
 import json
-import os
+from os import getenv
 from typing import Any, Dict, List, Optional
 
-from agno.tools.toolkit import Toolkit
+from agno.tools import Toolkit
 from agno.utils.log import logger
 
 try:
@@ -16,25 +16,26 @@ class SlackTools(Toolkit):
     def __init__(
         self,
         token: Optional[str] = None,
-        send_message: bool = True,
-        send_message_thread: bool = True,
-        list_channels: bool = True,
-        get_channel_history: bool = True,
+        enable_send_message: bool = True,
+        enable_send_message_thread: bool = True,
+        enable_list_channels: bool = True,
+        enable_get_channel_history: bool = True,
+        all: bool = False,
         **kwargs,
     ):
-        self.token: Optional[str] = token or os.getenv("SLACK_TOKEN")
+        self.token: Optional[str] = token or getenv("SLACK_TOKEN")
         if self.token is None or self.token == "":
             raise ValueError("SLACK_TOKEN is not set")
         self.client = WebClient(token=self.token)
 
         tools: List[Any] = []
-        if send_message:
+        if enable_send_message or all:
             tools.append(self.send_message)
-        if send_message_thread:
+        if enable_send_message_thread or all:
             tools.append(self.send_message_thread)
-        if list_channels:
+        if enable_list_channels or all:
             tools.append(self.list_channels)
-        if get_channel_history:
+        if enable_get_channel_history or all:
             tools.append(self.get_channel_history)
 
         super().__init__(name="slack", tools=tools, **kwargs)

agno/tools/sleep.py

@@ -5,9 +5,10 @@ from agno.utils.log import log_info
 
 
 class SleepTools(Toolkit):
-    def __init__(self, **kwargs):
+    def __init__(self, enable_sleep: bool = True, all: bool = False, **kwargs):
         tools = []
-        tools.append(self.sleep)
+        if all or enable_sleep:
+            tools.append(self.sleep)
 
         super().__init__(name="sleep", tools=tools, **kwargs)
 

agno/tools/spider.py

@@ -7,16 +7,33 @@ except ImportError:
 
 from typing import Any, List, Optional
 
-from agno.tools.toolkit import Toolkit
+from agno.tools import Toolkit
 from agno.utils.log import log_info, logger
 
 
 class SpiderTools(Toolkit):
+    """
+    Spider is a toolkit for web searching, scraping, and crawling.
+
+    Args:
+        enable_search (bool): Enable web search functionality. Default is True.
+        enable_scrape (bool): Enable web scraping functionality. Default is True.
+        enable_crawl (bool): Enable web crawling functionality. Default is True.
+        all (bool): Enable all tools. Overrides individual flags when True. Default is False.
+        max_results (Optional[int]): Default maximum number of results.
+        url (Optional[str]): Default URL for operations.
+        optional_params (Optional[dict]): Additional parameters for operations.
+    """
+
     def __init__(
         self,
         max_results: Optional[int] = None,
         url: Optional[str] = None,
         optional_params: Optional[dict] = None,
+        enable_search: bool = True,
+        enable_scrape: bool = True,
+        enable_crawl: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.max_results = max_results
@@ -24,9 +41,12 @@ class SpiderTools(Toolkit):
         self.optional_params = optional_params or {}
 
         tools: List[Any] = []
-        tools.append(self.search)
-        tools.append(self.scrape)
-        tools.append(self.crawl)
+        if enable_search or all:
+            tools.append(self.search)
+        if enable_scrape or all:
+            tools.append(self.scrape)
+        if enable_crawl or all:
+            tools.append(self.crawl)
 
         super().__init__(name="spider", tools=tools, **kwargs)
 

agno/tools/sql.py

@@ -25,9 +25,10 @@ class SQLTools(Toolkit):
         schema: Optional[str] = None,
         dialect: Optional[str] = None,
         tables: Optional[Dict[str, Any]] = None,
-        list_tables: bool = True,
-        describe_table: bool = True,
-        run_sql_query: bool = True,
+        enable_list_tables: bool = True,
+        enable_describe_table: bool = True,
+        enable_run_sql_query: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         # Get the database engine
@@ -53,11 +54,11 @@ class SQLTools(Toolkit):
         self.tables: Optional[Dict[str, Any]] = tables
 
         tools: List[Any] = []
-        if list_tables:
+        if enable_list_tables or all:
             tools.append(self.list_tables)
-        if describe_table:
+        if enable_describe_table or all:
             tools.append(self.describe_table)
-        if run_sql_query:
+        if enable_run_sql_query or all:
             tools.append(self.run_sql_query)
 
         super().__init__(name="sql_tools", tools=tools, **kwargs)

agno/tools/tavily.py

@@ -15,12 +15,13 @@ class TavilyTools(Toolkit):
     def __init__(
         self,
         api_key: Optional[str] = None,
-        search: bool = True,
+        enable_search: bool = True,
+        enable_search_context: bool = False,
+        all: bool = False,
         max_tokens: int = 6000,
         include_answer: bool = True,
         search_depth: Literal["basic", "advanced"] = "advanced",
         format: Literal["json", "markdown"] = "markdown",
-        use_search_context: bool = False,
         **kwargs,
     ):
         self.api_key = api_key or getenv("TAVILY_API_KEY")
@@ -34,8 +35,9 @@ class TavilyTools(Toolkit):
         self.format: Literal["json", "markdown"] = format
 
         tools: List[Any] = []
-        if search:
-            if use_search_context:
+
+        if enable_search or all:
+            if enable_search_context:
                 tools.append(self.web_search_with_tavily)
             else:
                 tools.append(self.web_search_using_tavily)

agno/tools/telegram.py

@@ -1,4 +1,4 @@
-import os
+from os import getenv
 from typing import Any, List, Optional, Union
 
 import httpx
@@ -10,15 +10,23 @@ from agno.utils.log import log_debug, logger
 class TelegramTools(Toolkit):
     base_url = "https://api.telegram.org"
 
-    def __init__(self, chat_id: Union[str, int], token: Optional[str] = None, **kwargs):
-        self.token = token or os.getenv("TELEGRAM_TOKEN")
+    def __init__(
+        self,
+        chat_id: Union[str, int],
+        token: Optional[str] = None,
+        enable_send_message: bool = True,
+        all: bool = False,
+        **kwargs,
+    ):
+        self.token = token or getenv("TELEGRAM_TOKEN")
         if not self.token:
             logger.error("TELEGRAM_TOKEN not set. Please set the TELEGRAM_TOKEN environment variable.")
 
         self.chat_id = chat_id
 
         tools: List[Any] = []
-        tools.append(self.send_message)
+        if all or enable_send_message:
+            tools.append(self.send_message)
 
         super().__init__(name="telegram", tools=tools, **kwargs)
 

agno/tools/todoist.py

@@ -1,5 +1,5 @@
 import json
-import os
+from os import getenv
 from typing import Any, Dict, List, Optional
 
 from agno.tools import Toolkit
@@ -17,48 +17,28 @@ class TodoistTools(Toolkit):
     def __init__(
         self,
         api_token: Optional[str] = None,
-        create_task: bool = True,
-        get_task: bool = True,
-        update_task: bool = True,
-        close_task: bool = True,
-        delete_task: bool = True,
-        get_active_tasks: bool = True,
-        get_projects: bool = True,
         **kwargs,
     ):
         """Initialize the Todoist toolkit.
 
         Args:
             api_token: Optional Todoist API token. If not provided, will look for TODOIST_API_TOKEN env var
-            create_task: Whether to register the create_task function
-            get_task: Whether to register the get_task function
-            update_task: Whether to register the update_task function
-            close_task: Whether to register the close_task function
-            delete_task: Whether to register the delete_task function
-            get_active_tasks: Whether to register the get_active_tasks function
-            get_projects: Whether to register the get_projects function
         """
-        self.api_token = api_token or os.getenv("TODOIST_API_TOKEN")
+        self.api_token = api_token or getenv("TODOIST_API_TOKEN")
         if not self.api_token:
             raise ValueError("TODOIST_API_TOKEN not set. Please set the TODOIST_API_TOKEN environment variable.")
 
         self.api = TodoistAPI(self.api_token)
 
-        tools: List[Any] = []
-        if create_task:
-            tools.append(self.create_task)
-        if get_task:
-            tools.append(self.get_task)
-        if update_task:
-            tools.append(self.update_task)
-        if close_task:
-            tools.append(self.close_task)
-        if delete_task:
-            tools.append(self.delete_task)
-        if get_active_tasks:
-            tools.append(self.get_active_tasks)
-        if get_projects:
-            tools.append(self.get_projects)
+        tools: List[Any] = [
+            self.create_task,
+            self.get_task,
+            self.update_task,
+            self.close_task,
+            self.delete_task,
+            self.get_active_tasks,
+            self.get_projects,
+        ]
 
         super().__init__(name="todoist", tools=tools, **kwargs)
 

agno/tools/toolkit.py

@@ -131,7 +131,7 @@ class Toolkit:
                 requires_confirmation=tool_name in self.requires_confirmation_tools,
                 external_execution=tool_name in self.external_execution_required_tools,
                 stop_after_tool_call=tool_name in self.stop_after_tool_call_tools,
-                show_result=tool_name in self.show_result_tools,
+                show_result=tool_name in self.show_result_tools or tool_name in self.stop_after_tool_call_tools,
             )
             self.functions[f.name] = f
             log_debug(f"Function: {f.name} registered with {self.name}")

agno/tools/trafilatura.py

@@ -63,6 +63,13 @@ class TrafilaturaTools(Toolkit):
         max_tree_size: Optional[int] = None,
         max_crawl_urls: int = 10,
         max_known_urls: int = 100000,
+        # Tool enable flags for <6 functions
+        enable_extract_text: bool = True,
+        enable_extract_metadata_only: bool = True,
+        enable_html_to_text: bool = True,
+        enable_extract_batch: bool = True,
+        enable_crawl_website: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.output_format = output_format
@@ -80,12 +87,21 @@ class TrafilaturaTools(Toolkit):
         self.max_crawl_urls = max_crawl_urls
         self.max_known_urls = max_known_urls
 
-        tools: List[Callable] = [self.extract_text, self.extract_metadata_only, self.html_to_text, self.extract_batch]
-
-        if not SPIDER_AVAILABLE:
-            logger.warning("Web crawling requested but spider module not available. Skipping crawler tool.")
-        else:
-            tools.append(self.crawl_website)
+        tools: List[Callable] = []
+        if all or enable_extract_text:
+            tools.append(self.extract_text)
+        if all or enable_extract_metadata_only:
+            tools.append(self.extract_metadata_only)
+        if all or enable_html_to_text:
+            tools.append(self.html_to_text)
+        if all or enable_extract_batch:
+            tools.append(self.extract_batch)
+
+        if all or enable_crawl_website:
+            if not SPIDER_AVAILABLE:
+                logger.warning("Web crawling requested but spider module not available. Skipping crawler tool.")
+            else:
+                tools.append(self.crawl_website)
 
         super().__init__(name="trafilatura_tools", tools=tools, **kwargs)
 

agno/tools/trello.py

@@ -17,13 +17,6 @@ class TrelloTools(Toolkit):
         api_key: Optional[str] = None,
         api_secret: Optional[str] = None,
         token: Optional[str] = None,
-        create_card: bool = True,
-        get_board_lists: bool = True,
-        move_card: bool = True,
-        get_cards: bool = True,
-        create_board: bool = True,
-        create_list: bool = True,
-        list_boards: bool = True,
         **kwargs,
     ):
         self.api_key = api_key or getenv("TRELLO_API_KEY")
@@ -39,21 +32,15 @@ class TrelloTools(Toolkit):
             logger.error(f"Error initializing Trello client: {e}")
             self.client = None
 
-        tools: List[Any] = []
-        if create_card:
-            tools.append(self.create_card)
-        if get_board_lists:
-            tools.append(self.get_board_lists)
-        if move_card:
-            tools.append(self.move_card)
-        if get_cards:
-            tools.append(self.get_cards)
-        if create_board:
-            tools.append(self.create_board)
-        if create_list:
-            tools.append(self.create_list)
-        if list_boards:
-            tools.append(self.list_boards)
+        tools: List[Any] = [
+            self.create_card,
+            self.get_board_lists,
+            self.move_card,
+            self.get_cards,
+            self.create_board,
+            self.create_list,
+            self.list_boards,
+        ]
 
         super().__init__(name="trello", tools=tools, **kwargs)
 

agno/tools/twilio.py

@@ -22,6 +22,10 @@ class TwilioTools(Toolkit):
         region: Optional[str] = None,
         edge: Optional[str] = None,
         debug: bool = False,
+        enable_send_sms: bool = True,
+        enable_get_call_details: bool = True,
+        enable_list_messages: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         """Initialize the Twilio toolkit.
@@ -84,9 +88,12 @@ class TwilioTools(Toolkit):
             self.client.http_client.logger.setLevel(logging.INFO)
 
         tools: List[Any] = []
-        tools.append(self.send_sms)
-        tools.append(self.get_call_details)
-        tools.append(self.list_messages)
+        if all or enable_send_sms:
+            tools.append(self.send_sms)
+        if all or enable_get_call_details:
+            tools.append(self.get_call_details)
+        if all or enable_list_messages:
+            tools.append(self.list_messages)
 
         super().__init__(name="twilio", tools=tools, **kwargs)
 

agno/tools/user_control_flow.py

@@ -9,6 +9,8 @@ class UserControlFlowTools(Toolkit):
         self,
         instructions: Optional[str] = None,
         add_instructions: bool = True,
+        enable_get_user_input: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         """A toolkit that provides the ability for the agent to interrupt the agent run and interact with the user."""
@@ -17,7 +19,10 @@ class UserControlFlowTools(Toolkit):
             self.instructions = self.DEFAULT_INSTRUCTIONS
         else:
             self.instructions = instructions
-        tools = [self.get_user_input]
+
+        tools = []
+        if all or enable_get_user_input:
+            tools.append(self.get_user_input)
 
         super().__init__(
             name="user_control_flow_tools",

agno/tools/valyu.py

@@ -12,9 +12,34 @@ except ImportError:
 
 
 class ValyuTools(Toolkit):
+    """
+    Valyu is a toolkit for academic and web search capabilities.
+
+    Args:
+        api_key (Optional[str]): Valyu API key. Retrieved from VALYU_API_KEY env variable if not provided.
+        enable_academic_search (bool): Enable academic sources search functionality. Default is True.
+        enable_web_search (bool): Enable web search functionality. Default is True.
+        enable_paper_search (bool): Enable search within paper functionality. Default is True.
+        all (bool): Enable all tools. Overrides individual flags when True. Default is False.
+        text_length (int): Maximum length of text content per result. Default is 1000.
+        max_results (int): Maximum number of results to return. Default is 10.
+        relevance_threshold (float): Minimum relevance score for results. Default is 0.5.
+        content_category (Optional[str]): Content category for filtering.
+        search_start_date (Optional[str]): Start date for search filtering (YYYY-MM-DD).
+        search_end_date (Optional[str]): End date for search filtering (YYYY-MM-DD).
+        search_domains (Optional[List[str]]): List of domains to search within.
+        sources (Optional[List[str]]): List of specific sources to search.
+        max_price (float): Maximum price for API calls. Default is 30.0.
+        tool_call_mode (bool): Enable tool call mode. Default is False.
+    """
+
     def __init__(
         self,
         api_key: Optional[str] = None,
+        enable_academic_search: bool = True,
+        enable_web_search: bool = True,
+        enable_paper_search: bool = True,
+        all: bool = False,
         text_length: int = 1000,
         max_results: int = 10,
         relevance_threshold: float = 0.5,
@@ -43,11 +68,15 @@ class ValyuTools(Toolkit):
         self.sources = sources
         self.tool_call_mode = tool_call_mode
 
-        super().__init__(
-            name="valyu_search",
-            tools=[self.search_academic_sources, self.search_web, self.search_within_paper],
-            **kwargs,
-        )
+        tools: List[Any] = []
+        if all or enable_academic_search:
+            tools.append(self.search_academic_sources)
+        if all or enable_web_search:
+            tools.append(self.search_web)
+        if all or enable_paper_search:
+            tools.append(self.search_within_paper)
+
+        super().__init__(name="valyu_search", tools=tools, **kwargs)
 
     def _parse_results(self, results: List[Any]) -> str:
         parsed_results = []

agno/tools/visualization.py

@@ -9,29 +9,21 @@ from agno.utils.log import log_info, logger
 class VisualizationTools(Toolkit):
     def __init__(
         self,
-        bar_chart: bool = True,
-        line_chart: bool = True,
-        pie_chart: bool = True,
-        scatter_plot: bool = True,
-        histogram: bool = True,
-        enable_all: bool = False,
         output_dir: str = "charts",
+        enable_create_bar_chart: bool = True,
+        enable_create_line_chart: bool = True,
+        enable_create_pie_chart: bool = True,
+        enable_create_scatter_plot: bool = True,
+        enable_create_histogram: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         """
         Initialize the VisualizationTools toolkit.
 
         Args:
-            bar_chart (bool): Enable bar chart creation. Default is True.
-            line_chart (bool): Enable line chart creation. Default is True.
-            pie_chart (bool): Enable pie chart creation. Default is True.
-            scatter_plot (bool): Enable scatter plot creation. Default is True.
-            histogram (bool): Enable histogram creation. Default is True.
-            enable_all (bool): Enable all chart types. Default is False.
             output_dir (str): Directory to save charts. Default is "charts".
         """
-        super().__init__(**kwargs)
-
         # Check if matplotlib is available
         try:
             import matplotlib
@@ -47,20 +39,19 @@ class VisualizationTools(Toolkit):
 
         self.output_dir = output_dir
 
-        # Register functions based on enabled chart types
-        if enable_all:
-            bar_chart = line_chart = pie_chart = scatter_plot = histogram = True
-
-        if bar_chart:
-            self.register(self.create_bar_chart)
-        if line_chart:
-            self.register(self.create_line_chart)
-        if pie_chart:
-            self.register(self.create_pie_chart)
-        if scatter_plot:
-            self.register(self.create_scatter_plot)
-        if histogram:
-            self.register(self.create_histogram)
+        tools: List[Any] = []
+        if enable_create_bar_chart or all:
+            tools.append(self.create_bar_chart)
+        if enable_create_line_chart or all:
+            tools.append(self.create_line_chart)
+        if enable_create_pie_chart or all:
+            tools.append(self.create_pie_chart)
+        if enable_create_scatter_plot or all:
+            tools.append(self.create_scatter_plot)
+        if enable_create_histogram or all:
+            tools.append(self.create_histogram)
+
+        super().__init__(name="visualization_tools", tools=tools, **kwargs)
 
     def _normalize_data_for_charts(
         self, data: Union[Dict[str, Any], List[Dict[str, Any]], List[Any], str]

agno/tools/webbrowser.py

@@ -7,11 +7,12 @@ from agno.tools import Toolkit
 class WebBrowserTools(Toolkit):
     """Tools for opening a page on the web browser"""
 
-    def __init__(self):
+    def __init__(self, enable_open_page: bool = True, all: bool = False, **kwargs):
         tools: List[Any] = []
-        tools.append(self.open_page)
+        if all or enable_open_page:
+            tools.append(self.open_page)
 
-        super().__init__(name="webbrowser_tools", tools=tools)
+        super().__init__(name="webbrowser_tools", tools=tools, **kwargs)
 
     def open_page(self, url: str, new_window: bool = False):
         """Open a URL in a browser window

agno/tools/webex.py

@@ -1,8 +1,8 @@
 import json
-import os
+from os import getenv
 from typing import Any, List, Optional
 
-from agno.tools.toolkit import Toolkit
+from agno.tools import Toolkit
 from agno.utils.log import logger
 
 try:
@@ -14,19 +14,23 @@ except ImportError:
 
 class WebexTools(Toolkit):
     def __init__(
-        self, send_message: bool = True, list_rooms: bool = True, access_token: Optional[str] = None, **kwargs
+        self,
+        enable_send_message: bool = True,
+        enable_list_rooms: bool = True,
+        all: bool = False,
+        access_token: Optional[str] = None,
+        **kwargs,
     ):
-        if access_token is None:
-            access_token = os.getenv("WEBEX_ACCESS_TOKEN")
+        access_token = access_token or getenv("WEBEX_ACCESS_TOKEN")
         if access_token is None:
             raise ValueError("Webex access token is not set. Please set the WEBEX_ACCESS_TOKEN environment variable.")
 
         self.client = WebexAPI(access_token=access_token)
 
         tools: List[Any] = []
-        if send_message:
+        if all or enable_send_message:
             tools.append(self.send_message)
-        if list_rooms:
+        if all or enable_list_rooms:
             tools.append(self.list_rooms)
 
         super().__init__(name="webex", tools=tools, **kwargs)