Skip to content

Commit 6ae4622

Browse files
Kludexfelixweinberger
authored andcommitted
Add missing paramters
1 parent ed5dfb2 commit 6ae4622

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
@@ -527,6 +527,8 @@ async def streamablehttp_client(
527527
timeout: float | timedelta = 30,
528528
sse_read_timeout: float | timedelta = 60 * 5,
529529
terminate_on_close: bool = True,
530+
httpx_client_factory: McpHttpClientFactory = create_mcp_http_client,
531+
auth: httpx.Auth | None = None,
530532
) -> AsyncGenerator[
531533
tuple[
532534
MemoryObjectReceiveStream[SessionMessage | Exception],
@@ -535,5 +537,7 @@ async def streamablehttp_client(
535537
],
536538
None,
537539
]:
538-
async with streamable_http_client(url, headers, timeout, sse_read_timeout, terminate_on_close) as streams:
540+
async with streamable_http_client(
541+
url, headers, timeout, sse_read_timeout, terminate_on_close, httpx_client_factory, auth
542+
) as streams:
539543
yield streams

0 commit comments

Comments
 (0)