$j8NdZddlmZddlmZmZddlmZddlm Z m Z m Z ddl Z ddl Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZddlZdd lmZddlm Z m!Z!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-e rddl.m/Z/m0Z0e,e1Z2GddZ3dS)z-Transport-related methods for FastMCP Server.) annotations) AwaitableCallable)partial) TYPE_CHECKINGAnyLiteralN)NotificationOptions) stdio_server) Middleware)Request)Response) BaseRouteRoute) EventStore)StarletteWithLifespancreate_sse_appcreate_streamable_http_app)Provider)FastMCPProvider)_WrappedProvider)log_server_banner) get_loggertemporary_log_level)FastMCP TransportceZdZdZ d1d2d Z d1d2dZ d3d4dZd5dZ d6d7dZ d8d9d+Z d:d;d0Z dS)<TransportMixinzMixin providing transport-related methods for FastMCP. Includes HTTP/stdio/SSE transport handling and custom HTTP routes. Nselfr transportTransport | None show_banner bool | Nonetransport_kwargsrreturnNonecK|tjj}|tjj}|dvrt d||dkr|jdd|i|d{VdS|dvr|jd||d|d{VdSt d|) a3Run the FastMCP server asynchronously. Args: transport: Transport protocol to use ("stdio", "http", "sse", or "streamable-http") show_banner: Whether to display the server banner. If None, uses the FASTMCP_SHOW_SERVER_BANNER setting (default: True). N>ssehttpstdiostreamable-httpUnknown transport: r*r">r(r)r+)r r")fastmcpsettingsshow_server_bannerr ValueErrorrun_stdio_asyncrun_http_asyncrr r"r$s g/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/mixins/transport.py run_asynczTransportMixin.run_async+s,  !*=K  (2I G G G>9>>?? ?   &$&'"         < < <%$%#'#          >9>>?? ?c Ttjt|j|fd|i|dS)aIRun the FastMCP server. Note this is a synchronous function. Args: transport: Transport protocol to use ("http", "stdio", "sse", or "streamable-http") show_banner: Whether to display the server banner. If None, uses the FASTMCP_SHOW_SERVER_BANNER setting (default: True). r"N)anyiorunrr6r4s r5r:zTransportMixin.runMsP    ( #        r7Tpathstrmethods list[str]name str | Noneinclude_in_schemabool^Callable[[Callable[[Request], Awaitable[Response]]], Callable[[Request], Awaitable[Response]]]c$dfd }|S)a Decorator to register a custom HTTP route on the FastMCP server. Allows adding arbitrary HTTP endpoints outside the standard MCP protocol, which can be useful for OAuth callbacks, health checks, or admin APIs. The handler function must be an async function that accepts a Starlette Request and returns a Response. Args: path: URL path for the route (e.g., "/auth/callback") methods: List of HTTP methods to support (e.g., ["GET", "POST"]) name: Optional name for the route (to reference this route with Starlette's reverse URL lookup feature) include_in_schema: Whether to include in OpenAPI schema, defaults to True Example: Register a custom HTTP route for a health check endpoint: ```python @server.custom_route("/health", methods=["GET"]) async def health_check(request: Request) -> Response: return JSONResponse({"status": "ok"}) ``` fn(Callable[[Request], Awaitable[Response]]r%c `jt||S)N)endpointr=r?rA)_additional_http_routesappendr)rErAr=r?r;rs r5 decoratorz.TransportMixin.custom_route..decoratorsH  ( / /#&7    Ir7)rErFr%rFr-)rr;r=r?rArKs````` r5 custom_routezTransportMixin.custom_routedsBD          r7list[BaseRoute]ct|j}dd}|jD]N}||}t|tr,||jO|S)aGet all additional HTTP routes including from mounted servers. Collects custom HTTP routes registered via ``@server.custom_route()`` from this server **and** from any FastMCP servers reachable through mounted providers (recursively). This ensures that routes defined on a child server are forwarded to the parent's HTTP app when using ``server.mount(child)``. Note: When path collisions occur between a parent and a mounted child, the parent's routes take precedence because they appear first in the returned list. Returns: List of Starlette Route objects providerrr%cht|tr|j}t|t|S)z:Unwrap _WrappedProvider layers to find the inner provider.) isinstancer_inner)rOs r5_unwrap_providerzDTransportMixin._get_additional_http_routes.._unwrap_providers6X'788 +#?X'788 +Or7)rOrr%r)listrI providersrQrextendserver_get_additional_http_routes)rroutesrSrOinners r5rXz*TransportMixin._get_additional_http_routess"#'t'C"D"D      J JH$$X..E%11 J elFFHHIII r7F log_level statelessc Kddlm}m}|rt||d} t |5|4d{Vt 4d{V\}}|rdnd} td|j d | |j |||j td  | d{Vdddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYdddn #1swxYwY||dS#||wxYw)zRun the server using stdio transport. Args: show_banner: Whether to display the server banner log_level: Log level for the server stateless: Whether to run in stateless mode (no session initialization) r)reset_transport set_transportrWr*N (stateless)Starting MCP server z with transport 'stdio'T) tools_changed)notification_options)r\)fastmcp.server.contextr^r_rr_lifespan_managerr loggerinfor? _mcp_serverr:create_initialization_optionsr ) rr"r[r\r^r_token read_stream write_streammodes r5r2zTransportMixin.run_stdio_asyncs JIIIIIII  + T * * * * g&& #$Y//  1133+~~1L+|1:B~~ ]49]]W[]]#.22'( ,JJ5H266"6"6"K '03                         & OE " " " " "OOE " " " "swE D4D(A:C4"D4 C> >DC> DD4 D! !D4$D! %D4( E 4D88E ;D8<E Er))Literal['http', 'streamable-http', 'sse']hostport int | Noneuvicorn_configdict[str, Any] | None middlewarelist[ASGIMiddleware] | None json_responsestateless_httpc K| | | } | tjj} | r|dkrtd||ntjj}||ntjj}||ntjj} ||||| | } |rt||pi}dddd }| |d |vr d |vr| |d <t|5| 4d{Vtj| f||d |}tj|}t!| jd dd}| rdnd}t&d|jd||d|d|d| |d{Vdddd{Vn#1d{VswxYwYddddS#1swxYwYdS)apRun the server using HTTP transport. Args: transport: Transport protocol to use - "http" (default), "streamable-http", or "sse" host: Host address to bind to (defaults to settings.host) port: Port to bind to (defaults to settings.port) log_level: Log level for the server (defaults to settings.log_level) path: Path for the endpoint (defaults to settings.streamable_http_path or settings.sse_path) uvicorn_config: Additional configuration for the Uvicorn server middleware: A list of middleware to apply to the app json_response: Whether to use JSON response format (defaults to settings.json_response) stateless_http: Whether to use stateless HTTP (defaults to settings.stateless_http) stateless: Alias for stateless_http for CLI consistency Nr(z-SSE transport does not support stateless mode)r;r rvrxryr`onzwebsockets-sansio)timeout_graceful_shutdownlifespanws log_configr[)rqrrr;rb/rarcz with transport z on http://:)r.r/ryr1rqrrr[lowerhttp_apprupdaterrguvicornConfigServergetattrstatelstriprhrir?serve)rr"r rqrrr[r;rtrvrxryr\default_log_level_to_useappuvicorn_config_from_user config_kwargsconfigrWros r5r3zTransportMixin.run_http_asyncs0:  ^%;&N  !$-9>>?? ?r7)NN) rrr r!r"r#r$rr%r&)NT) rrr;r<r=r>r?r@rArBr%rC)rrr%rM)TNF) rrr"rBr[r@r\rBr%r&) Tr)NNNNNNNNN)rrr"rBr rprqr@rrrsr[r@r;r@rtrurvrwrxr#ryr#r\r#r%r&)NNNNr)NN)rrr;r@rvrwrxr#ryr#r rprrrrsr%r) __name__ __module__ __qualname____doc__r6r:rLrXr2r3rr-r7r5rr%s;'+#' @ @ @ @ @H'+#'     6 "& 00000d    H! $ (#(#(#(#(#X!?E $0426%)&*!%O%O%O%O%O%f 26%)&*?E)-%)>@>@>@>@>@>@>@r7r)4r __future__rcollections.abcrr functoolsrtypingrrr r9rmcp.server.lowlevel.serverr mcp.server.stdior starlette.middlewarer ASGIMiddlewarestarlette.requestsr starlette.responsesrstarlette.routingrrr.fastmcp.server.event_storerfastmcp.server.httprrrfastmcp.server.providers.baser)fastmcp.server.providers.fastmcp_providerr)fastmcp.server.providers.wrapped_providerrfastmcp.utilities.clirfastmcp.utilities.loggingrrfastmcp.server.serverrrrrhrr-r7r5rs 33""""""////////.......... ::::::))))))======&&&&&&((((((........111111 322222EEEEEEFFFFFF333333EEEEEEEE988888888 H  L@L@L@L@L@L@L@L@L@L@r7