i7 dZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZmZmZddlm Z ddl!m"Z"m#Z#m$Z$ejJe&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)TransportSecuritySettings)INVALID_REQUEST ErrorData JSONRPCErrorceZdZdZ 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 5. Idle session cleanup via optional timeout 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. security_settings: Optional transport security settings. retry_interval: Retry interval in milliseconds to suggest to clients in SSE retry field. Used for SSE polling behavior. session_idle_timeout: Optional idle timeout in seconds for stateful sessions. If set, sessions that receive no HTTP requests for this duration will be automatically terminated and removed. When retry_interval is also configured, ensure the idle timeout comfortably exceeds the retry interval to avoid reaping sessions during normal SSE polling gaps. Default is None (no timeout). A value of 1800 (30 minutes) is recommended for most deployments. Nc6||dkr td|r | td||_||_||_||_||_||_||_tj|_ i|_ d|_ tj|_d|_y)Nrz9session_idle_timeout must be a positive number of secondsz7session_idle_timeout is not supported in stateless modeF) ValueError RuntimeErrorapp event_store json_response statelesssecurity_settingsretry_intervalsession_idle_timeoutanyioLock_session_creation_lock_server_instances _task_group _run_lock _has_started)selfrrrrrr r!s b/root/aria/mcps/aria-brain/venv/lib/python3.12/site-packages/mcp/server/streamable_http_manager.py__init__z%StreamableHTTPSessionManager.__init__As +0D0IXY Y -9XY Y&*"!2,$8!',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(rr"create_task_groupr&loggerinfo cancel_scopecancelr%clear)r)tgs r*runz StreamableHTTPSessionManager.runbs7&>> % %  "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().)r&rr_handle_stateless_request_handle_stateful_request)r)scopereceivesends r*handle_requestz+StreamableHTTPSessionManager.handle_requestsc    #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 requestN)mcp_session_idis_json_response_enabledrr task_statuscKj4d{}|\}}|j jj||jj dd{dddd{y7j7#t $rt jdY5wxYw7-#1d{7swYyxYww)NTrzStateless session crashed)connectstartedrr5create_initialization_options Exceptionr/ exception)rAstreams 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)rATaskStatus[None]) r/debugrrrr"TASK_STATUS_IGNOREDr&startr< terminate)r)r9r:r;rMrLs` @r*r7z6StreamableHTTPSessionManager._handle_stateless_requests  NO6%)%7%7"44  KPJcJc B B+++$$%9:::++E7DAAA&&((( ; B )s6A=CB:CB<C4B>5C<C>Cc Kt||}|jjt}||jvrj|}t j d|j<j0tjjz|j_ |j|||d{y|*t j dj4d{tj}t!|j"j$j&j( j*J j j*<t j-d|tj.dd fd}j0Jj0j3|d{ j|||d{dddd{yt5dd t7t8d  } t;| j=d d t>j@d} | |||d{y77e77z7l#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 transport)r>r?rrr z'Created new transport with session ID: r@c&Kj4d{}|\}}|j tj}j-tj jz|_|_|5jj||jjdd{ddd|jrojJtjdjdjj!jdj#d{jr_jjvrGj(s;tjdjdjj= dddd{y77#1swYxYw7#t$$r&tj'djdYwxYw#jrajjvrHj(s;tjdjdjj=wwwwxYw7#1d{7swYyxYww)NFrCzSession z idle timeoutz crashedzCleaning up crashed session z from active instances.)rDrEr" CancelScoper! current_timedeadline idle_scoperr5rFcancelled_caughtr>r/r0r%poprRrGrH is_terminated)rArIrJrKrXrLr)s r* run_serverzIStreamableHTTPSessionManager._handle_stateful_request..run_serversy-557'Z'Z74;1 \#++-$Z */):):);X;XZ^ _&4&>&>&@ @ @ !/ = =$2$A$ATE[E[$[(6(D(D & $B'5'D'D&EF8%8!" %)$:$:>;X;X$YO'Z'Z'Z!"""!A(a",,x8U8U7VV^-_`a!/ = =$2$A$ATE[E[$[(6(D(D & $B'5'D'D&EF8%8!" %)$:$:>;X;X$Y)E%\!>='Z'Z'Z'ZsJGJI<AG;G=G>GA>GGGA+I<0 J;I:<JGG G,HHHHA/I77I<:J<JJ J Jz2.0z server-errorzSession not found)codemessage)jsonrpciderrorT)by_alias exclude_nonezapplication/json)content status_code media_type)rArNreturnNone)!r headersgetrr%r/rOrXr!r"rVrWr<r$rhexrrrrr r>r0rPr&rQrrrr model_dump_jsonr NOT_FOUND) r)r9r:r;requestrequest_mcp_session_id transportnew_session_idr\error_responseresponserLs ` @r*r8z5StreamableHTTPSessionManager._handle_stateful_requestsj%)!(!4!45J!K " -2HDLbLb2b../EFI LLL M##/D4M4M4Y050B0B0DtG`G`0` $$-**5'4@ @ @  ! ) LL1 222? J? J!&!>#1-1-?-? $ 0 0&*&<&<#'#6#6 "&44@@@HV&&~'D'DE EnEUVWINHaHa(Z(ZV''333&&,,Z888%33E7DIII? J? J? JH*!(/N &66SW6X&00-H 5'40 0 0o A ? Jx9J? J? J? J? Jb 1sB>I5I-I50I1I54CIIII I$ I5/I0AI5 I3 I5I5III5I0$I' %I0,I5)NFFNNN)rzMCPServer[Any, Any]rzEventStore | Nonerboolrrtrz TransportSecuritySettings | Noner z int | Noner!z float | None)rgzAsyncIterator[None])r9r r:r r;r rgrh) __name__ __module__ __qualname____doc__r+ contextlibasynccontextmanagerr5r<r7r8r,r*rrs  J*.#>B%)-1" "'" "  " < "#"+"B##&/$&/PFFF F  F2/)/)/) /)  /)bo1o1o1 o1  o1r,r))rx __future__rryloggingcollections.abcrhttprtypingruuidrr" anyio.abcrstarlette.requestsr starlette.responsesr starlette.typesr r r mcp.server.lowlevel.serverr MCPServermcp.server.streamable_httprrrmcp.server.transport_securityr mcp.typesrrr getLoggerrur/rr{r,r*rsf5") &(00: D>>   8 $f1f1r,