App diagnose chat

helm/robusta/values.yaml

@@ -90,6 +90,7 @@ lightActions:
 - holmes_conversation
 - holmes_issue_chat
 - holmes_chat
+- holmes_workload_chat
 
 # install prometheus, alert-manager, and grafana along with Robusta?
 enablePrometheusStack: false

src/robusta/core/model/base_params.py

@@ -236,6 +236,20 @@ class HolmesWorkloadHealthParams(HolmesParams):
     silent_healthy: bool = False
 
 
+class HolmesWorkloadHealthChatParams(HolmesParams):
+    """
+    :var ask: User's prompt for holmes
+    :var workload_health_result: Result from the workload health check
+    :var resource: The resource related to the initial investigation
+    :var conversation_history: List of previous user prompts and responses.
+    """
+
+    ask: str
+    workload_health_result: HolmesInvestigationResult
+    resource: ResourceInfo
+    conversation_history: Optional[list[dict]] = None
+
+
 class NamespacedResourcesParams(ActionParams):
     """
     :var name: Resource name

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

@@ -10,6 +10,7 @@
     HolmesIssueChatParams,
     HolmesWorkloadHealthParams,
     ResourceInfo,
+    HolmesWorkloadHealthChatParams
 )
 from robusta.core.model.events import ExecutionBaseEvent
 from robusta.core.playbooks.actions_registry import action
@@ -26,6 +27,7 @@
     HolmesRequest,
     HolmesResult,
     HolmesResultsBlock,
+    HolmesWorkloadHealthRequest,
 )
 from robusta.core.schedule.model import FixedDelayRepeat
 from robusta.integrations.kubernetes.autogenerated.events import KubernetesAnyChangeEvent
@@ -354,3 +356,55 @@ def holmes_chat(event: ExecutionBaseEvent, params: HolmesChatParams):
             raise ActionException(ErrorCodes.HOLMES_REQUEST_ERROR, "Holmes internal configuration error.")
         else:
             raise ActionException(ErrorCodes.HOLMES_UNEXPECTED_ERROR, "An unexpected error occured.")
+
+
+@action
+def holmes_workload_chat(event: ExecutionBaseEvent, params: HolmesWorkloadHealthChatParams):
+    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 = HolmesWorkloadHealthRequest(
+            ask=params.ask,
+            conversation_history=params.conversation_history,
+            workload_health_result=params.workload_health_result,
+            resource=params.resource
+        )
+        result = requests.post(f"{holmes_url}/api/workload_health_chat", data=holmes_req.json())
+        result.raise_for_status()
+
+        holmes_result = HolmesChatResult(**json.loads(result.text))
+
+        finding = Finding(
+            title=f"AI Chat for Health Check of {params.resource}",
+            aggregation_key="HolmesWorkloadConversationResult",
+            subject=FindingSubject(
+                name=params.resource.name if params.resource else "",
+                namespace=params.resource.namespace if params.resource else "",
+                subject_type=FindingSubjectType.from_kind(params.resource.kind)
+                if params.resource
+                else FindingSubjectType.TYPE_NONE,
+                node=params.resource.node if params.resource else "",
+                container=params.resource.container if params.resource else "",
+            ),
+            finding_type=FindingType.AI_ANALYSIS,
+            failure=False,
+        )
+        finding.add_enrichment(
+            [HolmesChatResultsBlock(holmes_result=holmes_result)], enrichment_type=EnrichmentType.ai_analysis
+        )
+
+        event.add_finding(finding)
+
+    except Exception as e:
+        logging.exception(f"Failed to get holmes chat for health check of {params.resource}", exc_info=True)
+        if isinstance(e, requests.ConnectionError):
+            raise ActionException(ErrorCodes.HOLMES_CONNECTION_ERROR, "Holmes endpoint is currently unreachable.")
+        elif isinstance(e, requests.HTTPError):
+            if e.response.status_code == 401 and "invalid_api_key" in e.response.text:
+                raise ActionException(ErrorCodes.HOLMES_REQUEST_ERROR, "Holmes invalid api key.")
+
+            raise ActionException(ErrorCodes.HOLMES_REQUEST_ERROR, "Holmes internal configuration error.")
+        else:
+            raise ActionException(ErrorCodes.HOLMES_UNEXPECTED_ERROR, "An unexpected error occured.")

src/robusta/core/reporting/holmes.py

@@ -72,3 +72,8 @@ class HolmesChatResult(BaseModel):
 
 class HolmesChatResultsBlock(BaseBlock):
     holmes_result: Optional[HolmesChatResult]
+
+
+class HolmesWorkloadHealthRequest(HolmesChatRequest):
+    workload_health_result: HolmesInvestigationResult
+    resource: ResourceInfo