j 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(dS)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 eZdZdZ ddddddeeefdzdedzdee dzde d df d Z e d d d ede d e efdZddd edzd eefdZddd ededeeefdzd eeezfdZ dddd edzdeeezdzd eefdZddZdeedzdedzdedzd dfdZdS) MultiServerMCPClientzClient for connecting to multiple MCP servers. Loads LangChain-compatible tools, prompts and resources from MCP servers. NF) callbackstool_interceptorstool_name_prefix connectionsrrrreturncf||ni|_|p t|_|pg|_||_dS)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 C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain_mcp_adapters/client.py__init__zMultiServerMCPClient.__init__3sFr'2KK #1ikk!2!8b 0T)auto_initialize server_namer%cK||jvr)>)@)@$A$AEEE S// !44# <<<5  "  [ )             +((*********MMMM                               s"C  CCr+c :K|z||jvr+>???????  $ $E   U # # # #r$ arguments prompt_namerEcK||4d{V}t|||d{Vcdddd{VS#1d{VswxYwYdS)z%Get a prompt from a given MCP server.NrD)r4r)r!r&rFrEr4s r" get_promptzMultiServerMCPClient.get_prompts<< ,, T T T T T T T(+SSSSSSSSS T T T T T T T T T T T T T T T T T T T T T T T T T T T T T TsA  AAurisrJc K||jvr._load_resources_from_servers||D)) D D D D D D DW/dCCCCCCCCC D D D D D D D D D D D D D D D D D D D D D D D D D D D D D DsA  AAcJg|]}tj| S)r8r9).0r?rMs r" z6MultiServerMCPClient.get_resources..s@      ; ;D A A B B   r$) rr.r/r0r4rstrrr8r;r<) r!r&rJr3r4 all_resources load_tasksresources_list resourcesrMs ` ` @r" get_resourcesz"MultiServerMCPClient.get_resourcess6$  "$"222III(,T-=-B-B-D-D(E(EIII!oo%||K00 D D D D D D DG/dCCCCCCCCC D D D D D D D D D D D D D D D D D D D D D D D D D D D D DC DDJ D D D D D D D%'     (    '~z:::::::' , ,I   + + + +s(B BBc.Ktt)zAsync context manager entry point. Raises: NotImplementedError: Context manager support has been removed. NotImplementedErrorASYNC_CONTEXT_MANAGER_ERROR)r!s r" __aenter__zMultiServerMCPClient.__aenter__s ""=>>>r$exc_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. rY)r!r]r^r_s r" __aexit__zMultiServerMCPClient.__aexit__s ""=>>>r$)N)rr)__name__ __module__ __qualname____doc__dictrRrr r.rboolr#rrr r4r rCrr rrHrrWr\type BaseExceptionrrarOr$r"rr-sX59=1'+>B!& =1=1=1#z/*T1=1t# =1  34t; =1  =1 =1=1=1=1~ !% #### # } % ####J<@000cDj0DN0000n,0 T T T T T S>D( T lY& ' T T T T#'((, (((4Z(DIo$ ( d ((((T?????}%,?%?$ ?  ??????r$r)r rrrrrr))rer8collections.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.toolsrr[r__all__rOr$r"rys ))))))******......;;;;;;;;))))))GGGGGGGGCCCCCC::::::??????8777773j?j?j?j?j?j?j?j?Z   r$