ÁjAvdZddlmZddlZddlZddlmZmZddlm Z ddl m Z  ddl m Z mZmZmZmZmZmZmZmZmZddlmZd d lmZmZmZmZd d l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(d dl)m*Z+gdZ,dddddZ-d+dZ.e/hdZ0Gdde1Z2Gdde3Z4d,dZ5d-dZ6Gdd e7Z8dd! d.d"Z9dd! d/d#Z:dd! d0d$Z;dd! d1d%Z< d2d&Z=d3d'Z>ddddddd( d4d)Z?ddddddd( d5d*Z@y#e$rZed edZ[wwxYw)6aGHelpers for integrating MCP (Model Context Protocol) SDK types with the Anthropic SDK. These helpers reduce boilerplate when converting between MCP types and Anthropic API types. Usage:: from anthropic.lib.tools.mcp import mcp_tool, async_mcp_tool, mcp_message This module requires the ``mcp`` package to be installed. ) annotationsN)AnyIterable)urlparse)Literal) Tool TextContent ContentBlock ImageContent PromptMessageCallToolResultEmbeddedResourceReadResourceResultBlobResourceContentsTextResourceContents) ClientSessionz~The `mcp` package is required to use MCP helpers. Install it with: pip install anthropic[mcp]. Requires Python 3.10 or higher.)BetaBase64PDFSourceParamBetaPlainTextSourceParamBetaBase64ImageSourceParamBetaCacheControlEphemeralParam) ToolErrorBetaFunctionToolBetaAsyncFunctionToolBetaFunctionToolResultType beta_toolbeta_async_tool) tag_helper)Content)mcp_toolasync_mcp_tool mcp_content mcp_messagemcp_resource_to_contentmcp_resource_to_fileUnsupportedMCPValueError inputSchemamimeTypeisErrorstructuredContent) input_schema mime_typeis_errorstructured_contentc` t||S#t$rt|t|cYSwxYwN)getattrAttributeError _MCP_V1_NAMES)objnames >/tmp/pip-target-k_scjv0h/lib/python/anthropic/lib/tools/mcp.py_mcp_field_v1_or_v2r9Rs51sD!! 1sM$/001s --> image/gif image/png image/jpeg image/webpceZdZdZy) _TaggedDictzA dict subclass that can carry a ``_stainless_helper`` attribute. Behaves identically to a regular dict for serialization and isinstance checks, but allows attaching tracking metadata that won't appear in JSON output. N__name__ __module__ __qualname____doc__r8r?r?`srFr?ceZdZdZy) _TaggedTuplezBA tuple subclass that can carry a ``_stainless_helper`` attribute.Nr@rErFr8rHrHhsLrFrHc|tvSr2)_SUPPORTED_IMAGE_TYPESr.s r8_is_supported_image_typerLls . ..rFcX|duxs%|jdxs|dk(xs t|S)Ntext/application/pdf) startswithrLrKs r8 _is_supported_resource_mime_typerQps?T /    ( / ) ) / $I . rFceZdZdZy)r(zURaised when an MCP value cannot be converted to a format supported by the Claude API.Nr@rErFr8r(r(~s_rFr( cache_controlct|tr-td|jd}|||d<t |d|St|t r^t |d}t|std|tdtd|j| d }|||d<t |d|St|trt|j| St|d t|j }td |)zConvert a single MCP content block to an Anthropic content block. Handles text, image, and embedded resource content types. Raises :class:`UnsupportedMCPValueError` for audio and resource_link types. text)typerVrTr$r.zUnsupported image MIME type: imagebase64rWdata media_typerWsourcerSrWzUnsupported MCP content type: ) isinstancer r?rVr r r9rLr(rr[r_resource_contents_to_blockresourcer3rWrA)contentrTblockr. image_block content_types r8r$r$s';'VW\\BC  $%2E/ "5-( '<('= ' 2*-J9++VW W!4! (     $+8K (; .'+,*7+;+;=YY7FDM,B,BCL "%CL>#R SSrFct|d}|lt|rat|tst d|j t dtd|j|d}|||d<t|d|S|d k(rat|tst d |j t d td|jd d}|||d<t|d|S||jd rzt|tr |j}n.tj|jj!d }t d t#d|dd}|||d<t|d|St d|d|j )zK|]\}}t|s|ywr2)rQ).0cr.s r8 z*mcp_resource_to_content.."so|q)CcdmCnosNz>No supported MIME type found in resource contents. Available: z, rS)contentsr(r9nextzipjoinr`)rrrTrw mime_types supportedms r8r&r&s ??&'_``?EO!%a5OJOos6??J?o I!+=Aq}a= =&LTYYWaMbLc d   'y NNP >sB (B%0B%c|js td|jd}t|j}t |j }|r|j dddnd}t|tr tj|j}n|jjd}t||t|df}t!|d |S) zConvert MCP resource contents to a file tuple for ``files.upload()``. Returns a ``(filename, content_bytes, mime_type)`` tuple compatible with the SDK's ``FileTypes``. rtr/rNrhr.r')ryr(strrirpathrsplitr_rrYrkrjrVencoderHr9r )rrrauri_strrr7 content_bytes file_tuples r8r'r'.s ??&'_``q!H(,,G G  ! !D&*4;;sA r "D(01((7  ,,W5 t]4GR]4^_`Jz12 rFc*t|dr,t|jDcgc] }t|c}t|d}|js|t j |S|jDcgc] }t|c}Scc}wcc}w)zQConvert MCP ``CallToolResult`` to a value suitable for returning from ``call()``.r/r0)r9rrbr$jsondumps)rritemr0s r8_convert_tool_resultrPs}6:.v~~FtT*FGG-V5IJ >>0<zz,--*0.. 9$K  99G :s B 6B)rT defer_loadingallowed_callerseager_input_streaminginput_examplesstrictc  ddl |j d  fd }t| |jt |d|||||| } t | d| S)aConvert an MCP tool to a sync runnable tool for ``tool_runner()``. Example:: from anthropic.lib.tools.mcp import mcp_tool tools_result = await mcp_client.list_tools() runner = client.beta.messages.tool_runner( model="claude-sonnet-4-20250514", max_tokens=1024, tools=[mcp_tool(t, mcp_client) for t in tools_result.tools], messages=[{"role": "user", "content": "Use the available tools"}], ) Args: tool: An MCP tool definition from ``client.list_tools()``. client: The MCP ``ClientSession`` used to call the tool. cache_control: Cache control configuration. defer_loading: If true, tool will not be included in initial system prompt. allowed_callers: Which callers may use this tool. eager_input_streaming: Enable eager input streaming for this tool. input_examples: Example inputs for the tool. strict: When true, guarantees schema validation on tool names and inputs. rNchjjj|}t|Sr2) from_threadrun call_toolr)kwargsrranyioclient tool_names r8call_mcpzmcp_tool..call_mcps.""&&v'7'7FK#F++rFr- r7 descriptionr-rTrrrrrr"rrreturnr)anyio.from_threadr7rrr9r ) toolrrTrrrrrrrrrrs ` @@r8r"r"bs_L I, $$(~>##'3% Fvz" MrFc  |j d fd }t| |jt|d|||||| } t | d| S)aConvert an MCP tool to an async runnable tool for ``tool_runner()``. Example:: from anthropic.lib.tools.mcp import async_mcp_tool tools_result = await mcp_client.list_tools() runner = await client.beta.messages.tool_runner( model="claude-sonnet-4-20250514", max_tokens=1024, tools=[async_mcp_tool(t, mcp_client) for t in tools_result.tools], messages=[{"role": "user", "content": "Use the available tools"}], ) Args: tool: An MCP tool definition from ``client.list_tools()``. client: The MCP ``ClientSession`` used to call the tool. cache_control: Cache control configuration. defer_loading: If true, tool will not be included in initial system prompt. allowed_callers: Which callers may use this tool. eager_input_streaming: Enable eager input streaming for this tool. input_examples: Example inputs for the tool. strict: When true, guarantees schema validation on tool names and inputs. c\Kj|d{}t|S7w)N)r7 arguments)rr)rrrrrs r8rz async_mcp_tool..call_mcps0''Y&'II#F++Js ,*,r-rr"r)r7rrr9r ) rrrTrrrrrrrrrs ` @r8r#r#s\L I, $$(~>##'3% Fvz" MrF)r6rr7rrr)r.rrbool)r.z str | Nonerr)rbr rT%BetaCacheControlEphemeralParam | Noner BetaContent)raz+TextResourceContents | BlobResourceContentsrTrrr)rqr rTrrzdict[str, Any])rrrrTrrr)rrrrz$tuple[str | None, bytes, str | None])rrr rr)rrrrrTrr bool | Nonerolist[Literal['direct', 'code_execution_20250825', 'code_execution_20260120', 'code_execution_20260521']] | Nonerrr"Iterable[dict[str, object]] | NonerrrzBetaFunctionTool[Any])rrrrrTrrrrrrrrrrrrzBetaAsyncFunctionTool[Any])ArD __future__rrrYtypingrr urllib.parsertyping_extensionsr mcp.typesrr r r r r rrrrmcp.client.sessionr ImportError_err types.betarrrr_beta_functionsrrrrrr_stainless_helpersr 'types.beta.beta_tool_result_block_paramr!r__all__r5r9 frozensetrJdictr?tuplerHrLrQ Exceptionr(r$r`r%r&r'rr"r#rErFr8rs # !%   1  ,M $"-  1##YZ$M5M/`y`<@)T )T9)T )T^<@Ch9Ch9Ch Ch\<@   9  0<@O O9O O6 )D :,<@!%)-9=; ; ;9 ;  ;  ;';7; ;;D<@!%)-9=9 9 99 9  9  9'979 9 9{   I  sD%%D8* D33D8