~h->UddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2e rddl3m4Z4e2e5Z6e ddZ7de8d<GddeZ9ed%dZ:Gdd Z; d&d!Z< d' d(d"Z= d) d*d#Z> d+ d,d$Z?y)-) annotations)AsyncGeneratorCallable Generator)asynccontextmanagercontextmanager) ContextVar) TYPE_CHECKING)AuthContextMiddleware)BearerAuthBackendRequireAuthMiddleware)create_auth_routes)LifespanResultT)SseServerTransport) EventStore)StreamableHTTPSessionManager) Starlette) Middleware)AuthenticationMiddleware)Request)Response) BaseRouteMountRoute)LifespanReceiveScopeSend) OAuthProvider) get_logger)FastMCP http_requestN)defaultzContextVar[Request | None]_current_http_requestceZdZeddZy)StarletteWithLifespanc.|jjSN)routerlifespan_context)selfs O/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/server/http.pylifespanzStarletteWithLifespan.lifespan*s{{+++N)returnr)__name__ __module__ __qualname__propertyr-r.r,r&r&)s ,,r.r&c#Ktj|} |tj|y#tj|wxYwwr()r$setreset)requesttokens r,set_http_requestr:/s= ! % %g .E+ ##E*##E*sA 3A A  A ceZdZdZdZdZy)RequestContextMiddlewarez= Middleware that stores each request in a ContextVar c||_yr(app)r+r?s r,__init__z!RequestContextMiddleware.__init__=s r.cK|ddk(r9tt|5|j|||d{dddy|j|||d{y7)#1swYyxYw7w)Ntypehttp)r:rr?)r+scopereceivesends r,__call__z!RequestContextMiddleware.__call__@sq =F "!'%.1 5hhugt444 5 5((5'40 0 05 5 5 1s7A0A"A A"A0A.A0 A""A+'A0N)r0r1r2__doc__r@rGr4r.r,r<r<8s1r.r<c g}g}g}ttt|ttg}|jxsg}|j t ||j|j|j|j|||fS)zSet up authentication middleware and routes if auth is enabled. Args: auth: The OAuthProvider authorization server provider Returns: Tuple of (middleware, auth_routes, required_scopes) )provider)backend)rJ issuer_urlservice_documentation_urlclient_registration_optionsrevocation_options) rrr r required_scopesextendrrLrMrNrO)auth middleware auth_routesrPs r, setup_auth_middleware_and_routesrUHs$&J#%K!#O  $%t4  () J**0bO&*&D&D(,(H(H#66   {O 33r.c\|jttt||||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 routesrSdebugr-)appendrr<r&rWs r,create_base_appr[ns1$j!9:;   r.c g}g}t|dfd |rt|\} } } |j| |j| |jt |t | dg|jt |t j| nJdfd } |jt || dg|jt |j|r|j||jj|r|j|t|||} | j_ || 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 auth provider 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)Nr) connect_sse _mcp_serverruncreate_initialization_optionsr)rDrErFstreamsserversses r, handle_ssez"create_sse_app..handle_sses??5'48  G$$((  ""@@B     z        zsVB!A=B!?BA?B# B!.B/B!?BB!B B BB!GET)endpointmethodsr>cpK|j|j|jd{S7wr()rDrE_send)r8rfs r, sse_endpointz$create_sse_app..sse_endpoints(#GMM7??GMMRR RRs ,646)rXrSrY)rDrrErrFrr/r)r8rr/r) rrUrQrZrr rhandle_post_message_additional_http_routesr[statefastmcp_serverpath)rd message_pathsse_pathrRrYrXrS server_routesserver_middlewareauth_middlewarerTrPrlr?rfres` @@r,create_sse_apprwsd.&(M*, \ *C ,T 2 6o [)  1 .z?K    )#*A*A?S   S  %    ++  V$778  , $ C &CIICIIN Jr.c Lg} g} t|j||| dfd } |rXt|\} } }| j| | j| | j t |t | |n| j t || |r| j|| j|j|r| j|tdfd }t| | ||}||j_ ||j_ |S)azReturn 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 session management auth: Optional auth provider 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 Returns: A Starlette application with StreamableHTTP support )r? event_store json_response statelesscK j|||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: T)exc_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: )handle_request RuntimeErrorstrloggererror)rDrErFenew_error_messagesession_managers r,handle_streamable_httpz:create_streamable_http_app..handle_streamable_https !00F F F 1vQQ >qcBT<"#&7%88KA3#OPVWW) s0A0#!#A0# A-cKj4d{ddddd{y77#1d{7swYyxYwwr()ra)r?rs r,r-z,create_streamable_http_app..lifespanWsE"&&(          s:A 3A 7 A 5A A A A A A rW)rDrrErrFrr/None)r?rr/zAsyncGenerator[None, None]) rr`rUrQrZrr rnrr[rorprq)rdstreamable_http_pathryrRrzstateless_httprYrXrSrtrurrvrTrPr-r?rs @r,create_streamable_http_apprsQ6&(M*,3   # O&.2 8 ,T 2 6o [)  1  $)*@/R    $*  V$778  , $  C &CII)CIIN Jr.)r8rr/zGenerator[Request, None, None])rRrr/z3tuple[list[Middleware], list[BaseRoute], list[str]])FN) rXzlist[BaseRoute]rSzlist[Middleware]rYboolr-zCallable | Noner/r&)NFNN)rdFastMCP[LifespanResultT]rrrrsrrROAuthProvider | NonerYrrXlist[BaseRoute] | NonerSlist[Middleware] | Noner/r&)NNFFFNN)rdrrrryzEventStore | NonerRrrzrrrrYrrXrrSrr/r&)@ __future__rcollections.abcrrr contextlibrr contextvarsr typingr 'mcp.server.auth.middleware.auth_contextr &mcp.server.auth.middleware.bearer_authr r mcp.server.auth.routesrmcp.server.lowlevel.serverrmcp.server.ssermcp.server.streamable_httpr"mcp.server.streamable_http_managerrstarlette.applicationsrstarlette.middlewarer#starlette.middleware.authenticationrstarlette.requestsrstarlette.responsesrstarlette.routingrrrstarlette.typesrrrrfastmcp.server.auth.authrfastmcp.utilities.loggingr fastmcp.server.serverr!r0rr$__annotations__r&r:r<rUr[rwrr4r.r,rs"??:" I66-1K,+H&(55::20- H 5? 51 ,I, ++ 1 1 #4 #48#4R $       @"&%)*.e $eee  e  e # e(eeV&*!% %)*.v $vv#v  v  v  v v #v(vvr.