From 3dcad57481a2dcc97694692d83806d009167d4ac Mon Sep 17 00:00:00 2001
From: Ruben Fiszel <ruben@windmill.dev>
Date: Sun, 9 Nov 2025 10:53:37 +0000
Subject: [PATCH] add debug_sse_stream

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

diff --git a/backend/windmill-worker/src/ai/sse.rs b/backend/windmill-worker/src/ai/sse.rs
index fcafc9cd7e..42c5d73603 100644
--- a/backend/windmill-worker/src/ai/sse.rs
+++ b/backend/windmill-worker/src/ai/sse.rs
@@ -41,6 +41,12 @@ pub struct OpenAISSEEvent {
     pub choices: Option<Vec<OpenAIChoice>>,
 }
 
+lazy_static::lazy_static! {
+    static ref DEBUG_SSE_STREAM: bool = std::env::var("DEBUG_SSE_STREAM")
+        .unwrap_or("false".to_string())
+        .parse::<bool>()
+        .unwrap_or(false);
+}
 pub trait SSEParser {
     async fn parse_event_data(&mut self, data: &str) -> Result<(), Error>;
 
@@ -54,6 +60,9 @@ pub trait SSEParser {
 
             // Convert chunk to string and add to buffer
             let chunk_str = String::from_utf8_lossy(&chunk);
+            if *DEBUG_SSE_STREAM {
+                tracing::info!("SSE chunk: {}", chunk_str);
+            }
             buffer.push_str(&chunk_str);
 
             // Process complete lines from buffer