{h d Z ddlZddlZddlZddlmZmZ ddlmZm Z m
Z
ddl m Z ddl
mZ ddlmZ ddlZddlmZmZ dd lmZ dd
lmZ dd lmZ dd lm Z dd
l m Z m Z m 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/m0Z0m1Z1m2Z2m3Z3m4Z4 ejj e6 Z7dZ8dZ9dZ:dZ;dZ.Z `a a ,(@% '45FG
!r0 c | j S )z7Check if this transport has been explicitly terminated.)rR )r7 s r1
is_terminatedz+StreamableHTTPServerTransport.is_terminated s r0
error_message status_code
error_codeheadersc dt i}|r|j | | j r| j |t < t ddt || }t
|j dd || S ) z6Create an error response with a simple string message. Content-Typez2.0z server-error)coder' )jsonrpciderrorTby_alias exclude_nonerW rY )CONTENT_TYPE_JSONupdaterH MCP_SESSION_ID_HEADERr r r model_dump_json)r7 rV rW rX rY response_headerserror_responses r1 _create_error_responsez4StreamableHTTPServerTransport._create_error_response s +,=> # #G , 6:6I6I 2 3 & %
* *Dt * L#$
r0 response_messagec dt i}|r|j | | j r| j |t < t |r|j dd nd|| S )z,Create a JSON response from a JSONRPCMessager[ Tr` Nrc )rd re rH rf r rg )r7 rk rW rY rh s r1 _create_json_responsez3StreamableHTTPServerTransport._create_json_response sh +,=> # #G , 6:6I6I 2 3Rb , ,d , Nhl#$
r0 requestc @ |j j t S )z,Extract the session ID from request headers.)rY getrf )r7 rn s r1 _get_session_idz-StreamableHTTPServerTransport._get_session_id s ""#899r0
event_messagec | d|j j dd d}|j r|j |d< |S )z2Create event data dictionary from an EventMessage.r' Tr` )eventdatar^ )r' rg r( )r7 rr
event_datas r1 _create_event_dataz0StreamableHTTPServerTransport._create_event_data sI !))994VZ9[
! ! , 5 5Jt r0
request_idc K || j v rn | j | d j d{ | j | d j d{ | j j |d yy7 J7 $# t $ r t j d Y Bw xY w# | j j |d w xY ww)z/Clean up memory streams for a given request ID.r N z4Error closing memory streams - may already be closed)rQ aclose Exceptionloggerdebugpop)r7 rx s r1 _clean_up_memory_streamsz6StreamableHTTPServerTransport._clean_up_memory_streams s .. .
<++J7:AACCC++J7:AACCC %%))*d; / DC
U ST
U
%%))*d;sV C #B B 'B B B " C B B B%"B( $B%%B( ( CC scopereceivesendc K t || }|j dk( }| j j || d{ }|r |||| d{ y| j r3| j dt j } |||| d{ y|j dk( r | j |||| d{ y|j dk( r| j || d{ y|j dk( r| j || d{ y| j || d{ y7 7 7 7 z7 R7 *7 w)z6Application entry point that handles all HTTP requestsPOST)is_postNz&Not Found: Session has been terminatedGETDELETE) r methodrG validate_requestrR rj r
NOT_FOUND_handle_post_request_handle_get_request_handle_delete_request_handle_unsupported_request)r7 r r r rn r ri responses r1 handle_requestz,StreamableHTTPServerTransport.handle_request s6 %) ..F* #~~ > >wPW > XX 6 6 6 228$$H 5'40 0 0 >>V #++E7GTJ J J
^^u
$**7D9 9 9
^^x
'--gt< < <227DA A A+ Y 6
1
K 9 < As{ E D6+E =D8>)E 'D:()E D<E ,D>-E 4E 6E 8E :E <E >E c |j j dd }|j d D cg c] }|j }}t d |D }t d |D }||fS c c}w )z6Check if the request accepts the required media types.accept ,c 3 F K | ] }|j t y wN)
startswithrd .0
media_types r1 zFStreamableHTTPServerTransport._check_accept_headers..) s _Jz,,->?_ !c 3 F K | ] }|j t y wr )r CONTENT_TYPE_SSEr s r1 r zFStreamableHTTPServerTransport._check_accept_headers..* s ]*j++,<=]r rY rp splitstripany)r7 rn
accept_headerr accept_typeshas_jsonhas_sses r1 _check_accept_headersz3StreamableHTTPServerTransport._check_accept_headers$ ss ++Hb9
=J=P=PQT=UVz
((*V V_R^__]P\]] Ws A1c |j j dd }|j d d j d D cg c] }|j }}t d |D S c c}w )z2Check if the request has the correct Content-Type.z content-typer ;r r c 3 . K | ]
}|t k( y wr )rd )r parts r1 r zDStreamableHTTPServerTransport._check_content_type..3 s L4,,Ls r )r7 rn content_typer content_type_partss r1 _check_content_typez1StreamableHTTPServerTransport._check_content_type. sf **>2> 7C7I7I#7Nq7Q7W7WX[7\ ]tdjjl ] ]L9KLLL ^s A-c F K j }| t d j | \ }}|r|s3 j dt j
} |||| d{ y j
| s3 j dt j } |||| d{ y|j d{ } t j | }
t j |
} t' | j( t* xr | j( j, dk( }
|
ra j. rp j1 | }|r]| j. k7 rN j dt j2 } |||| d{ y j5 || d{ syt' | j( t* se j7 dt j8 } |||| d{ t; | }t= | |
}|j? | d{ yt | j( j@ tC jD tF d jH < jH d jJ r7t; | }t= | |
}|j? | d{ d}2 3 d{ }t' |jL j( tN tP z r|jL } n7tR jU d
|jL j( j, y|r$ j7 | } |||| d{ nGtR jW d j dt jX } |||| d{ ja d{ ytC jD tb t t f d \ } fd}ddtd d j. rtf j. ini }ti ||| } tC jj 4 d{ }|jm |||| t; | }t= | |
}|j? | d{ ddd d{ yy7 @7 7 # t j $ rN} j dt | t j t } |||| d{ 7 Y d} ~ yd} ~ ww xY w# t" $ rN} j dt | t j t$ } |||| d{ 7 Y d} ~ yd} ~ ww xY w7 7 7 7 V7 7 6 E7 %7 # tZ $ rQ tR j] d j dt jX t^ } |||| d{ 7 Y 7w xY w7 '# ja d{ 7 w xY w7 7 {7 n# 1 d{ 7 sw Y ~xY w# tZ $ rd tR j] d jo d{ 7 |jo d{ 7 ja d{ 7 Y yw xY w# tZ $ r}tR j] d j d| t jX t^ } |||| d{ 7 |r#|j? t[ | d{ 7 Y d}~yd}~ww xY 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)metadatar rz z
received: z1No response message received before stream closedz.Error processing request: No response receivedz Error processing JSON responsezError processing requestc K 4 d { 4 d { 2 3 d { } j | }j | d { t | j j t
t z s_ d d d d { d d d d { t j d j d { y 7 7 7 7 6 Y7 M# 1 d { 7 sw Y ]xY w7 T# 1 d { 7 sw Y dxY w# t $ r t j d Y w xY w7 a# t j d j d { 7 w xY ww)NzError in SSE writerzClosing SSE writer) rw r
isinstancer' rootr" r r| r} exceptionr~ r )rr rv rx request_stream_readerr7 sse_stream_writers r1
sse_writerzFStreamableHTTPServerTransport._handle_post_request..sse_writer sN H#4 * *6K * *7L
*
*m-1-D-D]-S
&7&<&$>$3l$B$" %* * * * * %9: " ; ;J GGG# * *
* !I 8M * * * * * * * * % @ (()>?@ H %9: " ; ;J GGGs E D C D C/CC/CCC
C&C C
/C<C= C/C C/
D C-D )E D(E D C/CCCC/C* C! C* &C/-D /D5C8 6D=D D%"D* $D%%D* (E **EEEE no-cache, no-transform
keep-alivez
Cache-Control
Connectionr[ contentdata_sender_callablerY zSSE response errorzError handling POST requestz Error handling POST request: )8rC rO r rj r
NOT_ACCEPTABLEr UNSUPPORTED_MEDIA_TYPEbodyjsonloadsJSONDecodeErrorr. BAD_REQUESTr r model_validater
r r r r! r rH rq r _validate_request_headersrm ACCEPTEDr r r r^ anyiocreate_memory_object_streamr& rQ rI r' r" r r} r~ r_ INTERNAL_SERVER_ERRORr| r r r dictr rf r create_task_group
start_soonr{ )r7 r rn r r writerr r r r raw_messageer' is_initialization_requestrequest_session_idr session_messagerk rr sse_stream_readerr rY tgerrrx r r s ` @@@r1 r z2StreamableHTTPServerTransport._handle_post_request5 s )) > ab b@ $ : :7 C Hg 66e -- ugt444 ++G4 66S 55 ugt444 ! 'D
"jj .
(77 D )37<< (P(xU\UaUaUhUhlxUx %(&&)-)=)=g)F& *.@DDWDW.W#'#>#> F & 0 0$ 'ugt <<< 99'4HHH gllN; 55 '' ugt444 1I"08"L kk/222 W\\__-J050Q0QR^0_`a0bD ! !* -$($9$9*$Ea$H !,,0I"08"L kk/222%D (,$ 0E [ [m%m&;&;&@&@/T`B`a/N#O &ugt <<< %XY#'#>#> L & < <$ 'ugt <<< 77
CCC 8=7X7XY]^acf^fYg7hij7k4!#4H4 &>".$4 HLGZGZ-t/B/BC`b / -)3 # D $ 6 6 8 ; ;B
hwE#8#Q*88*T $kk/ ::: ; ; ;E 5 5 ( ''
66 s1vh7OQ[QgQgitu ugt444
#
66(Q1 **"
ugt444
0 =H 5
3 3[/D = = 9$$%EF # : :2"88& H
#5'48889 D$77
CCCZ;
; ; ; ; ; ! D$$%9:+22444+22444 77
CCC D
: ;22/u500 H
5'40 0 0 kk)C.111
s \!AZ &R&'Z +\!,>Z *R)+Z /\!0Z R,Z R/ T 4BZ U-Z \! Z !U0"Z &\!'AZ .U3/1Z U6!Z %\!&BZ :U9;Z V U?U< U? BV V AV %V&V *Z >W%?Z \!A,Z 1X& X
X&
AXX
X X& X
X& $\!&Z )Z ,Z /T>T TT Z
\! TZ U* >U%U U% Z $\!%U**Z 0Z 3Z 6Z 9Z <U??V V V AW"WW" W( !W""W( %Z (X=X >XZ X&
X
X& X#X X# X& &1ZYZ0Y31Z
Z
ZZ \!ZZ \ A
\)[,*$\\\\!\ \!c ~ K j }| t d j | \ }}|sG j dt j
} ||j |j | d{ y j || d{ sy|j j t x}r j ||| d{ yddt d} j r j |t < t j" v rG j dt j$ } ||j |j | d{ yt' j( t* t, t, f d \ } fd }
t/ | |
|
} ||j |j | d{ y7 H7 07 7 u7 # t0 $ rh t2 j5 d j7 d{ 7 | j7 d{ 7 j9 t d{ 7 Y yw xY ww) 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.
Nr z4Not Acceptable: Client must accept text/event-streamr r r z4Conflict: Only one SSE stream is allowed per sessionr c K t j t d j t < j t d } 4 d { | 4 d { | 2 3 d { }j | }j
| d { 47 E7 <7 37 6 d d d d { 7 n# 1 d { 7 sw Y nxY wd d d d { 7 n# 1 d { 7 sw Y nxY wn$# t $ r t j d Y nw xY wt j d j t d { 7 y # t j d j t d { 7 w xY ww)Nr rz z Error in standalone SSE writerz Closing standalone SSE writer) r r r& rQ GET_STREAM_KEYrw r r| r} r r~ r )standalone_stream_readerrr rv r7 r s r1 standalone_sse_writerzPStreamableHTTPServerTransport._handle_get_request..standalone_sse_writer0 se
D 9>8Y8YZf8ghi8j%%n5+/+@+@ +PQR+S(,
A
A.F
A
A/G A Am &*%<%<]%K
/ 4 4Z @@@
A
AA A 0H
A
A
A
A
A
A
A
A
A
A
C !AB
C <=33NCCC <=33NCCCs
E<A C, BC, CBC B/ B $B
%B (&B/B
B/C, CB B/ B/ C(B+)C/C 5B86C =C C, CC, C( C C($C, +E , D
E D
E -E<=E >E<.E92E53E99E<r z Error in standalone SSE response) rC rO r rj r
r r r r rY rp LAST_EVENT_ID_HEADER_replay_eventsr rH rf r rQ CONFLICTr r r r. r r| r} r r{ r ) r7 rn r r _r r r: rY r r r s ` @r1 r z1StreamableHTTPServerTransport._handle_get_request s% )) > ab b //8
722F))H 7=='//4@ @ @ 33GTBBB $OO / /0D E E= E%%mWdC C C 6& ,
-1-@-@G) * T 2 2 222F##H 7=='//4@ @ @ 05/P/PQUVY[^V^Q_/`ab/c,, D2 '%!6
@7=='//4@ @ @C
A C
D$
AN
A @ ? @#**, , ,#**, , ,//? ? ? @s A1H=5F=6H=G E E<E E
E E E9E4/E<4E99E<c K t dd}| j r| j |t < | j dt j
| } ||j |j | d{ y7 w)z Handle unsupported HTTP methods.zGET, POST, DELETE)r[ AllowzMethod Not Allowed)rY N)rd rH rf rj r
r r r )r7 rn r rY r s r1 r z9StreamableHTTPServerTransport._handle_unsupported_request sq .(
-1-@-@G) *.. ) ) /
w}}goot<<:><>>c K | j sy| j | }|sG| j dt j } ||j
|j | d{ y|| j k7 rG| j dt j } ||j
|j | d{ yy7 \7 w)z'Validate the session ID in the request.Tz Bad Request: Missing session IDNFr )rH rq rj r
r r r r )r7 rn r r r s r1 r z/StreamableHTTPServerTransport._validate_session s "" " 1 1' : "221&&H 7=='//4@ @ @ !4!4 422:$$H 7=='//4@ @ @
A
As% A"C$C%AC:C;CCc : K |j j t }|t }|t vrfdj t }| j
d| dd| z t j } ||j |j | d{ yy7 w)z4Validate the protocol version header in the request.Nz, z+Bad Request: Unsupported protocol version: z. zSupported versions: FT) rY rp MCP_PROTOCOL_VERSION_HEADERr r joinrj r
r r r )r7 rn r protocol_versionsupported_versionsr s r1 r z8StreamableHTTPServerTransport._validate_protocol_version s #??../JK # 9 #> >!%+F!G 22=>N=OrR();(<=>&&H
7=='//4@ @ @
As BBBBr: c Z K j sy ddt d} j r j |t < t j
t t t f d \ } fd}t ||| } ||j |j | d{ j d{ |j d{ y7 5# t $ r t j d Y Sw xY w7 C7 -# j d{ 7 |j d{ 7 w xY w# t $ rd t j d j d t j" t$ } ||j |j | d{ 7 Y yw xY ww)
z
Replays events that would have been sent after the specified event ID.
Only used when resumability is enabled.
Nr r r r c K 4 d { dt dd ffd } j | d { }|r|j vr|t j t d j |< j | d }|4 d { |2 3 d { }j | }j
| d { 4d d d d { y 7 7 7 O7 F7 6 d d d d { 7 .# 1 d { 7 sw Y @xY w7 7# 1 d { 7 sw Y y xY w# t $ r t j d Y y w xY ww)Nrr r5 c f K j | }j | d { y 7 wr )rw r )rr rv r7 r s r1
send_eventzWStreamableHTTPServerTransport._replay_events..replay_sender..send_event s+ )-)@)@ )OJ"3"8"8"D D Ds &1/1r rz zError in replay sender)
r& r= rQ r r rw r r| r} r ) r r4
msg_readerrr rv rJ r: r7 r s r1
replay_senderzCStreamableHTTPServerTransport._replay_events..replay_sender sd ? 0 M MEL ET E
+6*I*I-Yc*d$d %$:O:O)O?D?`?`am?nop?qD 1 1) <)-)>)>y)I!)LJ (2 M M;E !M !M-151H1H 1WJ*;*@*@*L$L$L'M M M %e M!M %M ?s ED CD %D
CAD
C D
C3C C
C &C3?C
C3 D DD ED D
D
C C3 C3! D
,C/-D
3D 9C<:D D
D
D D D D E D E =E?E Er zError in replay responsezError replaying events)rP r rH rf r r r r. r r r r| r} r r{ rj r
r r )
r7 r: rn r rY r r r rJ r s
`` @@r1 r z,StreamableHTTPServerTransport._replay_events s
'' < A!9 * 0G ""151D1D-. 493T3TUYZ]_bZbUc3def3g 0 0
?4 +)%2 H
1 w}}gootDDD (..000'..000 E
= !;<
= 10 (..000'..000 A 5 622(00 H
7=='//4@ @ @ As F+A+D; !C $C%C )D; <D =D; DD; F+C C=:D <C==D D; D; D8DD81D42D88D; ;A$F( F" F(%F+'F((F+c n K t j t t z d \ }}t j t d \ }| _ | _ _ | _ t j 4 d{ } fd}|j | ||f t j j D ] } j | d{ j j |j d{ |j d{ j d{ |j d{ ddd d{ y7 7 7 a7 K7 57 # t $ r"}t j# d| Y d}~Cd}~ww xY w# t j j D ] } j | d{ 7 j j |j d{ 7 |j d{ 7 j d{ 7 |j d{ 7 w # t $ r"}t j# d| Y d}~w d}~ww xY wxY w7 4# 1 d{ 7 sw Y yxY ww)zContext manager that provides read and write streams for a connection.
Yields:
Tuple of (read_stream, write_stream) for bidirectional communication
r Nc K 2 3 d { } | j }d }t |j t t z r"t |j j }|}n[| j Ot | j t r5| j j t | j j }||nt }d }j r?j j || d { }t j d| d| |j v r6 j | d j! t# || d { Pt- j d| d k7 g7 7 &# t$ j& t$ j( f$ r j j+ |d Y w xY w6 y # t. $ r t j1 d Y y w xY ww)NzStored z from r zRequest stream z not found
for message. Still processing message as the client
might reconnect and replay.zError in message router)r' r r r" r r. r^ r r related_request_idr rP r8 r} r~ rQ r r& r BrokenResourceErrorClosedResourceErrorr loggingr| r )r r' target_request_id response_idrequest_stream_idr( r7 write_stream_readers r1 message_routerz=StreamableHTTPServerTransport.connect..message_router= s 2@1D / / o"1"9"9,0)%gllOl4RS*-glloo*>K 1< - !0 8 8 D$.$3$<$<$9%" %4$<$<$O$O$[478P8P8c8c4d 1ARA^,=dr)
$( ,,-1->->-J-JK\^e-f'fH "LL78*FCTBU)V W,0E0EE S&*&;&;?@s G%G F?E2F?C$G 3E54-G ".E9E7E9 G 2F?5G 7E99?F<8G ;F<<G G% G" G%!G""G%r )r r r r| rC rD rF rE r r r rQ r r r r{ r} r~ ) r7 read_stream_writer read_stream write_streamr r
r4 r r s ` @r1 connectz%StreamableHTTPServerTransport.connect s $ +0*K*KN]fLf*ghi*j'K,1,M,Mn,]^_,`) ) $6 '$7! ) **, J @ J @3
@l
MM. )
@!<//!%d&;&;&@&@&B!C CI 77 BBBC%%++-@,33555%,,...-44666&--///OJ @ J @ J @~ C
6.6/ @LL#:1#!>??@ "&d&;&;&@&@&B!C CI 77 BBBC%%++-@,33555%,,...-44666&--/// @LL#:1#!>??@QJ @ J @ J @ J @sD A:J5>E/?J5J F)":J E1 J >E;E3E;)E5*E;E7E;E9E; J5)J *J51J 3E;5E;7E;9E;; F&F! J !F&&J );J$G' %!JI,H
I,3H6
4I, I
I,%I(
&I,+J, J 5J
JJ JJ J5 J2&J) 'J2.J5)FNN)r5 N)7r) r* r+ r, rC r r r| r- rD r rE rF r r. boolr3 r rS propertyrU r r
intr r rj r r rm r rq r& rw r# r r r r r tupler r r r r r r r r r r r r r r/ r0 r1 rB rB z sk VZ/0JKdRYQUL+NY,FG$NUCGM).9D@GMQ3NCdJQ**
*/)->B %!d
%! #'%! $& %!
5t; %!
%!N t *)-
c3h $&
D #---)-
(4/
c3h $&
(:w :3: : c3h < = = =QU = w d t w d t : t PT ,EA# EA EAt EAX\ EAN e@
%ny&@ A "> 2
4
e@ e@r0 rB )Gr, r r reabcr r collections.abcr r r
contextlibr dataclassesr httpr
r anyio.streams.memoryr r pydanticr
sse_starletter starlette.requestsr starlette.responsesr starlette.typesr r r mcp.server.transport_securityr r mcp.shared.messager r mcp.shared.versionr mcp.typesr r r r r r r r r! r" r# getLoggerr) r} rf r r rd r r compilerM r. r> r? r& r@ r3 rB r/ r0 r1 r( s # ? ? * ! R $ - & ( 0 0 E :
8 $ ) 4 & ' & RZZ 12
, 4 89
#
#
LK @ K @r0