o i# @sddlZddlZddlZddlmZddlmZddlmZm Z ddl Z ddl Z ddl m Z ddlmZmZddlmZddlmZmZddlmZdd lmZdd lmZmZmZmZdd lm Z e!e"Z#ej$d krpgd ngdZ%dZ&de'e(e(ffddZ)GdddeZ*eej+fde*de fddZ,de(de(fddZ-dej+dfde(de.e(de'e(e(fdBde dee(BdBf dd Z/d%d!e eBd"e0ddfd#d$Z1dS)&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@returncCs<i}tD]}tj|}|durq|drq|||<q|S)zo Returns a default environment object including only environment variables deemed safe to inherit. Nz())DEFAULT_INHERITED_ENV_VARSosenvironget startswith)envkeyvaluer._/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mcp/client/stdio/__init__.pyget_default_environment3s   r0c@s~eZdZUeed< eedZeeed< dZe eefdBed< dZ ee BdBed< dZ eed< d Z ed ed <dS) StdioServerParameterscommand)default_factoryargsNr+cwdzutf-8encodingstrict)r7ignorereplaceencoding_error_handler)__name__ __module__ __qualname__str__annotations__r listr4r+dictr5rr6r:rr.r.r.r/r1Hs  r1servererrlogcs&td\}td\}z#tj}t|jjdur&itjnt|jdIdHWn$t yV| IdH| IdH IdH IdHwfdd}fdd}t 4IdH}4IdH| || |zq||fVWj rz j IdHWn tyYnwzttIdHWdn1swYWntytIdHYn tyYnw| IdH| IdH IdH IdHnpj r z j IdHWn ty YnwzttIdHWdn 1s&wYWnty=tIdHYn tyFYnw| IdH| IdH IdH IdHwWdIdHn 1IdHsuwYWdIdHdS1IdHswYdS)z Client transport for stdio: this will connect to a server by spawning a process and communicating with it over stdin/stdout. rN)r2r4r+rCr5c s*jsJdzy4IdHed}tjjjd2zJ3dHW}||d}|}|D]6}ztj|}Wnt yX}zt d |IdHWYd}~q/d}~wwt |} |IdHq/q6WdIdHWdS1IdHszwYWdStjytjIdHYdSw)Nz Opened process is missing stdoutr6errors z+Failed to parse JSONRPC message from server)stdoutr r6r:splitpoptypesJSONRPCMessagemodel_validate_json Exceptionlogger exceptionsendranyioClosedResourceErrorlowlevel checkpoint)bufferchunklineslinemessageexcsession_message)processread_stream_writerrBr.r/ stdout_readers:  2z#stdio_client..stdout_readerc sjsJdzE4IdH12z 3dHW}|jjddd}j|djjjdIdHq6WdIdHWdS1IdHsFwYWdStjy`tj IdHYdSw)NzOpened process is missing stdinT)by_alias exclude_nonerGrE) stdinrZmodel_dump_jsonrQencoder6r:rRrSrTrU)r\json)r]rBwrite_stream_readerr.r/ stdin_writers" 2 z"stdio_client..stdin_writer)rRcreate_memory_object_stream_get_executable_commandr2#_create_platform_compatible_processr4r+r0r5OSErroraclosecreate_task_group start_soonrbrN fail_afterPROCESS_TERMINATION_TIMEOUTwait TimeoutError_terminate_process_treeProcessLookupError)rBrC read_stream write_streamr2r_rgtgr.)r]r^rBrfr/ stdio_clientis            \rxr2cCstjdkr 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)r2r.r.r/ris rir4r+r5csJtjdkrt|||||IdH}|Stj|g||||ddIdH}|S)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+stderrr5start_new_session)ryrzrrR open_process)r2r4r+rCr5r]r.r.r/rjs   rjr]timeout_secondscsBtjdkrt||IdHdSt|tsJt||IdHdS)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)ryrzr isinstancerr )r]r~r.r.r/rss  rs)r$)2loggingr'ry contextlibrpathlibrtypingrrrRanyio.lowlevel anyio.abcranyio.streams.memoryrranyio.streams.textr pydanticr r mcp.typesrKmcp.os.posix.utilitiesr mcp.os.win32.utilitiesr rrrmcp.shared.messager getLoggerr;rOrzr&rprAr>r0r1r{rxrir@rjfloatrsr.r.r.r/sT         !q