From 14a339fcc858572c7d1d2a4d0530893b45d9c76e Mon Sep 17 00:00:00 2001 From: vikram singh shekhawat Date: Tue, 11 Nov 2025 23:33:50 +0530 Subject: [PATCH] [Test] Handle streaming chunks with null content in case of stream end. (#10862) Co-authored-by: svc_repro_tool --- test/srt/test_sagemaker_server.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/srt/test_sagemaker_server.py b/test/srt/test_sagemaker_server.py index 81ab9790c..7f54b0919 100644 --- a/test/srt/test_sagemaker_server.py +++ b/test/srt/test_sagemaker_server.py @@ -128,6 +128,10 @@ class TestSageMakerServer(CustomTestCase): is_firsts[index] = False continue + # Skip chunks that are just empty placeholders, usually at stream end/stop + if data.get("content") is None: + continue + if logprobs: assert line.get("choices")[0].get("logprobs") assert isinstance(