Skip to content

Commit c16f58e

Browse files
committed
fix: streaming
1 parent d76a748 commit c16f58e

File tree

1 file changed

+0
-38
lines changed

1 file changed

+0
-38
lines changed

src/openai/_streaming.py

-38
Original file line numberDiff line numberDiff line change
@@ -76,25 +76,6 @@ def __stream__(self) -> Iterator[_T]:
7676

7777
yield process_data(data=data, cast_to=cast_to, response=response)
7878

79-
else:
80-
data = sse.json()
81-
82-
if sse.event == "error" and is_mapping(data) and data.get("error"):
83-
message = None
84-
error = data.get("error")
85-
if is_mapping(error):
86-
message = error.get("message")
87-
if not message or not isinstance(message, str):
88-
message = "An error occurred during streaming"
89-
90-
raise APIError(
91-
message=message,
92-
request=self.response.request,
93-
body=data["error"],
94-
)
95-
96-
yield process_data(data={"data": data, "event": sse.event}, cast_to=cast_to, response=response)
97-
9879
# Ensure the entire stream is consumed
9980
for _sse in iterator:
10081
...
@@ -177,25 +158,6 @@ async def __stream__(self) -> AsyncIterator[_T]:
177158

178159
yield process_data(data=data, cast_to=cast_to, response=response)
179160

180-
else:
181-
data = sse.json()
182-
183-
if sse.event == "error" and is_mapping(data) and data.get("error"):
184-
message = None
185-
error = data.get("error")
186-
if is_mapping(error):
187-
message = error.get("message")
188-
if not message or not isinstance(message, str):
189-
message = "An error occurred during streaming"
190-
191-
raise APIError(
192-
message=message,
193-
request=self.response.request,
194-
body=data["error"],
195-
)
196-
197-
yield process_data(data={"data": data, "event": sse.event}, cast_to=cast_to, response=response)
198-
199161
# Ensure the entire stream is consumed
200162
async for _sse in iterator:
201163
...

0 commit comments

Comments
 (0)