i?3dZddlmZddlmZmZddlmZddlm 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&m'Z'e rddl(m)Z)m*Z*e&e+Z,GddZ-y)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)log_server_banner) get_loggertemporary_log_level)FastMCP Transportc*eZdZdZ d d dZ d d dZ d d dZddZ d ddZ d ddZ d dd Z y)TransportMixinzMixin providing transport-related methods for FastMCP. Includes HTTP/stdio/SSE transport handling and custom HTTP routes. NcJK|tjj}|tjj}|dvrt d||dk(r|j dd|i|d{y|dvr|j d||d|d{yt d|757w) 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 show_banner>rrr ) transportr")fastmcpsettingsshow_server_bannerr# ValueErrorrun_stdio_asyncrun_http_asyncselfr#r"transport_kwargss o/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/mixins/transport.py run_asynczTransportMixin.run_async(s  !**==K  ((22I G G29+>? ?  &$&&'"  < <%$%%#'#   29+>? ?   s$A'B#)B*!B# B! B#!B#c \tjt|j|fd|i|y)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)anyiorunrr/r+s r.r2zTransportMixin.runJs6   ( #   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"}) ``` c Zjjt||S)N)endpointmethodsnameinclude_in_schema)_additional_http_routesappendr)fnr9r7r8pathr,s r. decoratorz.TransportMixin.custom_route..decorators7  ( ( / /#&7  Ir3)r<(Callable[[Request], Awaitable[Response]]returnr?r$)r,r=r7r8r9r>s````` r. custom_routezTransportMixin.custom_routeas&D 8  5  r3c,t|jS)aGet all additional HTTP routes including from providers. Returns a list of all custom HTTP routes from this server and from all providers that have HTTP routes (e.g., FastMCPProvider). Returns: List of Starlette BaseRoute objects )listr:)r,s r._get_additional_http_routesz*TransportMixin._get_additional_http_routessD0011r3c Kddlm}m}|r t||d} t |5|j 4d{t 4d{\}}|rdnd} tjd|jd | |jj|||jjtd  | d{dddd{dddd{ddd||y77797+#1d{7swY;xYw72#1d{7swYBxYw#1swYFxYw#||wxYww)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_transportserverrN (stateless)Starting MCP server z with transport 'stdio'T) tools_changed)notification_options) stateless)fastmcp.server.contextrFrGrr_lifespan_managerr loggerinfor8 _mcp_serverr2create_initialization_optionsr ) r,r" log_levelrOrFrGtoken read_stream write_streammodes r.r)zTransportMixin.run_stdio_asyncs, J  T *g& #$Y/1133+~~1L+|1:~ 2499-?VW[V\]#..22'( ,,JJ5H266"K '03    .~430& E "%4-  .~~~43330/& E "sE ED8DD8D#D D#A5D D D  D#D D## D8.D!/D83E; ED8D#D  D# D DD D#!D8#D5 )D,*D5 1D88E=E EEc K| | | } | tjj} | r|dk(r td|xstjj}|xstjj }|xstjj j} |j|||| | } |r t||xsi}dddd }|j|d |vr d |vr| |d <t|5|j4d{tj| f||d |}tj|}t!| j"d dj%d}| rdnd}t&j)d|j*d||d|d|d| |j-d{dddd{dddy777#1d{7swY!xYw#1swYyxYww)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 Nrz-SSE transport does not support stateless mode)r=r# middleware json_responsestateless_httprHronzwebsockets-sansio)timeout_graceful_shutdownlifespanws log_configrV)hostportr=rK/rJrLz with transport z on http://:)r%r&r^r(rdrerVlowerhttp_apprupdaterrQuvicornConfigServergetattrstatelstriprRrSr8serve)r,r"r#rdrerVr=uvicorn_configr\r]r^rOdefault_log_level_to_useappuvicorn_config_from_user config_kwargsconfigrIrZs r.r*zTransportMixin.run_http_asyncs:  ^%;&N  !$--<? ?r3)NN) r,rr#zTransport | Noner" bool | Noner-rr@None)NT) r,rr=strr7z list[str]r8 str | Noner9boolr@z^Callable[[Callable[[Request], Awaitable[Response]]], Callable[[Request], Awaitable[Response]]])r,rr@zlist[BaseRoute])TNF) r,rr"rrVrrOrr@r) TrNNNNNNNNN)r,rr"rr#)Literal['http', 'streamable-http', 'sse']rdrre int | NonerVrr=rrrzdict[str, Any] | Noner\list[ASGIMiddleware] | Noner]rr^rrOrr@r)NNNNrNN)r,rr=rr\rr]rr^rr#rrzzEventStore | Noner{rr@r) __name__ __module__ __qualname____doc__r/r2rArDr)r*rir$r3r.rr"s?'+#' @ @# @! @ @  @H'+#'  # !     6 "& 0000 0  0   0d 2! $ (#(#(#(# (#  (#X!?E $0426%)&*!%M%M%M%=M% M%  M%  M%M%.M%0M%#M%$M%M% M%b 26%)&*?E)-%)<@<@<@0<@# <@ $ <@ = <@'<@#<@ <@r3r).r __future__rcollections.abcrr functoolsrtypingrrr r1rkmcp.server.lowlevel.serverr mcp.server.stdior starlette.middlewarer ASGIMiddlewarestarlette.requestsr starlette.responsesrstarlette.routingrrr%fastmcp.server.event_storerfastmcp.server.httprrrfastmcp.utilities.clirfastmcp.utilities.loggingrrfastmcp.server.serverrrrrRrr$r3r.rse3"/.. :)=&(.1 4E8 H q@q@r3