^sjdZddlZddlmZddlmZddlmZddlm Z m Z ddl Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZej:eZGd dZ y)a SSE Server Transport Module This module implements a Server-Sent Events (SSE) transport layer for MCP servers. Example usage: ``` # Create an SSE transport at an endpoint sse = SseServerTransport("/messages") # Create Starlette routes for SSE and message handling routes = [ Route("/sse", endpoint=handle_sse), Route("/messages", endpoint=handle_messages, methods=["POST"]) ] # Define handler functions async def handle_sse(request): async with sse.connect_sse( request.scope, request.receive, request._send ) as streams: await app.run( streams[0], streams[1], app.create_initialization_options() ) async def handle_messages(request): await sse.handle_post_message(request.scope, request.receive, request._send) # Create and run Starlette app starlette_app = Starlette(routes=routes) uvicorn.run(starlette_app, host="0.0.0.0", port=port) ``` See SseServerTransport class documentation for more details. N)asynccontextmanager)Any)quote)UUIDuuid4)MemoryObjectReceiveStreamMemoryObjectSendStream)ValidationError)EventSourceResponse)Request)Response)ReceiveScopeSendceZdZUdZeed<eeee je zfed<deddffd Z e ded ed efd Zded ed eddfd ZxZS) SseServerTransporta SSE server transport for MCP. This class provides _two_ ASGI applications, suitable to be used with a framework like Starlette and a server like Hypercorn: 1. connect_sse() is an ASGI application which receives incoming GET requests, and sets up a new SSE stream to send server messages to the client. 2. handle_post_message() is an ASGI application which receives incoming POST requests, which should contain client messages that link to a previously-established SSE session. _endpoint_read_stream_writersendpointreturnNcnt|||_i|_tj d|y)z Creates a new SSE server transport, which will direct the client to POST messages to the relative or absolute URL given. z.SseServerTransport initialized with endpoint: N)super__init__rrloggerdebug)selfr __class__s A/root/.hermes/venv/lib/python3.12/site-packages/mcp/server/sse.pyrzSseServerTransport.__init__Is3 !$&! EhZPQscopereceivesendc( K|ddk7r tjdtdtjdt j d\}}t j d\}t }t|jd|j ||j|<tjd|t j dtttf\ } fd } t j4d{} t|| } tjd | j!| |||tjd ||fdddd{y7g7#1d{7swYyxYww) Ntypehttpz%connect_sse received non-HTTP requestz)connect_sse can only handle HTTP requestszSetting up SSE connectionrz ?session_id=zCreated new session with ID: c Ktjd4d{4d{jddd{tjd23d{}tjd|jd|jdddd{O777q7P7 6dddd{7n#1d{7swYnxYwdddd{7y#1d{7swYyxYww) NzStarting SSE writerr)eventdatazSent endpoint event: zSending message via SSE: messageT)by_alias exclude_none)rrr"model_dump_json)r) session_urisse_stream_writerwrite_stream_readers r sse_writerz2SseServerTransport.connect_sse..sse_writerms  LL. /(  *=  ',,z;-WXXX 4[MBC%8  'LL#)B: *B>-AC.B< /C4D 6C88C:B><C>C? C8 C  C8C# CC# C8& D 1C42D 8D >D ?D D )contentdata_sender_callablezStarting SSE response taskzYielding read and write streams)rerror ValueErrorranyiocreate_memory_object_streamrrrhexrdictstrrcreate_task_groupr start_soon)rr r!r"read_stream_writer read_stream write_stream session_idsse_stream_readerr0tgresponser-r.r/s @@@r connect_ssezSseServerTransport.connect_sseTsa =F " LL@ AHI I 01+0*K*KA*N'K,1,M,Ma,P) )W t~~./|JNN;KL 0B!!*- 4ZLAB/4/P/P tCH~0 ,, "**, . .*) H LL5 6 MM(E7D 9 LL: ; - - . . . . . . .sID FE9FAE=( F3E;4F;F=FF F FcKtjdt||}|jj d}|4tj dt dd}||||d{S t|}tjd||jj |}|s7tj d |t d d }||||d{S|jd{} tjd|  tjj| } tjd| tjd| t dd}||||d{|j!| d{y77#t$r;tj d |t d d}||||d{7cYSwxYw77#t$r\} tjd| t dd}||||d{7|j!| d{7Yd} ~ yd} ~ wwxYw77w)NzHandling POST messager?z#Received request without session_idzsession_id is requiredi) status_code)r7zParsed session ID: zReceived invalid session ID: zInvalid session IDzCould not find session for ID: zCould not find sessionizReceived JSON: zValidated client message: zFailed to parse message: zCould not parse messagezSending message to writer: Accepted)rrr query_paramsgetwarningr rr4rjsontypesJSONRPCMessagemodel_validater r3r") rr r!r"requestsession_id_paramrBr?writerrKr)errs rhandle_post_messagez&SseServerTransport.handle_post_messages-  ,-%)"//33LA  # NN@ A 8cJH!%$77 7 8"23J LL.zl; < **..z: NNrisU"H* R$-&(00   8 $z#z#r