o j7@sdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZmZmZddlm Z ddl!m"Z"m#Z#m$Z$e%e&Z'GdddZ(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 JSONRPCErrorc@sVeZdZdZ      d$d%ddZejd&ddZd'ddZd'd d!Z d'd"d#Z 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 | NonecCs|dur |dkr td|r|durtd||_||_||_||_||_||_||_t |_ i|_ d|_ t |_d|_dS)Nrz9session_idle_timeout must be a positive number of secondsz7session_idle_timeout is not supported in stateless modeF) ValueError RuntimeErrorrrrrrrr!anyioLock_session_creation_lock_server_instances _task_group _run_lock _has_started)selfrrrrrrr!r-U/home/kuhnn/.local/lib/python3.10/site-packages/mcp/server/streamable_http_manager.py__init__As      z%StreamableHTTPSessionManager.__init__returnAsyncIterator[None]c Cs|j4IdH|jrtdd|_WdIdHn 1IdHs#wYt4IdH?}||_tdzdVWtd|j d|_|j ntd|j d|_|j wWdIdHdS1IdHsuwYdS)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,tgr-r-r.runbs,(     .z StreamableHTTPSessionManager.runscoper receiver sendr NonecsJ|jdur td|jr||||IdHdS||||IdHdS)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<r-r-r.handle_requests  z+StreamableHTTPSessionManager.handle_requestcs~tdtdjdjdtjdd fdd}jdus#Jj|IdH |||IdH IdHdS) 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_statusrDTaskStatus[None]c s4IdH@}|\}}|zjj||jddIdHWnty2tdYn wWdIdHdSWdIdHdS1IdHsOwYdS)NTrzStateless session crashed)connectstartedrr9create_initialization_options Exceptionr3 exception)rDstreams read_stream write_streamhttp_transportr,r-r.run_stateless_servers$  .zTStreamableHTTPSessionManager._handle_stateless_request..run_stateless_server)rDrE) r3debugrrrr%TASK_STATUS_IGNOREDr)startr@ terminate)r,r:r;r<rQr-rOr.r>s z6StreamableHTTPSessionManager._handle_stateless_requestc st||}|jt}|dur>|jvr>j|}td|jdur2jdur2t j|j_ | |||IdHdS|durtdj 4IdHXtj}t|jjjjdjdusgJjj<td|t jddfd d }jdusJj|IdH |||IdHWdIdHdS1IdHswYdStd dttddd} t| jdddtj dd} | |||IdHdS)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)rArBrrrz'Created new transport with session ID: rCrDrEr0r=c s4IdH}|\}}|zz[t}jdur(tj|_|_|jj ||j ddIdHWdn1sEwY|j rmj dusTJt dj djj dIdHWntyt dj dYnwWj rj jvrjst dj djj =nj rj jvrjst dj djj =wWdIdHdS1IdHswYdS)NFrFzSession z idle timeoutz crashedzCleaning up crashed session z from active instances.)rGrHr% CancelScoper! current_timedeadline idle_scoperr9rIcancelled_caughtrAr3r4r(poprUrJrK is_terminated)rDrLrMrNrYrOr-r. run_serversf     .zIStreamableHTTPSessionManager._handle_stateful_request..run_serverz2.0z server-errorzSession not found)codemessage)jsonrpciderrorT)by_alias exclude_nonezapplication/json)content status_code media_type)rDrEr0r=)!rheadersgetrr(r3rRrYr!r%rWrXr@r'rhexrrrrrrAr4rSr)rTrrrr model_dump_jsonr NOT_FOUND) r,r:r;r<requestrequest_mcp_session_id transportnew_session_idr]error_responseresponser-rOr.r?sV      +.D z5StreamableHTTPSessionManager._handle_stateful_request)NFFNNN)rrrrrrrrrrrr r!r")r0r1)r:r r;r r<r r0r=) __name__ __module__ __qualname____doc__r/ contextlibasynccontextmanagerr9r@r>r?r-r-r-r.rs% !  ( 1r))rv __future__rrwloggingcollections.abcrhttprtypingruuidrr% anyio.abcrstarlette.requestsrstarlette.responsesr starlette.typesr r r mcp.server.lowlevel.serverr MCPServermcp.server.streamable_httprrrmcp.server.transport_securityr mcp.typesrrr getLoggerrsr3rr-r-r-r.s&