~h7 rdZddlZddlZddlmZmZddlmZmZddl m Z ddl m Z ddl mZmZmZmZddlZddlmZmZdd lmZdd lmZdd lmZGd d eZe d deeeefzdee zdzdeeee ejBjDfdffdZ#GddZ$e%dk(rGddeZ&e$edddgedddgge&5Z'e(e'e(e'dddie(e'dddiddddZ)ejTe)yy#1swY%xYw)!zCore module for the MCPAdapt library. This module contains the core functionality for the MCPAdapt library. It provides the basic interfaces and classes for adapting tools from MCP to the desired Agent framework. N)ABCabstractmethod)AsyncExitStackasynccontextmanager) timedelta)partial)AnyAsyncGeneratorCallable Coroutine) ClientSessionStdioServerParameters) sse_client) stdio_client)streamablehttp_clientceZdZdZedeedzgejjfdejjde fdZ deedzge e e ejjffdejjde fdZy) ToolAdapterzMA basic interface for adapting tools from MCP to the desired Agent framework.funcNmcp_toolreturncy)akAdapt a single tool from MCP to the desired Agent framework. The MCP protocol will provide a name, description and inputSchema in JSON Schema format. This needs to be adapted to the desired Agent framework. Note that the function is synchronous (not a coroutine) you can use :meth:`ToolAdapter.async_adapt` if you need to use the tool asynchronously. Args: func: The function to be called (will call the tool via the MCP protocol). mcp_tool: The tool to adapt. Returns: The adapted tool in the agentic framework of choice. Nselfrrs I/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/mcpadapt/core.pyadaptzToolAdapter.adapts* afuncctd)a9Adapt a single tool from MCP to the desired Agent framework. The MCP protocol will provide a name, description and inputSchema in JSON Schema format. This needs to be adapted to the desired Agent framework. Note that the function is asynchronous (a coroutine) you can use :meth:`ToolAdapter.adapt` if you need to use the tool synchronously. Args: afunc: The coroutine to be called. mcp_tool: The tool to adapt. Returns: The adapted tool in the agentic framework of choice. z;Async adaptation is not supported for this Agent framework.)NotImplementedErrorrrrs r async_adaptzToolAdapter.async_adapt0s(" I  r)__name__ __module__ __qualname____doc__rr dictmcptypesCallToolResultToolr rr r"rrrrrsW t }cii&>&>>? ))..    ,  yc399;S;S1S'TTU ))..   rr serverparamsclient_session_timeout_secondsrcKt|tr t|}nkt|trC|j dd}|dk(r t d i|}n8|dk(r t d i|}n'td|dtdt|dd}t|tr t|}nt|tr|}|4d{^}}}t|||4d{}|jd{|jd{} || jfdddd{dddd{y77h7R7<7#1d{7swY-xYw7$#1d{7swYyxYww) a/Async context manager that yields tools from an MCP server. Note: the session can be then used to call tools on the MCP server but it's async. Use MCPAdapt instead if you need to use the tools synchronously. Args: serverparams: Parameters passed to either the stdio client or sse client. * if StdioServerParameters, run the MCP server using the stdio protocol. * if dict, assume the dict corresponds to parameters to an sse MCP server. client_session_timeout_seconds: Timeout for MCP ClientSession calls Yields: A tuple of (MCP Client Session, list of MCP tools) available on the MCP server. Usage: >>> async with mcptools(StdioServerParameters(command="uv", args=["run", "src/echo.py"])) as (session, tools): >>> print(tools) transportssezstreamable-httpz:Invalid transport, expected sse or streamable-http found ``zDInvalid serverparams, expected StdioServerParameters or dict found `N)secondsr) isinstancerrr'poprr ValueErrortypefloatrr initialize list_toolstools) r,r-clientr/timeoutreadwrite_sessionr:s rmcptoolsrAIs., 56l+ L$ ' $$[%8  /,/F + +*:\:FLYKWXY RSWXdSeRffg h  G0%8$BC 2I >0 ' '*$     ' '$$& & &!,,..E5;;& & ' ' ' ' ' ' '. ' ' ' ' ' ' ' 'sB?E7EE7E"EE"E 2E3E  E E  E"+E ,E"0 E7;E <E7E"E E  E" E EE E" E7"E4(E+ )E40E7c eZdZdZ ddeeeefzeeeeefzzde de de e zdzfdZ dZd eefd Zd Zd Zd ZdZd eefdZd eefdZdZy)MCPAdapta.The main class for adapting MCP tools to the desired Agent framework. This class can be used either as a sync or async context manager. If running synchronously, it will run the MCP server in a separate thread and take care of making the tools synchronous without blocking the server. If running asynchronously, it will use the async context manager and return async tools. Dependening on what your Agent framework supports choose the approriate method. If async is supported it is recommended. Important Note: adapters need to implement the async_adapt method to support async tools. Usage: >>> # sync usage >>> with MCPAdapt(StdioServerParameters(command="uv", args=["run", "src/echo.py"]), SmolAgentAdapter()) as tools: >>> print(tools) >>> # sync usage by start ... close pattern >>> adapter = MCPAdapt(StdioServerParameters(command="uv", args=["run", "src/echo.py"]), SmolAgentAdapter()) >>> adapter.start() >>> print(adapter.tools()) # get latest tools >>> adapter.close() >>> # sync usage with streamable-http >>> with MCPAdapt({"url": "http://127.0.0.1:8000/mcp", "transport": "streamable-http"}), SmolAgentAdapter()) as tools: >>> print(tools) >>> # async usage >>> async with MCPAdapt(StdioServerParameters(command="uv", args=["run", "src/echo.py"]), SmolAgentAdapter()) as tools: >>> print(tools) >>> # async usage with sse >>> async with MCPAdapt({"url": "http://127.0.0.1:8000/sse"}, SmolAgentAdapter()) as tools: >>> print(tools) r,adapterconnect_timeoutr-NcHt|tr||_n|g|_||_g|_g|_t j|_d|_ tj|_ tj|jd|_||_||_y)a Manage the MCP server / client lifecycle and expose tools adapted with the adapter. Args: serverparams (StdioServerParameters | dict[str, Any] | list[StdioServerParameters | dict[str, Any]]): MCP server parameters (stdio or sse). Can be a list if you want to connect multiple MCPs at once. adapter (ToolAdapter): Adapter to use to convert MCP tools call into agentic framework tools. connect_timeout (int): Connection timeout in seconds to the mcp server (default is 30s). client_session_timeout_seconds: Timeout for MCP ClientSession calls Raises: TimeoutError: When the connection to the mcp server time out. NT)targetdaemon)r3listr,rDsessions mcp_toolsasyncionew_event_looplooptask threadingEventreadyThread _run_loopthreadrEr-)rr,rDrEr-s r__init__zMCPAdapt.__init__s. lD ) ,D !-D  .0 57**,  __& &&dnnTJ ..L+rctjjfd}jj|_ jj jy#tj $rYywxYw)zARuns the event loop in a separate thread (for synchronous usage).c Kt4d{}jDcgc]/}|jt|jd{1}}t |Dcgc] }t |c}\__jjtjjd{dddd{y77cc}wcc}w7#7#1d{7swYyxYwwN)rr,enter_async_contextrAr-ziprIrJrKrRsetrLrQwait)stackparams connectionscrs rsetupz!MCPAdapt._run_loop..setups%' - -5 #'"3"3  33 )L)LM  CF{BS0TQa0T- t~  mmo**,,, - - - 1U, - - - -sDCDC.-C C C  C.%C%7AC.C*C. DC,DC  C.,D.D4C7 5D<DN)rLset_event_looprN create_taskrOrun_until_completeCancelledError)rrbs` rrTzMCPAdapt._run_loopsgtyy) -II))%'2   II ( ( 3%%   s %A22BBrc js td d dtdtdzdtj j ffd }g}dtdttj jfd}jD]R}|jtj||jjj gT|_t#jj Dcgc]=\}}|D]3}j$j't)|||j*|5?c}}}Scc}}}w) aReturns the tools from the MCP server adapted to the desired Agent framework. This is what is yielded if used as a context manager otherwise you can access it directly via this method. Only use this when you start the client in synchronous context or by :meth:`start`. An equivalent async method is available if your Agent framework supports it: see :meth:`atools`. zSession not initializedNname argumentsrctj|j||jj SrY)rLrun_coroutine_threadsafe call_toolrNresult)r@rhrirs r_sync_call_toolz'MCPAdapt.tools.._sync_call_tools533!!$ 2DIIfh rr@cRK|jd{jS7wrY)r9r:)r@s r _list_toolsz#MCPAdapt.tools.._list_toolss!!,,..55 5.s '%'r<rY)rJ RuntimeErrorstrr'r(r)r*r rIr+extendrLrkrNrmrErKr[rDrrrh)rrnrKrpr@r:tools` rr:zMCPAdapt.toolss;}}89 9:>  +/$;  YY % % 13  6} 6ciinn9M 6}} G   44#G,diifT%9%9f:  ##&dmmT^^"D    LL  wKT R R   sAE c|jj|jj|jst d|jdy)z(Start the client in synchronous context.rqz)Couldn't connect to the MCP server after z secondsN)rUstartrRr]rE TimeoutErrorrs rrwzMCPAdapt.startsR zzt';';<;DHc}}}S7cc}wcc}}}ww)aReturns the tools from the MCP server adapted to the desired Agent framework. This is what is yielded if used as an async context manager otherwise you can access it directly via this method. Only use this when you start the client in asynchronous context. An equivalent sync method is available if your Agent framework supports it: see :meth:`tools`. N) rJr9r:rKr[rDr"rrlrh)rsr@r:rus ratoolszMCPAdapt.atools3sAE N1/66N#&dmmT^^"D    LL $ $WW->-> %JD Q Q  0N s2CB8B6 B8,C&A B=1C6B88 Cc hKt|_|jDcgc]9}|jjt ||j d{;}}t |Dcgc] }t|c}\|_|_ |jd{S7Lcc}wcc}w7wrY) r _ctxmanagerr,rZrAr-r[rIrJrKr)rr_r`ras r __aenter__zMCPAdapt.__aenter__Gs)+ ++  ""66!D!DE    ;>{:K(LQa(L% t~[[]""  )M"s@B27B&B$ B& B2*B+<#B2B0 B2$B&& B2cZK|jj|||d{y7wrY)r __aexit__rs rrzMCPAdapt.__aexit__Us$((7FCCCs !+)+))r#r$r%r&rr'rsr rIrintr7rrVrTr:rwrrrrrrrrrrCrCs&\ "CD*M+ sCx. $tCH~5 67*M  *M  *M). (9D(@*MX,* tCy* X d3i ( #$s) #DrrC__main__ceZdZdeedzgej jfdej jfdZ deedzge e e ej jffdej jfdZ y) DummyAdapterrNrc|SrYrrs rrzDummyAdapter.adapt\s Krrc|SrYrr!s rr"zDummyAdapter.async_adaptcs Lr) r#r$r%r r'r(r)r*r+rr r r"rrrrr[s D4K=#))*B*BBC iinn   yc3993K3K)KLL  iinn  rruvrun src/echo.pycommandargstexthelloworldcfKttdddgtdddggt4d{}t|t|dddid{t|ddd id{dddd{y7\7977 #1d{7swYyxYww) Nrrrrrrrrr)rCrrprint) dummy_toolss rmainrws%d%9OP%d%9OP  N  ; ;  +   A'899 :  A'899 : ; ; ;:9 ; ; ; ;sh5B1BB1!BB B:B ;B B1BB1BBB1B."B% #B.*B1)r)+r&rLrPabcrr contextlibrrdatetimer functoolsrtypingr r r r r(r rmcp.client.ssermcp.client.stdiormcp.client.streamable_httprrr'rsr7tuplerIr)r+rArCr#rrrrrrrrrsv #:;; 4%)<0 #0 f@A6''$sCx.86'$)I$5$<6'E-ciinn)==>DE6'6'rSDSDl z{"  !$e]5K L !$e]5K L     1  k nk!nfg./0 nk!nfg./0 1 ;GKKU& 1 1s /D--D6