added new relic support

holmes/plugins/toolsets/__init__.py

@@ -9,6 +9,7 @@
 from holmes.plugins.toolsets.grafana.common import GrafanaConfig
 from holmes.plugins.toolsets.grafana.toolset_grafana_loki import GrafanaLokiToolset
 from holmes.plugins.toolsets.grafana.toolset_grafana_tempo import GrafanaTempoToolset
+from holmes.plugins.toolsets.newrelic import NewRelicToolset
 from holmes.plugins.toolsets.kafka import KafkaToolset
 from holmes.core.tools import Toolset, YAMLToolset
 from holmes.plugins.toolsets.opensearch import OpenSearchToolset
@@ -39,7 +40,7 @@ def load_toolsets_from_file(path: str, silent_fail: bool = False, is_default: bo
 
 def load_python_toolsets(dal:Optional[SupabaseDal]) -> List[Toolset]:
     logging.debug("loading python toolsets")
-    toolsets: list[Toolset] = [InternetToolset(), FindingsToolset(dal), KafkaToolset(),GrafanaLokiToolset(), GrafanaTempoToolset(),]
+    toolsets: list[Toolset] = [InternetToolset(), FindingsToolset(dal), KafkaToolset(), GrafanaLokiToolset(), GrafanaTempoToolset(), NewRelicToolset()]
 
     opensearch = OpenSearchToolset()
     toolsets.append(opensearch)

holmes/plugins/toolsets/newrelic.py

@@ -0,0 +1,153 @@
+import os
+import requests
+import json
+import logging
+from typing import Any
+from holmes.core.tools import StaticPrerequisite, Tool, ToolParameter, Toolset, ToolsetTag
+
+class GetLogs(Tool):
+    def __init__(self, api_key: str, account_id: int):
+        super().__init__(
+            name="newrelic_get_logs",
+            description="Retrieve logs from New Relic",
+            parameters={
+                "app": ToolParameter(
+                    description="The application name to filter logs",
+                    type="string",
+                    required=True,
+                ),
+                "since": ToolParameter(
+                    description="Time range to fetch logs (e.g., '1 hour ago')",
+                    type="string",
+                    required=True,
+                ),
+            },
+        )
+        self._api_key = api_key
+        self._account_id = account_id
+
+    def invoke(self, params: Any) -> str:
+        app = params.get("app")
+        since = params.get("since")
+
+        query = {
+            "query": f"""
+            {{
+                actor {{
+                    account(id: {self._account_id}) {{
+                        nrql(query: \"SELECT * FROM Log WHERE app = '{app}' SINCE {since}\") {{
+                            results
+                        }}
+                    }}
+                }}
+            }}
+            """
+        }
+
+        url = "https://api.newrelic.com/graphql"
+        headers = {
+            "Content-Type": "application/json",
+            "Api-Key": self._api_key,
+        }
+
+        response = requests.post(url, headers=headers, json=query)
+        logging.info(f"Getting new relic logs for app {app} since {since}")
+        if response.status_code == 200:
+            data = response.json()
+            return json.dumps(data, indent=2)
+        else:
+            return f"Failed to fetch logs. Status code: {response.status_code}\n{response.text}"
+
+    def get_parameterized_one_liner(self, params) -> str:
+        return f"newrelic GetLogs(app='{params.get('app')}', since='{params.get('since')}')"
+
+
+
+class GetTraces(Tool):
+    def __init__(self, api_key: str, account_id: int):
+        super().__init__(
+            name="newrelic_get_traces",
+            description="Retrieve traces from New Relic",
+            parameters={
+                "duration": ToolParameter(
+                    description="Minimum trace duration in seconds",
+                    type="number",
+                    required=True,
+                ),
+                "trace_id": ToolParameter(
+                    description="Specific trace ID to fetch details (optional)",
+                    type="string",
+                    required=False,
+                ),
+            },
+        )
+        self._api_key = api_key
+        self._account_id = account_id
+
+    def invoke(self, params: Any) -> str:
+        duration = params.get("duration")
+        trace_id = params.get("trace_id")
+
+        if trace_id:
+            query_string = f"SELECT * FROM Span WHERE trace.id = '{trace_id}' and duration.ms > {duration * 1000} and span.kind != 'internal'"
+        else:
+            query_string = f"SELECT * FROM Span WHERE duration.ms > {duration * 1000} and span.kind != 'internal'"
+
+        query = {
+            "query": f"""
+            {{
+                actor {{
+                    account(id: {self._account_id}) {{
+                        nrql(query: \"{query_string}\") {{
+                            results
+                        }}
+                    }}
+                }}
+            }}
+            """
+        }
+
+        url = "https://api.newrelic.com/graphql"
+        headers = {
+            "Content-Type": "application/json",
+            "Api-Key": self._api_key,
+        }
+
+        response = requests.post(url, headers=headers, json=query)
+        logging.info(f"Getting newrelic traces longer than {duration}s")
+        if response.status_code == 200:
+            data = response.json()
+            return json.dumps(data, indent=2)
+        else:
+            return f"Failed to fetch traces. Status code: {response.status_code}\n{response.text}"
+
+    def get_parameterized_one_liner(self, params) -> str:
+        if "trace_id" in params and params["trace_id"]:
+            return f"newrelic GetTraces(trace_id='{params.get('trace_id')}')"
+        return f"newrelic GetTraces(duration={params.get('duration')})"
+
+
+
+class NewRelicToolset(Toolset):
+    def __init__(self):
+        api_key = os.environ.get("NEW_RELIC_API_KEY")
+        account_id = os.environ.get("NEW_RELIC_ACCOUNT_ID")
+        tools = [
+            GetLogs(api_key, account_id),
+            GetTraces(api_key, account_id),
+        ]
+
+        super().__init__(
+            name="newrelic",
+            description="Toolset for interacting with New Relic to fetch logs and traces",
+            docs_url="https://docs.newrelic.com/docs/apis/nerdgraph-api/",
+            icon_url="https://upload.wikimedia.org/wikipedia/commons/4/4d/New_Relic_logo.svg",
+            prerequisites=[
+                StaticPrerequisite(
+                    enabled=bool(api_key and account_id),
+                    disabled_reason="API key or account ID is not configured",
+                )
+            ],
+            tools=tools,
+            tags=[ToolsetTag.CORE],
+        )