Skip to content

Commit d1ac8d6

Browse files
mous222dsp-ant
andauthored
fix: prevent async generator cleanup errors in StreamableHTTP transport (#1271)
Co-authored-by: David Soria Parra <[email protected]>
1 parent 0926613 commit d1ac8d6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/mcp/client/streamable_http.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,7 @@ async def _handle_resumption_request(self, ctx: RequestContext) -> None:
248248
ctx.metadata.on_resumption_token_update if ctx.metadata else None,
249249
)
250250
if is_complete:
251+
await event_source.response.aclose()
251252
break
252253

253254
async def _handle_post_request(self, ctx: RequestContext) -> None:
@@ -330,6 +331,7 @@ async def _handle_sse_response(
330331
# If the SSE event indicates completion, like returning respose/error
331332
# break the loop
332333
if is_complete:
334+
await response.aclose()
333335
break
334336
except Exception as e:
335337
logger.exception("Error reading SSE stream:")

0 commit comments

Comments
 (0)