From 60da45a6ee8deb487450f2e9eff3ade613dc2c7c Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Mon, 18 May 2026 18:35:52 +0000 Subject: [PATCH] fix(llm_gateway): use envoy body_size for response body replacement Co-authored-by: Musa --- crates/llm_gateway/src/stream_context.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/crates/llm_gateway/src/stream_context.rs b/crates/llm_gateway/src/stream_context.rs index 969dc266..0453afe5 100644 --- a/crates/llm_gateway/src/stream_context.rs +++ b/crates/llm_gateway/src/stream_context.rs @@ -1206,7 +1206,7 @@ impl HttpContext for StreamContext { String::from_utf8_lossy(&body) ); // Forward the error response as-is - let replace_size = if self.streaming_response { + let replace_size = if body_size > 0 { body_size } else { body.len() @@ -1248,7 +1248,7 @@ impl HttpContext for StreamContext { return Action::Continue; } - let replace_size = if self.streaming_response { + let replace_size = if body_size > 0 { body_size } else { body.len()