holmesgpt 0.13.3__py3-none-any.whl → 0.14.0a0__py3-none-any.whl

holmes/plugins/toolsets/grafana/trace_parser.py

@@ -187,7 +187,7 @@ def format_traces_list(trace_data: Dict) -> str:
                 else "\n"
             )
             trace_str += f"\tstartTime={unix_nano_to_rfc3339(int(trace.get('startTimeUnixNano')))}"
-            trace_str += f" rootServiceName={trace.get('trootServiceName')}"
+            trace_str += f" rootServiceName={trace.get('rootServiceName')}"
             trace_str += f" rootTraceName={trace.get('rootTraceName')}"
             traces_str.append(trace_str)
         return "\n".join(traces_str)

holmes/plugins/toolsets/newrelic.py

@@ -79,7 +79,7 @@ class GetLogs(BaseNewRelicTool):
 
         try:
             logging.info(f"Getting New Relic logs for app {app} since {since}")
-            response = requests.post(url, headers=headers, json=query)
+            response = requests.post(url, headers=headers, json=query)  # type: ignore[arg-type]
 
             if response.status_code == 200:
                 return success(response.json())
@@ -164,7 +164,7 @@ class GetTraces(BaseNewRelicTool):
 
         try:
             logging.info(f"Getting New Relic traces with duration > {duration}s")
-            response = requests.post(url, headers=headers, json=query)
+            response = requests.post(url, headers=headers, json=query)  # type: ignore[arg-type]
 
             if response.status_code == 200:
                 return success(response.json())
@@ -197,7 +197,7 @@ class NewRelicToolset(Toolset):
         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/",
+            docs_url="https://holmesgpt.dev/data-sources/builtin-toolsets/newrelic/",
             icon_url="https://companieslogo.com/img/orig/NEWR-de5fcb2e.png?t=1720244493",
             prerequisites=[CallablePrerequisite(callable=self.prerequisites_callable)],
             tools=[

{holmesgpt-0.13.3.dist-info → holmesgpt-0.14.0a0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: holmesgpt
-Version: 0.13.3
+Version: 0.14.0a0
 Summary: 
 Author: Natan Yellin
 Author-email: natan@robusta.dev
@@ -35,7 +35,7 @@ Requires-Dist: mcp (==v1.12.2)
 Requires-Dist: openai (>=1.6.1,<1.100.0)
 Requires-Dist: opensearch-py (>=2.8.0,<3.0.0)
 Requires-Dist: postgrest (==0.16.8)
-Requires-Dist: prometrix (==0.2.3)
+Requires-Dist: prometrix (==0.2.5)
 Requires-Dist: prompt-toolkit (>=3.0.51,<4.0.0)
 Requires-Dist: protobuf (>=6.31.1)
 Requires-Dist: pydantic (>=2.7,<3.0)

{holmesgpt-0.13.3.dist-info → holmesgpt-0.14.0a0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 holmes/.git_archival.json,sha256=PbwdO7rNhEJ4ALiO12DPPb81xNAIsVxCA0m8OrVoqsk,182
-holmes/__init__.py,sha256=8wTYyPOjl8qDr6BiYOrAsU76iYg_9luKrV-TylK8LQw,257
+holmes/__init__.py,sha256=6EFHBvUBpblW6V0uBPWb-MI_VRb2nAh-QMWNn_XRRVQ,263
 holmes/clients/robusta_client.py,sha256=lHV02aKJmAb08jVhjQ6jWvhhmjQ86Ajbo_lgbhqfn3s,1221
 holmes/common/env_vars.py,sha256=-P834z71-S7cklgTB7DX0OuTLzxwmAT3WETt3svBUY0,3013
 holmes/common/openshift.py,sha256=akbQ0GpnmuzXOqTcotpTDQSDKIROypS9mgPOprUgkCw,407
@@ -20,13 +20,13 @@ holmes/core/runbooks.py,sha256=Oj5ICmiGgaq57t4erPzQDvHQ0rMGj1nhiiYhl8peH3Q,939
 holmes/core/safeguards.py,sha256=SAw-J9y3uAehJVZJYsFs4C62jzLV4p_C07F2jUuJHug,4895
 holmes/core/supabase_dal.py,sha256=-AtQ6oDyIpoc2a46xA26e8416kUD4tj2Q3p4P42PFP4,21291
 holmes/core/todo_tasks_formatter.py,sha256=USyJZcoX6zoxID1UV-abAKdaWFYLO6QJd-UKryJAurI,1487
-holmes/core/tool_calling_llm.py,sha256=hdGmkm8fMaa7zRy4x2F8xMTYo1b1vpaHtQnfxXtG8VQ,41175
+holmes/core/tool_calling_llm.py,sha256=B1D-0QyVfYW-p3pfbbDo4DRoBEpzV7uN6IiH48i61XQ,42018
 holmes/core/tools.py,sha256=Xj7Yoztf9OqrGMSwhWBNmIwoZ147P6qi7U6vt0PNgBQ,21380
 holmes/core/tools_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 holmes/core/tools_utils/tool_executor.py,sha256=wLuyr1O8zkNXaZ_MzWm5hUDcOI3tB9I4YiserDhIu-E,2375
 holmes/core/tools_utils/toolset_utils.py,sha256=SvWzen8Fg_TB_6Idw1hK0nCPrJL40ueWVMfsv8Kh2RY,2363
 holmes/core/toolset_manager.py,sha256=PVmhRaSZntcJ_-jE3W2vHwZgAXIprPIgq0YMGYhmcZs,18451
-holmes/core/tracing.py,sha256=j7tC4xuPHTU3s1m8Y7kClNfGOSBTHBINFkr1NmYe51Y,8986
+holmes/core/tracing.py,sha256=o1vTLCbnFvBOVnQKqRaj4HDNYGYAd4ub2IgkV1E3qY4,9130
 holmes/interactive.py,sha256=l1kh0ABXO_RWIzkEuLkedM-73k1ezENAvXMrgNeqsKA,41836
 holmes/main.py,sha256=e_J4B1oMPbBiKz8cf-m_UJbhdk98AgAPS1LsU67PEe8,34813
 holmes/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -149,24 +149,24 @@ holmes/plugins/toolsets/datadog/datadog_metrics_instructions.jinja2,sha256=_59Dz
 holmes/plugins/toolsets/datadog/datadog_rds_instructions.jinja2,sha256=lbI6lkcMjG1CVKeL0XDkq5zl-fRDVxZ_t_CjS-Uh_OM,3119
 holmes/plugins/toolsets/datadog/datadog_traces_formatter.py,sha256=uTtWTrsbvO9cZcUDskJE9p5sEscieXwhEpxvRKkaiEw,10275
 holmes/plugins/toolsets/datadog/instructions_datadog_traces.jinja2,sha256=9j3-46UNE35DE2xBDTCRt1EedgNdgRXuC1u-X3yB-9I,1487
-holmes/plugins/toolsets/datadog/toolset_datadog_general.py,sha256=vT6E6tBBQM70o793c3OXRgzdode91dlFMnh2_XNn8Ug,27274
-holmes/plugins/toolsets/datadog/toolset_datadog_logs.py,sha256=2WbfvJRSj1sIvTshq1zdWvcpF5LmmnuBV-jE0xs0si8,9954
-holmes/plugins/toolsets/datadog/toolset_datadog_metrics.py,sha256=cngHD2vEmqLJV0afg-JSlfPz43DtAY3yCTnlM-kQWss,24810
+holmes/plugins/toolsets/datadog/toolset_datadog_general.py,sha256=ZYMqyycrnbr530xuF0WYiKyAbCdxeDPKYIvw0ito1Dc,27265
+holmes/plugins/toolsets/datadog/toolset_datadog_logs.py,sha256=lpZYyrMeg9225gY8vuwAGGQb0XzfhYpdrVyVNE0PANE,9940
+holmes/plugins/toolsets/datadog/toolset_datadog_metrics.py,sha256=_3C-x-P4KmaKmuqNQI9vlI7n-aqU30a8Qabu3xPhGOA,24793
 holmes/plugins/toolsets/datadog/toolset_datadog_rds.py,sha256=5nxqZ93Fx8AmGr7KvF76erk-YsFDIL35KhEJrWsklcQ,29015
-holmes/plugins/toolsets/datadog/toolset_datadog_traces.py,sha256=yFkEzignhXVvC3KP-PmamhyssyEhKSWa2o8M0lYJiKU,26059
+holmes/plugins/toolsets/datadog/toolset_datadog_traces.py,sha256=1NXO1gnqSB2qKFSRw3M9kex5mVTiOX393CBeyQ_TXfE,26044
 holmes/plugins/toolsets/docker.yaml,sha256=O0Q0z0kZS8_QBEhwrUfbXdGUn1nP1K9k0FlQd6EZVJ4,1559
-holmes/plugins/toolsets/git.py,sha256=7D_LEtzHgrcQmIWOG6-hhSCESdZLNb4IWfDhrvxeGP8,32001
+holmes/plugins/toolsets/git.py,sha256=JVDHl_im9GZepk2Ir4AAaeMbDVqfkMzrsUgR-GaGDck,32029
 holmes/plugins/toolsets/grafana/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 holmes/plugins/toolsets/grafana/base_grafana_toolset.py,sha256=AjvbS4txSo97YqeeFlXUnu95oPFt8rB-CD1Ccf92a04,1760
-holmes/plugins/toolsets/grafana/common.py,sha256=BoPZwZ3_MSWGYhJ7lBOuWb292r4qfCiMBOXuXEKY8xA,2103
+holmes/plugins/toolsets/grafana/common.py,sha256=ycAuGVGL0iAUh1K_cukG2UyTHWDXhj1gBOYkU3A1Ofo,2439
 holmes/plugins/toolsets/grafana/grafana_api.py,sha256=nq7KWoUAfFxp1-3VqKadPNKbTifLTgKUAZOFUH5QWUY,1665
+holmes/plugins/toolsets/grafana/grafana_tempo_api.py,sha256=bcCPEwOwm1Og9LYR6PlAtjGgQopvySNjf5nTwILuFt4,15697
 holmes/plugins/toolsets/grafana/loki_api.py,sha256=f7oTzfhJ1LojsPoAfsKt32ADWffLEywBJQWG9eyfb7I,2529
-holmes/plugins/toolsets/grafana/tempo_api.py,sha256=UbLfyzA5TbP-5jx4Dkc20xLlR-6Z4z-4nPKrMlzgCRU,3565
 holmes/plugins/toolsets/grafana/toolset_grafana.py,sha256=GsCxsaWNXzOoA6eoaH9LN6ezUcxdFSgg1iWpqqvcVX0,4882
 holmes/plugins/toolsets/grafana/toolset_grafana_loki.py,sha256=Xc80raV16wlt05hdPslkb8ru40h2GYFc2Y80bR1bvlM,3846
-holmes/plugins/toolsets/grafana/toolset_grafana_tempo.jinja2,sha256=AKRXmFTq6BTovJ6aC7dSbqfURRu0y6NTMLyp1ZJeVN4,1436
-holmes/plugins/toolsets/grafana/toolset_grafana_tempo.py,sha256=A34_xlssBxbC3KmTLjwkvdCM1-5KR0oO9KO-c6irj2g,22719
-holmes/plugins/toolsets/grafana/trace_parser.py,sha256=O6fJqUwvpCKL5hAdDtPIZ3LkpTbcDSVPLlLIdNqAbuk,7025
+holmes/plugins/toolsets/grafana/toolset_grafana_tempo.jinja2,sha256=VU49G5fxIJFny2wS2gvRBiyuCXEnvB7JqrJGLF6i0dY,6490
+holmes/plugins/toolsets/grafana/toolset_grafana_tempo.py,sha256=bgydZ-jAOxQANdSFN_p_7B7wurcNKVTs2RbqVeMwNEU,43794
+holmes/plugins/toolsets/grafana/trace_parser.py,sha256=8PjqPGDGo9uB2Z8WWWknMKdhcqlqZEVncQCCkl2F06A,7024
 holmes/plugins/toolsets/helm.yaml,sha256=-IPDChKMHcxGbzA0z9GKczRshL-mD24cHpBizfNM1jM,1604
 holmes/plugins/toolsets/internet/internet.py,sha256=zuI6tTdGF1ahg16xsIisRw3EPT6Tb1dCPlZ8bvCVry8,7749
 holmes/plugins/toolsets/internet/notion.py,sha256=dIeXJUCOZBdGmcgJJWeH2RcTyXdPsfNONj7T0_5iap4,4757
@@ -182,7 +182,7 @@ holmes/plugins/toolsets/logging_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCe
 holmes/plugins/toolsets/logging_utils/logging_api.py,sha256=seW5DNSzREQx4nmkt9NEqN21j8jTPNSJyOkCrdcdWoQ,11615
 holmes/plugins/toolsets/logging_utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 holmes/plugins/toolsets/mcp/toolset_mcp.py,sha256=tYc_fjBE3ycM2HPJjwW_EviSb2B8Psf3pj_AnMQZ01g,4874
-holmes/plugins/toolsets/newrelic.py,sha256=diC1zNhV9awH0in7WdOL_L5PaLFQ_u_F-JCSHptGmKY,7843
+holmes/plugins/toolsets/newrelic.py,sha256=0WyGqtr836J3mEvpC0j3fKa4VzRb3YgLYm68q_-G6Mc,7906
 holmes/plugins/toolsets/opensearch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 holmes/plugins/toolsets/opensearch/opensearch.py,sha256=rLbwYQJbKsPRA57rbo9Kae9v7SNd8B0GZqHVUk3UJWQ,8616
 holmes/plugins/toolsets/opensearch/opensearch_logs.py,sha256=-WlOPzjLwQuEoKi_4r5KZZNnKuvBZO0EFO2NnFflf1M,5859
@@ -227,8 +227,8 @@ holmes/utils/pydantic_utils.py,sha256=g0e0jLTa8Je8JKrhEP4N5sMxj0_hhPOqFZr0Vpd67s
 holmes/utils/stream.py,sha256=Z9duSET8uXx-1dqPBjGuDKBxkJMaMsCrOqOXrVbtjGw,2806
 holmes/utils/tags.py,sha256=SU4EZMBtLlIb7OlHsSpguFaypczRzOcuHYxDSanV3sQ,3364
 holmes/version.py,sha256=uDRPOvVaHreROj_9HPe81RVpTzHcG8ojpGTsnJIlQOM,5220
-holmesgpt-0.13.3.dist-info/LICENSE.txt,sha256=RdZMj8VXRQdVslr6PMYMbAEu5pOjOdjDqt3yAmWb9Ds,1072
-holmesgpt-0.13.3.dist-info/METADATA,sha256=825O5gscyikyfjYadAlKfq7eMGY6PBzRj2kx4H8Tc18,16237
-holmesgpt-0.13.3.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
-holmesgpt-0.13.3.dist-info/entry_points.txt,sha256=JdzEyZhpaYr7Boo4uy4UZgzY1VsAEbzMgGmHZtx9KFY,42
-holmesgpt-0.13.3.dist-info/RECORD,,
+holmesgpt-0.14.0a0.dist-info/LICENSE.txt,sha256=RdZMj8VXRQdVslr6PMYMbAEu5pOjOdjDqt3yAmWb9Ds,1072
+holmesgpt-0.14.0a0.dist-info/METADATA,sha256=TT4hUs10TqA-shY5RUPfIWdPhiza6qETumt_fFJvASI,16239
+holmesgpt-0.14.0a0.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
+holmesgpt-0.14.0a0.dist-info/entry_points.txt,sha256=JdzEyZhpaYr7Boo4uy4UZgzY1VsAEbzMgGmHZtx9KFY,42
+holmesgpt-0.14.0a0.dist-info/RECORD,,

holmes/plugins/toolsets/grafana/tempo_api.py

@@ -1,124 +0,0 @@
-import requests  # type: ignore
-from typing import Dict, List, Optional
-import backoff
-
-from holmes.plugins.toolsets.grafana.common import build_headers
-from holmes.plugins.toolsets.grafana.trace_parser import process_trace
-
-
-def execute_tempo_query_with_retry(
-    base_url: str,
-    api_key: Optional[str],
-    headers: Optional[Dict[str, str]],
-    query_params: dict,
-    retries: int = 3,
-    timeout: int = 5,
-):
-    """
-    Execute a Tempo API query through Grafana with retries and timeout.
-
-    Args:
-        tempo_datasource_uid: The UID of the Tempo datasource.
-        query_params: Query parameters for the API.
-        retries: Number of retries for the request.
-        timeout: Timeout for each request in seconds.
-
-    Returns:
-        List of trace results.
-    """
-    url = f"{base_url}/api/search"
-
-    @backoff.on_exception(
-        backoff.expo,  # Exponential backoff
-        requests.exceptions.RequestException,  # Retry on request exceptions
-        max_tries=retries,  # Maximum retries
-        giveup=lambda e: isinstance(e, requests.exceptions.HTTPError)
-        and e.response.status_code < 500,
-    )
-    def make_request():
-        response = requests.post(
-            url,
-            headers=build_headers(api_key=api_key, additional_headers=headers),
-            json=query_params,
-            timeout=timeout,  # Set timeout for the request
-        )
-        response.raise_for_status()  # Raise an error for non-2xx responses
-        return response.json()
-
-    try:
-        return make_request()
-    except requests.exceptions.RequestException as e:
-        raise Exception(f"Request to Tempo API failed after retries: {e}")
-
-
-def query_tempo_traces(
-    base_url: str,
-    api_key: Optional[str],
-    headers: Optional[Dict[str, str]],
-    query: Optional[str],
-    start: int,
-    end: int,
-    limit: int,
-) -> Dict:
-    query_params = {
-        "start": str(start),
-        "end": str(end),
-        "limit": str(limit),
-    }
-
-    if query:
-        query_params["q"] = query
-    data = execute_tempo_query_with_retry(
-        base_url=base_url,
-        api_key=api_key,
-        headers=headers,
-        query_params=query_params,
-    )
-    return data
-
-
-def query_tempo_trace_by_id(
-    base_url: str,
-    api_key: Optional[str],
-    headers: Optional[Dict[str, str]],
-    trace_id: str,
-    key_labels: List[str],
-    retries: int = 3,
-    timeout: int = 5,
-) -> str:
-    """
-    Query Tempo for a specific trace by its ID with retries and backoff.
-
-    Args:
-        tempo_datasource_id: The ID of the Tempo datasource.
-        trace_id: The trace ID to retrieve.
-        retries: Number of retries for the request.
-        timeout: Timeout for each request in seconds.
-
-    Returns:
-        A formatted trace details string
-    """
-    url = f"{base_url}/api/traces/{trace_id}"
-
-    @backoff.on_exception(
-        backoff.expo,
-        requests.exceptions.RequestException,
-        max_tries=retries,
-        giveup=lambda e: isinstance(e, requests.exceptions.HTTPError)
-        and e.response.status_code < 500,
-    )
-    def make_request():
-        response = requests.get(
-            url,
-            headers=build_headers(api_key=api_key, additional_headers=headers),
-            timeout=timeout,
-        )
-        response.raise_for_status()
-        return process_trace(response.json(), key_labels)
-
-    try:
-        return make_request()
-    except requests.exceptions.RequestException as e:
-        raise Exception(
-            f"Failed to retrieve trace by ID after retries: {e} \n for URL: {url}"
-        )