Skip to content

Commit d6c2fa5

Browse files
simplify streaming transport
1 parent 7db7680 commit d6c2fa5

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/integrations/mcp/streaming_asgi_transport.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,8 @@ async def run_app():
6666
await self.app(scope, receive, send)
6767

6868
class StreamingBodyStream(AsyncByteStream):
69-
def __init__(self, receiver, task):
69+
def __init__(self, receiver):
7070
self.receiver = receiver
71-
self.task = task
7271

7372
async def __aiter__(self):
7473
try:
@@ -77,7 +76,8 @@ async def __aiter__(self):
7776
except anyio.EndOfStream:
7877
pass
7978

80-
stream = StreamingBodyStream(body_receiver, asyncio.create_task(run_app()))
79+
stream = StreamingBodyStream(body_receiver)
8180
response = Response(status_code=200, headers=[], stream=stream)
8281

82+
asyncio.create_task(run_app())
8383
return response

0 commit comments

Comments
 (0)