Ki$ ddlZddlZddlZddlmZddlmZddlmZm Z ddl Z ddl Z ddl m Z ddlmZmZddlmZddlmZmZddlmZdd lmZdd lmZmZmZmZdd lm Z ejBe"Z#ejHd k(rgd ngdZ%dZ&de'e(e(ffdZ)GddeZ*eejVfde*de fdZ,de(de(fdZ-dejVdfde(de.e(de'e(e(fdzde dee(zdzf dZ/d de ezde0ddfdZ1y)!N)asynccontextmanager)Path)LiteralTextIO)Process)MemoryObjectReceiveStreamMemoryObjectSendStream)TextReceiveStream) BaseModelField)terminate_posix_process_tree)FallbackProcesscreate_windows_processget_windows_executable_commandterminate_windows_process_tree)SessionMessagewin32) APPDATA HOMEDRIVEHOMEPATH LOCALAPPDATAPATHPATHEXTPROCESSOR_ARCHITECTURE SYSTEMDRIVE SYSTEMROOTTEMPUSERNAME USERPROFILE)HOMELOGNAMErSHELLTERMUSER@returnci}tD];}tjj|}|%|j dr7|||<=|S)zo Returns a default environment object including only environment variables deemed safe to inherit. z())DEFAULT_INHERITED_ENV_VARSosenvironget startswith)envkeyvalues e/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/mcp/client/stdio/__init__.pyget_default_environmentr13sS C)  s# =    D ! C  JceZdZUeed< eeZeeed< dZe eefdzed< dZ ee zdzed< dZ eed< d Z ed ed <y) StdioServerParameterscommand)default_factoryargsNr-cwdzutf-8encodingstrict)r:ignorereplaceencoding_error_handler)__name__ __module__ __qualname__str__annotations__r listr7r-dictr8rr9r=rr2r0r4r4Hs} L4D1D$s)1;!%Cc3h$ % "Ctd !AHc FNG$ABMr2r4servererrlogc 6 Ktjd\ }tjd\} tj}t |j j itj n t|jd{ fd} fd}tj4d{}4d{|j||j| ||fjr# jjd{ tjt 5j#d{ddd|jd{|jd{ jd{ jd{dddd{dddd{y7P#t$rf|jd{7|jd{7 jd{7 jd{7wxYw777<#t$rYFwxYw7#1swYxYw#t$$rt'd{7Y?t($rYJwxYw7;7&77#jr4 jjd{7n#t$rYnwxYw tjt 5j#d{7dddn #1swYnxYwn-#t$$rt'd{7Ynt($rYnwxYw|jd{7|jd{7 jd{7 jd{7wxYw7#1d{7swYxYw7#1d{7swYyxYww)z Client transport for stdio: this will connect to a server by spawning a process and communicating with it over stdin/stdout. rN)r5r7r-rGr8cKjsJd 4d{d}tj j j23d{}||zj d}|j }|D]F} t jj|}t|}j|d{Hy77v#t$r9}tjdj|d{7Yd}~d}~wwxYw7P6dddd{7y#1d{7swYyxYw#tj$r*tj j#d{7YywxYww)Nz Opened process is missing stdoutr9errors z+Failed to parse JSONRPC message from server)stdoutr r9r=splitpoptypesJSONRPCMessagemodel_validate_json Exceptionlogger exceptionsendranyioClosedResourceErrorlowlevel checkpoint) bufferchunklineslinemessageexcsession_messageprocessread_stream_writerrFs r0 stdout_readerz#stdio_client..stdout_readersg~~AAA~ .) G G#4NN#__!88$GG% $e^2248E"YY[F % G%&+&:&:&N&Nt&TG +9*A055oFFF G GG )%",,-Z["4"9"9#">>>$% G#$ G G G G G((( ...++- - - .sE=D=C D=/D(DC D*D(C#D(D D( D= D D )D D D D( D D(D( D=!D$"D='E=(D:.D1 /D:6D=9E=:D==4E:1E42E:7E=9E::E=cKjsJd 4d{23d{}|jjdd}jj|dzj j j d{r7x7o7 6dddd{7y#1d{7swYyxYw#tj$r*tjjd{7YywxYww)NzOpened process is missing stdinT)by_alias exclude_nonerMrK) stdinr`model_dump_jsonrWencoder9r=rXrYrZr[)rbjsonrcrFwrite_stream_readers r0 stdin_writerz"stdio_client..stdin_writers}}???} .*  -@/*22BBD_cBdD!--,,,,%+__#)#@#@- .A     (( ...++- - - .sDCBCB0BBBA$B0BB0CBB0B0 C)B,*C/D0C6B9 7C>CDC4D9C<:D?DDD)rXcreate_memory_object_stream_get_executable_commandr5#_create_platform_compatible_processr7r-r1r8OSErroraclosecreate_task_group start_soonrirT fail_afterPROCESS_TERMINATION_TIMEOUTwait TimeoutError_terminate_process_treeProcessLookupError) rFrG read_stream write_streamr5rerntgrcrdrms ` @@@r0 stdio_clientris',&G&G&J# (-(I(I!(L%L%)&..9<@F @V<+-<<\s\u   .6." !"/"/%'"/"/ m$ l# /|+ + }}!--..000 %%&AB)!,,.(()$$& & &%%' ' '$++- - -%,,. . .E"/"/"/"/"/u    """!!### '')))!((*** f"/"/1 ))) 7.g666%   ' ' - .+}}!--..000  %%&AB)!,,.(())) 7.g666%  $$& & &%%' ' '$++- - -%,,. . .E"/"/"/"/"/"/"/"/s1PA(G3 G0!G3%&P I% PPI(P#O+>K O+I./I+0I.5JJ"I> #J'J/O+J=O+KO+2K3O+ K O+ PO(P P*P+P0G33I"HI"(H+)I"II"II""P(P+I.. I;7O+:I;;O+>JJ JJ:%J( &J:+O+.J:6O+9J::O+O+O+O+ O%K:3K6 4K:9O%: L O%L O% M#M 7L:8M = MM MO%M= *M-+M= 0O%2M= :O%<M= =O%N O%,N/ -O%O O%O! O%%O+(P+O> 1O42O> 9PPP P PPr5cBtjdk(r t|S|S)z Get the correct executable command normalized for the current platform. Args: command: Base command (e.g., 'uvx', 'npx') Returns: str: Platform-appropriate command r)sysplatformr)r5s r0rprps  ||w-g66r2r7r-r8cKtjdk(rt|||||d{}|Stj|g||||dd{}|S7-7w)z Creates a subprocess in a platform-compatible way. Unix: Creates process in a new session/process group for killpg support Windows: Creates process in a Job Object for reliable child termination rNT)r-stderrr8start_new_session)rrrrX open_process)r5r7r-rGr8rcs r0rqrqsq ||w.wc63OO N**  t "    NP s!&AA&AAAArctimeout_secondscKtjdk(rt||d{yt|tsJt ||d{y7,7w)af Terminate a process and all its children using platform-specific methods. Unix: Uses os.killpg() for atomic process group termination Windows: Uses Job Objects via pywin32 for reliable child process cleanup Args: process: The process to terminate timeout_seconds: Timeout in seconds before force killing (default: 2.0) rN)rrr isinstancerr )rcrs r0rzrzsS ||w,WoFFF'7+++*7ODDD G Es!#AA&A A AA)r%)2loggingr)r contextlibrpathlibrtypingrrrXanyio.lowlevel anyio.abcranyio.streams.memoryrr anyio.streams.textr pydanticr r mcp.typesrQmcp.os.posix.utilitiesr mcp.os.win32.utilitiesrrrrmcp.shared.messager getLoggerr>rUrr(rwrDrAr1r4rrrprCrqfloatrzrEr2r0rsY *" R0%? .   8 $$ ||w  >!("c3h*IBGJzzn/4n/fn/n/b S S &"&ZZ!   s) c3h$    d  6E7_+DEW\EgkEr2