j iQddZddlmZmZddlmZmZmZmZddl 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;n #e<$rdZ;YnwxYwee ze zZ=e;re>e=e ze:zZ?n e>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*ZLdS)+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<dS)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__C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain_mcp_adapters/tools.pyr-r-Fs2S#X&&&&&r7r-contentreturnct|trt|jSt|trt |j|jSt|trd|j}t|t|trf|jpd}|r8| dr#t t|j |Stt|j |St|tr|j}t|t"rt|jSt|t$rL|jpd}|r+| drt |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 ';''4 gl3333'<((S!AQRRRR'<((' L9@9I L L "#&&&'<((L$,  Q--h77 Q%#gk*:*:iPPP P S%5%5KKKK'+,, # h 4 5 5 9$(-888 8 h 4 5 5 P )1TI UY11(;; U))TTTT$HMYOOO OJh1HJJoo ?tG}}'= ? ?C S//r7call_tool_resultNcdt|tr|dfStrt|tr|dfSd|jD}|jrg}|D]}t|t r||-t|trB| ddkr)|| dd|rd |nt |}t|d}|j t|j }||fS)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. Nc,g|]}t|Sr6)rK).0r9s r8 z-_convert_call_tool_result..s0333  )11333r7rHr< )r.)r@r LANGGRAPH_PRESENTr+r9isErrorr4appendr3getjoinrstructuredContentr-)rL tool_content error_partsitem error_msgartifacts r8_convert_call_tool_resultr^sh@"K00&%%&Z(8'BB&%%33'/333L  '   9 9D$$$ 9""4((((D$'' 9DHHV,<,<,F,F""488FB#7#7888.9PDIIk***s.wrapped_handlers,*\#x888888888r7)reversedr&r(rrr')r_r`handler interceptorcurrent_handlerrgs r8_build_interceptor_chainrlsG &#$566 & &K%O5@$  9 9' 91 9#'()4E*FF 9 #  9 9 9 9&GG Nr7sessioncKd}g}d} |dz }|tkrd}t|||d{V}|jr||j|jsn|j}n|S)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. NrTz3Reached max of 1000 iterations while listing tools.)cursor)MAX_ITERATIONS RuntimeError list_toolstoolsextend nextCursor)rmcurrent_cursor all_tools iterationsrJlist_tools_page_results r8_list_all_toolsr{s"&N!IJ;a  & &GCs## #'.'9'9'9'P'P!P!P!P!P!P!P ! ' ;   39 : : :&0  /:!;" r7 connection callbacksr` server_nametool_name_prefixtoolr}r~rrc d}t| d dttdztfdtt t fdtttdzfffd }tdd} j j ni} | d| ini} i| | pd} j } |rr d j } t| jpd 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*t jn tdtdt f fd }t | }t j| pdd|}||d{V}t|S) 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:c\K|j}|j} }|j}|< :t } ddvr! di}i|||d<|}d}|d}t |t | 4d{V } | d{V | || j d{V} n#t$r } | }Yd} ~ nd} ~ wwxYwdddd{Vn#1d{VswxYwY||n#|| j d{V} | S)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) nameargsrr3rVrIr* initialize call_toolr Exception)rr tool_argseffective_connectionmodified_headersupdated_connectionexisting_headerscaptured_exceptionrJ tool_sessionrLer}rrms r8 execute_toolzKconvert_mcp_tool_to_langchain_tool..call_tool..execute_toolIs I I#-  ' + 0F%)*%5%5"k*/ (2~~i'D'D$5*5*5&y1,>(!% '/IC$S//))(  / / / / / / / /!&11333333333/1=1G1G%%.;.M2H22,,,,,,(( %///-.******/ / / / / / / / / / / / / / / / / / / / / / / / / / / /(&1,,2*1):):&3&E*;**$$$$$$ $ #s<C3$#CC3 CCC3CC33 C=C=unknown)rrrrr) to_mcp_formatr#rr%r&r'rlr^) rrrrirrLrr~r}rrmrr`s @r8rz5convert_mcp_tool_to_langchain_tool..call_tool1s&$  # #'K49UUU $    H $(:H $?PH $H $H $H $H $H $H $H $V+<9JKK$#0y    ")!1!1111111()9:::r7meta_meta_rQcontent_and_artifact)r description args_schema coroutineresponse_formatmetadatarf)rIrobjectrr3r4rtupleConvertedToolResultr-getattr annotations model_dumprrr inputSchema) rmrr}r~r`rrrJrrbaser lc_tool_names `````` r8"convert_mcp_tool_to_langchain_toolrsy::-Hoo@Dm;m;6D=/*;*;;<m;#s(^m; "Od$:: ;m;m;m;m;m;m;m;m;m;m;m;^ 4 & &D,0,<,H4  & & ( ( (bD".GT??BD$$'4H9L4K4%33 33  $*$.    r7cKd}t|$tn t}d}t|t |4d{V }|d{Vt |d{V} dddd{Vn#1d{VswxYwYnt d{V} fd| DS)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)rrrrc <g|]}t|S)r|)r)rOrr~r}rrmr`rs r8rPz"load_mcp_tools..sO     +  !/#-      r7)rIrr#r%r*rr{) rmr}r~r`rrrJrrrts `````` r8load_mcp_toolsrs6:-Hoo   K(P(P(PQQQ __   ACS// !! m    8 8 8 8 8 8 8 8 ))++ + + + + + + +),77777777E  8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 &g........             s?0C C C cdtdtfdfdt|jDS)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|]C}t|tp)t|tot|tVDdSrf)r@rrH issubclass)rOargs r8 zD_get_injected_args.._is_injected_arg_type..sd   sO , , L3%%J*S/*J*J      r7roN)anyr)rs r8_is_injected_arg_typez1_get_injected_args.._is_injected_arg_typesA   qrr*      r7c0g|]\}}||Sr6r6)rOfield field_infors r8rPz&_get_injected_args..s>    E:  , ,     r7)rHboolrritems)rrs @r8_get_injected_argsrsg T d        !>t?O!P!P!V!V!X!X   r7c(tjtsd}t|j}djjD}tj dfi|dti}t|}dtttfdtffd }t}t!|d krd }t#|t%|j j||d S) 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.c(i|]\}}||j|fSr6) annotation)rOrrs r8 zto_fastmcp.."s6 E:  %z2r7 Arguments__base__) arg_modelrr:c>K|d{VSrf)ainvoke)rrs r8fnzto_fastmcp..fn-s+\\),,,,,,,,,r7rz9LangChain tools with injected arguments are not supportedT)rrr parameters fn_metadatais_async)rrr! TypeErrortool_call_schemamodel_json_schema model_fieldsrr"rrrr3r4rrlenrC FastMCPToolr)rrJrfield_definitionsrrr injected_argss` r8 to_fastmcpr s^ d& 2 2 = nn&88::J!%!6!C!I!I!K!K 9#4?KI333K-d38n-------'t,,M =AI!#&&&  Y$    r7)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+rS ImportErrorToolMessageContentBlocklistrrqr-rKrr^rlr{r4rrrrrr6r7r8rsi 0///////666666666666////// DCCCCC888888QQQQQQQQ                    &%%%%%,,,,,,,,VVVVVVVVVV GFFFFFFF''''''+->>AQQF67+EO67+E ' ' ' ' 'i ' ' '0 00000f?"'?" $ 6 67?"?"?"?"D./;L1MMN/047!"I.?$@@AD"="T']""""R%)"&:>""aaa T !a aT! a 4 a /047 ataaaaaaN%)"&:>""=== T !=T!=4 = /047 = t == (^====@X$s)2/X/+//////sBB%$B%