Update src/mcp.rs

src/mcp.rs

@@ -3,6 +3,7 @@ use crate::parser::{DocumentParser, ParsedDocument};
 use anyhow::Result;
 use axum::{
     extract::State,
+    http::{header, HeaderMap, HeaderValue},
     response::sse::{Event, KeepAlive, Sse},
     routing::{get, post},
     Json, Router,
@@ -344,7 +345,12 @@ async fn handle_http_rpc(
 
 async fn handle_sse(
     State(_server): State<Arc<McpServer>>,
-) -> Sse<impl Stream<Item = Result<Event, Infallible>>> {
+) -> (HeaderMap, Sse<impl Stream<Item = Result<Event, Infallible>>>) {
+    let mut headers = HeaderMap::new();
+    headers.insert("x-accel-buffering", HeaderValue::from_static("no"));
+    headers.insert(header::CACHE_CONTROL, HeaderValue::from_static("no-cache"));
+    headers.insert(header::CONNECTION, HeaderValue::from_static("keep-alive"));
+
     let initial = tokio_stream::iter(vec![
         Ok(Event::default().event("endpoint").data("/rpc")),
     ]);
@@ -354,5 +360,5 @@ async fn handle_sse(
 
     let stream = initial.chain(interval);
 
-    Sse::new(stream).keep_alive(KeepAlive::default())
+    (headers, Sse::new(stream).keep_alive(KeepAlive::default()))
 }