From 01a5fba937fa9d1ad3035b8cf7bda0a7f3a10ad2 Mon Sep 17 00:00:00 2001
From: MengMa <zhangmeng889@sina.com>
Date: Sun, 27 Sep 2026 18:08:37 +0800
Subject: [PATCH] test(proxy): cover late SSE dict payloads

Assert a dict, a Response body, and a dict body attribute all become SSE data frames.
---
 .../proxy/common_utils/test_sse_keepalive.py  | 39 ++++++++++++++++++-
 1 file changed, 38 insertions(+), 1 deletion(-)

diff --git a/tests/test_litellm/proxy/common_utils/test_sse_keepalive.py b/tests/test_litellm/proxy/common_utils/test_sse_keepalive.py
index 228fd5bcae6..b754b31762d 100644
--- a/tests/test_litellm/proxy/common_utils/test_sse_keepalive.py
+++ b/tests/test_litellm/proxy/common_utils/test_sse_keepalive.py
@@ -3,7 +3,7 @@ from collections.abc import AsyncGenerator
 from typing import Final, cast
 
 import pytest
-from fastapi.responses import StreamingResponse
+from fastapi.responses import Response, StreamingResponse
 
 from litellm.proxy.common_request_processing import create_response
 from litellm.types.utils import ModelResponse
@@ -408,3 +408,40 @@ def test_ttft_interval_resolves_through_the_deployments_it_could_land_on(
     deployments, global_interval, expected, why
 ):
     assert resolve_ttft_keepalive_interval(deployments, global_interval) == expected, why
+
+
+@pytest.mark.asyncio
+async def test_relay_late_response_serializes_a_dict_payload_as_sse():
+    from litellm.proxy.common_request_processing import _relay_late_response
+
+    frames: Final = [chunk async for chunk in _relay_late_response({"error": {"message": "late"}})]
+
+    assert frames == [
+        b'data: {"error":{"message":"late"}}\n\n',
+        b"data: [DONE]\n\n",
+    ]
+
+
+@pytest.mark.asyncio
+async def test_relay_late_response_keeps_a_response_body():
+    from litellm.proxy.common_request_processing import _relay_late_response
+
+    frames: Final = [chunk async for chunk in _relay_late_response(Response(content=b'{"ok":true}'))]
+
+    assert frames == [
+        b'data: {"ok":true}\n\n',
+        b"data: [DONE]\n\n",
+    ]
+
+
+@pytest.mark.asyncio
+async def test_relay_late_response_serializes_a_dict_body_attribute():
+    from litellm.proxy.common_request_processing import _relay_late_response
+
+    class _DictBody:
+        body = {"ok": True}
+
+    frames: Final = [chunk async for chunk in _relay_late_response(_DictBody())]
+
+    assert frames[0] == b'data: {"ok":true}\n\n'
+    assert frames[1] == b"data: [DONE]\n\n"