)`i/dZddlmZddlZddlZddlmZddlmZddl 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 JSONRPCErrorc`eZdZdZ dddZejddZd 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 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. NFappMCPServer[Any, Any] event_storeEventStore | None json_responsebool statelesssecurity_settings TransportSecuritySettings | Noneretry_interval int | Nonec||_||_||_||_||_||_t j|_i|_ d|_ t j|_ d|_ dS)NF) rrrrrr!anyioLock_session_creation_lock_server_instances _task_group _run_lock _has_started)selfrrrrrr!s v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mcp/server/streamable_http_manager.py__init__z%StreamableHTTPSessionManager.__init__<so&*"!2,',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* RuntimeErrorr$create_task_groupr(loggerinfo cancel_scopecancelr'clear)r+tgs r,runz StreamableHTTPSessionManager.runVsz&> % % % % % % % %  "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(r2r_handle_stateless_request_handle_stateful_request)r+r;r<r=s r,handle_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 Er.cdKtdtdjdjt 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_statusrGTaskStatus[None]cK4d{V}|\}}| j||jdd{Vn*#t $rt dYnwxYwdddd{VdS#1d{VswxYwYdS)NTrzStateless session crashed)connectstartedrr:create_initialization_options Exceptionr4 exception)rGstreams read_stream write_streamhttp_transportr+s r,run_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<)rGrH) r4debugrrrr$TASK_STATUS_IGNOREDr(startrB terminate)r+r;r<r=rTrSs` @r,r@z6StreamableHTTPSessionManager._handle_stateless_requests  NOOO6%)%7"4    KPJc B B B B B B B B B+++$$%9:::::::::++E7DAAAAAAAAA&&(((((((((((r.c Kt||}|jt}|O|jvrFj|}t d||||d{VdS|t dj4d{Vtj }t|j j jj jJ j j<t d|t$jdd fd }jJj|d{V |||d{Vdddd{VdS#1d{VswxYwYdSt-d d t/t0d} t3| ddt6jd} | |||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)rDrErrr!z'Created new transport with session ID: rFrGrHr/r>cK4d{V}|\}}| j||jdd{Vn>#t $r1}t djd|dYd}~nd}~wwxYwjrEjj vr7j s0t djdj j=nQ#jrEjj vr7j s0t djdj j=wxYw dddd{VdS#1d{VswxYwYdS) NFrJzSession z crashed: T)exc_infozCleaning up crashed session z from active instances.) rKrLrr:rMrNr4errorrDr' is_terminatedr5)rGrPrQrRerSr+s r, run_serverzIStreamableHTTPSessionManager._handle_stateful_request..run_servers-5577ZZZZZZZ74;1 \#++---Z"&(,, + , $ F F H H*/ #/## )"LL W>+H W WTU W W)-)!/ = 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 Y7ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZsNE%;A54D5 B0?'B+&D+B00D3A E%AEE%% E/2E/z2.0z server-errorzSession not found)codemessage)jsonrpcidr\T)by_alias exclude_nonezapplication/json)content status_code media_type)rGrHr/r>)r headersgetrr'r4rUrBr&rhexrrrrr!rDr5r$rVr(rWrrrr model_dump_jsonr NOT_FOUND) r+r;r<r=requestrequest_mcp_session_id transportnew_session_idr_error_responseresponserSs ` @r,rAz5StreamableHTTPSessionManager._handle_stateful_requestsR%))!(!4!45J!K!K " -2HDLb2b2b./EFI LLL M M M**5'4@@ @ @ @ @ @ @ @ F ! ) LL1 2 2 223 J3 J3 J3 J3 J3 J3 J3 J!&!>#1-1-? $ 0&*&<#'#6 """&4@@@HV&~'DE VnVVWWWINHaZZZZZZZZZ>'333&,,Z888888888%33E7DIIIIIIIIIg3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 J3 Jp*!(/N &66SW6XX&0-H (5'400 0 0 0 0 0 0 0 0 0s/C F  FF)NFFNN) rrrrrrrrrr r!r")r/r0)r;r r<r r=r r/r>) __name__ __module__ __qualname____doc__r- contextlibasynccontextmanagerr:rBr@rAr.r,rrs@*.#>B%)"""""4#&/&/&/$#&/PFFFF2/)/)/)/)b`1`1`1`1`1`1r.r))rw __future__rrxloggingcollections.abcrhttprtypingruuidrr$ anyio.abcrstarlette.requestsr starlette.responsesr starlette.typesr r r mcp.server.lowlevel.serverr MCPServermcp.server.streamable_httprrrmcp.server.transport_securityr mcp.typesrrr getLoggerrtr4rrzr.r,rs55"""""")))))) &&&&&&((((((0000000000:::::: DCCCCC>>>>>>>>>>  8 $ $K1K1K1K1K1K1K1K1K1K1r.