[ǻi= UdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZddlmZmZddlmZdd lmZdd lmZdd lmZmZe rdd lmZdd lmZddlZddl m!Z!ejDe#Z$ejJdZ& d/dZ'e dZ(dZ)dZ*de+d<dZ,dZ-e dZ.e dZ/GddeZ0GddeZ1GddeZ2Gd d!eZ3Gd"d#eZ4e1e2ze3ze4zZ5edde)e*dd$ d0d%Z6ede,e-dddd& d1d'Z7ede.e/d(dddd) d2d*Z8edd+ d3d,Z9edd- d4d.Z:y)5zSession 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 \$\{([^}]+)\}c0tjd|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}`). c~tjj|jd|jdS)Nr)osenvirongetgroup)ms ]/opt/lhia/marcimex/agent/venv/lib/python3.12/site-packages/langchain_mcp_adapters/sessions.pyz"_expand_env_vars...s% qwwqz1771:(N)_BRACED_VAR_REsub)values r_expand_env_varsr"$s   NPU VVr)strictignorereplacezutf-8r#EncodingErrorHandlerDEFAULT_ENCODING_ERROR_HANDLERi,)secondsc.eZdZdZ d ddZy)McpHttpClientFactoryzFProtocol for creating httpx.AsyncClient instances for MCP connections.Ncy)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)selfheaderstimeoutauths r__call__zMcpHttpClientFactory.__call__@s r)NNN)r0dict[str, str] | Noner1zhttpx.Timeout | Noner2httpx.Auth | Nonereturnzhttpx.AsyncClient)__name__ __module__ __qualname____doc__r3r.rrr,r,=s;P*.(,"&  & &     rr,cneZdZUdZded<ded< ded< ded < d ed < d ed < ded< ded<y)StdioConnectionz=Configuration for stdio transport connections to MCP servers.zLiteral['stdio'] transportstrcommand 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_kwargsNr7r8r9r:__annotations__r.rrr<r<SsTG L4 O; ++  ('A >=76Drr<cneZdZUdZded<ded< ded< ded < ded < ded < d ed < ded<y) SSEConnectionzHConfiguration for Server-Sent Events (SSE) transport connections to MCP.zLiteral['sse']r=r>urlrFr0zNotRequired[float]r1sse_read_timeoutrG(NotRequired[McpHttpClientFactory | None]httpx_client_factoryNotRequired[httpx.Auth]r2NrHr.rrrKrKsQR H4 //3  )( 76DBB: !!6rrKczeZdZUdZded<ded< ded< ded < ded < d ed < ded < ded< ded<y)StreamableHttpConnectionz7Connection configuration for Streamable HTTP transport.zLiteral['streamable_http']r=r>rLrFr0zNotRequired[timedelta]r1rMzNotRequired[bool]terminate_on_closerGrNrOrPr2NrHr.rrrRrRsVA)) H0 /// ##,,>*)466DBB: !!6rrRc2eZdZUdZded<ded< ded<y) WebsocketConnectionzAConfiguration for WebSocket transport connections to MCP servers.zLiteral['websocket']r=r>rLrFrGNrHr.rrrUrUsK## H:66CrrU)rBrCrDrErGc:K|-|jDcic]\}}|t|c}}nd} | E| jD]2\}}tj|stj d||4t ||| |||} t| 4d{\} } t| | fi|xsi4d{} | dddd{dddd{ycc}}w7M727#1d{7swY/xYw7&#1d{7swYyxYww)aCreate a new session to an MCP server using stdio. Args: command: Command to execute. args: Arguments for the command. env: Environment variables for the command. Values containing `${VAR}` references are expanded from the current environment. Only braced syntax is supported; bare `${VAR}` is **not** expanded so that literal dollar signs in passwords or other values are never silently corrupted. Only values (not keys) are expanded; `${command}` and `${args}` are passed through unchanged. If not specified, inherits a subset of the current environment. The details are implemented in the MCP sdk. cwd: Working directory for the command. encoding: Character encoding. encoding_error_handler: How to handle encoding errors. session_kwargs: Additional keyword arguments to pass to the ClientSession. Yields: An initialized ClientSession. Nz2env[%r] contains unexpanded variable reference: %r)r?rArBrCrDrE) itemsr"rsearchloggerwarningr r r )r?rArBrCrDrErGkv resolved_env server_paramsreadwritesessions r_create_stdio_sessionrbs2J=@OCIIK8DAq Q  8QU &&( DAq$$Q'H!Q  *   5 M ]#'4edEKi}|||d<t|||||fd|i|4d{\} } } t| | fi|xsi4d{} | dddd{dddd{y7H7,7#1d{7swY)xYw7 #1d{7swYyxYww)aCreate a new session to an MCP server using Streamable HTTP. Args: url: URL of the endpoint to connect to. headers: HTTP headers to send to the endpoint. timeout: HTTP timeout. sse_read_timeout: How long the client will wait for a new event before disconnecting. terminate_on_close: Whether to terminate the session on close. session_kwargs: Additional keyword arguments to pass to the ClientSession. httpx_client_factory: Custom factory for httpx.AsyncClient (optional). auth: Authentication for the HTTP client. Yields: An initialized ClientSession. NrOr2)rr ) rLr0r1rMrSrGrOr2rdr_r`_ras r_create_streamable_http_sessionrh;s:F')=%&             dE1dEstreamable-httphttpstreamable_httpz:'url' parameter is required for Streamable HTTP connectionstdior?z4'command' parameter is required for stdio connectionrAz1'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'r.) ValueErrorrWrrqrrrerhrbrn) connectionrormr=r[r\paramsras rcreate_sessionr|s"*$ Y o;'I)//1 Ftq!Q+5Ead FF F #)::.>#C  ) ) 5.. # $%7 8  - - 922 # $%; <E  BCS/ !&00  GM    D D  NCS/ !2G H!F?)H!9G:H!=G H!G;H! G/ H!G3 H!G1 *H! H H!H  H!H  H!?H!GG GH!H!G, G# !G,(H!1H!3H9G< :HH! H! HH HH!)r!r>r6r>)r?r>rAr@rBr4rCzstr | Path | NonerDr>rEz&Literal['strict', 'ignore', 'replace']rGdict[str, Any] | Noner6AsyncIterator[ClientSession])rLr>r0r}r1floatrMrrGr}rOMcpHttpClientFactory | Noner2r5r6r~)rLr>r0r}r1rrMrrSboolrGr}rOrr2r5r6r~)rLr>rGr}r6r~)rz Connectionroz_MCPCallbacks | Noner6r~);r: __future__rloggingrre contextlibrdatetimertypingrrrr mcpr r mcp.client.sser mcp.client.stdior mcp.client.streamable_httprtyping_extensionsrrcollections.abcrpathlibrhttpx langchain_mcp_adapters.callbacksr getLoggerr7rYcompilerr"r&DEFAULT_ENCODINGr'rIDEFAULT_HTTP_TIMEOUTDEFAULT_SSE_READ_TIMEOUTDEFAULT_STREAMABLE_HTTP_TIMEOUT(DEFAULT_STREAMABLE_HTTP_SSE_READ_TIMEOUTr,r<rKrRrUrrbrerhrnr|r.rrrs # *884%)<4->   8 $,-= W<=7? 4?!"+B"7+4V+D( 8 ,,Ei,E^ 7I 7F7y7> D) Dm#&>>ATT  "&!$ ',0: : :  :  :  ::*:"::z&*)6,08<"$ $#$ $  $ * $6$ $"$$N&*8"J#,08<", ,#, ,  ,  ,*,6, ,",,^-1 *" DEIGG.BG!GGr