holmesgpt 0.16.2a0__py3-none-any.whl → 0.18.4__py3-none-any.whl

holmes/plugins/prompts/__init__.py

@@ -1,8 +1,9 @@
 import os
 import os.path
+from datetime import datetime, timezone
 from typing import Optional
+
 from jinja2 import Environment, FileSystemLoader
-from datetime import datetime, timezone
 
 THIS_DIR = os.path.abspath(os.path.dirname(__file__))
 

holmes/plugins/prompts/_fetch_logs.jinja2

@@ -1,5 +1,5 @@
 {%- set loki_ts = toolsets | selectattr("name", "equalto", "grafana/loki") | first -%}
-{%- set coralogix_ts = toolsets | selectattr("name", "equalto", "coralogix/logs") | first -%}
+{%- set coralogix_ts = toolsets | selectattr("name", "equalto", "coralogix") | first -%}
 {%- set k8s_base_ts = toolsets | selectattr("name", "equalto", "kubernetes/logs") | selectattr("fetch_pod_logs", "defined") | first -%}
 {%- set k8s_yaml_ts = toolsets | selectattr("name", "equalto", "kubernetes/logs") | rejectattr("fetch_pod_logs", "defined") | first -%}
 {%- set opensearch_ts = toolsets | selectattr("name", "equalto", "opensearch/logs") | first -%}
@@ -25,9 +25,9 @@
 ** If there are too many logs, or not enough, narrow or widen the timestamps
 * If you are not provided with time information. Ignore start_timestamp and end_timestamp.
 {%- elif coralogix_ts and coralogix_ts.status == "enabled" -%}
-### coralogix/logs
-#### Coralogix Logs Toolset
-Tools to search and fetch logs from Coralogix.
+### coralogix
+#### Coralogix Toolset
+Tools to search and fetch logs, traces, metrics, and other telemetry data from Coralogix.
 {% include '_default_log_prompt.jinja2' %}
 {%- elif k8s_base_ts and k8s_base_ts.status == "enabled" -%}
 {% include '_default_log_prompt.jinja2' %}
@@ -77,7 +77,7 @@ DO NOT use `--tail` or `| tail` when calling `kubectl logs` because you may miss
 ** 'kubernetes/logs'
 ** 'grafana/loki'
 ** 'opensearch/logs'
-** 'coralogix/logs'
+** 'coralogix'
 ** 'datadog/logs'
 ** 'openshift/logs'
 ** 'newrelic'

holmes/plugins/prompts/_runbook_instructions.jinja2

@@ -7,7 +7,8 @@
 {%- if available -%}
 # Runbook Selection
 
-You (HolmesGPT) have access to runbooks with step-by-step troubleshooting instructions. If one of the following runbooks relates to the user's issue, you MUST fetch it with the fetch_runbook tool.
+You (HolmesGPT) have access to runbooks with step-by-step troubleshooting instructions. 
+If one of the following runbooks relates to the user's issue or match one of the alerts or symptoms listed in the runbook entry, you MUST fetch it with the fetch_runbook tool.
 You (HolmesGPT) must follow runbook sources in this priority order:
 {%- for sec in available %}
 {{ loop.index }}) {{ sec.title }} (priority #{{ loop.index }})

holmes/plugins/prompts/base_user_prompt.jinja2

@@ -0,0 +1,7 @@
+{{ user_prompt }}
+
+{% if runbooks_enabled %}
+{% include '_runbook_instructions.jinja2' %}
+{% endif %}
+
+{% include '_current_date_time.jinja2' %}

holmes/plugins/prompts/conversation_history_compaction.jinja2

@@ -1,4 +1,5 @@
-Your task is to create a detailed summary of the conversation so far, paying close attention to the user's explicit requests and your previous actions.
+This conversation history is too long and no longer fits your context window. 
+Your task is to create a detailed summary of the conversation so far. This will help continuing the investigation after summarisation is done.
 This summary should be thorough in capturing technical details, code patterns, and architectural decisions that would be essential for continuing development work without losing context.
 
 Before providing your final summary, wrap your analysis in <analysis> tags to organize your thoughts and ensure you've covered all necessary points. In your analysis process:

holmes/plugins/prompts/generic_ask.jinja2

@@ -37,5 +37,3 @@ Validation error led to unhandled Java exception causing a crash.
 {% if system_prompt_additions %}
 {{ system_prompt_additions }}
 {% endif %}
-
-{% include '_current_date_time.jinja2' %}

holmes/plugins/prompts/generic_ask_conversation.jinja2

@@ -30,5 +30,3 @@ Relevant logs:
 ```
 
 Validation error led to unhandled Java exception causing a crash.
-
-{% include '_current_date_time.jinja2' %}

holmes/plugins/prompts/generic_ask_for_issue_conversation.jinja2

@@ -48,5 +48,3 @@ Relevant logs:
 ```
 
 Validation error led to unhandled Java exception causing a crash.
-
-{% include '_current_date_time.jinja2' %}

holmes/plugins/prompts/generic_investigation.jinja2

@@ -40,5 +40,3 @@ Use these rules when deciding how to apply them:
 * Remove unnecessary words
 
 {% include 'investigation_output_format.jinja2' %}
-
-{% include '_current_date_time.jinja2' %}

holmes/plugins/prompts/investigation_procedure.jinja2

@@ -91,7 +91,8 @@ YOU MUST COMPLETE EVERY SINGLE TASK before providing your final answer. NO EXCEP
 **Task Status Check Example:**
 Before final answer, confirm you see something like:
 [✓] completed - Task 1
-[✓] completed - Task 2[✓] completed - Task 3
+[✓] completed - Task 2
+[✓] completed - Task 3
 [✓] completed - Investigation Verification
 
 If you see ANY `[ ] pending` or `[~] in_progress` tasks, DO NOT provide final answer.

holmes/plugins/prompts/kubernetes_workload_ask.jinja2

@@ -75,5 +75,3 @@ Here are issues and configuration changes that happend to this kubernetes worklo
 {{ a }}
 {% endfor %}
 {% endif %}
-
-{% include '_current_date_time.jinja2' %}

holmes/plugins/prompts/kubernetes_workload_chat.jinja2

@@ -36,5 +36,3 @@ User: Why did the workload-example app crash?
 
 AI: `workload-example-1299492-d9g9d` crashed due to email validation error during HTTP request for /api/create_user
 Relevant logs:
-
-{% include '_current_date_time.jinja2' %}

holmes/plugins/runbooks/__init__.py

@@ -4,7 +4,8 @@ import os
 import os.path
 from datetime import date
 from pathlib import Path
-from typing import List, Optional, Pattern, Union, Tuple, TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional, Pattern, Tuple, Union
+
 import yaml
 from pydantic import BaseModel, PrivateAttr
 
@@ -24,6 +25,7 @@ class RobustaRunbookInstruction(BaseModel):
     symptom: str
     title: str
     instruction: Optional[str] = None
+    alerts: List[str] = []
 
     """
     Custom YAML dumper to represent multi-line strings in literal block style due to instructions often being multi-line.
@@ -54,7 +56,7 @@ class RobustaRunbookInstruction(BaseModel):
         return f"{self.id}"
 
     def to_prompt_string(self) -> str:
-        return f"id='{self.id}' | title='{self.title}' | symptom='{self.symptom}'"
+        return f"id='{self.id}' | title='{self.title}' | symptom='{self.symptom}' | relevant alerts={', '.join(self.alerts)}"
 
     def pretty(self) -> str:
         try:
@@ -149,13 +151,14 @@ class RunbookCatalog(BaseModel):
             parts.append("Here are MD runbooks:")
             parts.extend(f"* {e.to_prompt_string()}" for e in md)
         if robusta:
-            parts.append("Here are Robusta runbooks:")
+            parts.append("\nHere are Robusta runbooks:")
             parts.extend(f"* {e.to_prompt_string()}" for e in robusta)
         return "\n".join(parts)
 
 
 def load_runbook_catalog(
     dal: Optional["SupabaseDal"] = None,
+    custom_catalog_paths: Optional[List[Union[str, Path]]] = None,
 ) -> Optional[RunbookCatalog]:  # type: ignore
     dir_path = os.path.dirname(os.path.realpath(__file__))
     catalog = None
@@ -172,6 +175,32 @@ def load_runbook_catalog(
             f"Unexpected error while loading runbook catalog from {catalogPath}: {e}"
         )
 
+    # Append custom catalog files if provided
+    if custom_catalog_paths:
+        for custom_catalog_path in custom_catalog_paths:
+            try:
+                custom_catalog_path_str = str(custom_catalog_path)
+                if not os.path.isfile(custom_catalog_path_str):
+                    logging.warning(
+                        f"Custom catalog file not found: {custom_catalog_path_str}"
+                    )
+                    continue
+
+                with open(custom_catalog_path_str) as file:
+                    custom_catalog_dict = json.load(file)
+                    custom_catalog = RunbookCatalog(**custom_catalog_dict)
+
+                    if catalog:
+                        catalog.catalog.extend(custom_catalog.catalog)
+                    else:
+                        catalog = custom_catalog
+            except json.JSONDecodeError as e:
+                logging.error(f"Error decoding JSON from {custom_catalog_path}: {e}")
+            except Exception as e:
+                logging.error(
+                    f"Unexpected error while loading custom catalog from {custom_catalog_path}: {e}"
+                )
+
     # Append additional runbooks from SupabaseDal if provided
     if dal:
         try:

holmes/plugins/sources/github/__init__.py

@@ -1,9 +1,11 @@
 import logging
 from typing import List
+
+import requests  # type: ignore
+
+from holmes.core.issue import Issue
 from holmes.core.tool_calling_llm import LLMResult
 from holmes.plugins.interfaces import SourcePlugin
-from holmes.core.issue import Issue
-import requests  # type: ignore
 
 
 class GitHubSource(SourcePlugin):

holmes/plugins/sources/prometheus/models.py

@@ -2,6 +2,7 @@ import html
 from datetime import datetime, timedelta
 from typing import Dict, List, Optional, Union
 from urllib.parse import parse_qs, unquote, urlparse
+
 from pydantic import BaseModel, computed_field
 
 

holmes/plugins/toolsets/__init__.py

@@ -8,16 +8,18 @@ from pydantic import ValidationError
 
 import holmes.utils.env as env_utils
 from holmes.common.env_vars import (
-    USE_LEGACY_KUBERNETES_LOGS,
     DISABLE_PROMETHEUS_TOOLSET,
+    USE_LEGACY_KUBERNETES_LOGS,
 )
 from holmes.core.supabase_dal import SupabaseDal
 from holmes.core.tools import Toolset, ToolsetType, ToolsetYamlFromConfig, YAMLToolset
 from holmes.plugins.toolsets.atlas_mongodb.mongodb_atlas import MongoDBAtlasToolset
 from holmes.plugins.toolsets.azure_sql.azure_sql_toolset import AzureSQLToolset
 from holmes.plugins.toolsets.bash.bash_toolset import BashExecutorToolset
-from holmes.plugins.toolsets.coralogix.toolset_coralogix_logs import (
-    CoralogixLogsToolset,
+from holmes.plugins.toolsets.connectivity_check import ConnectivityCheckToolset
+from holmes.plugins.toolsets.coralogix.toolset_coralogix import CoralogixToolset
+from holmes.plugins.toolsets.datadog.toolset_datadog_general import (
+    DatadogGeneralToolset,
 )
 from holmes.plugins.toolsets.datadog.toolset_datadog_logs import DatadogLogsToolset
 from holmes.plugins.toolsets.datadog.toolset_datadog_metrics import (
@@ -26,37 +28,32 @@ from holmes.plugins.toolsets.datadog.toolset_datadog_metrics import (
 from holmes.plugins.toolsets.datadog.toolset_datadog_traces import (
     DatadogTracesToolset,
 )
-from holmes.plugins.toolsets.datadog.toolset_datadog_rds import (
-    DatadogRDSToolset,
+from holmes.plugins.toolsets.elasticsearch.elasticsearch import (
+    ElasticsearchClusterToolset,
+    ElasticsearchDataToolset,
 )
-from holmes.plugins.toolsets.datadog.toolset_datadog_general import (
-    DatadogGeneralToolset,
+from holmes.plugins.toolsets.elasticsearch.opensearch_query_assist import (
+    OpenSearchQueryAssistToolset,
 )
 from holmes.plugins.toolsets.git import GitToolset
-from holmes.plugins.toolsets.grafana.toolset_grafana import GrafanaToolset
 from holmes.plugins.toolsets.grafana.loki.toolset_grafana_loki import GrafanaLokiToolset
+from holmes.plugins.toolsets.grafana.toolset_grafana import GrafanaToolset
 from holmes.plugins.toolsets.grafana.toolset_grafana_tempo import GrafanaTempoToolset
 from holmes.plugins.toolsets.internet.internet import InternetToolset
 from holmes.plugins.toolsets.internet.notion import NotionToolset
+from holmes.plugins.toolsets.investigator.core_investigation import (
+    CoreInvestigationToolset,
+)
 from holmes.plugins.toolsets.kafka import KafkaToolset
 from holmes.plugins.toolsets.kubernetes_logs import KubernetesLogsToolset
 from holmes.plugins.toolsets.mcp.toolset_mcp import RemoteMCPToolset
 from holmes.plugins.toolsets.newrelic.newrelic import NewRelicToolset
-from holmes.plugins.toolsets.opensearch.opensearch import OpenSearchToolset
-from holmes.plugins.toolsets.opensearch.opensearch_logs import OpenSearchLogsToolset
-from holmes.plugins.toolsets.opensearch.opensearch_query_assist import (
-    OpenSearchQueryAssistToolset,
-)
-from holmes.plugins.toolsets.opensearch.opensearch_traces import OpenSearchTracesToolset
 from holmes.plugins.toolsets.rabbitmq.toolset_rabbitmq import RabbitMQToolset
 from holmes.plugins.toolsets.robusta.robusta import RobustaToolset
 from holmes.plugins.toolsets.runbook.runbook_fetcher import RunbookToolset
 from holmes.plugins.toolsets.servicenow_tables.servicenow_tables import (
     ServiceNowTablesToolset,
 )
-from holmes.plugins.toolsets.investigator.core_investigation import (
-    CoreInvestigationToolset,
-)
 
 THIS_DIR = os.path.abspath(os.path.dirname(__file__))
 
@@ -78,13 +75,16 @@ def load_toolsets_from_file(
     return toolsets
 
 
-def load_python_toolsets(dal: Optional[SupabaseDal]) -> List[Toolset]:
+def load_python_toolsets(
+    dal: Optional[SupabaseDal],
+    additional_search_paths: Optional[List[str]] = None,
+) -> List[Toolset]:
     logging.debug("loading python toolsets")
     toolsets: list[Toolset] = [
         CoreInvestigationToolset(),  # Load first for higher priority
         InternetToolset(),
+        ConnectivityCheckToolset(),
         RobustaToolset(dal),
-        OpenSearchToolset(),
         GrafanaLokiToolset(),
         GrafanaTempoToolset(),
         NewRelicToolset(),
@@ -95,18 +95,17 @@ def load_python_toolsets(dal: Optional[SupabaseDal]) -> List[Toolset]:
         DatadogGeneralToolset(),
         DatadogMetricsToolset(),
         DatadogTracesToolset(),
-        DatadogRDSToolset(),
-        OpenSearchLogsToolset(),
-        OpenSearchTracesToolset(),
         OpenSearchQueryAssistToolset(),
-        CoralogixLogsToolset(),
+        CoralogixToolset(),
         RabbitMQToolset(),
         GitToolset(),
         BashExecutorToolset(),
         MongoDBAtlasToolset(),
-        RunbookToolset(dal=dal),
+        RunbookToolset(dal=dal, additional_search_paths=additional_search_paths),
         AzureSQLToolset(),
         ServiceNowTablesToolset(),
+        ElasticsearchDataToolset(),
+        ElasticsearchClusterToolset(),
     ]
 
     if not DISABLE_PROMETHEUS_TOOLSET:
@@ -120,7 +119,10 @@ def load_python_toolsets(dal: Optional[SupabaseDal]) -> List[Toolset]:
     return toolsets
 
 
-def load_builtin_toolsets(dal: Optional[SupabaseDal] = None) -> List[Toolset]:
+def load_builtin_toolsets(
+    dal: Optional[SupabaseDal] = None,
+    additional_search_paths: Optional[List[str]] = None,
+) -> List[Toolset]:
     all_toolsets: List[Toolset] = []
     logging.debug(f"loading toolsets from {THIS_DIR}")
 
@@ -136,7 +138,9 @@ def load_builtin_toolsets(dal: Optional[SupabaseDal] = None) -> List[Toolset]:
         toolsets_from_file = load_toolsets_from_file(path, strict_check=True)
         all_toolsets.extend(toolsets_from_file)
 
-    all_toolsets.extend(load_python_toolsets(dal=dal))  # type: ignore
+    all_toolsets.extend(
+        load_python_toolsets(dal=dal, additional_search_paths=additional_search_paths)
+    )  # type: ignore
 
     # disable built-in toolsets by default, and the user can enable them explicitly in config.
     for toolset in all_toolsets:

holmes/plugins/toolsets/atlas_mongodb/mongodb_atlas.py

@@ -1,24 +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, StructuredToolResultStatus
-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
 
 

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):

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

@@ -1,22 +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,
+    StructuredToolResultStatus,
     ToolInvokeContext,
     ToolParameter,
-    StructuredToolResultStatus,
-)
-from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
-    BaseAzureSQLTool,
-    BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
 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
 
 

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

@@ -1,22 +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,
+    StructuredToolResultStatus,
     ToolInvokeContext,
     ToolParameter,
-    StructuredToolResultStatus,
+)
+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.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-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.utils import toolset_name_for_one_liner
 
 

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

@@ -1,20 +1,18 @@
 import logging
-from typing import Dict
 from datetime import datetime, timezone
+from typing import Dict, Tuple
 
 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
 
 

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

@@ -1,18 +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,
     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
 
 

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

@@ -1,22 +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,
+    StructuredToolResultStatus,
     ToolInvokeContext,
     ToolParameter,
-    StructuredToolResultStatus,
+)
+from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
+from holmes.plugins.toolsets.azure_sql.apis.storage_analysis_api import (
+    StorageAnalysisAPI,
 )
 from holmes.plugins.toolsets.azure_sql.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.storage_analysis_api import (
-    StorageAnalysisAPI,
-)
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner