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

holmes/plugins/toolsets/kubernetes_logs.py

@@ -63,15 +63,15 @@ class KubernetesLogsToolset(BasePodLoggingToolset):
         super().__init__(
             name="kubernetes/logs",
             description="Read Kubernetes pod logs using a unified API",
-            docs_url="https://docs.robusta.dev/master/configuration/holmesgpt/toolsets/kubernetes.html#logs",
+            docs_url="https://holmesgpt.dev/data-sources/builtin-toolsets/kubernetes/",
             icon_url="https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRPKA-U9m5BxYQDF1O7atMfj9EMMXEoGu4t0Q&s",
             prerequisites=[prerequisite],
             is_default=True,
-            tools=[
-                PodLoggingTool(self),
-            ],
+            tools=[],  # Initialize with empty tools first
             tags=[ToolsetTag.CORE],
         )
+        # Now that parent is initialized and self.name exists, create the tool
+        self.tools = [PodLoggingTool(self)]
         enabled, disabled_reason = self.health_check()
         prerequisite.enabled = enabled
         prerequisite.disabled_reason = disabled_reason

holmes/plugins/toolsets/kubernetes_logs.yaml

@@ -1,7 +1,7 @@
 toolsets:
   kubernetes/logs:
     description: "Read pod logs"
-    docs_url: "https://docs.robusta.dev/master/configuration/holmesgpt/toolsets/kubernetes.html#logs"
+    docs_url: "https://holmesgpt.dev/data-sources/builtin-toolsets/kubernetes/"
     icon_url: "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRPKA-U9m5BxYQDF1O7atMfj9EMMXEoGu4t0Q&s"
     tags:
       - core

holmes/plugins/toolsets/logging_utils/logging_api.py

@@ -18,6 +18,7 @@ from holmes.plugins.toolsets.utils import get_param_or_raise
 DEFAULT_LOG_LIMIT = 100
 SECONDS_PER_DAY = 24 * 60 * 60
 DEFAULT_TIME_SPAN_SECONDS = 7 * SECONDS_PER_DAY  # 1 week in seconds
+DEFAULT_GRAPH_TIME_SPAN_SECONDS = 1 * SECONDS_PER_DAY  # 1 day in seconds
 
 POD_LOGGING_TOOL_NAME = "fetch_pod_logs"
 
@@ -27,6 +28,9 @@ class LoggingCapability(str, Enum):
 
     REGEX_FILTER = "regex_filter"  # If not supported, falls back to substring matching
     EXCLUDE_FILTER = "exclude_filter"  # If not supported, parameter is not shown at all
+    HISTORICAL_DATA = (
+        "historical_data"  # Can fetch logs for pods no longer in the cluster
+    )
 
 
 class LoggingConfig(BaseModel):
@@ -78,9 +82,16 @@ class PodLoggingTool(Tool):
         parameters = self._get_tool_parameters(toolset)
 
         # Build description based on capabilities
-        description = "Fetch logs for a Kubernetes pod"
+        # Include the toolset name in the description
+        toolset_name = toolset.name if toolset.name else "logging backend"
+        description = f"Fetch logs for a Kubernetes pod from {toolset_name}"
         capabilities = toolset.supported_capabilities
 
+        if LoggingCapability.HISTORICAL_DATA in capabilities:
+            description += (
+                " (including historical data for pods no longer in the cluster)"
+            )
+
         if (
             LoggingCapability.REGEX_FILTER in capabilities
             and LoggingCapability.EXCLUDE_FILTER in capabilities
@@ -164,7 +175,9 @@ If you hit the log limit and see lots of repetitive INFO logs, use exclude_filte
 
         return params
 
-    def _invoke(self, params: dict) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         structured_params = FetchPodLogsParams(
             namespace=get_param_or_raise(params, "namespace"),
             pod_name=get_param_or_raise(params, "pod_name"),

holmes/plugins/toolsets/mcp/toolset_mcp.py

@@ -24,7 +24,9 @@ class RemoteMCPTool(Tool):
     url: str
     headers: Optional[Dict[str, str]] = None
 
-    def _invoke(self, params: Dict) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         try:
             return asyncio.run(self._invoke_async(params))
         except Exception as e:

holmes/plugins/toolsets/newrelic.py

@@ -10,7 +10,7 @@ from holmes.core.tools import (
 )
 from pydantic import BaseModel
 from holmes.core.tools import StructuredToolResult, ToolResultStatus
-from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
+from holmes.plugins.toolsets.utils import get_param_or_raise, toolset_name_for_one_liner
 
 
 class BaseNewRelicTool(Tool):
@@ -37,7 +37,9 @@ class GetLogs(BaseNewRelicTool):
             toolset=toolset,
         )
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         def success(msg: Any) -> StructuredToolResult:
             return StructuredToolResult(
                 status=ToolResultStatus.SUCCESS,
@@ -115,7 +117,9 @@ class GetTraces(BaseNewRelicTool):
             toolset=toolset,
         )
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         def success(msg: Any) -> StructuredToolResult:
             return StructuredToolResult(
                 status=ToolResultStatus.SUCCESS,
@@ -130,7 +134,7 @@ class GetTraces(BaseNewRelicTool):
                 params=params,
             )
 
-        duration = params.get("duration")
+        duration = get_param_or_raise(params, "duration")
         trace_id = params.get("trace_id")
 
         if trace_id:

holmes/plugins/toolsets/opensearch/opensearch.py

@@ -93,7 +93,9 @@ class ListShards(BaseOpenSearchTool):
             toolset=toolset,
         )
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         client = get_client(self.toolset.clients, host=params.get("host", ""))
         shards = client.client.cat.shards()
         return StructuredToolResult(
@@ -122,7 +124,9 @@ class GetClusterSettings(BaseOpenSearchTool):
             toolset=toolset,
         )
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         client = get_client(self.toolset.clients, host=params.get("host"))
         response = client.client.cluster.get_settings(
             include_defaults=True, flat_settings=True
@@ -153,7 +157,9 @@ class GetClusterHealth(BaseOpenSearchTool):
             toolset=toolset,
         )
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         client = get_client(self.toolset.clients, host=params.get("host", ""))
         health = client.client.cluster.health()
         return StructuredToolResult(
@@ -176,7 +182,9 @@ class ListOpenSearchHosts(BaseOpenSearchTool):
             toolset=toolset,
         )
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         hosts = [host for client in self.toolset.clients for host in client.hosts]
         return StructuredToolResult(
             status=ToolResultStatus.SUCCESS,
@@ -197,7 +205,7 @@ class OpenSearchToolset(Toolset):
             name="opensearch/status",
             enabled=False,
             description="Provide cluster metadata information like health, shards, settings.",
-            docs_url="https://docs.robusta.dev/master/configuration/holmesgpt/toolsets/opensearch.html",
+            docs_url="https://holmesgpt.dev/data-sources/builtin-toolsets/opensearch-status/",
             icon_url="https://opensearch.org/assets/brand/PNG/Mark/opensearch_mark_default.png",
             prerequisites=[CallablePrerequisite(callable=self.prerequisites_callable)],
             tools=[

holmes/plugins/toolsets/opensearch/opensearch_logs.py

@@ -42,16 +42,16 @@ class OpenSearchLogsToolset(BasePodLoggingToolset):
         super().__init__(
             name="opensearch/logs",
             description="OpenSearch integration to fetch logs",
-            docs_url="https://docs.robusta.dev/master/configuration/holmesgpt/toolsets/opensearch_logs.html",
+            docs_url="https://holmesgpt.dev/data-sources/builtin-toolsets/opensearch-logs/",
             icon_url="https://opensearch.org/wp-content/uploads/2025/01/opensearch_mark_default.png",
             prerequisites=[CallablePrerequisite(callable=self.prerequisites_callable)],
-            tools=[
-                PodLoggingTool(self),
-            ],
+            tools=[],  # Initialize with empty tools first
             tags=[
                 ToolsetTag.CORE,
             ],
         )
+        # Now that parent is initialized and self.name exists, create the tool
+        self.tools = [PodLoggingTool(self)]
 
     def get_example_config(self) -> Dict[str, Any]:
         example_config = OpenSearchLoggingConfig(

holmes/plugins/toolsets/opensearch/opensearch_traces.py

@@ -1,7 +1,6 @@
 import os
 import logging
 
-from typing import Any, Dict
 
 import requests  # type: ignore
 from cachetools import TTLCache  # type: ignore
@@ -20,7 +19,7 @@ from holmes.plugins.toolsets.opensearch.opensearch_utils import (
     get_search_url,
 )
 from holmes.core.tools import StructuredToolResult, ToolResultStatus
-from holmes.plugins.toolsets.utils import toolset_name_for_one_liner
+from holmes.plugins.toolsets.utils import get_param_or_raise, toolset_name_for_one_liner
 
 TRACES_FIELDS_CACHE_KEY = "cached_traces_fields"
 
@@ -35,7 +34,9 @@ class GetTracesFields(Tool):
         self._toolset = toolset
         self._cache = None
 
-    def _invoke(self, params: Dict) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         try:
             if not self._cache and self._toolset.opensearch_config.fields_ttl_seconds:
                 self._cache = TTLCache(
@@ -128,10 +129,12 @@ class TracesSearchQuery(Tool):
         self._toolset = toolset
         self._cache = None
 
-    def _invoke(self, params: Any) -> StructuredToolResult:
+    def _invoke(
+        self, params: dict, user_approved: bool = False
+    ) -> StructuredToolResult:
         err_msg = ""
         try:
-            body = json.loads(params.get("query"))
+            body = json.loads(get_param_or_raise(params, "query"))
             full_query = body
             full_query["size"] = int(
                 os.environ.get("OPENSEARCH_TRACES_SEARCH_SIZE", "5000")
@@ -196,7 +199,7 @@ class OpenSearchTracesToolset(BaseOpenSearchToolset):
         super().__init__(
             name="opensearch/traces",
             description="OpenSearch integration to fetch traces",
-            docs_url="https://docs.robusta.dev/master/configuration/holmesgpt/toolsets/opensearch-traces.html",
+            docs_url="https://holmesgpt.dev/data-sources/builtin-toolsets/opensearch-status/",
             icon_url="https://opensearch.org/assets/brand/PNG/Mark/opensearch_mark_default.png",
             prerequisites=[CallablePrerequisite(callable=self.prerequisites_callable)],
             tools=[