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

agno/tools/github.py

@@ -7,7 +7,6 @@ from agno.utils.log import log_debug, logger
 
 try:
     from github import Auth, Github, GithubException
-    from github.GithubObject import NotSet
 
 except ImportError:
     raise ImportError("`PyGithub` not installed. Please install using `pip install pygithub`")
@@ -18,45 +17,6 @@ class GithubTools(Toolkit):
         self,
         access_token: Optional[str] = None,
         base_url: Optional[str] = None,
-        search_repositories: bool = True,
-        list_repositories: bool = False,
-        get_repository: bool = False,
-        get_pull_request: bool = False,
-        get_pull_request_changes: bool = False,
-        create_issue: bool = False,
-        create_repository: bool = False,
-        delete_repository: bool = False,
-        get_repository_languages: bool = False,
-        list_branches: bool = False,
-        get_pull_request_count: bool = False,
-        get_repository_stars: bool = False,
-        get_pull_requests: bool = False,
-        get_pull_request_comments: bool = False,
-        create_pull_request_comment: bool = False,
-        edit_pull_request_comment: bool = False,
-        get_pull_request_with_details: bool = False,
-        get_repository_with_stats: bool = False,
-        list_issues: bool = False,
-        get_issue: bool = False,
-        comment_on_issue: bool = False,
-        close_issue: bool = False,
-        reopen_issue: bool = False,
-        assign_issue: bool = False,
-        label_issue: bool = False,
-        list_issue_comments: bool = False,
-        edit_issue: bool = False,
-        create_pull_request: bool = False,
-        create_file: bool = False,
-        get_file_content: bool = False,
-        update_file: bool = True,
-        delete_file: bool = False,
-        get_directory_content: bool = False,
-        get_branch_content: bool = False,
-        create_branch: bool = False,
-        set_default_branch: bool = False,
-        search_code: bool = False,
-        search_issues_and_prs: bool = False,
-        create_review_request: bool = False,
         **kwargs,
     ):
         self.access_token = access_token or getenv("GITHUB_ACCESS_TOKEN")
@@ -64,85 +24,47 @@ class GithubTools(Toolkit):
 
         self.g = self.authenticate()
 
-        tools: List[Any] = []
-        if search_repositories:
-            tools.append(self.search_repositories)
-        if list_repositories:
-            tools.append(self.list_repositories)
-        if get_repository:
-            tools.append(self.get_repository)
-        if get_pull_request:
-            tools.append(self.get_pull_request)
-        if get_pull_request_changes:
-            tools.append(self.get_pull_request_changes)
-        if create_issue:
-            tools.append(self.create_issue)
-        if create_repository:
-            tools.append(self.create_repository)
-        if delete_repository:
-            tools.append(self.delete_repository)
-        if list_branches:
-            tools.append(self.list_branches)
-        if get_repository_languages:
-            tools.append(self.get_repository_languages)
-        if get_pull_request_count:
-            tools.append(self.get_pull_request_count)
-        if get_repository_stars:
-            tools.append(self.get_repository_stars)
-        if get_pull_requests:
-            tools.append(self.get_pull_requests)
-        if get_pull_request_comments:
-            tools.append(self.get_pull_request_comments)
-        if create_pull_request_comment:
-            tools.append(self.create_pull_request_comment)
-        if edit_pull_request_comment:
-            tools.append(self.edit_pull_request_comment)
-        if get_pull_request_with_details:
-            tools.append(self.get_pull_request_with_details)
-        if get_repository_with_stats:
-            tools.append(self.get_repository_with_stats)
-        if list_issues:
-            tools.append(self.list_issues)
-        if get_issue:
-            tools.append(self.get_issue)
-        if comment_on_issue:
-            tools.append(self.comment_on_issue)
-        if close_issue:
-            tools.append(self.close_issue)
-        if reopen_issue:
-            tools.append(self.reopen_issue)
-        if assign_issue:
-            tools.append(self.assign_issue)
-        if label_issue:
-            tools.append(self.label_issue)
-        if list_issue_comments:
-            tools.append(self.list_issue_comments)
-        if edit_issue:
-            tools.append(self.edit_issue)
-        if create_pull_request:
-            tools.append(self.create_pull_request)
-        if create_file:
-            tools.append(self.create_file)
-        if get_file_content:
-            tools.append(self.get_file_content)
-        if update_file:
-            tools.append(self.update_file)
-        if delete_file:
-            tools.append(self.delete_file)
-        if get_directory_content:
-            tools.append(self.get_directory_content)
-        if get_branch_content:
-            tools.append(self.get_branch_content)
-        if create_branch:
-            tools.append(self.create_branch)
-        if set_default_branch:
-            tools.append(self.set_default_branch)
-        if search_code:
-            tools.append(self.search_code)
-        if search_issues_and_prs:
-            tools.append(self.search_issues_and_prs)
-        if create_review_request:
-            tools.append(self.create_review_request)
+        tools: List[Any] = [
+            self.search_repositories,
+            self.list_repositories,
+            self.get_repository,
+            self.get_pull_request,
+            self.get_pull_request_changes,
+            self.create_issue,
+            self.create_repository,
+            self.delete_repository,
+            self.list_branches,
+            self.get_repository_languages,
+            self.get_pull_request_count,
+            self.get_repository_stars,
+            self.get_pull_requests,
+            self.get_pull_request_comments,
+            self.create_pull_request_comment,
+            self.edit_pull_request_comment,
+            self.get_pull_request_with_details,
+            self.get_repository_with_stats,
+            self.list_issues,
+            self.get_issue,
+            self.comment_on_issue,
+            self.close_issue,
+            self.reopen_issue,
+            self.assign_issue,
+            self.label_issue,
+            self.list_issue_comments,
+            self.edit_issue,
+            self.create_pull_request,
+            self.create_file,
+            self.get_file_content,
+            self.update_file,
+            self.delete_file,
+            self.get_directory_content,
+            self.get_branch_content,
+            self.create_branch,
+            self.set_default_branch,
+            self.search_code,
+            self.search_issues_and_prs,
+            self.create_review_request,
+        ]
 
         super().__init__(name="github", tools=tools, **kwargs)
 
@@ -428,7 +350,7 @@ class GithubTools(Toolkit):
             logger.error(f"Error getting pull request changes: {e}")
             return json.dumps({"error": str(e)})
 
-    def create_issue(self, repo_name: str, title: str, body: Optional[str] = NotSet) -> str:
+    def create_issue(self, repo_name: str, title: str, body: Optional[str] = None) -> str:
         """Create an issue in a repository.
 
         Args:
@@ -442,7 +364,7 @@ class GithubTools(Toolkit):
         log_debug(f"Creating issue in repository: {repo_name}")
         try:
             repo = self.g.get_repo(repo_name)
-            issue = repo.create_issue(title=title, body=body)
+            issue = repo.create_issue(title=title, body=body)  # type: ignore
             issue_info = {
                 "id": issue.id,
                 "number": issue.number,
@@ -697,8 +619,8 @@ class GithubTools(Toolkit):
         self,
         repo_name: str,
         issue_number: int,
-        title: Optional[str] = NotSet,
-        body: Optional[str] = NotSet,
+        title: Optional[str] = None,
+        body: Optional[str] = None,
     ) -> str:
         """Edit the title or body of an issue.
 
@@ -715,7 +637,7 @@ class GithubTools(Toolkit):
         try:
             repo = self.g.get_repo(repo_name)
             issue = repo.get_issue(number=issue_number)
-            issue.edit(title=title, body=body)
+            issue.edit(title=title, body=body)  # type: ignore
             return json.dumps({"message": f"Issue #{issue_number} updated."}, indent=2)
         except GithubException as e:
             logger.error(f"Error editing issue: {e}")
@@ -1324,7 +1246,7 @@ class GithubTools(Toolkit):
         path: str,
         content: str,
         message: str,
-        branch: Optional[str] = NotSet,
+        branch: Optional[str] = None,
     ) -> str:
         """Create a new file in a repository.
 
@@ -1350,7 +1272,7 @@ class GithubTools(Toolkit):
 
             # Extract relevant information
             file_info = {
-                "path": result["content"].path,
+                "path": result["content"].path,  # type: ignore
                 "sha": result["content"].sha,
                 "url": result["content"].html_url,
                 "commit": {

agno/tools/gmail.py

@@ -101,17 +101,6 @@ class GmailTools(Toolkit):
 
     def __init__(
         self,
-        get_latest_emails: bool = True,
-        get_emails_from_user: bool = True,
-        get_unread_emails: bool = True,
-        get_starred_emails: bool = True,
-        get_emails_by_context: bool = True,
-        get_emails_by_date: bool = True,
-        get_emails_by_thread: bool = True,
-        create_draft_email: bool = True,
-        send_email: bool = True,
-        send_email_reply: bool = True,
-        search_emails: bool = True,
         creds: Optional[Credentials] = None,
         credentials_path: Optional[str] = None,
         token_path: Optional[str] = None,
@@ -122,17 +111,6 @@ class GmailTools(Toolkit):
         """Initialize GmailTools and authenticate with Gmail API
 
         Args:
-            get_latest_emails (bool): Enable getting latest emails. Defaults to True.
-            get_emails_from_user (bool): Enable getting emails from specific user. Defaults to True.
-            get_unread_emails (bool): Enable getting unread emails. Defaults to True.
-            get_starred_emails (bool): Enable getting starred emails. Defaults to True.
-            get_emails_by_context (bool): Enable getting emails by context. Defaults to True.
-            get_emails_by_date (bool): Enable getting emails by date. Defaults to True.
-            get_emails_by_thread (bool): Enable getting emails by thread. Defaults to True.
-            create_draft_email (bool): Enable creating draft emails. Defaults to True.
-            send_email (bool): Enable sending emails. Defaults to True.
-            search_emails (bool): Enable searching emails. Defaults to True.
-            send_email_reply (bool): Enable sending email replies. Defaults to True.
             creds (Optional[Credentials]): Pre-fetched OAuth credentials. Use this to skip a new auth flow. Defaults to None.
             credentials_path (Optional[str]): Path to credentials file. Defaults to None.
             token_path (Optional[str]): Path to token file. Defaults to None.
@@ -146,55 +124,61 @@ class GmailTools(Toolkit):
         self.scopes = scopes or self.DEFAULT_SCOPES
         self.port = port
 
-        # Validate that required scopes are present for requested operations
-        if (create_draft_email or send_email) and "https://www.googleapis.com/auth/gmail.compose" not in self.scopes:
+        """ tools functions:
+         enable_get_latest_emails (bool): Enable getting latest emails.
+         enable_get_emails_from_user (bool): Enable getting emails from specific user.
+         enable_get_unread_emails (bool): Enable getting unread emails.
+         enable_get_starred_emails (bool): Enable getting starred emails.
+         enable_get_emails_by_context (bool): Enable getting emails by context.
+         enable_get_emails_by_date (bool): Enable getting emails by date.
+         enable_get_emails_by_thread (bool): Enable getting emails by thread.
+         enable_create_draft_email (bool): Enable creating draft emails.
+         enable_send_email (bool): Enable sending emails.
+         enable_send_email_reply (bool): Enable sending email replies.
+         all (bool): Enable all tools.
+        """
+
+        tools: List[Any] = [
+            # Reading emails
+            self.get_latest_emails,
+            self.get_emails_from_user,
+            self.get_unread_emails,
+            self.get_starred_emails,
+            self.get_emails_by_context,
+            self.get_emails_by_date,
+            self.get_emails_by_thread,
+            self.search_emails,
+            # Composing emails
+            self.create_draft_email,
+            self.send_email,
+            self.send_email_reply,
+        ]
+
+        super().__init__(name="gmail_tools", tools=tools, **kwargs)
+
+        # Validate that required scopes are present for requested operations (only check registered functions)
+        if (
+            "create_draft_email" in self.functions or "send_email" in self.functions
+        ) and "https://www.googleapis.com/auth/gmail.compose" not in self.scopes:
             raise ValueError(
                 "The scope https://www.googleapis.com/auth/gmail.compose is required for email composition operations"
             )
-
         read_operations = [
-            get_latest_emails,
-            get_emails_from_user,
-            get_unread_emails,
-            get_starred_emails,
-            get_emails_by_context,
-            get_emails_by_date,
-            get_emails_by_thread,
-            search_emails,
+            "get_latest_emails",
+            "get_emails_from_user",
+            "get_unread_emails",
+            "get_starred_emails",
+            "get_emails_by_context",
+            "get_emails_by_date",
+            "get_emails_by_thread",
+            "search_emails",
         ]
-
-        if any(read_operations):
+        if any(read_operation in self.functions for read_operation in read_operations):
             read_scope = "https://www.googleapis.com/auth/gmail.readonly"
             write_scope = "https://www.googleapis.com/auth/gmail.modify"
             if read_scope not in self.scopes and write_scope not in self.scopes:
                 raise ValueError(f"The scope {read_scope} is required for email reading operations")
 
-        tools: List[Any] = []
-        if get_latest_emails:
-            tools.append(self.get_latest_emails)
-        if get_emails_from_user:
-            tools.append(self.get_emails_from_user)
-        if get_unread_emails:
-            tools.append(self.get_unread_emails)
-        if get_starred_emails:
-            tools.append(self.get_starred_emails)
-        if get_emails_by_context:
-            tools.append(self.get_emails_by_context)
-        if get_emails_by_date:
-            tools.append(self.get_emails_by_date)
-        if get_emails_by_thread:
-            tools.append(self.get_emails_by_thread)
-        if create_draft_email:
-            tools.append(self.create_draft_email)
-        if send_email:
-            tools.append(self.send_email)
-        if send_email_reply:
-            tools.append(self.send_email_reply)
-        if search_emails:
-            tools.append(self.search_emails)
-
-        super().__init__(name="gmail_tools", tools=tools, **kwargs)
-
     def _auth(self) -> None:
         """Authenticate with Gmail API"""
         token_file = Path(self.token_path or "token.json")

agno/tools/google_bigquery.py

@@ -17,10 +17,11 @@ class GoogleBigQueryTools(Toolkit):
         dataset: str,
         project: Optional[str] = None,
         location: Optional[str] = None,
-        list_tables: Optional[bool] = True,
-        describe_table: Optional[bool] = True,
-        run_sql_query: Optional[bool] = True,
         credentials: Optional[Any] = None,
+        enable_list_tables: bool = True,
+        enable_describe_table: bool = True,
+        enable_run_sql_query: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.project = project or getenv("GOOGLE_CLOUD_PROJECT")
@@ -37,11 +38,11 @@ class GoogleBigQueryTools(Toolkit):
         self.client = bigquery.Client(project=self.project, credentials=credentials)
 
         tools: List[Any] = []
-        if list_tables:
+        if all or enable_list_tables:
             tools.append(self.list_tables)
-        if describe_table:
+        if all or enable_describe_table:
             tools.append(self.describe_table)
-        if run_sql_query:
+        if all or enable_run_sql_query:
             tools.append(self.run_sql_query)
 
         super().__init__(name="google_bigquery_tools", tools=tools, **kwargs)

agno/tools/google_maps.py

@@ -6,7 +6,7 @@ Prerequisites:
   You can obtain the API key from the Google Cloud Console:
   https://console.cloud.google.com/projectselector2/google/maps-apis/credentials
 
-- You also need to activate the Address Validation API for your .
+- You also need to activate the Address Validation API for your project.
   https://console.developers.google.com/apis/api/addressvalidation.googleapis.com
 
 """
@@ -29,14 +29,6 @@ class GoogleMapTools(Toolkit):
     def __init__(
         self,
         key: Optional[str] = None,
-        search_places: bool = True,
-        get_directions: bool = True,
-        validate_address: bool = True,
-        geocode_address: bool = True,
-        reverse_geocode: bool = True,
-        get_distance_matrix: bool = True,
-        get_elevation: bool = True,
-        get_timezone: bool = True,
         **kwargs,
     ):
         self.api_key = key or getenv("GOOGLE_MAPS_API_KEY")
@@ -46,23 +38,16 @@ class GoogleMapTools(Toolkit):
 
         self.places_client = places_v1.PlacesClient()
 
-        tools: List[Any] = []
-        if search_places:
-            tools.append(self.search_places)
-        if get_directions:
-            tools.append(self.get_directions)
-        if validate_address:
-            tools.append(self.validate_address)
-        if geocode_address:
-            tools.append(self.geocode_address)
-        if reverse_geocode:
-            tools.append(self.reverse_geocode)
-        if get_distance_matrix:
-            tools.append(self.get_distance_matrix)
-        if get_elevation:
-            tools.append(self.get_elevation)
-        if get_timezone:
-            tools.append(self.get_timezone)
+        tools: List[Any] = [
+            self.search_places,
+            self.get_directions,
+            self.validate_address,
+            self.geocode_address,
+            self.reverse_geocode,
+            self.get_distance_matrix,
+            self.get_elevation,
+            self.get_timezone,
+        ]
 
         super().__init__(name="google_maps", tools=tools, **kwargs)
 

agno/tools/googlesearch.py

@@ -29,6 +29,8 @@ class GoogleSearchTools(Toolkit):
         cache_results (bool): Enable caching of search results.
         cache_ttl (int): Time-to-live for cached results in seconds.
         cache_dir (Optional[str]): Directory to store cache files.
+        enable_google_search (bool): Enable Google search.
+        all (bool): Enable all tools.
     """
 
     def __init__(
@@ -38,6 +40,8 @@ class GoogleSearchTools(Toolkit):
         headers: Optional[Any] = None,
         proxy: Optional[str] = None,
         timeout: Optional[int] = 10,
+        enable_google_search: bool = True,
+        all: bool = False,
         **kwargs,
     ):
         self.fixed_max_results: Optional[int] = fixed_max_results
@@ -47,7 +51,8 @@ class GoogleSearchTools(Toolkit):
         self.timeout: Optional[int] = timeout
 
         tools = []
-        tools.append(self.google_search)
+        if all or enable_google_search:
+            tools.append(self.google_search)
 
         super().__init__(name="google_search_tools", tools=tools, **kwargs)
 
@@ -77,7 +82,7 @@ class GoogleSearchTools(Toolkit):
         log_debug(f"Searching Google [{language}] for: {query}")
 
         # Perform Google search using the googlesearch-python package
-        results = list(search(query, num_results=max_results, lang=language, proxy=self.proxy, advanced=True))
+        results = list(search(query, num=max_results, lang=language))
 
         # Collect the search results
         res: List[Dict[str, str]] = []

agno/tools/googlesheets.py

@@ -94,27 +94,29 @@ class GoogleSheetsTools(Toolkit):
         creds: Optional[Credentials] = None,
         creds_path: Optional[str] = None,
         token_path: Optional[str] = None,
-        read: bool = True,
-        create: bool = False,
-        update: bool = False,
-        duplicate: bool = False,
         oauth_port: int = 0,
+        enable_read_sheet: bool = True,
+        enable_create_sheet: bool = False,
+        enable_update_sheet: bool = False,
+        enable_create_duplicate_sheet: bool = False,
+        all: bool = False,
         **kwargs,
     ):
         """Initialize GoogleSheetsTools with the specified configuration.
 
         Args:
-            scopes (Optional[List[str]]): Custom OAuth scopes. If None, determined by operations.
+            scopes (Optional[List[str]]): Custom OAuth scopes. If None, uses write scope by default.
             spreadsheet_id (Optional[str]): ID of the target spreadsheet.
             spreadsheet_range (Optional[str]): Range within the spreadsheet.
             creds (Optional[Credentials]): Pre-existing credentials.
             creds_path (Optional[str]): Path to credentials file.
             token_path (Optional[str]): Path to token file.
-            read (bool): Enable read operations. Defaults to True.
-            create (bool): Enable create operations. Defaults to False.
-            update (bool): Enable update operations. Defaults to False.
-            duplicate (bool): Enable duplicate operations. Defaults to False.
             oauth_port (int): Port to use for OAuth authentication. Defaults to 0.
+            enable_read_sheet (bool): Enable reading from a sheet.
+            enable_create_sheet (bool): Enable creating a sheet.
+            enable_update_sheet (bool): Enable updating a sheet.
+            enable_create_duplicate_sheet (bool): Enable creating a duplicate sheet.
+            all (bool): Enable all tools.
         """
 
         self.spreadsheet_id = spreadsheet_id
@@ -128,19 +130,21 @@ class GoogleSheetsTools(Toolkit):
         # Determine required scopes based on operations if no custom scopes provided
         if scopes is None:
             self.scopes = []
-            if read:
+            if enable_read_sheet:
                 self.scopes.append(self.DEFAULT_SCOPES["read"])
-            if create or update or duplicate:
+            if enable_create_sheet or enable_update_sheet or enable_create_duplicate_sheet:
                 self.scopes.append(self.DEFAULT_SCOPES["write"])
             # Remove duplicates while preserving order
             self.scopes = list(dict.fromkeys(self.scopes))
         else:
             self.scopes = scopes
             # Validate that required scopes are present for requested operations
-            if (create or update or duplicate) and self.DEFAULT_SCOPES["write"] not in self.scopes:
+            if (enable_create_sheet or enable_update_sheet or enable_create_duplicate_sheet) and self.DEFAULT_SCOPES[
+                "write"
+            ] not in self.scopes:
                 raise ValueError(f"The scope {self.DEFAULT_SCOPES['write']} is required for write operations")
             if (
-                read
+                enable_read_sheet
                 and self.DEFAULT_SCOPES["read"] not in self.scopes
                 and self.DEFAULT_SCOPES["write"] not in self.scopes
             ):
@@ -149,13 +153,13 @@ class GoogleSheetsTools(Toolkit):
                 )
 
         tools: List[Any] = []
-        if read:
+        if all or enable_read_sheet:
             tools.append(self.read_sheet)
-        if create:
+        if all or enable_create_sheet:
             tools.append(self.create_sheet)
-        if update:
+        if all or enable_update_sheet:
             tools.append(self.update_sheet)
-        if duplicate:
+        if all or enable_create_duplicate_sheet:
             tools.append(self.create_duplicate_sheet)
 
         super().__init__(name="google_sheets_tools", tools=tools, **kwargs)

agno/tools/hackernews.py

@@ -10,16 +10,20 @@ from agno.utils.log import log_debug, logger
 class HackerNewsTools(Toolkit):
     """
     HackerNews is a tool for getting top stories from Hacker News.
+
     Args:
-        get_top_stories (bool): Whether to get top stories from Hacker News.
-        get_user_details (bool): Whether to get user details from Hacker News.
+        enable_get_top_stories (bool): Enable getting top stories from Hacker News. Default is True.
+        enable_get_user_details (bool): Enable getting user details from Hacker News. Default is True.
+        all (bool): Enable all tools. Overrides individual flags when True. Default is False.
     """
 
-    def __init__(self, get_top_stories: bool = True, get_user_details: bool = True, **kwargs):
+    def __init__(
+        self, enable_get_top_stories: bool = True, enable_get_user_details: bool = True, all: bool = False, **kwargs
+    ):
         tools: List[Any] = []
-        if get_top_stories:
+        if all or enable_get_top_stories:
             tools.append(self.get_top_hackernews_stories)
-        if get_user_details:
+        if all or enable_get_user_details:
             tools.append(self.get_user_details)
 
         super().__init__(name="hackers_news", tools=tools, **kwargs)

agno/tools/jina.py

@@ -25,9 +25,10 @@ class JinaReaderTools(Toolkit):
         search_url: str = "https://s.jina.ai/",
         max_content_length: int = 10000,
         timeout: Optional[int] = None,
-        read_url: bool = True,
-        search_query: bool = False,
         search_query_content: bool = True,
+        enable_read_url: bool = True,
+        enable_search_query: bool = False,
+        all: bool = False,
         **kwargs,
     ):
         self.api_key = api_key or getenv("JINA_API_KEY")
@@ -41,9 +42,9 @@ class JinaReaderTools(Toolkit):
         )
 
         tools: List[Any] = []
-        if read_url:
+        if all or enable_read_url:
             tools.append(self.read_url)
-        if search_query:
+        if all or enable_search_query:
             tools.append(self.search_query)
 
         super().__init__(name="jina_reader_tools", tools=tools, **kwargs)