[ǻiQ dZddlmZmZddlmZmZmZmZddl m Z ddl m Z m Z mZmZmZmZddlmZmZmZmZddlmZddlmZdd lmZdd lmZm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)dd l!mZ*dd l+m,Z,m-Z-dd l.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8 ddl9m:Z:dZ;ee ze zZ=e;r e>e=e ze:zZ?ne>e=e zZ?dZ@GddeZAde$de=fdZBde4deCe?eAdzffdZDdee3gee4fde>e5dzdee3gee4ffdZEdede>e*fd ZFdddddd!dedzd"e*d#e7dzd$e0dzde>e5dzd%eGdzd&eHdefd'ZIdddddd!dedzd#e7dzd$e0dzde>e5dzd%eGdzd&eHde>efd(ZJd"ede>eGfd)ZKd"edefd*ZLy#e<$rdZ;YwxYw)+zTools adapter for converting MCP tools to LangChain tools. This module provides functionality to convert MCP tools into LangChain-compatible tools, handle tool execution, and manage tool conversion between the two formats. ) AwaitableCallable) AnnotatedAny TypedDictget_args) ToolMessage)FileContentBlockImageContentBlockTextContentBlockcreate_file_blockcreate_image_blockcreate_text_block)BaseToolInjectedToolArgStructuredTool ToolException)get_all_basemodel_annotations) ClientSession)Tool) ArgModelBase FuncMetadata) AudioContentBlobResourceContents ContentBlockEmbeddedResource ImageContent ResourceLink TextContentTextResourceContents) BaseModel create_model)CallbackContext Callbacks _MCPCallbacks)MCPToolCallRequestMCPToolCallResultToolCallInterceptor) Connectioncreate_session)CommandTFic&eZdZUdZeeefed<y)MCPToolArtifactafArtifact returned from MCP tool calls. This TypedDict wraps the structured content from MCP tool calls, allowing for future extension if MCP adds more fields to tool results. Attributes: structured_content: The structured content returned by the MCP tool, corresponding to the structuredContent field in CallToolResult. structured_contentN)__name__ __module__ __qualname____doc__dictstrr__annotations__Z/opt/lhia/marcimex/agent/venv/lib/python3.12/site-packages/langchain_mcp_adapters/tools.pyr-r-FsS#X&r7r-contentreturnct|trt|jSt|tr!t |j |jSt|trd|j}t|t|trc|jxsd}|r1|jdr t t|j|Stt|j|St|tr|j }t|t"rt|jSt|t$rQ|jxsd}|r(|jdrt |j&|St|j&|Sdt)|j*}t-|dt)|j*}t-|) a|Convert any MCP content block to a LangChain content block. Args: content: MCP content object (TextContent, ImageContent, AudioContent, ResourceLink, or EmbeddedResource). Returns: LangChain content block dict. Raises: NotImplementedError: If AudioContent is passed. ValueError: If an unknown content type is passed. )text)base64 mime_typeziAudioContent conversion to LangChain content blocks is not yet supported. Received audio with mime type: Nzimage/)urlr>z Unknown embedded resource type: zUnknown MCP content type: ) isinstancerrr<rrdatamimeTyperNotImplementedErrorr startswithr4urir rresourcer rblobtyper/ ValueError)r9msgr>rFs r8 _convert_mcp_content_to_lc_blockrKTs ';' gll33'<(!AQAQRR'<( 99@9I9I8J L "#&&'<($$, --h7%#gkk*:iP P S%5KK'+,## h 4 5$(--8 8 h 4 5 ))1TIY11(;))TT$HMMYO O0h1H1H0IJo &tG}'='=&> ?C S/r7call_tool_resultNcJt|tr|dfStrt|tr|dfS|jDcgc] }t |}}|j rg}|D]k}t|tr|j|%t|ts6|jddk(sK|j|jddm|rdj|n t|}t|d}|jt|j}||fScc}w)aConvert MCP MCPToolCallResult to LangChain tool result format. Converts MCP content blocks to LangChain content blocks: - TextContent -> {"type": "text", "text": ...} - ImageContent -> {"type": "image", "base64": ..., "mime_type": ...} - ResourceLink (image/*) -> {"type": "image", "url": ..., "mime_type": ...} - ResourceLink (other) -> {"type": "file", "url": ..., "mime_type": ...} - EmbeddedResource (text) -> {"type": "text", "text": ...} - EmbeddedResource (blob) -> {"type": "image", ...} or {"type": "file", ...} - AudioContent -> raises NotImplementedError Args: call_tool_result: The result from calling an MCP tool. Can be either a CallToolResult (MCP format), a ToolMessage (LangChain format), or a Command (LangGraph format, if langgraph is installed). Returns: A tuple containing: - The content: either a string (single text), list of content blocks, ToolMessage, or Command - The artifact: MCPToolArtifact with structured_content if present, otherwise None Raises: ToolException: If the tool call resulted in an error. NotImplementedError: If AudioContent is encountered. NrHr< )r.)r@r LANGGRAPH_PRESENTr+r9rKisErrorr4appendr3getjoinrstructuredContentr-)rLr9 tool_content error_partsitem error_msgartifacts r8_convert_call_tool_resultr[s%@"K0%%Z(8'B%% (//3  )13L3    9D$$""4(D$'DHHV,<,F""488FB#78  9 /:DIIk*s.wrapped_handlers*#x8888s )reversedr&r(rrr')r\r]handler interceptorcurrent_handlerrds r8_build_interceptor_chainris~G#$56 &K%O5@$  9' 91 9#'()4E*FF 9 #  9&G & Nr7sessionc Kd}g}d} |dz }|tkDr d}t||j|d{}|jr|j |j|j s |S|j }x7Gw)aList all available tools from an MCP session with pagination support. Args: session: The MCP client session. Returns: A list of all available MCP tools. Raises: RuntimeError: If maximum iterations exceeded while listing tools. Nrz3Reached max of 1000 iterations while listing tools.)cursor)MAX_ITERATIONS RuntimeError list_toolstoolsextend nextCursor)rjcurrent_cursor all_tools iterationsrJlist_tools_page_results r8_list_all_toolsrxs"&N!IJ a  &GCs# #'.'9'9'9'P!P ! ' '   399 :&00  0::!  "Qs8BBAB connection callbacksr] server_nametool_name_prefixtoolrzr{r|r}c  d}t| d dttdztfdtt t fdtttdzfffd }tdd} jjjni} | d| ini} i| | xsd} j} |rrd j} t| jxsd j |d | S)aConvert an MCP tool to a LangChain tool. NOTE: this tool can be executed only in a context of an active MCP client session. Args: session: MCP client session tool: MCP tool to convert connection: Optional connection config to use to create a new session if a `session` is not provided callbacks: Optional callbacks for handling notifications and events tool_interceptors: Optional list of interceptors for tool call processing server_name: Name of the server this tool belongs to tool_name_prefix: If `True` and `server_name` is provided, the tool name will be prefixed w/ server name (e.g., `"weather_search"` instead of `"search"`) Returns: a LangChain tool N8Either a session or a connection config must be providedruntime argumentsr:c*K'jt jn tdtdt f fd }t | }t j| xsdd|}||d{}t|S7w) aExecute tool call with interceptor chain and return formatted result. Args: runtime: LangGraph tool runtime if available, otherwise None. **arguments: Tool arguments as keyword args. Returns: A tuple of (content, artifact) where: - content: string, list of strings/content blocks, ToolMessage, or Command - artifact: MCPToolArtifact with structured_content if present, else None N)r| tool_namecontextrequestr:cxK|j}|j} }|j}|1 /t } ddvr j di}i|||d<|}d}z| d}t |t | 4d{} | jd{ | j|| jd{} dddd{|| Sj|| jd{} | S77k7F#t$r } | }Yd} ~ Td} ~ wwxYw7P#1d{7swY`xYw7;w)aExecute the actual MCP tool call with optional session creation. Args: request: Tool call request with name, args, headers, and context. Returns: MCPToolCallResult from MCP SDK. Raises: ValueError: If neither session nor connection provided. RuntimeError: If tool call returns None. N transport)ssehttpstreamable_httpzstreamable-httpheaders-Either session or connection must be provided mcp_callbacks)progress_callback) nameargsrr3rSrIr* initialize call_toolr Exception)rr tool_argseffective_connectionmodified_headersupdated_connectionexisting_headerscaptured_exceptionrJ tool_sessionrLerzrrjs r8 execute_toolzKconvert_mcp_tool_to_langchain_tool..call_tool..execute_toolIs I I#-  ' + 0F%)*%5"k*/ (2~~i'D$5*5*5&y1,>(!% '/IC$S/))(  / /!&11333/1=1G1G%%.;.M.M2H2,( / /(&1,,$ # *1):):&3&E&E*;*$ $ #; /4, %/-.*/ / / / /.$sA>D:DD:D#DD#!D DD  D:D!+D:<D8=D:D#D  DDD#DD#!D:#D5)D, *D51D:unknown)rrr|rr) to_mcp_formatr#rr%r&r'rir[) rrrrfrrLrr{rzr|rjr~r]s @r8rz5convert_mcp_tool_to_langchain_tool..call_tool1s&$  # #'K499U $  H $(:H $?PH $V+<9JK$#0y  ")!11()9::2sA=BBBmeta_meta_rNcontent_and_artifact)r description args_schema coroutineresponse_formatmetadatarc)rIrobjectrr3r4rtupleConvertedToolResultr-getattr annotations model_dumprrr inputSchema) rjr~rzr{r]r|r}rJrrbaser lc_tool_names `````` r8"convert_mcp_tool_to_langchain_toolrs!::-Ho@Dm;6D=/*;;<m;#s(^m; "Od$:: ;m;m;^ 4 &D,0,<,<,H4   & & (bD".GT?BD$$'4H99LK%a {3  $$*$$.  r7c K|| d}t|||jt|n t}|a| d}t|t ||4d{}|j d{t |d{} dddd{nt |d{}  D cgc]} t|| |||||c} S7s7]7L7>#1d{7swY:xYw7Acc} ww)aLoad all available MCP tools and convert them to LangChain [tools](https://docs.langchain.com/oss/python/langchain/tools). Args: session: The MCP client session. If `None`, connection must be provided. connection: Connection config to create a new session if session is `None`. callbacks: Optional `Callbacks` for handling notifications and events. tool_interceptors: Optional list of interceptors for tool call processing. server_name: Name of the server these tools belong to. tool_name_prefix: If `True` and `server_name` is provided, tool names will be prefixed w/ server name (e.g., `"weather_search"` instead of `"search"`). Returns: List of LangChain [tools](https://docs.langchain.com/oss/python/langchain/tools). Tool annotations are returned as part of the tool metadata object. Raises: ValueError: If neither session nor connection is provided. Nr)r|rrrry)rIrr#r%r*rrxr) rjrzr{r]r|r}rJrrrqr~s r8load_mcp_toolsrs'6:-Ho   K(PQ _   ACS/ !! m  8 8 ))+ + +),77E  8 8 8 &g..   +  !/#-    8 ,7  8 8 8 8 / sAC6CC6"C6C7C C C C6CC6-C/.C66C1C6CCC6C, C# !C,(C61C6cdtdtfd}t|jj Dcgc]\}}||r|c}}Scc}}w)zExtract field names with InjectedToolArg annotation from tool schema. Args: tool: LangChain tool to inspect. Returns: List of field names marked as injected arguments. type_r:c>tdt|ddDS)z2Check if type annotation contains InjectedToolArg.c3K|]8}t|txs"t|txrt|t:ywrc)r@rrH issubclass).0args r8 zD_get_injected_args.._is_injected_arg_type..sC  sO , L3%J*S/*J L s>ArlN)anyr)rs r8_is_injected_arg_typez1_get_injected_args.._is_injected_arg_types)  qr*   r7)rHboolrritems)r~rfield field_infos r8_get_injected_argsrsW T d "?t?O?O!P!V!V!X  E:  ,   sAcFtjts d}t|jj }jj jDcic]\}}||j|f}}}tjdfi|dti}t|}dtttfdtffd }t!} t#| dkDr d }t%|t'|jj(||d Scc}}w) aConvert LangChain tool to FastMCP tool. Args: tool: LangChain tool to convert. Returns: FastMCP tool equivalent. Raises: TypeError: If args_schema is not BaseModel subclass. NotImplementedError: If tool has injected arguments. ziTool args_schema must be a subclass of pydantic.BaseModel. Tools with dict args schema are not supported. Arguments__base__) arg_modelrr:cBKj|d{S7wrc)ainvoke)rr~s r8fnzto_fastmcp..fn-s\\),,,,s rz9LangChain tools with injected arguments are not supportedT)rrr parameters fn_metadatais_async)rrr! TypeErrortool_call_schemamodel_json_schema model_fieldsr annotationr"rrrr3r4rrlenrC FastMCPToolr) r~rJrrrfield_definitionsrrr injected_argss ` r8 to_fastmcpr s0 d&& 2 = n&&88:J"&!6!6!C!C!I!I!K E:  %%z22 99+Y#4?KI3K-d38n--'t,M =AI!#&&  YY$$  's*D)Mr2collections.abcrrtypingrrrrlangchain_core.messagesr langchain_core.messages.contentr r r r rrlangchain_core.toolsrrrrlangchain_core.tools.basermcprmcp.server.fastmcp.toolsrr*mcp.server.fastmcp.utilities.func_metadatarr mcp.typesrrrrrrrr MCPToolpydanticr!r" langchain_mcp_adapters.callbacksr#r$r%#langchain_mcp_adapters.interceptorsr&r'r(langchain_mcp_adapters.sessionsr)r*langgraph.typesr+rP ImportErrorToolMessageContentBlocklistrrnr-rKrr[rirxr4rrrrrr6r7r8rs 066/ D8Q   &,VV G' +->>AQQ67+EO67+E 'i '0 00f?"'?" $ 6 67?"D./;L1MMN/047!"I.?$@@AD"="T']"R%)"&:>""a T !a aT! a 4 a /047 ataaaN%)"&:>""= T !=T!=4 = /047 = t == (^=@X$s)2/X/+/msE99FF