From 5be3feaaa2dacf452372d4e3fa2c2e9f15243f0d Mon Sep 17 00:00:00 2001
From: centdix <40307056+centdix@users.noreply.github.com>
Date: Mon, 26 Jan 2026 12:18:43 +0100
Subject: [PATCH] fix(aiagent) break on consecutive errrors (#7681)

---
 backend/windmill-worker/src/ai/sse.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/backend/windmill-worker/src/ai/sse.rs b/backend/windmill-worker/src/ai/sse.rs
index db5fe81a68..4c08a2e40c 100644
--- a/backend/windmill-worker/src/ai/sse.rs
+++ b/backend/windmill-worker/src/ai/sse.rs
@@ -55,10 +55,12 @@ pub trait SSEParser {
 
     async fn parse_events(&mut self, response: Response) -> Result<(), Error> {
         let mut stream = response.bytes_stream().eventsource();
+        let mut consecutive_errors = 0;
 
         while let Some(event) = stream.next().await {
             match event {
                 Ok(event) => {
+                    consecutive_errors = 0;
                     if *DEBUG_SSE_STREAM {
                         tracing::info!("SSE event: {:?}", event);
                     }
@@ -66,7 +68,15 @@ pub trait SSEParser {
                     self.parse_event_data(&event.data).await?;
                 }
                 Err(e) => {
+                    consecutive_errors += 1;
                     tracing::error!("Failed to parse SSE event: {}", e);
+                    if consecutive_errors >= 5 {
+                        tracing::error!(
+                            "Breaking SSE stream after {} consecutive parsing errors",
+                            consecutive_errors
+                        );
+                        break;
+                    }
                 }
             }
         }