i/ 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 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. Nc||_||_||_||_||_||_t j|_i|_ d|_ t j|_ d|_ y)NF) app event_store json_response statelesssecurity_settingsretry_intervalanyioLock_session_creation_lock_server_instances _task_group _run_lock _has_started)selfrrrrrrs r/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/mcp/server/streamable_http_manager.py__init__z%StreamableHTTPSessionManager.__init__<si&*"!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_groupr#loggerinfo cancel_scopecancelr"clear)r&tgs r'runz StreamableHTTPSessionManager.runVs&>>>  "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#r+r_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)connectstartedrr3create_initialization_options Exceptionr- exception)r?streams read_stream write_streamhttp_transportr&s r'run_stateless_serverzTStreamableHTTPSessionManager._handle_stateless_request..run_stateless_servers%--//7,3) \##%B((,,#$>>@"& ' 0// !B$$%@AB0///ssCBCB-:B-B.B2 C=B+>CBB(%B-'B((B-+C-B?3B6 4B?;C)r?TaskStatus[None]) r-debugrrrrTASK_STATUS_IGNOREDr#startr: terminate)r&r7r8r9rKrJs` @r'r5z6StreamableHTTPSessionManager._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|jjjj j J j j <t j#d|t$j&dd fd}j(Jj(j+|d{ j|||d{dddd{yt-dd t/t0d  } t3| j5d d t6j8d} | |||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=rrrz'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) NFrAzSession z crashed: T)exc_infozCleaning up crashed session z from active instances.) rBrCrr3rDrEr-errorr<r" is_terminatedr.)r?rGrHrIerJr&s r' run_serverzIStreamableHTTPSessionManager._handle_stateful_request..run_servers-557774;1 \#++-Z"&((,, + , $ F F H*/ #/#!/ = =$2$A$ATE[E[$[(6(D(D & $B'5'D'D&EF8%8!" %)$:$:>;X;X$Y7 877 )"LL"*>+H+H*ITUSV W)-)!/ = =$2$A$ATE[E[$[(6(D(D & $B'5'D'D&EF8%8!" %)$:$:>;X;X$Y)E%\!>% 8777sF8C.F8F#:C2-C0.C22A+F# F8(F!)F80C22 D,;'D'"D/'D,,D//A/FF#!F8#F5)F, *F51F8z2.0z server-errorzSession not found)codemessage)jsonrpcidrTT)by_alias exclude_nonezapplication/json)content status_code media_type)r?rLreturnNone)r headersgetrr"r-rMr:r!rhexrrrrrr<r.rrNr#rOrrrr model_dump_jsonr NOT_FOUND) r&r7r8r9requestrequest_mcp_session_id transportnew_session_idrWerror_responseresponserJs ` @r'r6z5StreamableHTTPSessionManager._handle_stateful_requests%)!(!4!45J!K " -2HDLbLb2b../EFI LLL M**5'4@ @ @  ! ) LL1 2222!&!>#1-1-?-? $ 0 0&*&<&<#'#6#6 "&44@@@HV&&~'D'DE EnEUVWINHaHaZZ>''333&&,,Z888%33E7DIIIg322p*!(/N &66SW6X&00-H 5'40 0 0W A 3`9Jg3222J 1sA6H-:H ;-H-(H )H-,CH<H=HHH H-'H(AH-H+H- H-HHH-H(H H($H-)NFFNN) rzMCPServer[Any, Any]rzEventStore | Nonerboolrrnrz TransportSecuritySettings | Nonerz int | None)razAsyncIterator[None])r7r r8r r9r rarb) __name__ __module__ __qualname____doc__r( contextlibasynccontextmanagerr3r:r5r6r)r'rrs@*.#>B%)" "'" "  " < "#"4##&/$&/PFFF F  F2/)/)/) /)  /)b`1`1`1 `1  `1r)r))rr __future__rrsloggingcollections.abcrhttprtypingruuidrr anyio.abcrstarlette.requestsr starlette.responsesr starlette.typesr r r mcp.server.lowlevel.serverr MCPServermcp.server.streamable_httprrrmcp.server.transport_securityr mcp.typesrrr getLoggerror-rrur)r'rsf5") &(00: D>>   8 $K1K1r)