holmesgpt 0.13.2__py3-none-any.whl → 0.18.4__py3-none-any.whl

holmes/plugins/toolsets/aks.yaml

@@ -7,6 +7,10 @@ toolsets:
       - command: "az account show"
       - command: "az aks --help"
       - command: "kubectl version --client"
+
+    # Note: Tools in this toolset use transformers with llm_summarize
+    # to automatically summarize large JSON outputs from Azure CLI commands
+    # when a fast model is configured, focusing on key configuration and status information.
     tools:
       - name: "cloud_provider"
         description: "Fetches the cloud provider of the kubernetes cluster, determined by the providerID of the nodes"
@@ -18,16 +22,61 @@ toolsets:
         user_description: "get AKS cluster {{ CLUSTER_NAME }} under resource group {{ RESOURCE_GROUP_NAME }} in subscription {{ SUBSCRIPTION_ID }}"
         command: |
           az aks show --resource-group {{ RESOURCE_GROUP_NAME }} --name {{ CLUSTER_NAME }} --subscription {{ SUBSCRIPTION_ID }}
+        transformers:
+          - name: llm_summarize
+            config:
+              input_threshold: 1500
+              prompt: |
+                Summarize this AKS cluster configuration focusing on:
+                - Cluster status, health state, and version information
+                - Node pool configuration and scaling settings
+                - Network configuration (subnet, load balancer, network policy)
+                - Add-ons and features enabled (monitoring, auto-scaling, etc.)
+                - Security settings (RBAC, managed identity, private cluster)
+                - Any warnings or deprecated configurations
+                - When possible, highlight settings that might impact performance or security
+                - Be concise: aim for ≤ 50% of the original length; avoid repeating defaults/healthy/unchanged details
+                - Prefer aggregates and counts; list only outliers and actionable items
+                - Keep grep-friendly: include exact field names/values that matter
       - name: "aks_list_clusters_by_rg"
         description: "Lists all AKS clusters under a specific resource group. Only run this tool when you need to get all clusters in a resource group, rather than a specific one."
         user_description: "list AKS clusters in resource group {{ RESOURCE_GROUP_NAME }} under subscription {{ SUBSCRIPTION_ID }}"
         command: |
           az aks list --resource-group {{ RESOURCE_GROUP_NAME }} --subscription {{ SUBSCRIPTION_ID }}
+        transformers:
+          - name: llm_summarize
+            config:
+              input_threshold: 1000
+              prompt: |
+                Summarize this AKS clusters list focusing on:
+                - Cluster names, locations, and current status
+                - Kubernetes versions and any version skew issues
+                - Node pool counts and VM SKUs
+                - Key configuration differences between clusters
+                - Any clusters in error or unusual states
+                - When possible, group clusters by similar configurations
+                - Be concise and avoid expansion: target ≤ 50% of input size; prefer counts + outliers over full listings
+                - Include grep-ready keys/values; avoid repeating entire objects or unchanged defaults
       - name: "aks_list_node_pools"
         description: "Lists node pools in an AKS cluster"
         user_description: "list node pools for AKS cluster {{ CLUSTER_NAME }} under resource group {{ RESOURCE_GROUP_NAME }}"
         command: |
           az aks nodepool list --resource-group {{ RESOURCE_GROUP_NAME }} --cluster-name {{ CLUSTER_NAME }} --subscription {{ SUBSCRIPTION_ID }}
+        transformers:
+          - name: llm_summarize
+            config:
+              input_threshold: 1200
+              prompt: |
+                Summarize this AKS node pools list focusing on:
+                - Node pool names, VM sizes, and current node counts
+                - Scaling configuration (min/max nodes, auto-scaling enabled)
+                - Node pool status and provisioning state
+                - OS types, Kubernetes versions, and any upgrade patterns
+                - Taints, labels, and specialized configurations
+                - Any node pools in error states or with scaling issues
+                - When possible, identify resource allocation patterns
+                - Strive for ≤ 50% of the original size; keep results compact and grep-friendly (one line per aggregate)
+                - Prioritize aggregates and actionable outliers over comprehensive details
       - name: "aks_show_node_pool"
         description: "Shows details of a specific node pool in an AKS cluster"
         user_description: "get node pool {{ NODE_POOL_NAME }} in AKS cluster {{ CLUSTER_NAME }} under resource group {{ RESOURCE_GROUP_NAME }}"
@@ -84,3 +133,18 @@ toolsets:
         user_description: "list NSG rules for NSG {{ NSG_NAME }} in resource group {{ RESOURCE_GROUP_NAME }} under subscription {{ SUBSCRIPTION_ID }}"
         command: |
             az network nsg rule list --resource-group {{ RESOURCE_GROUP_NAME }} --nsg-name {{ NSG_NAME }} --subscription {{ SUBSCRIPTION_ID }} --include-default -o table
+        transformers:
+          - name: llm_summarize
+            config:
+              input_threshold: 1000
+              prompt: |
+                Summarize these NSG rules focusing on:
+                - Allow vs deny rules and their priority order
+                - Inbound vs outbound traffic rules
+                - Port ranges and protocols affected
+                - Source and destination configurations
+                - Any overly permissive or potentially insecure rules
+                - Default rules vs custom rules
+                - When possible, highlight rules that might impact AKS connectivity
+                - Be concise: aim for ≤ 50% of the original text; prioritize aggregates and actionable outliers
+                - Include grep-ready keys/values; avoid repeating entire rules or unchanged defaults

holmes/plugins/toolsets/atlas_mongodb/mongodb_atlas.py

@@ -1,23 +1,25 @@
-import requests  # type: ignore
+import gzip
+import io
 import logging
-from typing import Any, Dict, Tuple, List
+import os
+from collections import Counter
+from datetime import datetime, timedelta, timezone
+from typing import Any, Dict, List, Tuple
+
+import requests  # type: ignore
+from pydantic import BaseModel, PrivateAttr
+from requests.auth import HTTPDigestAuth  # type: ignore
+
 from holmes.core.tools import (
     CallablePrerequisite,
+    StructuredToolResult,
+    StructuredToolResultStatus,
     Tool,
+    ToolInvokeContext,
     ToolParameter,
     Toolset,
     ToolsetTag,
 )
-
-from pydantic import BaseModel, PrivateAttr
-from holmes.core.tools import StructuredToolResult, ToolResultStatus
-from requests.auth import HTTPDigestAuth  # type: ignore
-import gzip
-import io
-from datetime import datetime, timedelta, timezone
-import os
-from collections import Counter
-
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 
 
@@ -41,7 +43,6 @@ class MongoDBAtlasToolset(Toolset):
     def __init__(self):
         super().__init__(
             prerequisites=[CallablePrerequisite(callable=self.prerequisites_callable)],
-            experimental=True,
             tools=[
                 ReturnProjectAlerts(toolset=self),
                 ReturnProjectProcesses(toolset=self),
@@ -66,8 +67,8 @@ class MongoDBAtlasToolset(Toolset):
                 {"Accept": "application/vnd.atlas.2025-03-12+json"}
             )
             self._session.auth = HTTPDigestAuth(
-                self.config.get("public_key"),
-                self.config.get("private_key"),
+                self.config.get("public_key"),  # type: ignore
+                self.config.get("private_key"),  # type: ignore
             )
             return True, ""
         except Exception:
@@ -90,15 +91,15 @@ class MongoDBAtlasBaseTool(Tool):
         if response.ok:
             res = response.json()
             return StructuredToolResult(
-                status=ToolResultStatus.SUCCESS
+                status=StructuredToolResultStatus.SUCCESS
                 if res.get(field, [])
-                else ToolResultStatus.NO_DATA,
+                else StructuredToolResultStatus.NO_DATA,
                 data=res,
                 params=params,
             )
         else:
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Failed {self.name}.\n{response.text}",
                 return_code=response.status_code,
                 params=params,
@@ -118,9 +119,7 @@ class ReturnProjectAlerts(MongoDBAtlasBaseTool):
         project_id = self.toolset.config.get("project_id", "")
         return f"{toolset_name_for_one_liner(self.toolset.name)}: Get Project Alerts ({project_id})"
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             url = "https://cloud.mongodb.com/api/atlas/v2/groups/{project_id}/alerts".format(
                 project_id=self.toolset.config.get("project_id")
@@ -130,7 +129,7 @@ class ReturnProjectAlerts(MongoDBAtlasBaseTool):
         except Exception as e:
             logging.exception(self.get_parameterized_one_liner(params))
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 data=f"Exception {self.name}: {str(e)}",
                 params=params,
             )
@@ -145,9 +144,7 @@ class ReturnProjectProcesses(MongoDBAtlasBaseTool):
         project_id = self.toolset.config.get("project_id", "")
         return f"{toolset_name_for_one_liner(self.toolset.name)}: Get Project Processes ({project_id})"
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             url = "https://cloud.mongodb.com/api/atlas/v2/groups/{project_id}/processes".format(
                 project_id=self.toolset.config.get("project_id")
@@ -157,7 +154,7 @@ class ReturnProjectProcesses(MongoDBAtlasBaseTool):
         except Exception as e:
             logging.exception(self.get_parameterized_one_liner(params))
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Exception {self.name}: {str(e)}",
                 params=params,
             )
@@ -180,9 +177,7 @@ class ReturnProjectSlowQueries(MongoDBAtlasBaseTool):
         process_id = params.get("process_id", "")
         return f"{toolset_name_for_one_liner(self.toolset.name)}: Get Slow Queries ({process_id})"
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             url = self.url.format(
                 project_id=self.toolset.config.get("project_id"),
@@ -193,7 +188,7 @@ class ReturnProjectSlowQueries(MongoDBAtlasBaseTool):
         except Exception as e:
             logging.exception(self.get_parameterized_one_liner(params))
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Exception {self.name}: {str(e)}",
                 params=params,
             )
@@ -209,9 +204,7 @@ class ReturnEventsFromProject(MongoDBAtlasBaseTool):
         project_id = self.toolset.config.get("project_id", "")
         return f"{toolset_name_for_one_liner(self.toolset.name)}: Get Project Events ({project_id})"
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         params.update({"itemsPerPage": 500})
         try:
             now_utc = datetime.now(timezone.utc)
@@ -230,14 +223,14 @@ class ReturnEventsFromProject(MongoDBAtlasBaseTool):
                 )
                 data = f"last 4 hours eventTypeName and # of occurrences list: {events_counter} \n to get more information about a given eventTypeName call atlas_return_events_type_from_project"
                 status = (
-                    ToolResultStatus.SUCCESS
+                    StructuredToolResultStatus.SUCCESS
                     if events_counter
-                    else ToolResultStatus.NO_DATA
+                    else StructuredToolResultStatus.NO_DATA
                 )
                 return StructuredToolResult(status=status, data=data, params=params)
             else:
                 return StructuredToolResult(
-                    status=ToolResultStatus.ERROR,
+                    status=StructuredToolResultStatus.ERROR,
                     error=f"Failed {self.name}. \n{response.text}",
                     return_code=response.status_code,
                     params=params,
@@ -245,7 +238,7 @@ class ReturnEventsFromProject(MongoDBAtlasBaseTool):
         except Exception as e:
             logging.exception(self.get_parameterized_one_liner(params))
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Exception {self.name}: {str(e)}",
                 params=params,
             )
@@ -268,9 +261,7 @@ class ReturnLogsForProcessInProject(MongoDBAtlasBaseTool):
         hostname = params.get("hostName", "")
         return f"{toolset_name_for_one_liner(self.toolset.name)}: Get Host Logs ({hostname})"
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         one_hour_ago = datetime.now(timezone.utc) - timedelta(hours=1)
         try:
             url = self.url.format(
@@ -287,11 +278,13 @@ class ReturnLogsForProcessInProject(MongoDBAtlasBaseTool):
                 with gzip.GzipFile(fileobj=io.BytesIO(response.content)) as gz:
                     text_data = gz.read().decode("utf-8")
                 return StructuredToolResult(
-                    status=ToolResultStatus.SUCCESS, data=text_data, params=params
+                    status=StructuredToolResultStatus.SUCCESS,
+                    data=text_data,
+                    params=params,
                 )
             else:
                 return StructuredToolResult(
-                    status=ToolResultStatus.ERROR,
+                    status=StructuredToolResultStatus.ERROR,
                     error=f"Failed {self.name}. \n{response.text}",
                     return_code=response.status_code,
                     params=params,
@@ -299,7 +292,7 @@ class ReturnLogsForProcessInProject(MongoDBAtlasBaseTool):
         except Exception as e:
             logging.exception(self.get_parameterized_one_liner(params))
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Exception {self.name}: {str(e)}",
                 params=params,
             )
@@ -322,9 +315,7 @@ class ReturnEventTypeFromProject(MongoDBAtlasBaseTool):
         event_type = params.get("eventType", "")
         return f"{toolset_name_for_one_liner(self.toolset.name)}: Get Event Details ({event_type})"
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             url = self.url.format(projectId=self.toolset.config.get("project_id"))
 
@@ -340,7 +331,7 @@ class ReturnEventTypeFromProject(MongoDBAtlasBaseTool):
         except Exception as e:
             logging.exception(self.get_parameterized_one_liner(params))
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Exception {self.name}: {str(e)}",
                 params=params,
             )

holmes/plugins/toolsets/azure_sql/apis/alert_monitoring_api.py

@@ -1,6 +1,7 @@
-from typing import Dict, List, Any
 import logging
-from datetime import datetime, timezone, timedelta
+from datetime import datetime, timedelta, timezone
+from typing import Any, Dict, List
+
 from azure.core.credentials import TokenCredential
 from azure.mgmt.monitor import MonitorManagementClient
 from azure.mgmt.resource import ResourceManagementClient

holmes/plugins/toolsets/azure_sql/apis/azure_sql_api.py

@@ -1,6 +1,7 @@
-from typing import Dict, List
 import logging
 import struct
+from typing import Dict, List
+
 from azure.core.credentials import TokenCredential
 from azure.mgmt.sql import SqlManagementClient
 

holmes/plugins/toolsets/azure_sql/apis/connection_failure_api.py

@@ -1,6 +1,7 @@
-from typing import Dict, List, Any
 import logging
-from datetime import datetime, timezone, timedelta
+from datetime import datetime, timedelta, timezone
+from typing import Any, Dict, List
+
 from azure.core.credentials import TokenCredential
 from azure.mgmt.monitor import MonitorManagementClient
 

holmes/plugins/toolsets/azure_sql/apis/connection_monitoring_api.py

@@ -1,8 +1,10 @@
-from typing import Dict, List
 import logging
 from datetime import datetime, timedelta
+from typing import Dict, List
+
 from azure.core.credentials import TokenCredential
 from azure.monitor.query import MetricsQueryClient
+
 from .azure_sql_api import AzureSQLAPIClient
 
 

holmes/plugins/toolsets/azure_sql/apis/storage_analysis_api.py

@@ -1,8 +1,10 @@
-from typing import Dict, List
 import logging
 from datetime import datetime, timedelta
+from typing import Dict, List
+
 from azure.core.credentials import TokenCredential
 from azure.monitor.query import MetricsQueryClient
+
 from .azure_sql_api import AzureSQLAPIClient
 
 

holmes/plugins/toolsets/azure_sql/azure_sql_toolset.py

@@ -1,19 +1,24 @@
-import os
 import logging
+import os
 from typing import Any, Dict, Tuple, Union
 
-from azure.identity import DefaultAzureCredential, ClientSecretCredential
+from azure.identity import ClientSecretCredential, DefaultAzureCredential
 
 from holmes.core.tools import (
     CallablePrerequisite,
     ToolsetTag,
 )
 from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
-from holmes.plugins.toolsets.consts import TOOLSET_CONFIG_MISSING_ERROR
 from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
-    BaseAzureSQLToolset,
     AzureSQLConfig,
     AzureSQLDatabaseConfig,
+    BaseAzureSQLToolset,
+)
+from holmes.plugins.toolsets.azure_sql.tools.analyze_connection_failures import (
+    AnalyzeConnectionFailures,
+)
+from holmes.plugins.toolsets.azure_sql.tools.analyze_database_connections import (
+    AnalyzeDatabaseConnections,
 )
 
 # Import all tool classes
@@ -23,24 +28,19 @@ from holmes.plugins.toolsets.azure_sql.tools.analyze_database_health_status impo
 from holmes.plugins.toolsets.azure_sql.tools.analyze_database_performance import (
     AnalyzeDatabasePerformance,
 )
-from holmes.plugins.toolsets.azure_sql.tools.analyze_database_connections import (
-    AnalyzeDatabaseConnections,
-)
 from holmes.plugins.toolsets.azure_sql.tools.analyze_database_storage import (
     AnalyzeDatabaseStorage,
 )
-from holmes.plugins.toolsets.azure_sql.tools.get_top_cpu_queries import GetTopCPUQueries
+from holmes.plugins.toolsets.azure_sql.tools.get_active_alerts import GetActiveAlerts
 from holmes.plugins.toolsets.azure_sql.tools.get_slow_queries import GetSlowQueries
+from holmes.plugins.toolsets.azure_sql.tools.get_top_cpu_queries import GetTopCPUQueries
 from holmes.plugins.toolsets.azure_sql.tools.get_top_data_io_queries import (
     GetTopDataIOQueries,
 )
 from holmes.plugins.toolsets.azure_sql.tools.get_top_log_io_queries import (
     GetTopLogIOQueries,
 )
-from holmes.plugins.toolsets.azure_sql.tools.get_active_alerts import GetActiveAlerts
-from holmes.plugins.toolsets.azure_sql.tools.analyze_connection_failures import (
-    AnalyzeConnectionFailures,
-)
+from holmes.plugins.toolsets.consts import TOOLSET_CONFIG_MISSING_ERROR
 
 
 class AzureSQLToolset(BaseAzureSQLToolset):
@@ -60,7 +60,6 @@ class AzureSQLToolset(BaseAzureSQLToolset):
             docs_url="https://kagi.com/proxy/png-clipart-microsoft-sql-server-microsoft-azure-sql-database-microsoft-text-logo-thumbnail.png?c=4Sg1bvcUGOrhnDzXgoBBa0G0j27ykgskX4a8cLrZp_quzqlpVGVG02OqQtezTxy7lB6ydmTKgbVAn_F7BxofxK6LKKUZSpjJ1huIAsXPVaXyakO4sWXFiX0Wz_8WjkA0AIlO_oFfW31AKaj5RcvGcr3siy0n5kW-GcqdpeBWsmm_huxUT6RycULFCDFBwuUzHvVl5TW3cYqlMxT8ecPZfg%3D%3D",
             icon_url="https://upload.wikimedia.org/wikipedia/commons/thumb/f/f7/Azure_SQL_Database_logo.svg/1200px-Azure_SQL_Database_logo.svg.png",
             tags=[ToolsetTag.CORE],
-            experimental=True,
             tools=[
                 AnalyzeDatabaseHealthStatus(self),
                 AnalyzeDatabasePerformance(self),

holmes/plugins/toolsets/azure_sql/tools/analyze_connection_failures.py

@@ -1,17 +1,22 @@
 import logging
-from typing import Dict, Tuple
 from datetime import datetime, timezone
+from typing import Dict, Tuple
 
-from holmes.core.tools import StructuredToolResult, ToolParameter, ToolResultStatus
-from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
-    BaseAzureSQLTool,
-    BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
+from holmes.core.tools import (
+    StructuredToolResult,
+    StructuredToolResultStatus,
+    ToolInvokeContext,
+    ToolParameter,
 )
 from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.apis.connection_failure_api import (
     ConnectionFailureAPI,
 )
+from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
+    BaseAzureSQLTool,
+    BaseAzureSQLToolset,
+)
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 
 
@@ -213,9 +218,7 @@ class AnalyzeConnectionFailures(BaseAzureSQLTool):
 
         return "\n".join(report_sections)
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             # Get configuration
             db_config = self.toolset.database_config()
@@ -242,7 +245,7 @@ class AnalyzeConnectionFailures(BaseAzureSQLTool):
             # Check for errors
             if "error" in analysis_data:
                 return StructuredToolResult(
-                    status=ToolResultStatus.ERROR,
+                    status=StructuredToolResultStatus.ERROR,
                     error=analysis_data["error"],
                     params=params,
                 )
@@ -253,7 +256,7 @@ class AnalyzeConnectionFailures(BaseAzureSQLTool):
             )
 
             return StructuredToolResult(
-                status=ToolResultStatus.SUCCESS,
+                status=StructuredToolResultStatus.SUCCESS,
                 data=report_text,
                 params=params,
             )
@@ -263,7 +266,7 @@ class AnalyzeConnectionFailures(BaseAzureSQLTool):
                 f"Error in analyze_connection_failures: {str(e)}", exc_info=True
             )
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=f"Failed to analyze connection failures: {str(e)}",
                 params=params,
             )

holmes/plugins/toolsets/azure_sql/tools/analyze_database_connections.py

@@ -1,17 +1,22 @@
 import logging
-from typing import Any, Dict, Tuple
 from datetime import datetime, timezone
+from typing import Any, Dict, Tuple
 
-from holmes.core.tools import StructuredToolResult, ToolParameter, ToolResultStatus
-from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
-    BaseAzureSQLTool,
-    BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
+from holmes.core.tools import (
+    StructuredToolResult,
+    StructuredToolResultStatus,
+    ToolInvokeContext,
+    ToolParameter,
 )
+from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.apis.connection_monitoring_api import (
     ConnectionMonitoringAPI,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
+from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
+    BaseAzureSQLTool,
+    BaseAzureSQLToolset,
+)
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 
 
@@ -151,9 +156,7 @@ class AnalyzeDatabaseConnections(BaseAzureSQLTool):
 
         return "\n".join(report_sections)
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             hours_back = params.get("hours_back", 2)
 
@@ -200,7 +203,7 @@ class AnalyzeDatabaseConnections(BaseAzureSQLTool):
             )
 
             return StructuredToolResult(
-                status=ToolResultStatus.SUCCESS,
+                status=StructuredToolResultStatus.SUCCESS,
                 data=report_text,
                 params=params,
             )
@@ -208,7 +211,7 @@ class AnalyzeDatabaseConnections(BaseAzureSQLTool):
             error_msg = f"Failed to generate connection report: {str(e)}"
             logging.error(error_msg)
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=error_msg,
                 params=params,
             )

holmes/plugins/toolsets/azure_sql/tools/analyze_database_health_status.py

@@ -1,16 +1,18 @@
 import logging
-from typing import Dict
 from datetime import datetime, timezone
+from typing import Dict, Tuple
 
-from holmes.core.tools import StructuredToolResult, ToolResultStatus
+from holmes.core.tools import (
+    StructuredToolResult,
+    StructuredToolResultStatus,
+    ToolInvokeContext,
+)
+from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
-from typing import Tuple
-
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 
 
@@ -131,9 +133,7 @@ class AnalyzeDatabaseHealthStatus(BaseAzureSQLTool):
 
         return "\n".join(report_sections)
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             db_config = self.toolset.database_config()
             client = self.toolset.api_client()
@@ -145,7 +145,7 @@ class AnalyzeDatabaseHealthStatus(BaseAzureSQLTool):
             report_text = self._build_health_report(health_data, db_config)
 
             return StructuredToolResult(
-                status=ToolResultStatus.SUCCESS,
+                status=StructuredToolResultStatus.SUCCESS,
                 data=report_text,
                 params=params,
             )
@@ -153,7 +153,7 @@ class AnalyzeDatabaseHealthStatus(BaseAzureSQLTool):
             error_msg = f"Failed to generate health report: {str(e)}"
             logging.error(error_msg)
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=error_msg,
                 params=params,
             )

holmes/plugins/toolsets/azure_sql/tools/analyze_database_performance.py

@@ -1,14 +1,18 @@
 import logging
-from typing import Any, Dict, List, Tuple, cast
 from datetime import datetime, timezone
+from typing import Any, Dict, List, Tuple, cast
 
-from holmes.core.tools import StructuredToolResult, ToolResultStatus
+from holmes.core.tools import (
+    StructuredToolResult,
+    StructuredToolResultStatus,
+    ToolInvokeContext,
+)
+from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 
 
@@ -192,9 +196,7 @@ class AnalyzeDatabasePerformance(BaseAzureSQLTool):
 
         return "\n".join(report_sections)
 
-    def _invoke(
-        self, params: dict, user_approved: bool = False
-    ) -> StructuredToolResult:
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
         try:
             db_config = self.toolset.database_config()
             client = self.toolset.api_client()
@@ -206,7 +208,7 @@ class AnalyzeDatabasePerformance(BaseAzureSQLTool):
             report_text = self._build_performance_report(performance_data, db_config)
 
             return StructuredToolResult(
-                status=ToolResultStatus.SUCCESS,
+                status=StructuredToolResultStatus.SUCCESS,
                 data=report_text,
                 params=params,
             )
@@ -214,7 +216,7 @@ class AnalyzeDatabasePerformance(BaseAzureSQLTool):
             error_msg = f"Failed to generate performance report: {str(e)}"
             logging.error(error_msg)
             return StructuredToolResult(
-                status=ToolResultStatus.ERROR,
+                status=StructuredToolResultStatus.ERROR,
                 error=error_msg,
                 params=params,
             )