o ho@sdZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z ddl mZddlmZddlZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZmZm Z ddl!m"Z"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4e5e6Z7dZ8dZ9dZ:dZ;dZdZ?e@ZAe@ZBeGdddZCe eCge dfZDGdddeZEGdddZFdS)z StreamableHTTP Server Transport Module This module implements an HTTP transport layer with Streamable HTTP. The transport handles bidirectional communication using HTTP requests and responses, with streaming support for long-running operations. N)ABCabstractmethod)AsyncGenerator AwaitableCallable)asynccontextmanager) dataclass) HTTPStatus)MemoryObjectReceiveStreamMemoryObjectSendStream)ValidationError)EventSourceResponse)Request)Response)ReceiveScopeSend)TransportSecurityMiddlewareTransportSecuritySettings)ServerMessageMetadataSessionMessage)SUPPORTED_PROTOCOL_VERSIONS) DEFAULT_NEGOTIATED_VERSIONINTERNAL_ERRORINVALID_PARAMSINVALID_REQUEST PARSE_ERROR ErrorData JSONRPCErrorJSONRPCMessageJSONRPCRequestJSONRPCResponse RequestIdzmcp-session-idzmcp-protocol-versionz last-event-idzapplication/jsonztext/event-stream _GET_streamz^[\x21-\x7E]+$c@s*eZdZUdZeed<dZedBed<dS) EventMessagezM A JSONRPCMessage with an optional event ID for stream resumability. messageNevent_id)__name__ __module__ __qualname____doc__r__annotations__r&strr-r-[/var/www/html/openai_agents/venv/lib/python3.10/site-packages/mcp/server/streamable_http.pyr$Gs r$c@sHeZdZdZedededefddZedede ded Bfd d Z d S) EventStorez? Interface for resumability support via event storage. stream_idr%returncdS)z Stores an event for later retrieval. Args: stream_id: ID of the stream the event belongs to message: The JSON-RPC message to store Returns: The generated event ID for the stored event Nr-)selfr0r%r-r-r. store_eventYs zEventStore.store_event last_event_id send_callbackNcr2)a2 Replays events that occurred after the specified event ID. Args: last_event_id: The ID of the last event the client received send_callback: A callback function to send events to the client Returns: The stream ID of the replayed events Nr-)r3r5r6r-r-r.replay_events_aftergszEventStore.replay_events_after) r'r(r)r*rStreamIdrEventIdr4 EventCallbackr7r-r-r-r.r/Ts r/c @seZdZUdZdZeeeBdBed<dZ e eeBdBed<dZ eedBed<dZ e edBed<e ed<   dBd edBd ed edBd edBd df ddZed efddZedfdedededeeefdBd ef ddZejdfdedBdedeeefdBd efddZded edBfddZde d eeeffdd Z!d!e"d dfd"d#Z#d$e$d%e%d&e&d dfd'd(Z'ded e(eeffd)d*Z)ded efd+d,Z*d$e$ded%e%d&e&d df d-d.Z+ded&e&d dfd/d0Z,ded&e&d dfd1d2Z-dCd3d4Z.ded&e&d dfd5d6Z/ded&e&d efd7d8Z0ded&e&d efd9d:Z1ded&e&d efd;d<Z2d=eded&e&d dfd>d?Z3e4d e5e(e eeBeefdffd@dAZ6dS)DStreamableHTTPServerTransportz HTTP server transport with event streaming support for MCP. Handles JSON-RPC messages in HTTP POST requests with SSE streaming. Supports optional JSON responses and session management. N_read_stream_writer _read_stream _write_stream_write_stream_reader _securityFmcp_session_idis_json_response_enabled event_storesecurity_settingsr1cCsF|dur t|s td||_||_||_t||_i|_d|_ dS)a Initialize a new StreamableHTTP server transport. Args: mcp_session_id: Optional session identifier for this connection. Must contain only visible ASCII characters (0x21-0x7E). is_json_response_enabled: If True, return JSON responses for requests instead of SSE streams. Default is False. event_store: Event store for resumability support. If provided, resumability will be enabled, allowing clients to reconnect and resume messages. security_settings: Optional security settings for DNS rebinding protection. Raises: ValueError: If the session ID contains invalid characters. NzASession ID must only contain visible ASCII characters (0x21-0x7E)F) SESSION_ID_PATTERN fullmatch ValueErrorrArB _event_storerr@_request_streams _terminated)r3rArBrCrDr-r-r.__init__s  z&StreamableHTTPServerTransport.__init__cCs|jS)z7Check if this transport has been explicitly terminated.)rJ)r3r-r-r. is_terminatedsz+StreamableHTTPServerTransport.is_terminated error_message status_code error_codeheaderscCsTdti}|r |||jr|j|t<tddt||dd}t|jddd||dS) z6Create an error response with a simple string message. Content-Typez2.0z server-error)coder%)jsonrpciderrorTby_alias exclude_nonerNrP)CONTENT_TYPE_JSONupdaterAMCP_SESSION_ID_HEADERrrrmodel_dump_json)r3rMrNrOrPresponse_headerserror_responser-r-r._create_error_responses$   z4StreamableHTTPServerTransport._create_error_responseresponse_messagecCsFdti}|r |||jr|j|t<t|r|jdddnd||dS)z,Create a JSON response from a JSONRPCMessagerQTrVNrY)rZr[rAr\rr])r3rarNrPr^r-r-r._create_json_responses  z3StreamableHTTPServerTransport._create_json_responserequestcCs |jtS)z,Extract the session ID from request headers.)rPgetr\)r3rcr-r-r._get_session_ids z-StreamableHTTPServerTransport._get_session_id event_messagecCs*d|jjdddd}|jr|j|d<|S)z2Create event data dictionary from an EventMessage.r%TrV)eventdatarT)r%r]r&)r3rf event_datar-r-r._create_event_datas  z0StreamableHTTPServerTransport._create_event_data request_idc s||jvrLz=z|j|dIdH|j|dIdHWnty/tdYn wW|j|ddSW|j|ddS|j|dwdS)z/Clean up memory streams for a given request ID.rNz4Error closing memory streams - may already be closed)rIaclose Exceptionloggerdebugpop)r3rkr-r-r._clean_up_memory_streamss  "z6StreamableHTTPServerTransport._clean_up_memory_streamsscopereceivesendcst||}|jdk}|jj||dIdH}|r#||||IdHdS|jr8|dtj}||||IdHdS|jdkrJ|||||IdHdS|jdkrZ| ||IdHdS|jdkrj| ||IdHdS| ||IdHdS)z6Application entry point that handles all HTTP requestsPOST)is_postNz&Not Found: Session has been terminatedGETDELETE) rmethodr@validate_requestrJr`r NOT_FOUND_handle_post_request_handle_get_request_handle_delete_request_handle_unsupported_request)r3rsrtrurcrwr_responser-r-r.handle_requests*     z,StreamableHTTPServerTransport.handle_requestcCsN|jdd}dd|dD}tdd|D}tdd|D}||fS) z6Check if the request accepts the required media types.acceptcSg|]}|qSr-strip.0 media_typer-r-r. 'zGStreamableHTTPServerTransport._check_accept_headers..,cs|]}|tVqdSN) startswithrZrr-r-r. )zFStreamableHTTPServerTransport._check_accept_headers..csrr)rCONTENT_TYPE_SSErr-r-r.r*rrPrdsplitany)r3rc accept_header accept_typeshas_jsonhas_sser-r-r._check_accept_headers$s z3StreamableHTTPServerTransport._check_accept_headerscCs>|jdd}dd|dddD}tdd |DS) z2Check if the request has the correct Content-Type.z content-typercSrr-rrpartr-r-r.r1rzEStreamableHTTPServerTransport._check_content_type..;rrcss|]}|tkVqdSr)rZrr-r-r.r3szDStreamableHTTPServerTransport._check_content_type..r)r3rc content_typecontent_type_partsr-r-r._check_content_type.sz1StreamableHTTPServerTransport._check_content_typec sZj}|dur tdze|\}}|r|s,dtj}||||IdHWdS|sDdtj}||||IdHWdS|IdH} zt | } Wn+t j y}} zdt | tj t}||||IdHWYd} ~ WdSd} ~ wwzt| } Wn*ty} zdt | tj t}||||IdHWYd} ~ WdSd} ~ wwt| jto| jjdk} | rjr|}|r|jkrdtj}||||IdHWdSn ||IdHsWdSt| jtsdtj}||||IdHt|d }t| |d }||IdHWdSt | jj t!j"t#d j$<j$d j%rt|d }t| |d }||IdHzzPd}2z!3dHW}t|j&jt't(Brc|j&}nt)*d |j&jjqN6|r|}||||IdHnt)+ddtj,}||||IdHWn t-yt).ddtj,t/}||||IdHYn wW0IdHWdSW0IdHWdS0IdHwt!j"t1t t fd \}fdd}ddt2djrt3jini}t4|||d}zBt!54IdH&}|6||||t|d }t| |d }||IdHWdIdHn1IdHs?wYWWdSWWdSt-yrt).d7IdH|7IdH0IdHYWdSwt-y}z-t).dd|tj,t/}||||IdH|r|t-|IdHWYd}~dSd}~ww)z2Handle POST requests containing JSON-RPC messages.NBNo read stream writer available. Ensure connect() is called first.zNNot Acceptable: Client must accept both application/json and text/event-streamz=Unsupported Media Type: Content-Type must be application/jsonz Parse error: zValidation error: initialize(Not Found: Invalid or expired session ID)request_context)metadatarrlz received: z1No response message received before stream closedz.Error processing request: No response receivedzError processing JSON responsezError processing requestc s6zz\4IdHI4IdH-2z3dHW}|}|IdHt|jjttBr1nq6WdIdHn 1IdHsCwYWdIdHn 1IdHsXwYWntylt dYnwWt d IdHdSWt d IdHdSt d IdHw)NzError in SSE writerzClosing SSE writer) rjru isinstancer%rootr!rrnro exceptionrprrrfrirkrequest_stream_readerr3sse_stream_writerr-r. sse_writers4 R    zFStreamableHTTPServerTransport._handle_post_request..sse_writerno-cache, no-transform keep-alivez Cache-Control ConnectionrQcontentdata_sender_callablerPzSSE response errorzError handling POST requestzError handling POST request: )8r<rGrr`r NOT_ACCEPTABLErUNSUPPORTED_MEDIA_TYPEbodyjsonloadsJSONDecodeErrorr, BAD_REQUESTrrmodel_validater rrrr rzrArer|_validate_request_headersrbACCEPTEDrrrurTanyiocreate_memory_object_streamr$rIrBr%r!rrorprUINTERNAL_SERVER_ERRORrnrrrrdictrr\r create_task_group start_soonrm)r3rsrcrtruwriterrrrr raw_messageer%is_initialization_requestrequest_session_idrsession_messagerarfsse_stream_readerrrPtgerrr-rr.r}5s             (  :  z2StreamableHTTPServerTransport._handle_post_requestc s|j}|dur td|\}}|s)dtj}||j|j|IdHdS||IdHs4dS|j t }rH |||IdHdSddt d}jrVj|t<tjvrodtj}||j|j|IdHdStjtttfd\} fd d } t| | |d }z||j|j|IdHWdStytd IdH| IdHtIdHYdSw) a Handle GET request to establish SSE. This allows the server to communicate to the client without the client first sending data via HTTP POST. The server can send JSON-RPC requests and notifications on this stream. Nrz4Not Acceptable: Client must accept text/event-streamrrrz4Conflict: Only one SSE stream is allowed per sessionrc sBzzbtjtdjt<jtd}4IdH>|4IdH"|2z3dHW}|}|IdHq$6WdIdHn 1IdHsIwYWdIdHn 1IdHs^wYWntyrt dYnwWt d tIdHdSWt d tIdHdSt d tIdHw)NrrlzError in standalone SSE writerzClosing standalone SSE writer) rrr$rIGET_STREAM_KEYrjrurnrorrprr)standalone_stream_readerrfrir3rr-r.standalone_sse_writer0s, R    zPStreamableHTTPServerTransport._handle_get_request..standalone_sse_writerrz Error in standalone SSE response)r<rGrr`r rrsrtrrPrdLAST_EVENT_ID_HEADER_replay_eventsrrAr\rrICONFLICTrrrr,r rnrorrmrr) r3rcrur_rrr5rPrrr-rr.r~sZ    z1StreamableHTTPServerTransport._handle_get_requestcs||js|dtj}||j|j|IdHdS|||IdHs#dS|IdH|dtj }||j|j|IdHdS)z8Handle DELETE requests for explicit session termination.z5Method Not Allowed: Session termination not supportedN) rAr`r METHOD_NOT_ALLOWEDrsrtr terminaterbOK)r3rcrurr-r-r.rXs z4StreamableHTTPServerTransport._handle_delete_requestc sd|_td|jt|j}|D] }||IdHq|jz:|j dur4|j IdH|j durA|j IdH|j durN|j IdH|j dur^|j IdHWdSWdSty{}ztd|WYd}~dSd}~ww)zTerminate the current session, closing all streams. Once terminated, all requests with this session ID will receive 404 Not Found. TzTerminating session: NError closing streams: )rJroinforAlistrIkeysrrclearr<rmr=r?r>rnrp)r3request_stream_keyskeyrr-r-r.ros*     z'StreamableHTTPServerTransport.terminatecsHtdd}|jr|j|t<|jdtj|d}||j|j|IdHdS)z Handle unsupported HTTP methods.zGET, POST, DELETE)rQAllowzMethod Not Allowed)rPN)rZrAr\r`r rrsrt)r3rcrurPrr-r-r.rs z9StreamableHTTPServerTransport._handle_unsupported_requestcs2|||IdHs dS|||IdHsdSdS)NFT)_validate_session_validate_protocol_version)r3rcrur-r-r.rs z7StreamableHTTPServerTransport._validate_request_headerscsx|jsdS||}|s!|dtj}||j|j|IdHdS||jkr:|dtj}||j|j|IdHdSdS)z'Validate the session ID in the request.TzBad Request: Missing session IDNFr)rArer`r rrsrtr|)r3rcrurrr-r-r.rs&  z/StreamableHTTPServerTransport._validate_sessioncsj|jt}|dur t}|tvr3dt}|d|dd|tj}||j |j |IdHdSdS)z4Validate the protocol version header in the request.Nz, z+Bad Request: Unsupported protocol version: z. zSupported versions: FT) rPrdMCP_PROTOCOL_VERSION_HEADERrrjoinr`r rrsrt)r3rcruprotocol_versionsupported_versionsrr-r-r.rs   z8StreamableHTTPServerTransport._validate_protocol_versionr5csLjsdSz{ddtd}jrj|t<tjtttfd\}fdd}t|||d}z@z ||j |j |IdHWnt yPt d YnwWIdH|IdHWdSWIdH|IdHWdSIdH|IdHwt yt d d tjt}||j |j |IdHYdSw) z Replays events that would have been sent after the specified event ID. Only used when resumability is enabled. Nrrrrc s$z4IdHmdtddffdd }|IdH}|ri|jvritjtdj|<j|d}|4IdH"|2z3dHW}|}|IdHq?6WdIdHn 1IdHsdwYWdIdHWdS1IdHs{wYWdStyt dYdSw)Nrfr1cs |}|IdHdSr)rjrurrr-r. send_events zWStreamableHTTPServerTransport._replay_events..replay_sender..send_eventrrlzError in replay sender) r$r7rIrrrjrurnror)rr0 msg_readerrfrirCr5r3rr-r. replay_senders$ *2 zCStreamableHTTPServerTransport._replay_events..replay_senderrzError in replay responsezError replaying events)rHrrAr\rrrr,r rsrtrnrorrmr`r rr)r3r5rcrurPrrrr-rr.rsP    z,StreamableHTTPServerTransport._replay_eventscstjttBd\}}tjtd\}|_|__|_t4IdH}fdd}| |zV||fVWt j D] } |IdHqDj z|IdH|IdHIdH|IdHWnjty}ztd|WYd}~n`d}~wwt j D] } |IdHqj z|IdH|IdHIdH|IdHWwty}ztd|WYd}~wd}~wwWdIdHdSWdIdHdS1IdHswYdS)zContext manager that provides read and write streams for a connection. Yields: Tuple of (read_stream, write_stream) for bidirectional communication rNc sDz2z3dHW}|j}d}t|jttBr t|jj}|}n|jdur7t|jtr7|jj dur7t|jj }|dur=|nt }d}j rYj ||IdH}t d|d||jvrzj|dt||IdHWqtjtjfyj|dYqwtd|dq6WdStyt dYdSw)NzStored z from rzRequest stream z not found for message. Still processing message as the client might reconnect and replay.zError in message router)r%rrr!rr,rTrrrelated_request_idrrHr4rorprIrur$rBrokenResourceErrorClosedResourceErrorrqloggingrnr)rr%target_request_id response_idrequest_stream_idr&r3write_stream_readerr-r.message_router=sJ     $   /z=StreamableHTTPServerTransport.connect..message_routerr)rrrrnr<r=r?r>rrrrIrrrrrmrorp)r3read_stream_writer read_stream write_streamrrr0rr-rr.connectsP 5   I.z%StreamableHTTPServerTransport.connect)FNN)r1N)7r'r(r)r*r<r rrnr+r=r r>r?rr,boolr/rrKpropertyrLrr intrrr`rrrbrrer$rjr"rrrrrrtuplerrr}r~rrrrrrrrrrr-r-r-r.r;zs  ' "   H\ G r;)Gr*rrreabcrrcollections.abcrrr contextlibr dataclassesrhttpr ranyio.streams.memoryr r pydanticr sse_starletter starlette.requestsrstarlette.responsesrstarlette.typesrrrmcp.server.transport_securityrrmcp.shared.messagerrmcp.shared.versionr mcp.typesrrrrrrrrr r!r" getLoggerr'ror\rrrZrrcompilerEr,r8r9r$r:r/r;r-r-r-r.sF         4   &