Commit 7e88d422

Robert Craigie <robert@craigie.dev>
2025-01-14 00:59:43
fix: streaming
1 parent babe65f
Changed files (1)
src
src/openai/_streaming.py
@@ -76,25 +76,6 @@ class Stream(Generic[_T]):
 
             yield process_data(data=data, cast_to=cast_to, response=response)
 
-            else:
-                data = sse.json()
-
-                if sse.event == "error" and is_mapping(data) and data.get("error"):
-                    message = None
-                    error = data.get("error")
-                    if is_mapping(error):
-                        message = error.get("message")
-                    if not message or not isinstance(message, str):
-                        message = "An error occurred during streaming"
-
-                    raise APIError(
-                        message=message,
-                        request=self.response.request,
-                        body=data["error"],
-                    )
-
-                yield process_data(data={"data": data, "event": sse.event}, cast_to=cast_to, response=response)
-
         # Ensure the entire stream is consumed
         for _sse in iterator:
             ...
@@ -177,25 +158,6 @@ class AsyncStream(Generic[_T]):
 
             yield process_data(data=data, cast_to=cast_to, response=response)
 
-            else:
-                data = sse.json()
-
-                if sse.event == "error" and is_mapping(data) and data.get("error"):
-                    message = None
-                    error = data.get("error")
-                    if is_mapping(error):
-                        message = error.get("message")
-                    if not message or not isinstance(message, str):
-                        message = "An error occurred during streaming"
-
-                    raise APIError(
-                        message=message,
-                        request=self.response.request,
-                        body=data["error"],
-                    )
-
-                yield process_data(data={"data": data, "event": sse.event}, cast_to=cast_to, response=response)
-
         # Ensure the entire stream is consumed
         async for _sse in iterator:
             ...