Skip to content

Commit b67a57f

Browse files
authored
Add return type to client
Update function signature to include return type. Signed-off-by: Dennis Traub <dennis.traub@gmail.com>
1 parent 82385bb commit b67a57f

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

mcp_proxy_for_aws/client.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,14 @@ def aws_iam_streamablehttp_client(
3434
sse_read_timeout: float | timedelta = 60 * 5,
3535
terminate_on_close: bool = True,
3636
httpx_client_factory: McpHttpClientFactory = create_mcp_http_client,
37-
):
37+
) -> AsyncGenerator[
38+
tuple[
39+
MemoryObjectReceiveStream[SessionMessage | Exception],
40+
MemoryObjectSendStream[SessionMessage],
41+
GetSessionIdCallback,
42+
],
43+
None,
44+
]:
3845
"""Create an AWS IAM-authenticated MCP streamable HTTP client.
3946
4047
This function creates a context manager for connecting to an MCP server using AWS IAM

0 commit comments

Comments
 (0)