j i= UdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZddlmZmZddlmZdd lmZdd lmZdd lmZmZe rdd lmZdd lmZddlZddl m!Z!ej"e#Z$ej%dZ& dOdZ'e dZ(dZ)dZ*de+d<dZ,dZ-e dZ.e dZ/GddeZ0Gdd eZ1Gd!d"eZ2Gd#d$eZ3Gd%d&eZ4e1e2ze3ze4zZ5edde)e*dd'dPd5Z6ede,e-dddd6dQd@Z7ede.e/dAddddBdRdFZ8eddGdSdHZ9eddIdTdNZ:dS)UzSession management for different MCP transport types. This module provides connection configurations and session management for various MCP transport types including stdio, SSE, WebSocket, and streamable HTTP. ) annotationsN)asynccontextmanager) timedelta) TYPE_CHECKINGAnyLiteralProtocol) ClientSessionStdioServerParameters) sse_client) stdio_client)streamablehttp_client) NotRequired TypedDict) AsyncIterator)Path) _MCPCallbacksz \$\{([^}]+)\}valuestrreturnc:td|S)avExpand `${VAR}` references in *value* using the current environment. Only braced syntax is expanded; bare `$VAR` references are left untouched so that literal dollar signs in passwords or other values are never silently corrupted by an unrelated environment variable. Undefined variables are preserved as-is (e.g. `${MISSING}` stays `${MISSING}`). ctj|d|dS)Nr)osenvirongetgroup)ms C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain_mcp_adapters/sessions.pyz"_expand_env_vars...s* qwwqzz1771::(N(N)_BRACED_VAR_REsub)rs r_expand_env_varsr$$s   NNPU V VVr!)strictignorereplacezutf-8r%EncodingErrorHandlerDEFAULT_ENCODING_ERROR_HANDLERi,)secondsc"eZdZdZ d d d ZdS)McpHttpClientFactoryzFProtocol for creating httpx.AsyncClient instances for MCP connections.Nheadersdict[str, str] | Nonetimeouthttpx.Timeout | Noneauthhttpx.Auth | Nonerhttpx.AsyncClientcdS)a Create an httpx.AsyncClient instance. Args: headers: HTTP headers to include in requests. timeout: Request timeout configuration. auth: Authentication configuration. Returns: Configured httpx.AsyncClient instance. N)selfr/r1r3s r__call__zMcpHttpClientFactory.__call__@s r!)NNN)r/r0r1r2r3r4rr5)__name__ __module__ __qualname____doc__r9r7r!rr.r.=s>PP*.(,"&        r!r.cpeZdZUdZded<ded< ded< ded < d ed < d ed < ded< ded<dS)StdioConnectionz=Configuration for stdio transport connections to MCP servers.zLiteral['stdio'] transportrcommand list[str]argsz"NotRequired[dict[str, str] | None]envzNotRequired[str | Path | None]cwdzNotRequired[str]encodingz!NotRequired[EncodingErrorHandler]encoding_error_handler"NotRequired[dict[str, Any] | None]session_kwargsNr:r;r<r=__annotations__r7r!rr?r?SsGGLLL4OOO;++++ ('''A >===7666DDr!r?cpeZdZUdZded<ded< ded< ded < ded < ded < d ed < ded<dS) SSEConnectionzHConfiguration for Server-Sent Events (SSE) transport connections to MCP.zLiteral['sse']r@rurlrHr/zNotRequired[float]r1sse_read_timeoutrI(NotRequired[McpHttpClientFactory | None]httpx_client_factoryNotRequired[httpx.Auth]r3NrJr7r!rrMrMsRR HHH4////3 )((( 7666DBBBB:!!!!66r!rMc|eZdZUdZded<ded< ded< ded < ded < d ed < ded < ded< ded<dS)StreamableHttpConnectionz7Connection configuration for Streamable HTTP transport.zLiteral['streamable_http']r@rrNrHr/zNotRequired[timedelta]r1rOzNotRequired[bool]terminate_on_closerIrPrQrRr3NrJr7r!rrTrTsAA)))) HHH0/////####,,,,>*)))46666DBBBB:!!!!66r!rTc4eZdZUdZded<ded< ded<dS) WebsocketConnectionzAConfiguration for WebSocket transport connections to MCP servers.zLiteral['websocket']r@rrNrHrINrJr7r!rrWrWs<KK#### HHH:6666CCr!rW)rDrErFrGrIrArCrBrDr0rEstr | Path | NonerFrG&Literal['strict', 'ignore', 'replace']rIdict[str, Any] | NoneAsyncIterator[ClientSession]cK|d|Dnd}|P|D];\}} t| rtd|| z)_create_stdio_session..s'888DAq Q  888r!z2env[%r] contains unexpanded variable reference: %r)rArCrDrErFrG)itemsr"searchloggerwarningr r r )rArCrDrErFrGrI resolved_envr`ra server_paramsreadwritesessions r_create_stdio_sessionrlseJ=@O88CIIKK8888QU &&((  DAq$$Q'' H!Q*   5 M ]##'4edE<AA> A$ $A>'A$ (A>> B B)rIcNK ddlm}n #t$rd}t|dwxYw||4d{V \}}t||fi|pi4d{V }|WVdddd{Vn#1d{VswxYwYdddd{VdS#1d{VswxYwYdS)aCCreate a new session to an MCP server using Websockets. Args: url: URL of the Websocket endpoint. session_kwargs: Additional keyword arguments to pass to the ClientSession. Yields: An initialized ClientSession. Raises: ImportError: If websockets package is not installed. r)websocket_clientzCould not import websocket_client. To use Websocket connections, please install the required dependency: 'pip install mcp[ws]' or 'pip install websockets'N)mcp.client.websocketrw ImportErrorr )rNrIrwmsgrirjrks r_create_websocket_sessionr{ks $)9999999 ))) @ #D( ) !.$dE<B B!B) mcp_callbacks connection Connectionr|_MCPCallbacks | Nonec&Kd|vrd}t||d}d|D}|G|di|d<|j|j|dd<|j|j|dd<|dkrSd |vrd }t|t di|4d{V}|WVdddd{VdS#1d{VswxYwYdS|d vrSd |vrd }t|t di|4d{V}|WVdddd{VdS#1d{VswxYwYdS|d krhd|vrd}t|d|vrd}t|tdi|4d{V}|WVdddd{VdS#1d{VswxYwYdS|dkrSd |vrd}t|tdi|4d{V}|WVdddd{VdS#1d{VswxYwYdSd|d}t|)aCreate a new session to an MCP server. Args: connection: Connection config to use to connect to the server mcp_callbacks: mcp sdk compatible callbacks to use for the ClientSession Raises: ValueError: If transport is not recognized ValueError: If required parameters for the specified transport are missing Yields: A ClientSession r@zConfiguration error: Missing 'transport' key in server configuration. Each server must include 'transport' with one of: 'stdio', 'sse', 'websocket', 'http'. Please refer to the langchain-mcp-adapters documentation for more details.c&i|]\}}|dk ||S)r@r7r^s rrbz"create_session..s( F F Ftq!Q+5E5Ea5E5E5Er!NrIlogging_callbackelicitation_callbacksserNz.'url' parameter is required for SSE connection>streamable-httphttpstreamable_httpz:'url' parameter is required for Streamable HTTP connectionstdiorAz4'command' parameter is required for stdio connectionrCz1'args' parameter is required for stdio connection websocketz4'url' parameter is required for Websocket connectionzUnsupported transport: z5. Must be one of: 'stdio', 'sse', 'websocket', 'http'r7) ValueErrorrcrrrrqrurlr{)r}r|rzr@paramsrks rcreate_sessionrs"*$$ Y oo;'I F Fz//11 F F FF #)::.>#C#C  ) 5. # $%7 8  - 92 # $%; <E   BCS// !&0000       GMMMM                               D D D   NCS// !2<rs #""""" ******88888888888844444444%%%%%%))))))<<<<<<44444444?------LLL>>>>>>  8 $ $,--= W W W W<=7?????!"+)B"7"7"7+49V+D+D+D(     8   ,,E,E,E,E,Ei,E,E,E^ 7 7 7 7 7I 7 7 7F77777y777> D D D D D) D D Dm#&>>ATT  "&!$ ',0::::::z&*)6,08<"$$$$$$N&*8"J#,08<",,,,,,^-1DEIGGGGGGGGr!