iKddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl Z ddl mZmZddlmZddlmZddlmZmZdd lmZdd lmZeeZGd d eZd edeede eefdzdedzdee zdzdede jBde jBdejDefdZ#GddeZ$GddeZ%GddeZ&GddeZ'Gdd eZ(Gd!d"eZ)y)#N) AsyncIterator)Path)TextIOcast) ClientSessionStdioServerParameters) stdio_client)Unpack)ClientTransport SessionKwargs) get_logger) UVEnvironmentceZdZdZ ddedeedeeefdzdedzdedzdee zdzf d Z e jd e ed eefd Zd e ed edzfd ZdZdZdZd efdZy)StdioTransportz 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_alivelog_filec||_||_||_||_|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. log_file: Optional path or file-like object where subprocess stderr will be written. Can be a Path or TextIO object. Defaults to sys.stderr if not provided. When a Path is provided, the file will be created if it doesn't exist, or appended to if it does. When set, server errors will be written to this file instead of appearing in the console. NT) rrrrrr_session _connect_taskanyioEvent _ready_event _stop_event)selfrrrrrrs o/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/client/transports/stdio.py__init__zStdioTransport.__init__sd6    J$  .2 26!KKM ;;=session_kwargsreturnc xK |jdi|d{tt|j|js|j d{yt jdy7\7#|js|j d{7wt jdwxYww)Nz6Stdio transport has keep_alive=True, not disconnecting)connectrrrr disconnectloggerdebug)rr"s rconnect_sessionzStdioTransport.connect_sessionGs W$,,00 0 0}dmm4 4??oo''' UV 1(??oo''' UVsIB:A:A6!A:B:A8B:6A:8B:: B7BB77B:c "K|jytj}tjt |j |j |j|j|j||j|j| |_|jjd{|jjr|jj}|||d{|_|jS7W7w)N) rrrrrr" ready_event stop_eventsession_future)rasyncioFuture create_task_stdio_transport_connect_taskrrrrrrrwaitdone exceptionr)rr"r.r5s rr&zStdioTransport.connectTs    ) 8?8H%00 ) YYHHHH- --++-    $$&&&    " " $**446I$,, }} '-s%B1D3D 4AD5D 6D DcK|jy|jj|jd{d|_tj|_tj|_y7>wN)rrsetrrrrs rr'zStdioTransport.disconnectxse    %      " ;;=!KKM !s7A:A8?A:c@K|jd{y7wr7)r'r9s rclosezStdioTransport.closesoos cn|jjs|jjyy)zcEnsure that we send a disconnection signal to the transport task if we are being garbage collected.N)ris_setr8r9s r__del__zStdioTransport.__del__s+&&(    ")r!cjd|jjd|jd|jdS)N) __class____name__rrr9s r__repr__zStdioTransport.__repr__s3''( 4<<. SU V r!)NNNN)rB __module__ __qualname____doc__strlistdictboolrrr contextlibasynccontextmanagerr r rrr*r&r'r;r>rCr%r!rrrs&*"&)-')')3i')#s(^d " ') 4Z ') 4K ')-$&')R## W &} 5 W } % W$ W" &} 5"  "H * #  # r!rrrrrrr"r,r-r.c K tj4d{} t||||} |tj} n3t |t r!| j|jd} n|} | jt| | d{} | \} }|j| jt| |fi|d{tjd|j|j!d{tjddddd{y777p71#tjdwxYw7(#1d{7swYyxYw#t"$r|jwxYww)zA standalone connection task for a stdio transport. It is not a part of the StdioTransport class to ensure that the connection task does not hold a reference to the Transport object.N)rrrra)errlogzStdio transport connectedzStdio transport disconnected)rKAsyncExitStackrsysstderr isinstancer enter_contextopenenter_async_contextr set_resultrr(r)r8r3 Exception)rrrrrr"r,r-r.stack server_paramslog_file_handle transport read_stream write_streams rr2r2sd',,..%! = 5# ! #&)jjO$/&+&9&9(--:L&MO'/O"'";"; G# -6) \))33%k<R>R  89!!oo''' ;$% 8$ !  r!)NFNNNNNN) rBrDrErFrGrHrJrrIr rrrss@rrrTs9 "&)-%)*.)-*."&I I 3i$I  I  $; I d I Cy4'I  $;I sCx.4'I 4KI I r!rceZdZdZ d dedeedzdedzdedzdeedzdedzd eeefdzd edzffd ZxZ S) UvxStdioTransportz0Transport for running commands via the uvx tool.N tool_name tool_argsrrr from_packagerrc |r't|jstd|g} |r| jd|g|r| jd|g|xsgD]} | jd| g| j ||r| j|d} |r/t j j} | j|t |)d| | ||||_ y)a Initialize a Uvx transport. Args: tool_name: Name of the tool to run via uvx tool_args: Arguments to pass to the tool project_directory: Project directory (for package resolution) python_version: Python version to use with_packages: Additional packages to include from_package: Package to install the tool from env_vars: Additional environment variables 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. rrz--fromrNuvxr) rrrrnrrrrrror r) rrrrrrrrruvx_argsrrrAs rr zUvxStdioTransport.__init__s8 T*;%<%C%C%E$/0A/BC  !  OOZ8 9  OOX|4 5 &B&C OOXsO ,'  "  OOI &%) **//#C JJx  !!  (r!)NNNNNNN rBrDrErFrGrHrIrJr rrrss@rrrs: '+(,%)*.#'*."&;(;(9t#;(: ;( d ;( Cy4' ;(Dj;(sCx.4';(4K;(;(r!rcpeZdZdZ d dedeedzdedzdeeefdzdededzf fd ZxZ S) NpxStdioTransportz0Transport for running commands via the npx tool.Npackagerrruse_package_lockrctjd td|r't|j st d|g}|r|j d|j ||r|j|d}|r/tjj}|j|t |5d||||||_y)a Initialize an Npx transport. Args: package: Name of the npm package to run args: Arguments to pass to the package command project_directory: Project directory with package.json env_vars: Additional environment variables use_package_lock: Whether to use package-lock.json (--prefer-offline) 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. npxNzCommand 'npx' not foundrz--prefer-offliner)shutilwhichrmrrrrrnrrrrror r) rrrrrrrnpx_argsrrAs rr zNpxStdioTransport.__init__s0 <<  &67 7 T*;%<%C%C%E$/0A/BC    OO. /   OOD ! **//#C JJx  !!   r!)NNNTNrrss@rrrsy: "&(,*.!%"&883i$8: 8 sCx.4' 8  84K88r!r)*r/rKrrrQcollections.abcrpathlibrtypingrrrmcprrmcp.client.stdior typing_extensionsr fastmcp.client.transports.baser r fastmcp.utilities.loggingr 6fastmcp.utilities.mcp_server_config.v1.environments.uvrrBr(rrGrHrIrr0r2r`rur{rrrr%r!rrs$ ) 4)$I0P H | _| ~5 5 s)5 c3h$ 5 t 5 Vmd" 5 " 55 5NN=15p2'>2'j'N':2'2'jL ~L ^>(>(B;;r!