{h+dZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZmZmZddlm Z ejBe"Z#GddZ$y)z/StreamableHTTP Session Manager for MCP servers.) annotationsN) AsyncIterator) HTTPStatus)Any)uuid4) TaskStatus)Request)Response)ReceiveScopeSend)Server)MCP_SESSION_ID_HEADER EventStoreStreamableHTTPServerTransport)TransportSecuritySettingsceZdZdZ d d dZej d dZ d dZ d dZ d dZ y) StreamableHTTPSessionManagera Manages StreamableHTTP sessions with optional resumability via event store. This class abstracts away the complexity of session management, event storage, and request handling for StreamableHTTP transports. It handles: 1. Session tracking for clients 2. Resumability via an optional event store 3. Connection management and lifecycle 4. Request handling and transport setup Important: Only one StreamableHTTPSessionManager instance should be created per application. The instance cannot be reused after its run() context has completed. If you need to restart the manager, create a new instance. Args: app: The MCP server instance event_store: Optional event store for resumability support. If provided, enables resumable connections where clients can reconnect and receive missed events. If None, sessions are still tracked but not resumable. json_response: Whether to use JSON responses instead of SSE streams stateless: If True, creates a completely fresh transport for each request with no session tracking or state persistence between requests. Nc||_||_||_||_||_t j |_i|_d|_ t j |_ d|_ y)NF) app event_store json_response statelesssecurity_settingsanyioLock_session_creation_lock_server_instances _task_group _run_lock _has_started)selfrrrrrs c/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/mcp/server/streamable_http_manager.py__init__z%StreamableHTTPSessionManager.__init__8sa&*"!2',jjl#KM !cK|j4d{|jr tdd|_dddd{tj4d{}||_t jd dt jd|jjd|_|jjdddd{y77#1d{7swYxYw7#t jd|jjd|_|jjwxYw7u#1d{7swYyxYww)aw Run the session manager with proper lifecycle management. This creates and manages the task group for all session operations. Important: This method can only be called once per instance. The same StreamableHTTPSessionManager instance cannot be reused after this context manager exits. Create a new instance if you need to restart. Use this in the lifespan context manager of your Starlette app: @contextlib.asynccontextmanager async def lifespan(app: Starlette) -> AsyncIterator[None]: async with session_manager.run(): yield NzyStreamableHTTPSessionManager .run() can only be called once per instance. Create a new instance if you need to run again.Tz&StreamableHTTP session manager startedz,StreamableHTTP session manager shutting down) r r! RuntimeErrorrcreate_task_grouprloggerinfo cancel_scopecancelrclear)r"tgs r#runz StreamableHTTPSessionManager.runPs7&>> % %  "Y!%D   % %**, / /!D  KK@ A / JK&&(#' &&,,. / / / % % % % % / JK&&(#' &&,,. / / / /sE-C&E-C* E-C(E-C?E-"EDAE E- E!E-(E-*C<0C3 1C<8E-AEEE-E*E! E*&E-cK|j td|jr|j|||d{y|j |||d{y7!7w)a Process ASGI request with proper session handling and transport setup. Dispatches to the appropriate handler based on stateless mode. Args: scope: ASGI scope receive: ASGI receive function send: ASGI send function Nz6Task group is not initialized. Make sure to use run().)rr'r_handle_stateless_request_handle_stateful_request)r"scopereceivesends r#handle_requestz+StreamableHTTPSessionManager.handle_requestysc    #WX X >>00F F F//wE E E G Es!:A"AA"A A" A"cKtjdtdjdjt j ddfd}jJjj|d{j|||d{jd{y7877 w)z Process request in stateless mode - creating a new transport for each request. Args: scope: ASGI scope receive: ASGI receive function send: ASGI send function z7Stateless mode: Creating new transport for this requestNmcp_session_idis_json_response_enabledrr task_statuscKj4d{}|\}}|j jj||jj dd{dddd{y7j7#t $rt jdY5wxYw7-#1d{7swYyxYww)NTrzStateless session crashed)connectstartedrr/create_initialization_options Exceptionr) exception)r<streams read_stream write_streamhttp_transportr"s r#run_stateless_serverzTStreamableHTTPSessionManager._handle_stateless_request..run_stateless_servers%--/ B B7,3) \##%B((,,#$>>@"& ' B B B !B$$%@AB B B B BssCBCB-:B-B.B2 C=B+>CBB(%B-'B((B-+C-B?3B6 4B?;C)r<TaskStatus[None]) r)debugrrrrTASK_STATUS_IGNOREDrstartr6 terminate)r"r3r4r5rHrGs` @r#r1z6StreamableHTTPSessionManager._handle_stateless_requests  NO6%)%7%7"44  KPJcJc B B+++$$%9:::++E7DAAA&&((( ; B )s6A=CB:CB<C4B>5C<C>Cc Kt||}|jjt}|N|jvr@j|}t j d|j|||d{y|t j dj4d{tj}t|jjj jJ j j<t j!d|t"j$dd fd}j&Jj&j)|d{ j|||d{dddd{yt+dt,j. } | |||d{y7V7+7d7K7=#1d{7swYyxYw7&w) z Process request in stateful mode - maintaining session state between requests. Args: scope: ASGI scope receive: ASGI receive function send: ASGI send function Nz1Session already exists, handling request directlyzCreating new transportr8z'Created new transport with session ID: r;ctKj4d{}|\}}|j jj||jj dd{jr_jjvrGjs;t jdjdjj= dddd{y77#t $r1}t jdjd|dYd}~d}~wwxYw#jrajjvrHjs;t jdjdjj=wwwwxYw7#1d{7swYyxYww) NFr>zSession z crashed: T)exc_infozCleaning up crashed session z from active instances.) r?r@rr/rArBr)errorr9r is_terminatedr*)r<rDrErFerGr"s r# run_serverzIStreamableHTTPSessionManager._handle_stateful_request..run_servers-557ZZ74;1 \#++-Z"&((,, + , $ F F H*/ #/#!/ = =$2$A$ATE[E[$[(6(D(D & $B'5'D'D&EF8%8!" %)$:$:>;X;X$Y7ZZZ )"LL"*>+H+H*ITUSV W)-)!/ = =$2$A$ATE[E[$[(6(D(D & $B'5'D'D&EF8%8!" %)$:$:>;X;X$Y)E%\!>%ZZZZsF8C.F8F#:C2-C0.C22A+F# F8(F!)F80C22 D,;'D'"D/'D,,D//A/FF#!F8#F5)F, *F51F8z)Bad Request: No valid session ID provided) status_code)r<rIreturnNone)r headersgetrrr)rJr6rrhexrrrrr9r*rrKrrLr r BAD_REQUEST) r"r3r4r5requestrequest_mcp_session_id transportnew_session_idrTresponserGs ` @r#r2z5StreamableHTTPSessionManager._handle_stateful_requests%)!(!4!45J!K " -2HDLbLb2b../EFI LLL M**5'4@ @ @  ! ) LL1 2222 J2 J!&!>#1-1-?-? $ 0 0&*&<&< "&44@@@HV&&~'D'DE EnEUVWINHaHaZZ>''333&&,,Z888%33E7DIIIe2 J2 J2 Jj ;&22H5'40 0 0 A 2 J^9Je2 J2 J2 J2 Jr 1sA6G3:G;-G3(G)G3,CG1G2G G G G3G-G3 G1 G3G3GGG3G."G% #G.*G3)NFFN) rzMCPServer[Any, Any]rzEventStore | Nonerboolrrarz TransportSecuritySettings | None)rVzAsyncIterator[None])r3r r4r r5r rVrW) __name__ __module__ __qualname____doc__r$ contextlibasynccontextmanagerr/r6r1r2r%r#rrs:*.#>B " "'" "  " < "0##&/$&/PFFF F  F2/)/)/) /)  /)bT1T1T1 T1  T1r%r)%re __future__rrfloggingcollections.abcrhttprtypingruuidrr anyio.abcrstarlette.requestsr starlette.responsesr starlette.typesr r r mcp.server.lowlevel.serverr MCPServermcp.server.streamable_httprrrmcp.server.transport_securityr getLoggerrbr)rrhr%r#rxs`5") &(00: D   8 $z1z1r%