ROB-3570: Add holmes_validate_toolset action for toolset validation

helm/robusta/values.yaml

@@ -97,6 +97,8 @@ lightActions:
 - holmes_conversation
 - holmes_issue_chat
 - holmes_chat
+- holmes_validate_toolset
+- holmes_refresh_toolsets
 - list_pods
 - kubectl_describe
 - fetch_resource_yaml

src/robusta/core/model/base_params.py

@@ -232,6 +232,20 @@ class HolmesConversationParams(HolmesParams):
     include_tool_call_results: bool = True
 
 
+class HolmesValidateToolsetParams(HolmesParams):
+    """
+    :var yaml_config: Raw YAML string containing the toolset configuration, including the 'holmes:' wrapper.
+    """
+
+    yaml_config: str
+
+
+class HolmesRefreshToolsetsParams(HolmesParams):
+    """Params for triggering a toolset status refresh via Holmes."""
+
+    pass
+
+
 class NamespacedResourcesParams(ActionParams):
     """
     :var name: Resource name

src/robusta/core/playbooks/internal/ai_integration.py

@@ -12,6 +12,8 @@
     HolmesChatParams,
     HolmesConversationParams,
     HolmesIssueChatParams,
+    HolmesRefreshToolsetsParams,
+    HolmesValidateToolsetParams,
     ResourceInfo,
 )
 from robusta.core.model.events import ExecutionBaseEvent
@@ -33,6 +35,8 @@
     HolmesRequest,
     HolmesResult,
     HolmesResultsBlock,
+    HolmesValidateToolsetRequest,
+    HolmesValidateToolsetResponse,
 )
 from robusta.core.reporting.utils import convert_svg_to_png
 from robusta.core.stream.utils import (
@@ -385,6 +389,48 @@ def holmes_chat(event: ExecutionBaseEvent, params: HolmesChatParams):
         handle_holmes_error(e)
 
 
+@action
+def holmes_validate_toolset(event: ExecutionBaseEvent, params: HolmesValidateToolsetParams):
+    holmes_url = HolmesDiscovery.find_holmes_url(params.holmes_url)
+    if not holmes_url:
+        raise ActionException(
+            ErrorCodes.HOLMES_DISCOVERY_FAILED,
+            "Robusta couldn't connect to the Holmes client.",
+        )
+
+    try:
+        holmes_req = HolmesValidateToolsetRequest(yaml_config=params.yaml_config)
+        url = f"{holmes_url}/api/toolsets/validate"
+        result = requests.post(url, data=holmes_req.json())
+        result.raise_for_status()
+        holmes_response = HolmesValidateToolsetResponse(**json.loads(result.text))
+        event.response = {"success": True, **holmes_response.dict()}
+
+    except Exception as e:
+        logging.exception("Failed to validate toolset via Holmes", exc_info=True)
+        handle_holmes_error(e)
+
+
+@action
+def holmes_refresh_toolsets(event: ExecutionBaseEvent, params: HolmesRefreshToolsetsParams):
+    holmes_url = HolmesDiscovery.find_holmes_url(params.holmes_url)
+    if not holmes_url:
+        raise ActionException(
+            ErrorCodes.HOLMES_DISCOVERY_FAILED,
+            "Robusta couldn't connect to the Holmes client.",
+        )
+
+    try:
+        url = f"{holmes_url}/api/toolsets/refresh"
+        result = requests.post(url)
+        result.raise_for_status()
+        event.response = {"success": True}
+
+    except Exception as e:
+        logging.exception("Failed to refresh toolsets via Holmes", exc_info=True)
+        handle_holmes_error(e)
+
+
 def stream_and_render_graphs(url, holmes_req, event):
     with requests.post(
         url,

src/robusta/core/reporting/holmes.py

@@ -93,3 +93,20 @@ class HolmesChatResult(BaseModel):
 
 class HolmesChatResultsBlock(BaseBlock):
     holmes_result: Optional[HolmesChatResult]
+
+
+class HolmesValidateToolsetRequest(BaseModel):
+    yaml_config: str = Field(
+        description="Raw YAML string containing the toolset configuration, including the 'holmes:' wrapper."
+    )
+
+
+class HolmesValidateToolsetResult(BaseModel):
+    toolset_name: str
+    status: str = Field(description="Toolset status: 'valid' or 'invalid'")
+    error: Optional[str] = None
+    description: Optional[str] = None
+
+
+class HolmesValidateToolsetResponse(BaseModel):
+    results: List[HolmesValidateToolsetResult]