~hMddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZmZmZmZmZmZddlZddlZddlZddlmZmZddlmZmZmZmZddlm Z!ddl"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)dd l*m+Z+dd l,m-Z-ddl.m Z ddl/m0Z0ddl1m2Z2m3Z3e0e4Z5eddZ6gdZ7GddedZ8GddejrZ:Gdde:Z;Gdde:Z<Gdde:Z=Gdd e:Z>Gd!d"e>Z?Gd#d$e>Z@Gd%d&e>ZAGd'd(e>ZBGd)d*e>ZCGd+d,e:ZDGd-d.e:ZEed/e6d0e6fd1ZFed/e d0eDfd2ZFed/e!d0eDfd3ZFed/e2d0eEfd4ZFed/eGeHefd0eEfd5ZFed/e%d0e) __class__r1rEs r=__repr__zClientTransport.__repr__^s4>>**+1--r<c Kyw)zClose the transport.Nr;rKs r=closezClientTransport.closebs  sauthoauthNc| tdy)Nz$This transport does not support auth) ValueErrorrErOs r= _set_authzClientTransport._set_authfs  CD D r<)r1r2r3r4abcabstractmethod contextlibasynccontextmanagerrr*rr rFstrrLrNhttpxAuthrrTr;r<r=rr=s ## &} 5 } %$,.#. Eejj77+;;cADHEr<cheZdZdZdeezfdZejde e de e fdZ defdZy) WSTransportzGTransport implementation that connects to an MCP server via WebSockets.urlctjdtdt|tr t |}t|t r|j ds td||_y)NzwWSTransport is a deprecated MCP transport and will be removed in a future version. Use StreamableHttpTransport instead.) stacklevelwszInvalid WebSocket URL provided.) warningswarnDeprecationWarning isinstancerrY startswithrRr^)rEr^s r=__init__zWSTransport.__init__nsW  F  c6 "c(C#s#3>>$+?>? ?r<r@rAc dK ddlm}||j4d{}|\}}t ||fi|4d{}|dddd{dddd{y#t$r tdwxYw7]7D71#1d{7swYAxYw78#1d{7swYyxYww)Nr)websocket_clientzxThe websocket transport is not available. Please install fastmcp[websockets] or install the websockets package manually.)mcp.client.websocketrj ImportErrorr^r )rEr@rj transport read_stream write_streamsessions r=rFzWSTransport.connect_session{s  = $DHH-  (1 %K$\-;          K             sB0A&B0A>B0BBBB BBB B0 B!B0&A;;B0BBB B  B BB0B-!B$ "B-)B0c"d|jdS)Nzr^rKs r=rLzWSTransport.__repr__s*488*C88r<N)r1r2r3r4rYrrhrWrXrr*rr rFrLr;r<r=r]r]ksUQ C&L ## &} 5 } %$"9#9r<r]c @eZdZdZ ddeezdeeefdzdeje dzezdzde je ze zdzdegejfdzf d Zdeje dzezdzfd Zej&d eed eefd Zd efdZy)rzOTransport implementation that connects to an MCP server via Server-Sent Events.Nr^headersrOrPsse_read_timeouthttpx_client_factorycNt|tr t|}t|tr|jds t d||_|xsi|_|j|||_t|ttzrtj|}||_ y)Nhttpz$Invalid HTTP/S URL provided for SSE.secondsrfrrYrgrRr^rurTrwintfloatr5r6rvrEr^rurOrvrws r=rhzSSETransport.__init__s c6 "c(C#s#3>>&+ACD D}"  t$8! &e 4'11:JK  0r<c|dk(rt|j}||_yt|tr t |}||_yNrPrr^rfrYrrOrSs r=rTzSSETransport._set_auth< 7??D c "d#D r<r@rAc Kddlm}i}t|jz|d<|j|jj |d<|j dd>&+AOP P}"  t$8! &e 4'11:JK  0r<c|dk(rt|j}||_yt|tr t |}||_yrrrSs r=rTz!StreamableHttpTransport._set_authrr<r@rAc >Kddlm}i}t|jz|d<|j|j|d<|j dd|j d|d<|j |j |d<||jfd|ji|4d{}|\}}}t||fi|4d{}|dddd{dddd{y7F7,7#1d{7swY)xYw7 #1d{7swYyxYww) Nr)streamablehttp_clientrurvr+rrwrO) mcp.client.streamable_httprrrurvrrwr^rOr ) rEr@rrrmrnro_rps r=rFz'StreamableHttpTransport.connect_sessionsA E(* $4#5 #D i   ,040E0EM, -   4d ; G'5'9'9:P'QM) $  $ $ 0484M4MM0 1( HH    +4 (Kq$\-;                  sB"D$C+%D(DC-DC1 DC/D D%D&D-D/D1D 7C:8D ?DDDD DDc"d|jdS)Nz#>r<r c eZdZdZ ddedeedeeefdzdedzdedzf dZe jd e e d e efd Zd e e d edzfd Zd ZdZd efdZy)r!z Base transport for connecting to an MCP server via subprocess with stdio. This is a base class that can be subclassed for specific command-based transports like Python, Node, Uvx, etc. Ncommandargsenvcwd keep_alivec||_||_||_||_|d}||_d|_d|_tj|_ tj|_ y)a\ Initialize a Stdio transport. Args: command: The command to run (e.g., "python", "node", "uvx") args: The arguments to pass to the command env: Environment variables to set for the subprocess cwd: Current working directory for the subprocess keep_alive: Whether to keep the subprocess alive between connections. Defaults to True. When True, the subprocess remains active after the connection context exits, allowing reuse in subsequent connections. NT) rrrrr_session _connect_taskanyioEvent _ready_event _stop_event)rErrrrrs r=rhzStdioTransport.__init__"s]*    J$.2 26!KKM ;;=r<r@rAc xK |jdi|d{|jJ|j|js|jd{ytj dy7\7#|js|jd{7wtj dwxYww)Nz6Stdio transport has keep_alive=True, not disconnectingr;)connectrr disconnectloggerdebugrDs r=rFzStdioTransport.connect_sessionDs W$,,00 0 0==, ,,-- ??oo''' UV 1 (??oo''' UVsIB:A:A6!A:B:A8B:6A:8B:: B7BB77B:cKjyfd}tj|_jj d{y7w)NcKddlm}tj4d{} t j j jj}|j||d{}|\}}|jt||fid{_ tjdjjj j#d{d_ tjddddd{y77778#d_ tjdwxYw7/#1d{7swYyxYww)Nr) stdio_client)rrrrzStdio transport connectedzStdio transport disconnected)mcp.client.stdiorrWAsyncExitStackr rrrrenter_async_contextr rrrrsetrwait)rstack server_paramsrmrnrorEr@s r=rz-StdioTransport.connect.._connect_taskXsG 5!002 A AeA$9 $ 499$((PTPXPX%M',&?&?$]3'!I1:-K*/*C*C%k<R>R+%DMLL!<=%%))+**//111%)DMLL!?@+ A A A !%2%)DMLL!?@+ A A A AsE+D*E+EAD39D-:(D3"D/#AD38D19D3=E E+$E%E+-D3/D31D33EEE+E(E E($E+)rasyncio create_taskrr)rEr@rs`` r=rzStdioTransport.connectRsM    )  A6%00A$$&&&sAAAAcK|jy|jj|jd{d|_tj|_tj|_y7>wN)rrrrrrrKs r=rzStdioTransport.disconnectwse    %      " ;;=!KKM !s7A:A8?A:c@K|jd{y7wr)rrKs r=rNzStdioTransport.closesoos cjd|jjd|jd|jdS)NrHz (command='z', args=z)>)rJr1rrrKs r=rLzStdioTransport.__repr__s3''( 4<<. SU V r<)NNN)r1r2r3r4rYlistrboolrhrWrXrr*rr rFrrrNrLr;r<r=r!r!s&*"& ) )3i )#s(^d " ) 4Z ) 4K )D## W &} 5 W } % W$ W#' &} 5#'  #'J *  # r<r!ceZdZdZdddej dfdeezdeedzde eefdzdedzdede dzf fd Z xZ S) r"z%Transport for running Python scripts.N script_pathrrr python_cmdrc8t|j}|jstd|t |j dst d|t |g}|r|j|t|%|||||||_ y)a Initialize a Python transport. Args: script_path: Path to the Python script to run args: Additional arguments to pass to the script env: Environment variables to set for the subprocess cwd: Current working directory for the subprocess python_cmd: Python command to use (default: "python") keep_alive: Whether to keep the subprocess alive between connections. Defaults to True. When True, the subprocess remains active after the connection context exits, allowing reuse in subsequent connections. Script not found: .pyNot a Python script: rrrrrN rresolveis_fileFileNotFoundErrorrYendswithrRextendsuperrhr) rErrrrrr full_argsrJs r=rhzPythonStdioTransport.__init__s.;'//1 ""$#&8 $FG G;((/4[MBC C%&    T " !  'r<) r1r2r3r4sys executablerYrrrrrh __classcell__rJs@r=r"r"s/ "&%).."&('4Z('3i$('#s(^d " (' 4Z ('  ('4K('('r<r"c peZdZdZ d deezdeedzdeeefdzdedzdedzf fd Z xZ S) r#zz2FastMCPTransport.connect_session..sFDKK3377#$ //MMO)-)>)> 8r<)rnror;)rrcreate_task_group start_soonr cancel_scopecancel) rEr@client_streamsserver_streams client_read client_writetgclient_sessionrrs ` @@r=rFz FastMCPTransport.connect_sessions078 - -=  (6 %K(6 %K..0 - -B -, $/%1 ) --(,, --OO**,% - - - - - ------OO**,% - - - - - - - -sD+CD+&DCDC?C,C -C0C 6 CCCC? D+C=,D0 D+;D<D+DCC C CC CC::C?=D?D DD DD+D(D D($D+c6d|jjdS)NzB#Br<r'cheZdZdZdeezfdZejde e de e fdZ defdZy) MCPConfigTransportaTransport for connecting to one or more MCP servers defined in an MCPConfig. This transport provides a unified interface to multiple MCP servers defined in an MCPConfig object or dictionary matching the MCPConfig schema. It supports two key scenarios: 1. If the MCPConfig contains exactly one server, it creates a direct transport to that server. 2. If the MCPConfig contains multiple servers, it creates a composite client by mounting all servers on a single FastMCP instance, with each server's name used as its mounting prefix. In the multi-server case, tools are accessible with the prefix pattern `{server_name}_{tool_name}` and resources with the pattern `protocol://{server_name}/path/to/resource`. This is particularly useful for creating clients that need to interact with multiple specialized MCP servers through a single interface, simplifying client code. Examples: ```python from fastmcp import Client from fastmcp.utilities.mcp_config import MCPConfig # Create a config with multiple servers config = { "mcpServers": { "weather": { "url": "https://weather-api.example.com/mcp", "transport": "streamable-http" }, "calendar": { "url": "https://calendar-api.example.com/mcp", "transport": "streamable-http" } } } # Create a client with the config client = Client(config) async with client: # Access tools with prefixes weather = await client.call_tool("weather_get_forecast", {"city": "London"}) events = await client.call_tool("calendar_list_events", {"date": "2023-06-01"}) # Access resources with prefixed URIs icons = await client.read_resource("weather://weather/icons/sunny") ``` configcddlm}t|trt j |}||_t|j jdk(r tdt|j jdk(rDt|j jjdj|_ yt}|j jjD]B\}}||j}|j!|tj"|Dt%||_ y)Nr)Clientz$No MCP servers defined in the configrm)prefixrr8)fastmcp.client.clientrrfrr from_dictrlen mcpServersrRrvalues to_transportrmritemsmountas_proxyr')rErrcomposite_serverr r server_clients r=rhzMCPConfigTransport.__init__s0 fd #((0F  t{{%% &! +CD D'' (A -!$++"8"8"?"?"AB1ERRTDN 'y  $ 6 6 < < >  f &1D1D1F G  &&(8(8(G'  .2BCDNr<r@rAc K|jjdi|4d{}|dddd{y77#1d{7swYyxYww)Nr;)rmrF)rEr@rps r=rFz"MCPConfigTransport.connect_sessionsU24>>11CNC  wM       s@!A>AA AAAAAA AAc"d|jdS)Nz>r<N)r1r2r3r4rrrhrWrXrr*rr rFrYrLr;r<r=rrs[-^Dy4/D6## &} 5 } %$ ?#?r<rrmrAcyrr;rs r=r(r(FIr<cyrr;rs r=r(r(s=@r<cyrr;rs r=r(r(sDGr<cyrr;rs r=r(r(sADr<cyrr;rs r=r(r(r'r<cyrr;rs r=r(r("s.1r<cyrr;rs r=r(r((s r<cyrr;rs r=r(r(0sSVr<ct|tr|St|ttzrt |}nt|t t zrut |jr\t |jdr t|}nt |jdr t|}ntd|t|tt zrDt |jdr*t|}|dk(r t|}n?t!|}n2t|t"t$zr t'| }ntd |t(j+d ||S) a: Infer the appropriate transport type from the given transport argument. This function attempts to infer the correct transport type from the provided argument, handling various input types and converting them to the appropriate ClientTransport subclass. The function supports these input types: - ClientTransport: Used directly without modification - FastMCP or FastMCP1Server: Creates an in-memory FastMCPTransport - Path or str (file path): Creates PythonStdioTransport (.py) or NodeStdioTransport (.js) - AnyUrl or str (URL): Creates StreamableHttpTransport (default) or SSETransport (for /sse endpoints) - MCPConfig or dict: Creates MCPConfigTransport, potentially connecting to multiple servers For HTTP URLs, they are assumed to be Streamable HTTP URLs unless they end in `/sse`. For MCPConfig with multiple servers, a composite client is created where each server is mounted with its name as prefix. This allows accessing tools and resources from multiple servers through a single unified client interface, using naming patterns like `servername_toolname` for tools and `protocol://servername/path` for resources. If the MCPConfig contains only one server, a direct connection is established without prefixing. Examples: ```python # Connect to a local Python script transport = infer_transport("my_script.py") # Connect to a remote server via HTTP transport = infer_transport("http://example.com/mcp") # Connect to multiple servers using MCPConfig config = { "mcpServers": { "weather": {"url": "http://weather.example.com/mcp"}, "calendar": {"url": "http://calendar.example.com/mcp"} } } transport = infer_transport(config) ``` rr)rrzUnsupported script type: ryssersr%z(Could not infer a valid transport from: zInferred transport: )rfrrrr'rrYrrr"r$rRrrgrrr rrrrr)rminferred_transportinferred_transport_types r=r(r(4s4h)_- Iw7 8-)< Itcz *tI/E/E/G y> " "5 )!5)!L  ^ $ $U +!3 !J 8 DE E Iv| ,Y1J1J61R"? "J "e +!-)!< !8Y!G  Iti/ 0/yACI;OPP LL'(:';<= r<)IrUrrWr5rrrrccollections.abcrrpathlibrtypingrrrr r r rrZ mcp.typesr8r r mcp.client.sessionrrrrmcp.server.fastmcprrmcp.shared.memoryrpydanticrtyping_extensionsrfastmcp.client.auth.bearerrfastmcp.client.auth.oauthrfastmcp.server.dependenciesrfastmcp.server.serverfastmcp.utilities.loggingrfastmcp.utilities.mcp_configrrr1rr__all__r*ABCrr]rr r!r"r#r$r%r&r'rr(rrYr;r<r=rDs  3CC 4WW8A$1+8)0Q H -5FG 1IU1+Ecgg+E\#9/#9LA3?A3HB>oB>Jr _r j+'>+'\'N'6''''T?#?#D;;|3B3BlS?S?l I/I4DI I @w@+;@ @ G~G2BG G DyD-?D D ItCH~I2DI I 11++1 1  --