From a1390dac859c733da543b7d86f9396b1606ec3a5 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Tue, 12 Nov 2024 10:56:22 -0800
Subject: [PATCH] fix int tests

---
 crates/prompt_gateway/src/http_context.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/prompt_gateway/src/http_context.rs b/crates/prompt_gateway/src/http_context.rs
index 2ef10025..3174a597 100644
--- a/crates/prompt_gateway/src/http_context.rs
+++ b/crates/prompt_gateway/src/http_context.rs
@@ -264,7 +264,7 @@ impl HttpContext for StreamContext {
             );
         }
 
-        if end_of_stream {
+        if end_of_stream && body_size == 0 {
             if let Some(traceparent) = self.traceparent.as_ref() {
                 let since_the_epoch_ns = SystemTime::now()
                     .duration_since(UNIX_EPOCH)