Merge branch 'main' of github.com:NZX/extra

sebastien · Sep 30, 2024 · 54c3a10 · 54c3a10
2 parents 267d2c1 + f0b14d3
commit 54c3a10
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/src/py/extra/client.py b/src/py/extra/client.py
@@ -434,21 +434,19 @@ async def OnRequest(
                 if atom is HTTPProcessingStatus.Complete:
                     status = atom
                 elif isinstance(atom, HTTPResponse):
+                    if atom.body:
+                        yield atom.body
                     res = atom
-                    if res.body:
-                        yield res.body
                     break
                 else:
                     yield atom
             iteration += 1
         if (
-            # We continue if we have streaming
-            streaming is True
-            or res
-            and res.body
-            and HTTPBody.HasRemaining(res.body)
-            or res
-            and res.headers.contentType in {"text/event-stream"}
+            # We continue if we have streaming or
+            status is HTTPProcessingStatus.Processing
+            or streaming is True
+            or (res and res.body and HTTPBody.HasRemaining(res.body))
+            or (res and res.headers.contentType in {"text/event-stream"})
         ):
             # TODO: We should swap out the body for a streaming body
             cxn.isStreaming = True