File tree 1 file changed +16
-2
lines changed
1 file changed +16
-2
lines changed Original file line number Diff line number Diff line change @@ -65,8 +65,15 @@ def __stream__(self) -> Iterator[_T]:
65
65
if sse .event is None :
66
66
data = sse .json ()
67
67
if is_mapping (data ) and data .get ("error" ):
68
+ message = None
69
+ error = data .get ("error" )
70
+ if is_mapping (error ):
71
+ message = error .get ("message" )
72
+ if not message or not isinstance (message , str ):
73
+ message = "An error occurred during streaming"
74
+
68
75
raise APIError (
69
- message = "An error occurred during streaming" ,
76
+ message = message ,
70
77
request = self .response .request ,
71
78
body = data ["error" ],
72
79
)
@@ -145,8 +152,15 @@ async def __stream__(self) -> AsyncIterator[_T]:
145
152
if sse .event is None :
146
153
data = sse .json ()
147
154
if is_mapping (data ) and data .get ("error" ):
155
+ message = None
156
+ error = data .get ("error" )
157
+ if is_mapping (error ):
158
+ message = error .get ("message" )
159
+ if not message or not isinstance (message , str ):
160
+ message = "An error occurred during streaming"
161
+
148
162
raise APIError (
149
- message = "An error occurred during streaming" ,
163
+ message = message ,
150
164
request = self .response .request ,
151
165
body = data ["error" ],
152
166
)
You can’t perform that action at this time.
0 commit comments