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

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

@@ -1,23 +1,21 @@
 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.alert_monitoring_api import (
+    AlertMonitoringAPI,
+)
+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.azure_sql.apis.alert_monitoring_api import (
-    AlertMonitoringAPI,
-)
-from typing import Tuple
-
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 
 

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

@@ -3,16 +3,16 @@ from typing import Dict, List, 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.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.utils import format_timing
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 

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

@@ -3,16 +3,16 @@ from typing import Dict, List, 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.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.utils import format_timing
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 

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

@@ -3,16 +3,16 @@ from typing import Dict, List, 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.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.utils import format_timing
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 

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

@@ -3,16 +3,16 @@ from typing import Dict, List, 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.azure_base_toolset import (
+    AzureSQLDatabaseConfig,
     BaseAzureSQLTool,
     BaseAzureSQLToolset,
-    AzureSQLDatabaseConfig,
 )
-from holmes.plugins.toolsets.azure_sql.apis.azure_sql_api import AzureSQLAPIClient
 from holmes.plugins.toolsets.azure_sql.utils import format_timing
 from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
 

holmes/plugins/toolsets/azure_sql/utils.py

@@ -20,38 +20,6 @@ def format_timing(microseconds: float) -> str:
         return f"{microseconds:.0f} μs"
 
 
-def format_size_bytes(bytes_value: float) -> str:
-    """Format byte values with appropriate units.
-
-    Args:
-        bytes_value: Size value in bytes
-
-    Returns:
-        Formatted string with appropriate units (GB, MB, KB, or B)
-    """
-    if bytes_value >= 1_073_741_824:  # >= 1 GB
-        return f"{bytes_value / 1_073_741_824:.2f} GB"
-    elif bytes_value >= 1_048_576:  # >= 1 MB
-        return f"{bytes_value / 1_048_576:.2f} MB"
-    elif bytes_value >= 1_024:  # >= 1 KB
-        return f"{bytes_value / 1_024:.2f} KB"
-    else:  # < 1 KB
-        return f"{bytes_value:.0f} B"
-
-
-def format_percentage(value: float, decimal_places: int = 2) -> str:
-    """Format percentage values.
-
-    Args:
-        value: Percentage value (0-100)
-        decimal_places: Number of decimal places to show
-
-    Returns:
-        Formatted percentage string
-    """
-    return f"{value:.{decimal_places}f}%"
-
-
 def safe_divide(numerator: float, denominator: float, default: float = 0.0) -> float:
     """Safely divide two numbers, handling division by zero.
 

holmes/plugins/toolsets/bash/argocd/__init__.py

@@ -1,13 +1,13 @@
 import argparse
 from typing import Any, Optional
 
-from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
-from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
-from holmes.plugins.toolsets.bash.common.stringify import escape_shell_args
 from holmes.plugins.toolsets.bash.argocd.constants import (
     ALLOWED_ARGOCD_COMMANDS,
     DENIED_ARGOCD_COMMANDS,
 )
+from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
+from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
+from holmes.plugins.toolsets.bash.common.stringify import escape_shell_args
 from holmes.plugins.toolsets.bash.common.validators import (
     validate_command_and_operations,
 )

holmes/plugins/toolsets/bash/aws/__init__.py

@@ -1,16 +1,16 @@
 import argparse
 from typing import Any, Optional
 
+from holmes.plugins.toolsets.bash.aws.constants import (
+    ALLOWED_AWS_COMMANDS,
+    DENIED_AWS_COMMANDS,
+)
 from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
 from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
 from holmes.plugins.toolsets.bash.common.stringify import escape_shell_args
 from holmes.plugins.toolsets.bash.common.validators import (
     validate_command_and_operations,
 )
-from holmes.plugins.toolsets.bash.aws.constants import (
-    ALLOWED_AWS_COMMANDS,
-    DENIED_AWS_COMMANDS,
-)
 
 
 class AWSCommand(BashCommand):

holmes/plugins/toolsets/bash/azure/__init__.py

@@ -1,16 +1,16 @@
 import argparse
 from typing import Any, Optional
 
+from holmes.plugins.toolsets.bash.azure.constants import (
+    ALLOWED_AZURE_COMMANDS,
+    DENIED_AZURE_COMMANDS,
+)
 from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
 from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
 from holmes.plugins.toolsets.bash.common.stringify import escape_shell_args
 from holmes.plugins.toolsets.bash.common.validators import (
     validate_command_and_operations,
 )
-from holmes.plugins.toolsets.bash.azure.constants import (
-    ALLOWED_AZURE_COMMANDS,
-    DENIED_AZURE_COMMANDS,
-)
 
 
 class AzureCommand(BashCommand):

holmes/plugins/toolsets/bash/bash_toolset.py

@@ -4,21 +4,20 @@ import os
 import random
 import re
 import string
-from typing import Dict, Any, Optional
+from typing import Any, Dict, Optional
 
 import sentry_sdk
 
-
 from holmes.common.env_vars import (
     BASH_TOOL_UNSAFE_ALLOW_ALL,
 )
 from holmes.core.tools import (
     CallablePrerequisite,
     StructuredToolResult,
+    StructuredToolResultStatus,
     Tool,
     ToolInvokeContext,
     ToolParameter,
-    StructuredToolResultStatus,
     Toolset,
     ToolsetTag,
 )

holmes/plugins/toolsets/bash/common/bash.py

@@ -1,11 +1,14 @@
 import subprocess
+
 from holmes.core.tools import StructuredToolResult, StructuredToolResultStatus
+from holmes.utils.memory_limit import check_oom_and_append_hint, get_ulimit_prefix
 
 
 def execute_bash_command(cmd: str, timeout: int, params: dict) -> StructuredToolResult:
     try:
+        protected_cmd = get_ulimit_prefix() + cmd
         process = subprocess.run(
-            cmd,
+            protected_cmd,
             shell=True,
             executable="/bin/bash",
             stdout=subprocess.PIPE,
@@ -16,16 +19,23 @@ def execute_bash_command(cmd: str, timeout: int, params: dict) -> StructuredTool
         )
 
         stdout = process.stdout.strip() if process.stdout else ""
-        result_data = f"{cmd}\n" f"{stdout}"
+        stdout = check_oom_and_append_hint(stdout, process.returncode)
+        result_data = f"{cmd}\n{stdout}"
 
-        status = StructuredToolResultStatus.ERROR
-        if process.returncode == 0 and stdout:
-            status = StructuredToolResultStatus.SUCCESS
-        elif not stdout:
-            status = StructuredToolResultStatus.NO_DATA
+        if process.returncode == 0:
+            status = (
+                StructuredToolResultStatus.SUCCESS
+                if stdout
+                else StructuredToolResultStatus.NO_DATA
+            )
+            error = None
+        else:
+            status = StructuredToolResultStatus.ERROR
+            error = f'Error: Command "{cmd}" returned non-zero exit status {process.returncode}'
 
         return StructuredToolResult(
             status=status,
+            error=error,
             data=result_data,
             params=params,
             invocation=cmd,
@@ -38,10 +48,10 @@ def execute_bash_command(cmd: str, timeout: int, params: dict) -> StructuredTool
             params=params,
         )
     except FileNotFoundError:
-        # This might occur if /bin/bash is not found, or if shell=False and command is not found
+        # This might occur if /bin/bash is not found, or command is not found
         return StructuredToolResult(
             status=StructuredToolResultStatus.ERROR,
-            error="Error: Bash executable or command not found. Ensure bash is installed and the command is valid.",
+            error="Error: Bash executable or command not found.",
             params=params,
         )
     except Exception as e:

holmes/plugins/toolsets/bash/common/bash_command.py

@@ -1,5 +1,5 @@
-from abc import ABC, abstractmethod
 import argparse
+from abc import ABC, abstractmethod
 from typing import Any, Optional
 
 from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig

holmes/plugins/toolsets/bash/common/stringify.py

@@ -1,5 +1,5 @@
-import shlex
 import re
+import shlex
 
 SAFE_SHELL_CHARS = frozenset(".-_=/,:")
 

holmes/plugins/toolsets/bash/kubectl/__init__.py

@@ -1,4 +1,5 @@
 from typing import Any, Optional
+
 from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
 from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
 from holmes.plugins.toolsets.bash.kubectl.constants import (
@@ -9,9 +10,9 @@ from holmes.plugins.toolsets.bash.kubectl.constants import (
 )
 from holmes.plugins.toolsets.bash.kubectl.kubectl_describe import KubectlDescribeCommand
 from holmes.plugins.toolsets.bash.kubectl.kubectl_events import KubectlEventsCommand
+from holmes.plugins.toolsets.bash.kubectl.kubectl_get import KubectlGetCommand
 from holmes.plugins.toolsets.bash.kubectl.kubectl_logs import KubectlLogsCommand
 from holmes.plugins.toolsets.bash.kubectl.kubectl_top import KubectlTopCommand
-from holmes.plugins.toolsets.bash.kubectl.kubectl_get import KubectlGetCommand
 
 
 class KubectlCommand(BashCommand):

holmes/plugins/toolsets/bash/kubectl/constants.py

@@ -1,6 +1,5 @@
 import re
 
-
 SAFE_NAME_PATTERN = re.compile(r"^[a-zA-Z0-9][a-zA-Z0-9\-_.]*$")
 SAFE_NAMESPACE_PATTERN = re.compile(r"^[a-z0-9][a-z0-9\-]*$")
 SAFE_SELECTOR_PATTERN = re.compile(r"^[a-zA-Z0-9\-_.=,!()]+$")

holmes/plugins/toolsets/bash/kubectl/kubectl_get.py

@@ -1,6 +1,8 @@
+import argparse
 from typing import Any, Optional
 
-import argparse
+from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
+from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
 from holmes.plugins.toolsets.bash.common.stringify import escape_shell_args
 from holmes.plugins.toolsets.bash.common.validators import (
     whitelist_validator,
@@ -9,9 +11,6 @@ from holmes.plugins.toolsets.bash.kubectl.constants import (
     VALID_RESOURCE_TYPES,
 )
 
-from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
-from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
-
 
 class KubectlGetCommand(BashCommand):
     def __init__(self):

holmes/plugins/toolsets/bash/parse_command.py

@@ -3,28 +3,27 @@ import logging
 import shlex
 from typing import Any, Optional
 
-from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
-from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
-from holmes.plugins.toolsets.bash.kubectl import KubectlCommand
+from holmes.plugins.toolsets.bash.argocd import ArgocdCommand
 from holmes.plugins.toolsets.bash.aws import AWSCommand
 from holmes.plugins.toolsets.bash.azure import AzureCommand
-from holmes.plugins.toolsets.bash.argocd import ArgocdCommand
+from holmes.plugins.toolsets.bash.common.bash_command import BashCommand
+from holmes.plugins.toolsets.bash.common.config import BashExecutorConfig
 from holmes.plugins.toolsets.bash.docker import DockerCommand
 from holmes.plugins.toolsets.bash.helm import HelmCommand
-
-# Utilities imports - all now use Command classes
-from holmes.plugins.toolsets.bash.utilities.wc import WCCommand
+from holmes.plugins.toolsets.bash.kubectl import KubectlCommand
+from holmes.plugins.toolsets.bash.utilities.base64_util import Base64Command
 from holmes.plugins.toolsets.bash.utilities.cut import CutCommand
-from holmes.plugins.toolsets.bash.utilities.sort import SortCommand
-from holmes.plugins.toolsets.bash.utilities.uniq import UniqCommand
+from holmes.plugins.toolsets.bash.utilities.grep import GrepCommand
 from holmes.plugins.toolsets.bash.utilities.head import HeadCommand
-from holmes.plugins.toolsets.bash.utilities.tail import TailCommand
-from holmes.plugins.toolsets.bash.utilities.tr import TrCommand
-from holmes.plugins.toolsets.bash.utilities.base64_util import Base64Command
 from holmes.plugins.toolsets.bash.utilities.jq import JqCommand
 from holmes.plugins.toolsets.bash.utilities.sed import SedCommand
-from holmes.plugins.toolsets.bash.utilities.grep import GrepCommand
+from holmes.plugins.toolsets.bash.utilities.sort import SortCommand
+from holmes.plugins.toolsets.bash.utilities.tail import TailCommand
+from holmes.plugins.toolsets.bash.utilities.tr import TrCommand
+from holmes.plugins.toolsets.bash.utilities.uniq import UniqCommand
 
+# Utilities imports - all now use Command classes
+from holmes.plugins.toolsets.bash.utilities.wc import WCCommand
 
 # All commands now use BashCommand classes
 AVAILABLE_COMMANDS: list[BashCommand] = [

holmes/plugins/toolsets/connectivity_check.py

@@ -0,0 +1,124 @@
+import socket
+from typing import Any, Dict, Literal
+
+from holmes.core.tools import (
+    StructuredToolResult,
+    StructuredToolResultStatus,
+    Tool,
+    ToolInvokeContext,
+    ToolParameter,
+    Toolset,
+    ToolsetTag,
+)
+from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
+
+BROWSER_LIKE_UA = (
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+    "AppleWebKit/537.36 (KHTML, like Gecko) "
+    "Chrome/120.0.0.0 Safari/537.36"
+)
+
+UserAgentMode = Literal["none", "browser"]
+
+
+def tcp_check(host: str, port: int, timeout: float) -> Dict[str, Any]:
+    if not (1 <= port <= 65535):
+        return {
+            "ok": False,
+            "error": "invalid port (must be 1-65535)",
+        }
+
+    try:
+        with socket.create_connection((host, port), timeout=timeout):
+            return {
+                "ok": True,
+            }
+    except (OSError, socket.timeout) as e:
+        return {
+            "ok": False,
+            "error": str(e),
+        }
+
+
+class TcpCheckTool(Tool):
+    toolset: "ConnectivityCheckToolset" = None  # type: ignore
+
+    def __init__(self, toolset: "ConnectivityCheckToolset"):
+        super().__init__(
+            name="tcp_check",
+            description="Check if a TCP socket can be opened to a host and port.",
+            parameters={
+                "host": ToolParameter(
+                    description="The hostname or IP address to connect to",
+                    type="string",
+                    required=True,
+                ),
+                "port": ToolParameter(
+                    description="The port to connect to",
+                    type="integer",
+                    required=True,
+                ),
+                "timeout": ToolParameter(
+                    description="Timeout in seconds (default: 3.0)",
+                    type="number",
+                    required=False,
+                ),
+            },
+        )
+        self.toolset = toolset
+
+    def _invoke(self, params: dict, context: ToolInvokeContext) -> StructuredToolResult:
+        host = params.get("host")
+        port = params.get("port")
+        if host is None:
+            return StructuredToolResult(
+                status=StructuredToolResultStatus.ERROR,
+                data={"error": "host parameter is required"},
+                params=params,
+            )
+        if port is None:
+            return StructuredToolResult(
+                status=StructuredToolResultStatus.ERROR,
+                data={"error": "port parameter is required"},
+                params=params,
+            )
+
+        result = tcp_check(
+            host=host,
+            port=int(port),
+            timeout=float(params.get("timeout", 3.0)),
+        )
+        return StructuredToolResult(
+            status=StructuredToolResultStatus.SUCCESS,
+            data=result,
+            params=params,
+        )
+
+    def get_parameterized_one_liner(self, params) -> str:
+        host = params.get("host", "<missing host>")
+        port = params.get("port", "<missing port>")
+        return (
+            f"{toolset_name_for_one_liner(self.toolset.name)}: "
+            f"TCP check {host}:{port}"
+        )
+
+
+class ConnectivityCheckToolset(Toolset):
+    def __init__(self):
+        super().__init__(
+            name="connectivity_check",
+            description="Check TCP connectivity to endpoints",
+            icon_url="https://platform.robusta.dev/demos/internet-access.svg",
+            tools=[
+                TcpCheckTool(self),
+            ],
+            tags=[
+                ToolsetTag.CORE,
+            ],
+            is_default=True,
+            enabled=True,
+            docs_url="https://holmesgpt.dev/data-sources/builtin-toolsets/connectivity-check/",
+        )
+
+    def get_example_config(self) -> Dict[str, Any]:
+        return {}