[ǻi)dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlmZddlmZmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z%dZ&GddZ'gdZ(y)zClient for connecting to multiple MCP servers and loading LC tools/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)CallbackContext Callbacks)ToolCallInterceptor)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 VeZdZdZ ddddddeeefdzdedzdee dzde d df d Z e d d d ede d e efdZddd edzd eefdZddd ededeeefdzd eeezfdZ dddd edzdeeezdzd eefdZddZdeedzdedzdedzd dfdZy) MultiServerMCPClientzClient for connecting to multiple MCP servers. Loads LangChain-compatible tools, prompts and resources from MCP servers. NF) callbackstool_interceptorstool_name_prefix connectionsrrrreturncd||ni|_|xs t|_|xsg|_||_y)aInitialize a `MultiServerMCPClient` with MCP servers connections. Args: connections: A `dict` mapping server names to connection configurations. If `None`, no initial connections are established. callbacks: Optional callbacks for handling notifications and events. tool_interceptors: Optional list of tool call interceptors for modifying requests and responses. tool_name_prefix: If `True`, tool names are prefixed with the server name using an underscore separator (e.g., `"weather_search"` instead of `"search"`). This helps avoid conflicts when multiple servers have tools with the same name. Defaults to `False`. !!! 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": "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)rr rrr)selfrrrrs [/opt/lhia/marcimex/agent/venv/lib/python3.12/site-packages/langchain_mcp_adapters/client.py__init__zMultiServerMCPClient.__init__3s<r'2K #1ik!2!8b 0T)auto_initialize server_namer%cK||jvr5d|dt|jjd}t||jj t |}t|j||4d{}|r|jd{|dddd{y7477 #1d{7swYyxYww)aqConnect 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 ''r&)context) mcp_callbacksN) rlistkeys ValueErrorr to_mcp_formatr r initialize)r!r&r%msgr-sessions r"r4zMultiServerMCPClient.sessionrs( d.. .4[MB$$()9)9)>)>)@$A#B!E S/ !44# <5 "   [ )    ((***M    +     sZBC B? CC$C% C. C9C:CCCC C CCr+c ~K|||jvr5d|dt|jjd}t|t d|j||j ||j |jd{Sg}g}|jjD]X\}}tjt d||j ||j |j}|j|Ztj|d{}|D]} |j| |S77 w)aGet 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. !!! note A new session will be created for each tool call Returns: A list of LangChain [tools](https://docs.langchain.com/oss/python/langchain/tools) Nr(r)r*) connectionrr&rr)rr.r/r0rrrritemsasyncio create_taskappendgatherextend) r!r&r3 all_toolsload_mcp_tool_tasksnamer6load_mcp_tool_task tools_listtoolss r" get_toolszMultiServerMCPClient.get_toolssZ  "$"2"228 F((,T-=-=-B-B-D(E'FaI!o%'++K8..'"&"8"8!%!6!6  %'   $ 0 0 6 6 8 ; D*!(!4!4)"nn $&*&<&<%)%:%:  "  & &'9 : ;#>>+>??  $E   U # $5.@s%BD=D9BD=D;D=;D= arguments prompt_namerEcK|j|4d{}t|||d{cdddd{S7+77 #1d{7swYyxYww)z%Get a prompt from a given MCP server.NrD)r4r)r!r&rFrEr4s r" get_promptzMultiServerMCPClient.get_promptsa<< , T T(+SS T T TS T T T TsQAAAA AA  AAAA A AA AAurisrJc`K||jvr5d|dtjjd}t|j |4d{}t |d{cdddd{Sdt dttffd }g}jDcgc]}tj||}}tj|d{} | D]} |j| |S777#1d{7swYxYwcc}w7>w) aGet resources from MCP server(s). Args: server_name: Optional name of the server to get resources from. If `None`, all resources from all servers will be returned. uris: Optional resource URI or list of URIs to load. If not provided, all resources will be loaded. Returns: A list of LangChain [Blob][langchain_core.documents.base.Blob] objects. Nr(r)r*rIr?rcKj|4d{}t|d{cdddd{S7*77 #1d{7swYyxYww)NrI)r4r)r?r4r!rJs r"_load_resources_from_serverzGMultiServerMCPClient.get_resources.._load_resources_from_servers^||D) D DW/dCC D D DC D D D DsQAAAA AA  AAAA A AA AA) rr.r/r0r4rstrrr8r9r;r<) r!r&rJr3r4rM all_resourcesr? load_tasksresources_list resourcess ` ` r" get_resourcesz"MultiServerMCPClient.get_resourcessQ$  "$"2"228 F((,T-=-=-B-B-D(E'FaI!o%||K0 D DG/dCC D D D DC DDJ D%' ((     ;D A B   '~~z::' ,I   + , DC D D D D ;srAD.D  D.#D4D5D8 D.D/D.4"D'D.-D,.D.DD.D$D D$ D.c(Kttw)zAsync context manager entry point. Raises: NotImplementedError: Context manager support has been removed. NotImplementedErrorASYNC_CONTEXT_MANAGER_ERROR)r!s 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. rU)r!rYrZr[s r" __aexit__zMultiServerMCPClient.__aexit__s ""=>>r$)N)rr)__name__ __module__ __qualname____doc__dictrNrr r.rboolr#rrr r4r rCrr rrHrrSrXtype BaseExceptionrr]r$r"rr-s59=1'+>B!& =1#z/*T1=1t# =1  34t; =1  =1 =1~ !% ## # } % ##J<@0cDj0DN0n,0 T T T S>D( T lY& ' T#'((, (4Z(DIo$ ( d (T??}%,?%?$ ?  ?r$r)r rrrrrr))rar8collections.abcr contextlibrtypesrtypingrlangchain_core.documents.baserlangchain_core.messagesrr langchain_core.toolsr mcpr langchain_mcp_adapters.callbacksr r #langchain_mcp_adapters.interceptorsrlangchain_mcp_adapters.promptsr langchain_mcp_adapters.resourcesrlangchain_mcp_adapters.sessionsrrrrrrrlangchain_mcp_adapters.toolsrrWr__all__rfr$r"rvse )*.;)GC:?83j?j?Z r$