kKj/UUddlmZddlmZmZmZmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.m/Z/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>erddl?m@Z@e>eAZBdZCGdde ZDGdd ZEd:d!ZFd;d"ZGd;d#ZHdd&ZKd:d'ZLd=d(ZMd?d)ZNd@d*ZOGd+d,ZPe d-d./ZQd0eRd1<Gd2d3e$ZSe dAd4ZTGd5d6ZU dB dCd7ZV dD dEd8ZW dF dGd9ZXy.)H) annotations)AsyncGeneratorCallable GeneratorSequence)asynccontextmanagercontextmanager) ContextVar) fnmatchcase) ip_address) TYPE_CHECKINGAny)urlsplit)uuid4)build_resource_metadata_url)LifespanResultT)SseServerTransport) EventStore)StreamableHTTPSessionManager)TransportSecuritySettings) Starlette)Headers) Middleware)Request)Response) BaseRouteMountRoute)ASGIAppLifespanReceiveScopeSend) AuthProvider)RequireAuthMiddleware)SessionScopedEventStore) get_logger)FastMCP)z 127.0.0.1 localhostz::1ceZdZdZ d dfd ZeddZejddZxZS) #FastMCPStreamableHTTPSessionManagerzGSession manager that scopes resumability storage per transport session.c>d|_t| ||||||y)N)app event_store json_response statelesssecurity_settingsretry_interval)_shared_event_storesuper__init__)selfr-r.r/r0r1r2 __class__s d/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/http.pyr5z,FastMCPStreamableHTTPSessionManager.__init__,s17;  #'/)  cn|jyt|jtjS)N) session_id)r3r&rhexr6s r8r.z/FastMCPStreamableHTTPSessionManager.event_store?s,  # # +'t'?'?EGKKXXr9c||_yN)r3)r6r.s r8r.z/FastMCPStreamableHTTPSessionManager.event_storeGs #. r9)NFFNN)r-rr.EventStore | Noner/boolr0rAr1z TransportSecuritySettings | Noner2 int | NonereturnNone)rCr@)r.r@rCrD) __name__ __module__ __qualname____doc__r5propertyr.setter __classcell__)r7s@r8r+r+)sQ *.#>B%)  '     <  #   &YY//r9r+c eZdZdZddZddZy)StreamableHTTPASGIAppz>ASGI application wrapper for Streamable HTTP server transport.c||_yr?)session_manager)r6rOs r8r5zStreamableHTTPASGIApp.__init__Os .r9c$K |j td|jj|||d{y7#t$rA}t|dk(r-tj d|dd}t|d||d}~wwxYww)Nz6Task group is not initialized. Make sure to use run().z(Original RuntimeError from mcp library: Texc_infoaFastMCP's StreamableHTTPSessionManager task group was not initialized. This commonly occurs when the FastMCP application's lifespan is not passed to the parent ASGI application (e.g., FastAPI or Starlette). Please ensure you are setting `lifespan=mcp_app.lifespan` in your parent app's constructor, where `mcp_app` is the application instance returned by `fastmcp_instance.http_app()`. \nFor more details, see the FastMCP ASGI integration documentation: https://gofastmcp.com/deployment/asgiz\nOriginal error: )rO RuntimeErrorhandle_requeststrloggererror)r6scopereceivesendenew_error_messages r8__call__zStreamableHTTPASGIApp.__call__Rs ##+"L&&55eWdK K K 1vQQ >qcBT<"#&7%88KA3#OPVWW) s8B7AAABA B  [w7& r9cX t|}|jxsdS#t$rYywxYw)Nrb)rrshostname)originparseds r8 _origin_hostrs6&! ?? b  s  ))c$||S|dk(ry|dk(ryy)NhttpPhttpsir`)schemeports r8 _origin_portrs&     r9c<d|vr|jdsd|dS|S)Nrgrcrd)rjrts r8_format_origin_hostrs' d{4??3/4&{ Kr9c|jjd} t|}|j}|j r |js|j S|js|js |jr|j S|j j }tt|j}t||}||d|S|d|d|S#t$r|j cYSwxYw)N/://rg)rhrstriprrrsrirrpathqueryfragmentrrpr)rrrrrnnormalized_ports r8_normalize_originrs \\^ " "3 'F&!{{ ==||~ {{fllfoo||~ ]] "F v? @D"640OTF##XSa0 11 ||~sC++DDcBt|jddd|S)Nrrr)rget)rXrns r8_request_originrs$  (F ;[9* r9cDeZdZdZ d ddZd dZd dZd dZy) HostOriginGuardMiddlewarezDValidate Host and Origin headers before requests reach MCP sessions.Ncb||_t|xsd|_t|xsd|_y)Nr`)r-tupler|r)r6r-r|rs r8r5z"HostOriginGuardMiddleware.__init__s/ "=#6B7$_%:;r9cK|ddk7r|j|||d{y|j|}t|}|jdd}t ||s t dd}||||d{y|jd }t ||} |r3|j|| |s t d d }||||d{y|j|||d{y77u7%7 w) Ntyper)rXrnrbzMisdirected Requesti) status_coderzForbidden Origini)r-_allowed_hosts_for_scoperrrrr_origin_allowed) r6rXrYrZr|headersrnresponserrequest_origins r8r]z"HostOriginGuardMiddleware.__call__s =F "((5'40 0 0 55e< &{{62&T=1 53GH5'40 0 0 X&(5 $..v~tL 2DH5'40 0 0 hhugt,,,' 1 1 1 -sGC8C0AC8<C2=AC8C4C8*C6+C82C84C86C8ctt}|j|j|j d}|r!|d}t |s|j |t|S)Nserverr)list DEFAULT_HOSTSextendr|rrxappendr)r6rXr|r server_hosts r8rz2HostOriginGuardMiddleware._allowed_hosts_for_scopes[]+ T//08$  )K' 4$$[1]##r9ct||jryt|}t|r t|ryt ||k(S)NT)rrrrur)r6rrrn origin_hosts r8rz)HostOriginGuardMiddleware._origin_allowedsB 64#7#7 8"6* [ ).?.E (N::r9)NN)r-rr|Sequence[str] | NonerrrCrDr^)rXr"rCztuple[str, ...])rrUrrUrnrUrCrA)rErFrGrHr5r]rrr`r9r8rrsGN /304 < <,<. <  <-. $;r9r http_requestN)defaultzContextVar[Request | None]_current_http_requestceZdZeddZy)StarletteWithLifespanc.|jjSr?)routerlifespan_contextr=s r8lifespanzStarletteWithLifespan.lifespan!s{{+++r9N)rCzLifespan[Starlette])rErFrGrIrr`r9r8rr s ,,r9rc#Ktj|} |tj|y#tj|wxYwwr?)rsetreset)requesttokens r8set_http_requestr&s= ! % %g .E+ ##E*##E*sA 3A A  A ceZdZdZdZdZy)RequestContextMiddlewarezV Middleware that stores each request in a ContextVar and sets transport type. c||_yr?r-)r6r-s r8r5z!RequestContextMiddleware.__init__4s r9cxK|ddk(rsddlm}m}t|djdd}|r||nd} t t |5|j|||d{ddd| ||yy|j|||d{y74#1swY3xYw#| ||wwxYw7#w)Nrrr)reset_transport set_transportr-transport_type)fastmcp.server.contextrrgetattrstaterrr-)r6rXrYrZrrrtransport_tokens r8r]z!RequestContextMiddleware.__call__7s =F " M%U5\%7%79I4PN?MmN;SWO 5%gen5((5'48886#.#O4/((5'40 0 0 965#.#O4/ 1sR7B:B(B%B&B*B(2"B:B8B:BB%!B(( B55B:Nr_r`r9r8rr/s1r9rc^|jdttt||||S)aRCreate a base Starlette app with common middleware and routes. Args: routes: List of routes to include in the app middleware: List of middleware to include in the app debug: Whether to enable debug mode lifespan: Optional lifespan manager for the app Returns: A Starlette application rroutes middlewaredebugr)insertrrrrs r8create_base_apprHs3$a$<=>   r9c g}g}t|d fd |r|j} |j|} |j| |j| |j |} | r t | nd} |j t|t|j| dg|j t|tj|j| nJd fd } |j t|| dg|j t|j|r|j||jj|r|j|td fd }t|||| }|j_||j_d |j_|S)aReturn an instance of the SSE server app. Args: server: The FastMCP server instance message_path: Path for SSE messages sse_path: Path for SSE connections auth: Optional authentication provider (AuthProvider) debug: Whether to enable debug mode routes: Optional list of custom routes middleware: Optional list of middleware Returns: A Starlette application with RequestContextMiddleware cFKj|||4d{}jj|d|djjd{dddd{t S7a7 7#1d{7swYt SxYww)Nrrf) connect_sse _mcp_serverruncreate_initialization_optionsr)rXrYrZstreamsrsses r8 handle_ssez"create_sse_app..handle_sses??5'488G$$((  ""@@B  98 z 9 9888 zsVB!A=B!?BA?B# B!.B/B!?BB!B B BB!mcp_pathNGETendpointmethodsrcpK|j|j|jd{S7wr?)rXrY_send)rrs r8 sse_endpointz$create_sse_app..sse_endpoints(#GMM7??GMMRR RRs ,646cKj4d{ddddd{y77#1d{7swYyxYwwr?)_lifespan_manager)r-rs r8rz create_sse_app..lifespans2++-- .------s:A 3A 7 A 5A A A A A A rr)rXr"rYr!rZr#rCr)rrrCrr-rrCzAsyncGenerator[None, None])rget_middleware get_routesr_get_resource_urlrrrr%required_scopesrhandle_post_message_get_additional_http_routesrrrfastmcp_serverrr)r message_pathsse_pathauthrrr server_routesserver_middlewareauth_middleware auth_routes resource_urlresource_metadata_urlrrr-rrs` @@r8create_sse_apprds.&(M*, \ *C --/ooxo8 [)  1--h7 9E ' 54   .(()     )++(()   S  %    ++  V$;;=>  , $  C &CIICIIN$CII Jr9c >g} g}td|r|j}|j|}| j||j||j |}|r t |nd}rddgngd}| j t|t|j||n%rddgnd}| j t|||r| j|| jj| r"|jdtt| | | r|j| td fd }t| ||| }|j _||j _d |j _|S) aReturn an instance of the StreamableHTTP server app. Args: server: The FastMCP server instance streamable_http_path: Path for StreamableHTTP connections event_store: Optional event store for SSE polling/resumability retry_interval: Optional retry interval in milliseconds for SSE polling. Controls how quickly clients should reconnect after server-initiated disconnections. Requires event_store to be set. Defaults to SDK default. auth: Optional authentication provider (AuthProvider) json_response: Whether to use JSON response format stateless_http: Whether to use stateless mode (new transport per request) debug: Whether to enable debug mode routes: Optional list of custom routes middleware: Optional list of middleware host_origin_protection: Whether to validate Host and Origin headers before requests reach the MCP endpoint. allowed_hosts: Additional hostnames that may appear in the Host header. allowed_origins: Additional browser origins trusted by the request guard. Configure CORS separately when browser JavaScript must read cross-origin responses. Returns: A Starlette application with StreamableHTTP support NrPOSTDELETE)rrrrr)r|rc RKtj_j4d{jj 4d{ dj}t |j jD]} |jd{ dddd{dddd{y77{7,#t$rtjddYhwxYw#j}t |j jD]B} |jd{7#t$rtjddY@wxYwwxYw7#1d{7swYxYw7#1d{7swYyxYww)N)r-r.r2r/r0z7Error terminating streamable-HTTP transport on shutdownTrQ) r+rrOrrr_server_instancesvalues terminate ExceptionrVr) r-sm transportr.r/r2rstateless_httpstreamable_http_apps r8rz,create_streamable_http_app..lifespanPsk.Q""#)'$ / +  $ $ & &  / / 3 3 5 5 )88"&b&:&:&A&A&C!DI'11333"E 6 5 ' & & 5 4$ U%)%)88"&b&:&:&A&A&C!DI'11333$ U%)%"E 6 5 5 5 ' & & &s 4F'CF'FCFE; D%2E;C+C ,C0E;3 F>E9?F F'FF'FC C= :E;<C= =E;3E64E E E E6 E1 .E60E1 1E66E;9F;F FF FF'F$F F$ F'rzstreamable-httpr)rMrrrrrrrr%rrrrrrrrrrr)rstreamable_http_pathr.r2rr/rrrrhost_origin_protectionr|rrrrrrr http_methodsrr-rs` `` `` @r8create_streamable_http_appr sP&(M*,05 --/oo/CoD [)  1--.BC 9E ' 54 #1VX 6O   $.'(() %  .<)  $,$  V$;;=>  )+ /    ,D $  C &CII)CIIN0CII Jr9)rnrUrCrU)rnrUrCrA)rnrUr| Sequence[str]rCrA)rrUrCrU)rrUrrBrCrB)rXr"rnrUrCrU)rrUrr rCrA)rrrCzGenerator[Request, None, None])FN) rzlist[BaseRoute]rzlist[Middleware]rrArzCallable | NonerCr)NFNN)rFastMCP[LifespanResultT]rrUrrUrAuthProvider | NonerrArlist[BaseRoute] | Nonerlist[Middleware] | NonerCr) NNNFFFNNTNN)rr rrUr.r@r2rBrrr/rArrArrArrrrr rAr|rrrrCr)Y __future__rcollections.abcrrrr contextlibrr contextvarsr fnmatchr ipaddressr typingr r urllib.parseruuidrmcp.server.auth.routesrmcp.server.lowlevel.serverrmcp.server.ssermcp.server.streamable_httpr"mcp.server.streamable_http_managerrmcp.server.transport_securityrstarlette.applicationsrstarlette.datastructuresrstarlette.middlewarerstarlette.requestsrstarlette.responsesrstarlette.routingrrrstarlette.typesrr r!r"r#fastmcp.server.authr$fastmcp.server.auth.middlewarer%fastmcp.server.event_storer&fastmcp.utilities.loggingr'fastmcp.server.serverr(rErVrr+rMrprurxrrrrrrrrr__annotations__rrrrrr r`r9r8r-sv"II:" %!>6-LC,,+&(55CC,@>0- H 1  /*F /F!!H"! 2.H8;8;v5? 51 ,I, ++118 $       @!%%)*.| $|||  |  | # |(||D&*!% $ %)*.#'*.,0[ $[[#[ [  [  [[ [ #[([![([*[[r9