i7dZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZmZmZddlm Z ddl!m"Z"m#Z#m$Z$ej%e&Z'GddZ(dS)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 JSONRPCErrorcbeZdZdZ dd dZejd!dZd"dZd"dZ d"dZ dS)#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. NFappMCPServer[Any, Any] event_storeEventStore | None json_responsebool statelesssecurity_settings TransportSecuritySettings | Noneretry_interval int | Nonesession_idle_timeout float | NonecF||dkrtd|r|td||_||_||_||_||_||_||_tj |_ i|_ d|_ tj |_d|_dS)Nrz9session_idle_timeout must be a positive number of secondsz7session_idle_timeout is not supported in stateless modeF) ValueError RuntimeErrorrrrrrr!r#anyioLock_session_creation_lock_server_instances _task_group _run_lock _has_started)selfrrrrrr!r#s i/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/mcp/server/streamable_http_manager.py__init__z%StreamableHTTPSessionManager.__init__As +0D0I0IXYY Y  Z-9XYY Y&*"!2,$8!',jll#KM !returnAsyncIterator[None]cK|j4d{V|jrtdd|_dddd{Vn#1d{VswxYwYtj4d{V}||_t d dWVt d|j d|_|j nX#t d|j d|_|j wxYw dddd{VdS#1d{VswxYwYdS)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.r'r(create_task_groupr,loggerinfo cancel_scopecancelr+clear)r/tgs r0runz StreamableHTTPSessionManager.runbsz&> % % % % % % % %  "Y!%D   % % % % % % % % % % % % % % % % % % % % % % % % % % %*,, / / / / / / /!D  KK@ A A A / JKKK&&(((#' &,,....  JKKK&&(((#' &,,..... / / / / / / / / / / / / / / / / / / / / / / / / / / / / / /s=A A  A *"E C&AE&AD;;E EEscoper receiver sendr NonecK|jtd|jr||||d{VdS||||d{VdS)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,r'r_handle_stateless_request_handle_stateful_request)r/r>r?r@s r0handle_requestz+StreamableHTTPSessionManager.handle_requests   #WXX X > F00FF F F F F F F F F F//wEE E E E E E E E E Er2cdKtdtdjdjt jddfd}jJj|d{V |||d{V d{VdS) 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_statusrJTaskStatus[None]cK4d{V}|\}}| j||jdd{Vn*#t $rt dYnwxYwdddd{VdS#1d{VswxYwYdS)NTrzStateless session crashed)connectstartedrr=create_initialization_options Exceptionr7 exception)rJstreams read_stream write_streamhttp_transportr/s r0run_stateless_serverzTStreamableHTTPSessionManager._handle_stateless_request..run_stateless_servers%--// B B B B B B B7,3) \##%%%B(,,#$>>@@"& ' !BBB$$%@AAAAAB B B B B B B B B B B B B B B B B B B B B B B B B B B B B B Bs4B2;A54B25$BB2BB22 B<?B<)rJrK) r7debugrrrr(TASK_STATUS_IGNOREDr,startrE terminate)r/r>r?r@rWrVs` @r0rCz6StreamableHTTPSessionManager._handle_stateless_requests  NOOO6%)%7"4    KPJc B B B B B B B B B+++$$%9:::::::::++E7DAAAAAAAAA&&(((((((((((r2c~ Kt||}|jt}||jvryj|}t d|j,j%tj jz|j_ | |||d{VdS|t dj 4d{Vtj}t!|jjjj jJ j j<t d|tjdd fd }jJj|d{V |||d{Vdddd{VdS#1d{VswxYwYdSt5d d t7t8d} t;| ddt>j d} | |||d{VdS)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)rGrHrrr!z'Created new transport with session ID: rIrJrKr3rAcjK4d{V}|\}}| tj}j'tjjz|_|_|5j ||j dd{Vdddn #1swxYwY|j rfj Jtdj djj dd{Vn3#t$$r&tdj dYnwxYwj rEj jvr7js0tdj djj =nQ#j rEj jvr7js0tdj djj =wxYw dddd{VdS#1d{VswxYwYdS)NFrMzSession z idle timeoutz crashedzCleaning up crashed session z from active instances.)rNrOr( CancelScoper# current_timedeadline idle_scoperr=rPcancelled_caughtrGr7r8r+popr[rQrR is_terminated)rJrSrTrUrarVr/s r0 run_serverzIStreamableHTTPSessionManager._handle_stateful_request..run_servers-5577'Z'Z'Z'Z'Z'Z'Z74;1 \#++---$Z */):)<);XZ^ _ _ _&4&>&>&@&@ @ @ @ @ @ @ @(aaa",,-_8U-_-_-_`````a!/ = Z$2$ATE[$[$[(6(D%\!' %8'5'D%8%8%8!"!"!" %)$:>;X$Y!/ = Z$2$ATE[$[$[(6(D%\!' %8'5'D%8%8%8!"!"!" %)$:>;X$Y Y Y Y Y YO'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'Z'ZsnH"AD=<r?r@requestrequest_mcp_session_id transportnew_session_idreerror_responseresponserVs ` @r0rDz5StreamableHTTPSessionManager._handle_stateful_requests%))!(!4!45J!K!K " -2HDLb2b2b./EFI LLL M M M#/D4M4Y050B0D0DtG`0` $-**5'4@@ @ @ @ @ @ @ @ F ! ) LL1 2 2 22? J? J? J? J? J? J? J? J!&!>#1-1-? $ 0&*&<#'#6 """&4@@@HV&~'DE VnVVWWWINHa(Z(Z(Z(Z(Z(Z(Z(Z(ZV'333&,,Z888888888%33E7DIIIIIIIII? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? J? JH*!(/N &66SW6XX&0-H (5'400 0 0 0 0 0 0 0 0 0s"C G G  G )NFFNNN)rrrrrrrrrr r!r"r#r$)r3r4)r>r r?r r@r r3rA) __name__ __module__ __qualname____doc__r1 contextlibasynccontextmanagerr=rErCrDr2r0rrs  J*.#>B%)-1"""""B#&/&/&/$#&/PFFFF2/)/)/)/)bo1o1o1o1o1o1r2r))r~ __future__rrloggingcollections.abcrhttprtypingruuidrr( anyio.abcrstarlette.requestsr starlette.responsesr starlette.typesr r r mcp.server.lowlevel.serverr MCPServermcp.server.streamable_httprrrmcp.server.transport_securityr mcp.typesrrr getLoggerr{r7rrr2r0rs55"""""")))))) &&&&&&((((((0000000000:::::: DCCCCC>>>>>>>>>>  8 $ $f1f1f1f1f1f1f1f1f1f1r2