chore: make linter happy

encode · Jun 10, 2024 · 6916ede · 6916ede
1 parent 4745366
commit 6916ede
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/starlette/middleware/base.py b/starlette/middleware/base.py
@@ -167,7 +167,11 @@ async def coro() -> None:
 
             assert message["type"] == "http.response.start"
 
-            async def body_stream() -> typing.AsyncGenerator[bytes, None]:
+            async def body_stream() -> (
+                typing.AsyncGenerator[
+                    typing.Union[bytes, typing.MutableMapping[str, typing.Any]], None
+                ]
+            ):
                 async with recv_stream:
                     async for message in recv_stream:
                         if message["type"] == "http.response.pathsend":

diff --git a/starlette/responses.py b/starlette/responses.py
@@ -207,7 +207,7 @@ def __init__(
         self.headers["location"] = quote(str(url), safe=":/%#?=@[]!$&'()*+,;")
 
 
-Content = typing.Union[str, bytes, memoryview]
+Content = typing.Union[str, bytes, memoryview, typing.MutableMapping[str, typing.Any]]
 SyncContentStream = typing.Iterable[Content]
 AsyncContentStream = typing.AsyncIterable[Content]
 ContentStream = typing.Union[AsyncContentStream, SyncContentStream]
@@ -254,7 +254,7 @@ async def stream_response(self, send: Send) -> None:
                 should_close_body = False
                 await send(chunk)
                 break
-            if not isinstance(chunk, (bytes, memoryview)):
+            if isinstance(chunk, str):
                 chunk = chunk.encode(self.charset)
             await send({"type": "http.response.body", "body": chunk, "more_body": True})