Skip to content

Commit 2d21186

Browse files
committed
fixup!
1 parent a3860ff commit 2d21186

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/openai/_streaming.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -59,9 +59,9 @@ def __stream__(self) -> Iterator[_T]:
5959
if sse.data.startswith("[DONE]"):
6060
break
6161

62-
data = sse.json()
63-
if sse.event is not None and sse.event == "error":
64-
if sse.event == "error" and is_mapping(data) and data.get("error"):
62+
if sse.event is not None and not sse.event.startswith("thread."):
63+
data = sse.json()
64+
if is_mapping(data) and data.get("error"):
6565
message = None
6666
error = data.get("error")
6767
if is_mapping(error):
@@ -74,11 +74,11 @@ def __stream__(self) -> Iterator[_T]:
7474
request=self.response.request,
7575
body=data["error"],
7676
)
77-
elif sse.event is not None and sse.event.startswith("thread."):
78-
# have to manually create part of the event since we don't have a full event
79-
yield process_data(data={"data": data, "event": sse.event}, cast_to=cast_to, response=response)
77+
78+
yield process_data(data=data, cast_to=cast_to, response=response)
8079
else:
81-
if is_mapping(data) and data.get("error"):
80+
data = sse.json()
81+
if sse.event == "error" and is_mapping(data) and data.get("error"):
8282
message = None
8383
error = data.get("error")
8484
if is_mapping(error):
@@ -91,8 +91,8 @@ def __stream__(self) -> Iterator[_T]:
9191
request=self.response.request,
9292
body=data["error"],
9393
)
94-
95-
yield process_data(data=data, cast_to=cast_to, response=response)
94+
# have to manually create part of the event since we don't have a full event
95+
yield process_data(data={"data": data, "event": sse.event}, cast_to=cast_to, response=response)
9696

9797
# Ensure the entire stream is consumed
9898
for _sse in iterator:

0 commit comments

Comments
 (0)