1313from typing import Any
1414
1515from anyio .streams .memory import MemoryObjectReceiveStream , MemoryObjectSendStream
16-
1716from mcp .client .session import ClientSession
1817from mcp .client .sse import sse_client
1918from mcp .client .streamable_http import streamablehttp_client
2221
2322class SimpleStreamablePrivateGateway :
2423 """Simple MCP private gateway client supporting StreamableHTTP and SSE transports.
25-
24+
2625 This client demonstrates how to use custom extensions (e.g., SNI hostname) for
2726 private gateway connectivity with both transport types.
2827 """
@@ -191,26 +190,26 @@ def get_user_input():
191190 print ("🚀 Simple Streamable Private Gateway" )
192191 print ("\n 📝 Server Configuration" )
193192 print ("=" * 50 )
194-
193+
195194 # Get server port
196195 server_port = input ("Server port [8081]: " ).strip () or "8081"
197-
196+
198197 # Get server hostname
199198 server_hostname = input ("Server hostname [mcp.deepwiki.com]: " ).strip () or "mcp.deepwiki.com"
200-
199+
201200 # Get transport type
202201 print ("\n Transport type:" )
203202 print (" 1. streamable-http (default)" )
204203 print (" 2. sse" )
205204 transport_choice = input ("Select transport [1]: " ).strip () or "1"
206-
205+
207206 if transport_choice == "2" :
208207 transport_type = "sse"
209208 else :
210209 transport_type = "streamable-http"
211-
210+
212211 print ("=" * 50 )
213-
212+
214213 return server_port , server_hostname , transport_type
215214
216215
@@ -219,7 +218,7 @@ async def main():
219218 try :
220219 # Get configuration from user input
221220 server_port , server_hostname , transport_type = get_user_input ()
222-
221+
223222 # Set URL endpoint based on transport type
224223 # StreamableHTTP servers typically use /mcp, SSE servers use /sse
225224 endpoint = "/mcp" if transport_type == "streamable-http" else "/sse"
@@ -232,7 +231,7 @@ async def main():
232231 # Start connection flow
233232 client = SimpleStreamablePrivateGateway (server_url , server_hostname , transport_type )
234233 await client .connect ()
235-
234+
236235 except KeyboardInterrupt :
237236 print ("\n \n 👋 Goodbye!" )
238237 except EOFError :
0 commit comments