workers-runtime-sdk 1.1.0__tar.gz → 1.1.1__tar.gz

{workers_runtime_sdk-1.1.0 → workers_runtime_sdk-1.1.1}/CHANGELOG.md

@@ -2,6 +2,18 @@
 
 <!-- version list -->
 
+## v1.1.1 (2026-03-18)
+
+### Bug Fixes
+
+- Fix Python ASGI adaptor to handle streaming responses correctly
+  ([#82](https://github.com/cloudflare/workers-py/pull/82),
+  [`d3ea87a`](https://github.com/cloudflare/workers-py/commit/d3ea87aff37c7a833f0602cc2a8018f1d5dde91b))
+
+- Fix streaming responses in asgi module ([#82](https://github.com/cloudflare/workers-py/pull/82),
+  [`d3ea87a`](https://github.com/cloudflare/workers-py/commit/d3ea87aff37c7a833f0602cc2a8018f1d5dde91b))
+
+
 ## v1.1.0 (2026-03-12)
 
 ### Features

{workers_runtime_sdk-1.1.0 → workers_runtime_sdk-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workers-runtime-sdk
-Version: 1.1.0
+Version: 1.1.1
 Summary: Python SDK for Cloudflare Workers
 Project-URL: Homepage, https://github.com/cloudflare/workers-py
 Project-URL: Bug Tracker, https://github.com/cloudflare/workers-py/issues

{workers_runtime_sdk-1.1.0 → workers_runtime_sdk-1.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "workers-runtime-sdk"
-version = "1.1.0"
+version = "1.1.1"
 description = "Python SDK for Cloudflare Workers"
 readme = "README.md"
 requires-python = ">=3.12"

{workers_runtime_sdk-1.1.0 → workers_runtime_sdk-1.1.1}/src/asgi.py

@@ -7,7 +7,7 @@ from typing import Any
 
 import js
 
-from workers import Context, Request
+from workers import Context, Request, wait_until
 
 ASGI = {"spec_version": "2.0", "version": "3.0"}
 logger = logging.getLogger("asgi")
@@ -116,7 +116,12 @@ async def start_application(app):
 
 
 async def process_request(
-    app: Any, req: "Request | js.Request", env: Any, ctx: Context
+    app: Any,
+    req: "Request | js.Request",
+    env: Any,
+    # added for waitUntil, but not used anymore
+    # TODO(later): remove this parameter after unvendoring Python SDK from workerd
+    ctx: Context | None,
 ) -> js.Response:
     from js import Object, Response, TransformStream
     from pyodide.ffi import create_proxy
@@ -124,9 +129,11 @@ async def process_request(
     status = None
     headers = None
     result = Future()
-    is_sse = False
     finished_response = Event()
 
+    # Streaming state — initialized lazily on first body chunk with more_body=True.
+    writer = None
+
     receive_queue = Queue()
     if req.body:
         async for data in req.body:
@@ -148,60 +155,52 @@ async def process_request(
             message = {"type": "http.disconnect"}
         return message
 
-    # Create a transform stream for handling streaming responses
-    transform_stream = TransformStream.new()
-    readable = transform_stream.readable
-    writable = transform_stream.writable
-    writer = writable.getWriter()
-
     async def send(got):
         nonlocal status
         nonlocal headers
-        nonlocal is_sse
+        nonlocal writer
 
         if got["type"] == "http.response.start":
             status = got["status"]
             # Like above, we need to convert byte-pairs into string explicitly.
             headers = [(k.decode(), v.decode()) for k, v in got["headers"]]
-            # Check if this is a server-sent events response
-            for k, v in headers:
-                if k.lower() == "content-type" and v.lower().startswith(
-                    "text/event-stream"
-                ):
-                    is_sse = True
-                    break
-            if is_sse:
-                # For SSE, create and return the response immediately after http.response.start
-                resp = Response.new(
-                    readable, headers=Object.fromEntries(headers), status=status
-                )
-                result.set_result(resp)
 
         elif got["type"] == "http.response.body":
             body = got["body"]
             more_body = got.get("more_body", False)
 
-            # Convert body to JS buffer
-            px = create_proxy(body)
-            buf = px.getBuffer()
-            px.destroy()
-
-            if is_sse:
-                # For SSE, write chunk to the stream
-                await writer.write(buf.data)
-                # If this is the last chunk, close the writer
+            if writer is not None:
+                # Already in streaming mode — write chunk to the stream.
+                with acquire_js_buffer(body) as jsbytes:
+                    await writer.write(jsbytes.slice())
                 if not more_body:
                     await writer.close()
                     finished_response.set()
+            elif more_body:
+                # First body chunk with more data coming — switch to streaming.
+                # Create a TransformStream so the runtime can start consuming
+                # body chunks as they are written.
+                transform_stream = TransformStream.new()
+                readable = transform_stream.readable
+                writer = transform_stream.writable.getWriter()
+                resp = Response.new(
+                    readable, headers=Object.fromEntries(headers), status=status
+                )
+                result.set_result(resp)
+                with acquire_js_buffer(body) as jsbytes:
+                    await writer.write(jsbytes.slice())
             else:
+                # Complete body in a single chunk
+                px = create_proxy(body)
+                buf = px.getBuffer()
+                px.destroy()
                 resp = Response.new(
                     buf.data, headers=Object.fromEntries(headers), status=status
                 )
                 result.set_result(resp)
-                await writer.close()
                 finished_response.set()
 
-    # Run the application in the background to handle SSE
+    # Run the application in the background
     async def run_app():
         try:
             await app(request_to_scope(req, env), receive, send)
@@ -212,7 +211,8 @@ async def process_request(
         except Exception as e:
             if not result.done():
                 result.set_exception(e)
-                await writer.close()  # Close the writer
+                if writer is not None:
+                    await writer.close()
                 finished_response.set()
             else:
                 # Response already sent — exception can't be propagated to the
@@ -220,22 +220,13 @@ async def process_request(
                 logger.exception("Exception in ASGI application after response started")
 
     # Create task to run the application in the background
-    app_task = create_task(run_app())
-
-    # Wait for the result (the response)
-    response = await result
+    app_task = create_proxy(create_task(run_app()))
+    wait_until(app_task)
 
-    # For non-SSE responses, we need to wait for the application to complete
-    if not is_sse:
-        await app_task
-    else:  # noqa: PLR5501
-        if ctx is not None:
-            ctx.waitUntil(create_proxy(app_task))
-        else:
-            raise RuntimeError(
-                "Server-Side-Events require ctx to be passed to asgi.fetch"
-            )
-    return response
+    try:
+        return await result
+    finally:
+        app_task.destroy()
 
 
 async def process_websocket(app: Any, req: "Request | js.Request") -> js.Response:

{workers_runtime_sdk-1.1.0 → workers_runtime_sdk-1.1.1}/src/workers/__init__.py

@@ -52,6 +52,7 @@ __all__ = [
     "python_from_rpc",
     "python_to_rpc",
     "waitUntil",
+    "wait_until",
 ]
 
 
@@ -59,7 +60,7 @@ def __getattr__(key):
     if key == "env":
         cloudflare_workers = import_from_javascript("cloudflare:workers")
         return cloudflare_workers.env
-    if key == "waitUntil":
+    if key in ("wait_until", "waitUntil"):
         cloudflare_workers = import_from_javascript("cloudflare:workers")
         return cloudflare_workers.waitUntil
     raise AttributeError(f"module {__name__!r} has no attribute {key!r}")

{workers_runtime_sdk-1.1.0 → workers_runtime_sdk-1.1.1}/uv.lock

@@ -210,7 +210,7 @@ wheels = [
 
 [[package]]
 name = "workers-runtime-sdk"
-version = "1.0.1"
+version = "1.1.0"
 source = { editable = "." }
 
 [package.dev-dependencies]