o ^۷iU= @sdZddlmZddlZddlZddlmZmZddlm Z ddl m Z z ddl m Z mZmZmZmZmZmZmZmZmZddlmZWneyUZzed edZ[wwd d lmZmZmZmZd d l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(d dl)m*Z+gdZ,e-hdZ.Gddde/Z0Gddde1Z2dQddZ3dRddZ4Gd d!d!e5Z6dd"dSd(d)Z7dd"dTd,d-Z8dd"dUd1d2Z9dd"dVd5d6Z:dWd8d9Z;dXddYdLdMZ=ddddddd>dZdOdPZ>dS)[aGHelpers 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> image/gif image/png image/jpeg image/webpc@eZdZdZdS) _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__r3r3M/home/ubuntu/vllm_env/lib/python3.10/site-packages/anthropic/lib/tools/mcp.pyr-Jr-c@r,) _TaggedTuplezBA tuple subclass that can carry a ``_stainless_helper`` attribute.Nr.r3r3r3r4r6Rr5r6 mime_typestrreturnboolcCs|tvSN)_SUPPORTED_IMAGE_TYPESr7r3r3r4_is_supported_image_typeVsr> str | NonecCs"|dup|dp|dkpt|S)Ntext/application/pdf) startswithr>r=r3r3r4 _is_supported_resource_mime_typeZsrCc@r,)r'zURaised when an MCP value cannot be converted to a format supported by the Claude API.Nr.r3r3r3r4r'hr5r' cache_controlcontentr rE%BetaCacheControlEphemeralParam | None BetaContentcCst|trtd|jd}|dur||d<t|d|St|trKt|js.td|jtdt d|j |jd d }|durD||d<t|d|St|t rWt |j |d 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)typerINrEr#zUnsupported image MIME type: imagebase64rJdata media_typerJsourcerDrJzUnsupported MCP content type: ) isinstancerr-rIrr r>mimeTyper'rrNr _resource_contents_to_blockresourcegetattrrJr/)rFrEblock image_block content_typer3r3r4r#qs4       r#rU+TextResourceContents | BlobResourceContentscCsD|j}|dur4t|r4t|tstd|jtdtd|j|dd}|dur-||d<t |d|S|d krat|tsEtd |jtd t d|jd dd}|durZ||d<t |d|S|dusj| d rt|t rs|j }n t|jd }td td|ddd}|dur||d<t |d|Std|d|j)zrRrr'urir-rblobrrrBrrIrL b64decodedecoder)rUrEr7rX pdf_blockrN text_blockr3r3r4rTsf         rTmessager dict[str, Any]cCs*t|jt|j|dgd}t|d|S)zCConvert an MCP prompt message to an Anthropic ``BetaMessageParam``.rD)rolerFr$)r-rer#rFr)rcrEresultr3r3r4r$s r$rfrcCs\|jstdtdd|jDd}|dur(dd|jD}tdd|t||d S) zConvert MCP resource contents to an Anthropic content block. Finds the first resource with a supported MIME type from the result's ``contents`` list. 6Resource contents array must contain at least one itemcss|] }t|jr|VqdSr;)rCrS.0cr3r3r4 sz*mcp_resource_to_content..NcSsg|] }|jdur|jqSr;)rSrhr3r3r4 sz+mcp_resource_to_content..z>No supported MIME type found in resource contents. Available: z, rD)contentsr'nextjoinrT)rfrE supported mime_typesr3r3r4r%s  r%$tuple[str | None, bytes, str | None]cCs|jstd|jd}t|j}t|j}|r |dddnd}t|tr.t |j }n|j d}t|||jf}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``. rgr/rNr\r&)rmr'r8r]rpathrsplitrRrrLr_r^rIencoder6rSr)rfrUuri_strruname content_bytes file_tupler3r3r4r&s      r&r rcCsF|jr tdd|jD|js|jdurt|jSdd|jDS)zQConvert MCP ``CallToolResult`` to a value suitable for returning from ``call()``.cSg|]}t|qSr3r#riitemr3r3r4rl;z(_convert_tool_result..NcSr|r3r}r~r3r3r4rlAr)isErrorrrFstructuredContentjsondumps)rfr3r3r4_convert_tool_result8s  r)rE defer_loadingallowed_callerseager_input_streaminginput_examplesstricttoolrclientrr bool | NonerTlist[Literal['direct', 'code_execution_20250825', 'code_execution_20260120']] | Nonerr"Iterable[dict[str, object]] | NonerBetaFunctionTool[Any]c sNddl|jd fdd }t||j|j||||||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. rNkwargsrr9rcsjj|}t|Sr;) from_threadrun call_toolrrrfanyior tool_namer3r4call_mcppszmcp_tool..call_mcp ry description input_schemarErrrrrr!rrr9r)anyio.from_threadryrr inputSchemar rrrErrrrrrrfr3rr4r!Is"# r!BetaAsyncFunctionTool[Any]c sD|jd fdd }t||j|j||||||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. rrr9rcsj|dIdH}t|S)N)ry arguments)rrrrrr3r4rsz async_mcp_tool..call_mcprr!Nr)ryrrrrrr3rr4r"s # r")r7r8r9r:)r7r?r9r:)rFr rErGr9rH)rUrZrErGr9rH)rcr rErGr9rd)rfrrErGr9rH)rfrr9rr)rfr r9r)rrrrrErGrrrrrrrrrrr9r)rrrrrErGrrrrrrrrrrr9r)?r2 __future__rrrLtypingrr urllib.parsertyping_extensionsr mcp.typesrrr r r r r rrrmcp.client.sessionr ImportError_err types.betarrrr_beta_functionsrrrrrr_stainless_helpersr'types.beta.beta_tool_result_block_paramr rH__all__ frozensetr<dictr-tupler6r>rC Exceptionr'r#rTr$r%r&rr!r"r3r3r3r4sj   0        .N  "?