Skip to content

Commit bcc4b19

Browse files
Kludexfelixweinberger
authored andcommitted
Add missing paramters
1 parent 433da3e commit bcc4b19

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/mcp/client/streamable_http.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -640,6 +640,8 @@ async def streamablehttp_client(
640640
timeout: float | timedelta = 30,
641641
sse_read_timeout: float | timedelta = 60 * 5,
642642
terminate_on_close: bool = True,
643+
httpx_client_factory: McpHttpClientFactory = create_mcp_http_client,
644+
auth: httpx.Auth | None = None,
643645
) -> AsyncGenerator[
644646
tuple[
645647
MemoryObjectReceiveStream[SessionMessage | Exception],
@@ -648,5 +650,7 @@ async def streamablehttp_client(
648650
],
649651
None,
650652
]:
651-
async with streamable_http_client(url, headers, timeout, sse_read_timeout, terminate_on_close) as streams:
653+
async with streamable_http_client(
654+
url, headers, timeout, sse_read_timeout, terminate_on_close, httpx_client_factory, auth
655+
) as streams:
652656
yield streams

0 commit comments

Comments
 (0)