fix responses stream_writer and respond_stream interfaces

livioribeiro · Dec 17, 2023 · 8c647fb · 8c647fb
1 parent 5da1be3
commit 8c647fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/asgikit/responses.py b/src/asgikit/responses.py
@@ -282,7 +282,7 @@ async def __listen_for_disconnect(receive):
 
 
 @asynccontextmanager
-async def stream_writer(response):
+async def stream_writer(response: Response):
     client_disconect = asyncio.create_task(
         __listen_for_disconnect(response.asgi.receive)
     )
@@ -300,7 +300,7 @@ async def write(data: bytes | str):
 
 
 async def respond_stream(
-    response: Response, stream: AsyncIterable[bytes], *, status=HTTPStatus.OK
+    response: Response, stream: AsyncIterable[bytes | str], *, status=HTTPStatus.OK
 ):
     await response.start(status)