Skip to content

Commit ca1812f

Browse files
committed
fix pr
1 parent c663707 commit ca1812f

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

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

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,11 +69,18 @@ def ask_holmes(event: ExecutionBaseEvent, params: AIInvestigateParams):
6969
)
7070

7171
if params.stream:
72-
with requests.post(f"{holmes_url}/api/stream/investigate", data=holmes_req.json(), stream=True) as resp:
72+
with requests.post(
73+
f"{holmes_url}/api/stream/investigate",
74+
data=holmes_req.json(),
75+
stream=True,
76+
headers={"Connection": "keep-alive"},
77+
) as resp:
78+
resp.raise_for_status()
7379
for line in resp.iter_content(
7480
chunk_size=None, decode_unicode=True
7581
): # Avoid streaming chunks from holmes. send them as they arrive.
76-
event.ws(data=line)
82+
if line:
83+
event.ws(data=line)
7784
return
7885

7986
else:

0 commit comments

Comments
 (0)