[Change] client: outputs the body when request, continues when large …

…body
NZX · Sep 30, 2024 · f0b14d3 · f0b14d3
1 parent d7f2ad7
commit f0b14d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/py/extra/client.py b/src/py/extra/client.py
@@ -436,14 +436,16 @@ async def OnRequest(
                     status = atom
                 elif isinstance(atom, HTTPResponse):
                     res = atom
+                    yield atom.body
                     break
                 else:
                     yield atom
             iteration += 1
         if (
+            status is HTTPProcessingStatus.Processing,
             streaming is True
             or res
-            and res.headers.contentType in {"text/event-stream"}
+            and res.headers.contentType in {"text/event-stream"},
         ):
             # TODO: We should swap out the body for a streaming body
             cxn.isStreaming = True