@@ -147,8 +147,9 @@ def request(
147
147
if stream != got_stream :
148
148
logger .error ("Unexpected response: %s" , resp )
149
149
raise RuntimeError (
150
- "Streaming response is expected, but got non-streaming response. This may indicate a bug in erniebot."
151
- )
150
+ f"A { 'streamed' if stream else 'non-streamed' } response was expected, "
151
+ f"but got a { 'streamed' if got_stream else 'non-streamed' } response. "
152
+ "This may indicate a bug in erniebot." )
152
153
return resp
153
154
154
155
@overload
@@ -226,8 +227,9 @@ async def arequest(
226
227
if stream != got_stream :
227
228
logger .error ("Unexpected response: %s" , resp )
228
229
raise RuntimeError (
229
- "Streaming response is expected, but got non-streaming response. This may indicate a bug in erniebot."
230
- )
230
+ f"A { 'streamed' if stream else 'non-streamed' } response was expected, "
231
+ f"but got a { 'streamed' if got_stream else 'non-streamed' } response. "
232
+ "This may indicate a bug in erniebot." )
231
233
except Exception as e :
232
234
await ctx .__aexit__ (None , None , None )
233
235
raise e
@@ -300,7 +302,6 @@ def request_raw(
300
302
raise errors .ConnectionError (
301
303
f"Error communicating with server: { e } " ) from e
302
304
303
- logger .debug ("API response body: %r" , result .content )
304
305
logger .debug ("API response headers: %r" , result .headers )
305
306
306
307
return result
@@ -340,7 +341,6 @@ async def arequest_raw(
340
341
raise errors .ConnectionError (
341
342
f"Error communicating with server: { e } " ) from e
342
343
343
- logger .debug ("API response body: %r" , result .content )
344
344
logger .debug ("API response headers: %r" , result .headers )
345
345
346
346
return result
@@ -406,10 +406,11 @@ def _validate_headers(
406
406
def _parse_line (self , line : bytes ) -> Optional [str ]:
407
407
if line :
408
408
if line .startswith (b"data: " ):
409
+ # Data-only messages
409
410
line = line [len (b"data: " ):]
410
411
return line .decode ('utf-8' )
411
412
else :
412
- # Filter out empty lines
413
+ # Filter out other lines
413
414
return None
414
415
return None
415
416
0 commit comments