6?idZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZmZmZmZmZddlm Z dZ!GddZ"gdZ#y)zClient for connecting to multiple MCP servers and loading LangChain-compatible resources. This module provides the MultiServerMCPClient class for managing connections to multiple MCP servers and loading tools, prompts, and resources from them. N) AsyncIterator)asynccontextmanager) TracebackType)Any)Blob) AIMessage HumanMessage)BaseTool) ClientSession)load_mcp_prompt)load_mcp_resources) ConnectionMcpHttpClientFactory SSEConnectionStdioConnectionStreamableHttpConnectionWebsocketConnectioncreate_session)load_mcp_toolsaAs of langchain-mcp-adapters 0.1.0, MultiServerMCPClient cannot be used as a context manager (e.g., async with MultiServerMCPClient(...)). Instead, you can do one of the following: 1. client = MultiServerMCPClient(...) tools = await client.get_tools() 2. client = MultiServerMCPClient(...) async with client.session(server_name) as session: tools = await load_mcp_tools(session)c "eZdZdZddeeefdzddfdZeddded e de e fd Z dd dedzde efd Zdd dededeeefdzde eezfdZdddedee ezdzde efdZddZdeedzdedzdedzddfdZy)MultiServerMCPClientzvClient for connecting to multiple MCP servers and loading LangChain-compatible tools, prompts and resources from them.N connectionsreturnc&|||_yi|_y)aInitialize a MultiServerMCPClient with MCP servers connections. Args: connections: A dictionary mapping server names to connection configurations. If None, no initial connections are established. Example: basic usage (starting a new session on each tool call) ```python from langchain_mcp_adapters.client import MultiServerMCPClient client = MultiServerMCPClient( { "math": { "command": "python", # Make sure to update to the full absolute path to your math_server.py file "args": ["/path/to/math_server.py"], "transport": "stdio", }, "weather": { # Make sure you start your weather server on port 8000 "url": "http://localhost:8000/mcp", "transport": "streamable_http", } } ) all_tools = await client.get_tools() ``` Example: explicitly starting a session ```python from langchain_mcp_adapters.client import MultiServerMCPClient from langchain_mcp_adapters.tools import load_mcp_tools client = MultiServerMCPClient({...}) async with client.session("math") as session: tools = await load_mcp_tools(session) ``` N)r)selfrs _/opt/tws/coop-23/cliente-mcp/venv/lib/python3.12/site-packages/langchain_mcp_adapters/client.py__init__zMultiServerMCPClient.__init__-sTBMAX+^`T)auto_initialize server_namerchK||jvr5d|dt|jjd}t|t |j|4d{}|r|j d{|dddd{y7477 #1d{7swYyxYww)aoConnect to an MCP server and initialize a session. Args: server_name: Name to identify this server connection auto_initialize: Whether to automatically initialize the session Raises: ValueError: If the server name is not found in the connections Yields: An initialized ClientSession "Couldn't find a server with name '', expected one of ''N)rlistkeys ValueErrorr initialize)rr rmsgsessions rr*zMultiServerMCPClient.sessionYs( d.. .6{mCWX\]a]m]m]r]r]tXuWvvwxCS/ !!$"2"2;"?@  G((***M   *    sZA B2"B#B2&B<B= B B2BB2BB2B/#B& $B/+B2)r cK|e||jvr5d|dt|jjd}t|t d|j|d{Sg}g}|jj D]3}t jt d|}|j|5t j|d{}|D]}|j||S77 w)a\Get a list of all tools from all connected servers. Args: server_name: Optional name of the server to get tools from. If None, all tools from all servers will be returned (default). NOTE: a new session will be created for each tool call Returns: A list of LangChain tools Nr"r#r$) connection) rr%r&r'rvaluesasyncio create_taskappendgatherextend) rr r) all_toolsload_mcp_tool_tasksr,load_mcp_tool_task tools_listtoolss r get_toolszMultiServerMCPClient.get_toolsvs  "$"2"22:;-G[\`aeaqaqavavax\y[zz{| o%'9I9I+9VWW W$&  **113 ;J!(!4!4^DU_5`!a   & &'9 : ;#>>+>??  $E   U # $X@s%A#C6%C2&A-C6C4C64C6 arguments prompt_namer:cK|j|4d{}t|||d{cdddd{S7+77 #1d{7swYyxYww)z%Get a prompt from a given MCP server.Nr9)r*r )rr r;r:r*s r get_promptzMultiServerMCPClient.get_promptsa<< , T T(+SS T T TS T T T TsQAAAA AA  AAAA A AA AAurisr?cK|j|4d{}t||d{cdddd{S7*77 #1d{7swYyxYww)a)Get resources from a given MCP server. Args: server_name: Name of the server to get resources from uris: Optional resource URI or list of URIs to load. If not provided, all resources will be loaded. Returns: A list of LangChain Blobs Nr>)r*r )rr r?r*s r get_resourcesz"MultiServerMCPClient.get_resourcess_ << , @ @+G$?? @ @ @? @ @ @ @sQAAAA AA  AAAA A AA AAc(Kttw)zAsync context manager entry point. Raises: NotImplementedError: Context manager support has been removed. NotImplementedErrorASYNC_CONTEXT_MANAGER_ERROR)rs r __aenter__zMultiServerMCPClient.__aenter__s ""=>>sexc_typeexc_valexc_tbc tt)aUAsync context manager exit point. Args: exc_type: Exception type if an exception occurred. exc_val: Exception value if an exception occurred. exc_tb: Exception traceback if an exception occurred. Raises: NotImplementedError: Context manager support has been removed. rC)rrGrHrIs r __aexit__zMultiServerMCPClient.__aexit__s ""=>>r)N)rr)__name__ __module__ __qualname____doc__dictstrrrrboolrr r*r%r r8rr rr=rrArFtype BaseExceptionrrKrrrr*sTA*aDj$9D$@*aD*aX !%   } % 8<@cDjDND,0 T T T S>D( T lY& ' T(, @@DIo$ @ d @&??}%,?%?$ ?  ?rr)rrrrrr)$rOr.collections.abcr contextlibrtypesrtypingrlangchain_core.documents.baserlangchain_core.messagesrr langchain_core.toolsr mcpr langchain_mcp_adapters.promptsr langchain_mcp_adapters.resourcesr langchain_mcp_adapters.sessionsrrrrrrrlangchain_mcp_adapters.toolsrrEr__all__rUrrrcs_ )*.;):?83_?_?D r