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

holmes/plugins/toolsets/grafana/common.py

@@ -3,12 +3,12 @@ from typing import Dict, Optional
 from pydantic import BaseModel
 import datetime
 
-from holmes.core.tools import StructuredToolResult, ToolResultStatus
+from holmes.core.tools import StructuredToolResult, StructuredToolResultStatus
 
 
 class GrafanaConfig(BaseModel):
     """A config that represents one of the Grafana related tools like Loki or Tempo
-    If `grafana_datasource_uid` is set, then it is assume that Holmes will proxy all
+    If `grafana_datasource_uid` is set, then it is assumed that Holmes will proxy all
     requests through grafana. In this case `url` should be the grafana URL.
     If `grafana_datasource_uid` is not set, it is assumed that the `url` is the
     systems' URL
@@ -61,8 +61,20 @@ def ensure_grafana_uid_or_return_error_result(
 ) -> Optional[StructuredToolResult]:
     if not config.grafana_datasource_uid:
         return StructuredToolResult(
-            status=ToolResultStatus.ERROR,
+            status=StructuredToolResultStatus.ERROR,
             error="This tool only works when the toolset is configued ",
         )
     else:
         return None
+
+
+class GrafanaTempoLabelsConfig(BaseModel):
+    pod: str = "k8s.pod.name"
+    namespace: str = "k8s.namespace.name"
+    deployment: str = "k8s.deployment.name"
+    node: str = "k8s.node.name"
+    service: str = "service.name"
+
+
+class GrafanaTempoConfig(GrafanaConfig):
+    labels: GrafanaTempoLabelsConfig = GrafanaTempoLabelsConfig()

holmes/plugins/toolsets/grafana/grafana_api.py

@@ -6,37 +6,59 @@ import backoff
 from holmes.plugins.toolsets.grafana.common import (
     GrafanaConfig,
     build_headers,
-    get_base_url,
 )
 
 
 @backoff.on_exception(
-    backoff.expo,  # Exponential backoff
-    requests.exceptions.RequestException,  # Retry on request exceptions
-    max_tries=5,  # Maximum retries
+    backoff.expo,
+    requests.exceptions.RequestException,
+    max_tries=2,
     giveup=lambda e: isinstance(e, requests.exceptions.HTTPError)
     and e.response.status_code < 500,
 )
+def _try_health_url(url: str, headers: dict) -> None:
+    response = requests.get(url, headers=headers, timeout=5)
+    response.raise_for_status()
+
+
 def grafana_health_check(config: GrafanaConfig) -> Tuple[bool, str]:
-    base_url = get_base_url(config)
-    url = f"{base_url}/{config.healthcheck}"
-    try:
-        headers_ = build_headers(api_key=config.api_key, additional_headers=None)
-
-        response = requests.get(url, headers=headers_, timeout=10)  # Added timeout
-        response.raise_for_status()
-        return True, ""
-    except Exception as e:
-        logging.error(f"Failed to fetch grafana health status at {url}", exc_info=True)
-        error_msg = f"Failed to fetch grafana health status at {url}. {str(e)}"
-
-        # Add helpful hint if this looks like a common misconfiguration
-        if config.grafana_datasource_uid and ":3100" in config.url:
-            error_msg += (
-                "\n\nPossible configuration issue: grafana_datasource_uid is set but URL contains port 3100 "
-                "(typically used for direct Loki connections). Please verify:\n"
-                "- If connecting directly to Loki: remove grafana_datasource_uid from config\n"
-                "- If connecting via Grafana proxy: ensure URL points to Grafana (usually port 3000)"
+    """
+    Tests a healthcheck url for grafna loki.
+    1. When using grafana as proxy, grafana_datasource_uid is provided, use the data source health url (docs are added).
+    2. When using loki directly there are two cases.
+        a. Using loki cloud, health check is provided on the base url.
+        b. Using local loki, uses url/healthcheck default is url/ready
+        c. This function tries both direct loki cases for the user.
+    """
+    health_urls = []
+    if config.grafana_datasource_uid:
+        # https://grafana.com/docs/grafana/latest/developers/http_api/data_source/#check-data-source-health
+        health_urls.append(
+            f"{config.url}/api/datasources/uid/{config.grafana_datasource_uid}/health"
+        )
+    else:
+        health_urls.append(f"{config.url}/{config.healthcheck}")
+        health_urls.append(config.url)  # loki cloud uses no suffix.
+    g_headers = build_headers(api_key=config.api_key, additional_headers=config.headers)
+
+    error_msg = ""
+    for url in health_urls:
+        try:
+            _try_health_url(url, g_headers)
+            return True, ""
+        except Exception as e:
+            logging.debug(
+                f"Failed to fetch grafana health status at {url}", exc_info=True
             )
+            error_msg += f"Failed to fetch grafana health status at {url}. {str(e)}\n"
+
+    # Add helpful hint if this looks like a common misconfiguration
+    if config.grafana_datasource_uid and ":3100" in config.url:
+        error_msg += (
+            "\n\nPossible configuration issue: grafana_datasource_uid is set but URL contains port 3100 "
+            "(typically used for direct Loki connections). Please verify:\n"
+            "- If connecting directly to Loki: remove grafana_datasource_uid from config\n"
+            "- If connecting via Grafana proxy: ensure URL points to Grafana (usually port 3000)"
+        )
 
-        return False, error_msg
+    return False, error_msg

holmes/plugins/toolsets/grafana/grafana_tempo_api.py

@@ -0,0 +1,454 @@
+"""Grafana Tempo API wrapper for querying traces and metrics."""
+
+import logging
+from typing import Any, Dict, Optional, Union
+from urllib.parse import quote
+
+import backoff
+import requests  # type: ignore
+
+from holmes.plugins.toolsets.grafana.common import (
+    GrafanaTempoConfig,
+    build_headers,
+    get_base_url,
+)
+
+
+logger = logging.getLogger(__name__)
+
+
+class TempoAPIError(Exception):
+    """Custom exception for Tempo API errors with detailed response information."""
+
+    def __init__(self, status_code: int, response_text: str, url: str):
+        self.status_code = status_code
+        self.response_text = response_text
+        self.url = url
+
+        # Try to extract error message from JSON response
+        try:
+            import json
+
+            error_data = json.loads(response_text)
+            # Tempo may return errors in different formats
+            error_message = (
+                error_data.get("error")
+                or error_data.get("message")
+                or error_data.get("errorType")
+                or response_text
+            )
+        except (json.JSONDecodeError, TypeError):
+            error_message = response_text
+
+        super().__init__(f"Tempo API error {status_code}: {error_message}")
+
+
+class GrafanaTempoAPI:
+    """Python wrapper for Grafana Tempo REST API.
+
+    This class provides a clean interface to all Tempo API endpoints.
+    """
+
+    def __init__(self, config: GrafanaTempoConfig):
+        """Initialize the Tempo API wrapper.
+
+        Args:
+            config: GrafanaTempoConfig instance with connection details
+        """
+        self.config = config
+        self.base_url = get_base_url(config)
+        self.headers = build_headers(config.api_key, config.headers)
+
+    def _make_request(
+        self,
+        endpoint: str,
+        params: Optional[Dict[str, Any]] = None,
+        path_params: Optional[Dict[str, str]] = None,
+        timeout: int = 30,
+        retries: int = 3,
+    ) -> Dict[str, Any]:
+        """Make HTTP request to Tempo API with retry logic.
+
+        Args:
+            endpoint: API endpoint path (e.g., "/api/echo")
+            params: Query parameters
+            path_params: Parameters to substitute in the endpoint path
+            timeout: Request timeout in seconds
+            retries: Number of retry attempts
+
+        Returns:
+            JSON response from the API
+
+        Raises:
+            Exception: If the request fails after all retries
+        """
+        # Format endpoint with path parameters
+        if path_params:
+            for key, value in path_params.items():
+                endpoint = endpoint.replace(f"{{{key}}}", quote(str(value), safe=""))
+
+        url = f"{self.base_url}{endpoint}"
+
+        @backoff.on_exception(
+            backoff.expo,
+            requests.exceptions.RequestException,
+            max_tries=retries,
+            giveup=lambda e: isinstance(e, requests.exceptions.HTTPError)
+            and getattr(e, "response", None) is not None
+            and e.response.status_code < 500,
+        )
+        def make_request():
+            # GET request with query parameters
+            response = requests.get(
+                url,
+                headers=self.headers,
+                params=params,
+                timeout=timeout,
+            )
+            response.raise_for_status()
+            return response.json()
+
+        try:
+            return make_request()
+        except requests.exceptions.HTTPError as e:
+            # Extract detailed error message from response
+            response = e.response
+            if response is not None:
+                logger.error(
+                    f"HTTP error {response.status_code} for {url}: {response.text}"
+                )
+                raise TempoAPIError(
+                    status_code=response.status_code,
+                    response_text=response.text,
+                    url=url,
+                )
+            else:
+                logger.error(f"Request failed for {url}: {e}")
+                raise
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Request failed for {url}: {e}")
+            raise
+
+    def query_echo_endpoint(self) -> bool:
+        """Query the echo endpoint to check Tempo status.
+
+        API Endpoint: GET /api/echo
+        HTTP Method: GET
+
+        Returns:
+            bool: True if endpoint returns 200 status code, False otherwise
+        """
+        url = f"{self.base_url}/api/echo"
+
+        try:
+            response = requests.get(
+                url,
+                headers=self.headers,
+                timeout=30,
+            )
+
+            # Just check status code, don't try to parse JSON
+            return response.status_code == 200
+
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Request failed for {url}: {e}")
+            return False
+
+    def query_trace_by_id_v2(
+        self,
+        trace_id: str,
+        start: Optional[int] = None,
+        end: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Query a trace by its ID.
+
+        API Endpoint: GET /api/v2/traces/{trace_id}
+        HTTP Method: GET
+
+        Args:
+            trace_id: The trace ID to retrieve
+            start: Optional start time in Unix epoch seconds
+            end: Optional end time in Unix epoch seconds
+
+        Returns:
+            dict: OpenTelemetry format trace data
+        """
+        params = {}
+        if start is not None:
+            params["start"] = str(start)
+        if end is not None:
+            params["end"] = str(end)
+
+        return self._make_request(
+            "/api/v2/traces/{trace_id}",
+            params=params,
+            path_params={"trace_id": trace_id},
+        )
+
+    def _search_traces_common(
+        self,
+        search_params: Dict[str, Any],
+        limit: Optional[int] = None,
+        start: Optional[int] = None,
+        end: Optional[int] = None,
+        spss: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Common search implementation for both tag and TraceQL searches.
+
+        Args:
+            search_params: The search-specific parameters (tags or q)
+            limit: Optional max number of traces to return
+            start: Optional start time in Unix epoch seconds
+            end: Optional end time in Unix epoch seconds
+            spss: Optional spans per span set
+
+        Returns:
+            dict: Search results with trace metadata
+        """
+        params = search_params.copy()
+
+        if limit is not None:
+            params["limit"] = str(limit)
+        if start is not None:
+            params["start"] = str(start)
+        if end is not None:
+            params["end"] = str(end)
+        if spss is not None:
+            params["spss"] = str(spss)
+
+        return self._make_request("/api/search", params=params)
+
+    def search_traces_by_tags(
+        self,
+        tags: str,
+        min_duration: Optional[str] = None,
+        max_duration: Optional[str] = None,
+        limit: Optional[int] = None,
+        start: Optional[int] = None,
+        end: Optional[int] = None,
+        spss: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Search for traces using tag-based search.
+
+        API Endpoint: GET /api/search
+        HTTP Method: GET
+
+        Args:
+            tags: logfmt-encoded span/process attributes (required)
+            min_duration: Optional minimum trace duration (e.g., "5s")
+            max_duration: Optional maximum trace duration
+            limit: Optional max number of traces to return
+            start: Optional start time in Unix epoch seconds
+            end: Optional end time in Unix epoch seconds
+            spss: Optional spans per span set
+
+        Returns:
+            dict: Search results with trace metadata
+        """
+        search_params = {"tags": tags}
+
+        # minDuration and maxDuration are only supported with tag-based search
+        if min_duration is not None:
+            search_params["minDuration"] = min_duration
+        if max_duration is not None:
+            search_params["maxDuration"] = max_duration
+
+        return self._search_traces_common(
+            search_params=search_params,
+            limit=limit,
+            start=start,
+            end=end,
+            spss=spss,
+        )
+
+    def search_traces_by_query(
+        self,
+        q: str,
+        limit: Optional[int] = None,
+        start: Optional[int] = None,
+        end: Optional[int] = None,
+        spss: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Search for traces using TraceQL query.
+
+        API Endpoint: GET /api/search
+        HTTP Method: GET
+
+        Note: minDuration and maxDuration are not supported with TraceQL queries.
+        Use the TraceQL query syntax to filter by duration instead.
+
+        Args:
+            q: TraceQL query (required)
+            limit: Optional max number of traces to return
+            start: Optional start time in Unix epoch seconds
+            end: Optional end time in Unix epoch seconds
+            spss: Optional spans per span set
+
+        Returns:
+            dict: Search results with trace metadata
+        """
+        return self._search_traces_common(
+            search_params={"q": q},
+            limit=limit,
+            start=start,
+            end=end,
+            spss=spss,
+        )
+
+    def search_tag_names_v2(
+        self,
+        scope: Optional[str] = None,
+        q: Optional[str] = None,
+        start: Optional[int] = None,
+        end: Optional[int] = None,
+        limit: Optional[int] = None,
+        max_stale_values: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Search for available tag names.
+
+        API Endpoint: GET /api/v2/search/tags
+        HTTP Method: GET
+
+        Args:
+            scope: Optional scope filter ("resource", "span", or "intrinsic")
+            q: Optional TraceQL query to filter tags
+            start: Optional start time in Unix epoch seconds
+            end: Optional end time in Unix epoch seconds
+            limit: Optional max number of tag names
+            max_stale_values: Optional max stale values parameter
+
+        Returns:
+            dict: Available tag names organized by scope
+        """
+        params = {}
+        if scope is not None:
+            params["scope"] = scope
+        if q is not None:
+            params["q"] = q
+        if start is not None:
+            params["start"] = str(start)
+        if end is not None:
+            params["end"] = str(end)
+        if limit is not None:
+            params["limit"] = str(limit)
+        if max_stale_values is not None:
+            params["maxStaleValues"] = str(max_stale_values)
+
+        return self._make_request("/api/v2/search/tags", params=params)
+
+    def search_tag_values_v2(
+        self,
+        tag: str,
+        q: Optional[str] = None,
+        start: Optional[int] = None,
+        end: Optional[int] = None,
+        limit: Optional[int] = None,
+        max_stale_values: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Search for values of a specific tag with optional TraceQL filtering.
+
+        API Endpoint: GET /api/v2/search/tag/{tag}/values
+        HTTP Method: GET
+
+        Args:
+            tag: The tag name to get values for (required)
+            q: Optional TraceQL query to filter tag values (e.g., '{resource.cluster="us-east-1"}')
+            start: Optional start time in Unix epoch seconds
+            end: Optional end time in Unix epoch seconds
+            limit: Optional max number of values
+            max_stale_values: Optional max stale values parameter
+
+        Returns:
+            dict: List of discovered values for the tag
+        """
+        params = {}
+        if q is not None:
+            params["q"] = q
+        if start is not None:
+            params["start"] = str(start)
+        if end is not None:
+            params["end"] = str(end)
+        if limit is not None:
+            params["limit"] = str(limit)
+        if max_stale_values is not None:
+            params["maxStaleValues"] = str(max_stale_values)
+
+        return self._make_request(
+            "/api/v2/search/tag/{tag}/values",
+            params=params,
+            path_params={"tag": tag},
+        )
+
+    def query_metrics_instant(
+        self,
+        q: str,
+        start: Optional[Union[int, str]] = None,
+        end: Optional[Union[int, str]] = None,
+        since: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """Query TraceQL metrics for an instant value.
+
+        Computes a single value across the entire time range.
+
+        API Endpoint: GET /api/metrics/query
+        HTTP Method: GET
+
+        Args:
+            q: TraceQL metrics query (required)
+            start: Optional start time (Unix seconds/nanoseconds/RFC3339)
+            end: Optional end time (Unix seconds/nanoseconds/RFC3339)
+            since: Optional duration string (e.g., "1h")
+
+        Returns:
+            dict: Single computed metric value
+        """
+        params = {"q": q}
+        if start is not None:
+            params["start"] = str(start)
+        if end is not None:
+            params["end"] = str(end)
+        if since is not None:
+            params["since"] = since
+
+        return self._make_request("/api/metrics/query", params=params)
+
+    def query_metrics_range(
+        self,
+        q: str,
+        step: Optional[str] = None,
+        start: Optional[Union[int, str]] = None,
+        end: Optional[Union[int, str]] = None,
+        since: Optional[str] = None,
+        exemplars: Optional[int] = None,
+    ) -> Dict[str, Any]:
+        """Query TraceQL metrics for a time series range.
+
+        Returns metrics computed at regular intervals over the time range.
+
+        API Endpoint: GET /api/metrics/query_range
+        HTTP Method: GET
+
+        Args:
+            q: TraceQL metrics query (required)
+            step: Optional time series granularity (e.g., "1m", "5m")
+            start: Optional start time (Unix seconds/nanoseconds/RFC3339)
+            end: Optional end time (Unix seconds/nanoseconds/RFC3339)
+            since: Optional duration string (e.g., "3h")
+            exemplars: Optional maximum number of exemplars to return
+
+        Returns:
+            dict: Time series of metric values
+        """
+        params = {"q": q}
+        if step is not None:
+            params["step"] = step
+        if start is not None:
+            params["start"] = str(start)
+        if end is not None:
+            params["end"] = str(end)
+        if since is not None:
+            params["since"] = since
+        if exemplars is not None:
+            params["exemplars"] = str(exemplars)
+
+        return self._make_request("/api/metrics/query_range", params=params)

holmes/plugins/toolsets/grafana/loki/instructions.jinja2

@@ -0,0 +1,9 @@
+Grafana Loki is a multi-tenant log aggregation system designed to store and query logs from all your applications and infrastructure.
+
+* Deleted K8s objects don’t delete their past logs. You can still find them by time-bounded queries on stable labels (e.g., namespace + app) or by regex on pod names.
+* If you can't find the kubernetes workload in the cluster YOU SHOULD still try to find logs from it using non specific values on labels.
+* If asked to check for logs, you must always try 1-2 of the best queries you can construct to search for the logs.
+
+Loki indexes log lines using labels to help find relevant log lines.
+For example a default Kubernetes labels setup would look like that
+{namespace="prod", app="backend-api", container="api", pod="backend-api-68b7d9df9c-xyz12", stream="stdout"}