From a60cd5f651566f9aae6c66ebcdedc59147d6b380 Mon Sep 17 00:00:00 2001
From: EfeDurmaz16 <efebarandurmaz05@gmail.com>
Date: Tue, 21 Jul 2026 00:29:46 +0300
Subject: [PATCH] fix(ui): emit safe errors for failed SSE streams

---
 .../ai.ui/ai-sdk/outbound-stream.mdx          | 25 +++++++++++++++
 src/ai/ui/ai_sdk/outbound_stream.py           | 23 +++++++++++---
 tests/ui/ai_sdk/test_outbound_stream.py       | 31 +++++++++++++++++++
 3 files changed, 75 insertions(+), 4 deletions(-)

diff --git a/docs/ai-python/content/docs/reference/ai.ui/ai-sdk/outbound-stream.mdx b/docs/ai-python/content/docs/reference/ai.ui/ai-sdk/outbound-stream.mdx
index 1271fbef..7c2afb1b 100644
--- a/docs/ai-python/content/docs/reference/ai.ui/ai-sdk/outbound-stream.mdx
+++ b/docs/ai-python/content/docs/reference/ai.ui/ai-sdk/outbound-stream.mdx
@@ -12,6 +12,31 @@ async for chunk in ai.ui.ai_sdk.to_sse(agent_stream):
     yield chunk
 ```
 
+If the agent stream raises, `to_sse` sends a safe `error` event and closes the
+stream with `[DONE]`. Pass `on_error` to log the exception or return a more
+specific client-facing message:
+
+```python
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def error_text(error: Exception) -> str:
+    logger.exception("Agent stream failed", exc_info=error)
+    return "The model is temporarily unavailable."
+
+
+async for chunk in ai.ui.ai_sdk.to_sse(
+    agent_stream,
+    on_error=error_text,
+):
+    yield chunk
+```
+
+Without `on_error`, the client receives `An error occurred.` so server-side
+details are not exposed.
+
 ```python
 async for part in ai.ui.ai_sdk.to_stream(agent_stream):
     yield part
diff --git a/src/ai/ui/ai_sdk/outbound_stream.py b/src/ai/ui/ai_sdk/outbound_stream.py
index f506aa80..1bbea4ba 100644
--- a/src/ai/ui/ai_sdk/outbound_stream.py
+++ b/src/ai/ui/ai_sdk/outbound_stream.py
@@ -16,7 +16,12 @@
 from .tool_utils import normalize_tool_input
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncGenerator, AsyncIterable
+    from collections.abc import AsyncGenerator, AsyncIterable, Callable
+
+
+def _default_error_text(_: Exception) -> str:
+    """Return a safe client-facing message for stream failures."""
+    return "An error occurred."
 
 
 def _tool_error_text(part: messages_.ToolResultPart) -> str:
@@ -621,8 +626,18 @@ async def to_stream(
 
 async def to_sse(
     events: AsyncIterable[events_.AgentEvent],
+    *,
+    on_error: Callable[[Exception], str] = _default_error_text,
 ) -> AsyncGenerator[str]:
-    """Convert an internal event stream into SSE strings."""
-    async for event in to_stream(events):
-        yield format_sse(event)
+    """Convert an internal event stream into SSE strings.
+
+    Stream failures become AI SDK UI ``error`` events.  The default message
+    avoids exposing server-side details; use ``on_error`` to log the exception
+    or return a more specific client-facing message.
+    """
+    try:
+        async for event in to_stream(events):
+            yield format_sse(event)
+    except Exception as error:
+        yield format_sse(ui_events.UIErrorEvent(error_text=on_error(error)))
     yield format_done_sse()
diff --git a/tests/ui/ai_sdk/test_outbound_stream.py b/tests/ui/ai_sdk/test_outbound_stream.py
index 02edd380..5eec3a4c 100644
--- a/tests/ui/ai_sdk/test_outbound_stream.py
+++ b/tests/ui/ai_sdk/test_outbound_stream.py
@@ -23,6 +23,11 @@ async def _gen(
         yield event
 
 
+async def _broken_gen() -> AsyncGenerator[agent_events_.AgentEvent]:
+    yield events_.TextStart(block_id="t1")
+    raise RuntimeError("sensitive provider detail")
+
+
 async def _collect(
     stream_events: list[agent_events_.AgentEvent],
 ) -> list[ui_events.UIMessageStreamEvent]:
@@ -102,6 +107,32 @@ async def test_to_sse_emits_data_prefixed_lines() -> None:
     assert lines[-1] == "data: [DONE]\n\n"
 
 
+async def test_to_sse_converts_stream_errors_to_safe_error_events() -> None:
+    lines = [line async for line in to_sse(_broken_gen())]
+
+    error = json.loads(lines[-2].removeprefix("data: ").rstrip())
+    assert error == {"type": "error", "errorText": "An error occurred."}
+    assert lines[-1] == "data: [DONE]\n\n"
+
+
+async def test_to_sse_supports_custom_error_messages() -> None:
+    errors: list[Exception] = []
+
+    def on_error(error: Exception) -> str:
+        errors.append(error)
+        return "The model is temporarily unavailable."
+
+    lines = [line async for line in to_sse(_broken_gen(), on_error=on_error)]
+
+    error = json.loads(lines[-2].removeprefix("data: ").rstrip())
+    assert error == {
+        "type": "error",
+        "errorText": "The model is temporarily unavailable.",
+    }
+    assert len(errors) == 1
+    assert str(errors[0]) == "sensitive provider detail"
+
+
 async def test_stream_start_uses_runtime_message_id() -> None:
     assistant = messages_.Message(
         id="assistant-runtime-id",