ROB-175 add keep alive headers (#1766)

* add keep alive headers

* use sse format for errors

Author: RoiGlinik

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

@@ -64,9 +64,12 @@ def ask_holmes(event: ExecutionBaseEvent, params: AIInvestigateParams):
         )
 
         if params.stream:
-            with requests.post(f"{holmes_url}/api/stream/investigate", data=holmes_req.json(), stream=True) as resp:
+            with requests.post(f"{holmes_url}/api/stream/investigate", data=holmes_req.json(), stream=True, headers={"Connection": "keep-alive"}) as resp:
+                resp.raise_for_status()
                 for line in resp.iter_content(chunk_size=None, decode_unicode=True): # Avoid streaming chunks from holmes. send them as they arrive.
-                    event.ws(data=line)
+                    if line:
+                        event.ws(data=line)
+
             return
 
         else:

src/robusta/integrations/receiver.py

@@ -195,7 +195,7 @@ def __exec_external_stream_request(self, action_request: ExternalActionRequest,
         res = self.event_handler.run_external_stream_action(action_request.body.action_name,
                                                             action_request.body.action_params,
                                                             lambda data: self.__stream_response(request_id=action_request.request_id, data=data))
-        res = "" if res.get("success") else json.dumps(res)
+        res = "" if res.get("success") else f"event: error\ndata: {json.dumps(res)}\n\n"
         self.__close_stream_response(action_request.request_id, res)
 
     def _process_action(self, action: ExternalActionRequest, validate_timestamp: bool) -> None: