ÁjHdZddlmZddlZddlmZmZmZmZddl m Z ddl m Z m Z dd lmZdd lmZdd lmZmZdd lmZgd ZGdde ZGdde ZedeZddZddZddZ ddZddZ d dZ!d!dZ"y)"a=Shared tool-dispatch helpers for the tool runners. Both ``client.beta.messages.tool_runner`` (the Messages tool runner) and ``client.beta.sessions.events.tool_runner`` (the sessions-side :class:`~anthropic.lib.tools._beta_session_runner.SessionToolRunner`) do the same three small things: index the supplied tools by name, run a runnable tool over a JSON input, and turn an exception raised by a tool into tool-result content. Those steps are factored out here so the two runners stay consistent instead of each carrying its own copy. Consumed by the runner helpers only. ) annotationsN)UnionTypeVarIterable Awaitable)Protocol) ToolErrorBetaFunctionToolResultType)BetaMessageParam)BetaContentBlockParam)Tool BetaRequestToolRemovalBlockParam)!BetaRequestToolAdditionBlockParam) tool_registrytool_error_contentrun_runnable_toolavailable_tool_namesc"eZdZdZeddZy) _NamedTooluHAnything with a ``name`` — the shape :func:`tool_registry` indexes on.cyN)selfs I/tmp/pip-target-k_scjv0h/lib/python/anthropic/lib/tools/_tool_dispatch.pynamez_NamedTool.name!sN)returnstr)__name__ __module__ __qualname____doc__propertyrrrrrrsR rrceZdZdZddZy) _CallableToolzhA runnable tool: ``call`` may be sync or async (it returns either the result or an awaitable of it).cyrr)rinputs rcallz_CallableTool.call)srN)r)objectrzHUnion[BetaFunctionToolResultType, Awaitable[BetaFunctionToolResultType]])r!r"r#r$r*rrrr'r'%s &srr' NamedToolT)boundcB|Dcic]}|j|c}Scc}w)zIndex ``tools`` by their ``name`` for O(1) dispatch lookup. On a duplicate name the later tool wins, matching a plain dict comprehension. )r)toolstools rrr/s ). .DIItO .. .sct|}|D]3}|d}|ddk7st|tr!|D]}t||5|S)uvFold mid-conversation ``tool_removal`` / ``tool_addition`` blocks over the locally runnable ``tool_names``. Only ``role: "system"`` messages carry these blocks, and only a ``tool_reference`` can name a locally runnable tool — MCP references are executed server-side, so they (and any unknown block/reference type) are ignored rather than raising. contentrolesystem)set isinstancer _apply_tool_change)messages tool_names availablemessager2blocks rrr7s_JI1)$ 6?h &*Wc*B  1E ui 0 1 1 rct|tsy|ddk(s|ddk(r t||y|ddk(r(|dD]}|ddk(s |ddk(st||!yy)a6Apply a single ``tool_removal`` / ``tool_addition`` block to ``available``. A ``mid_conv_system`` block's ``content`` is limited by the API schema to ``text`` / ``tool_addition`` / ``tool_removal``, so exactly one level is walked (no deeper nesting exists); every other block type is a no-op. Ntype tool_removal tool_additionmid_conv_systemr2)r6dict_apply_tool_reference_change)r<r:inners rr7r7Js{ eT "  V}&%-?*J$UI6 v+ +9% ?EV}.%-?2R,UI> ? rc|t|d}|y|ddk(r|j|y|j|y)zGFold one ``tool_removal`` / ``tool_addition`` block into ``available``.r0Nr>r?)_referenced_tool_namediscardadd)r<r:rs rrCrCas@ !v /D | V}&$ drc|ddk(r|dSy)a1The locally runnable tool name a tool-change reference resolves to. Only ``tool_reference`` names a runnable tool; ``mcp_tool_reference`` / ``mcp_toolset_reference`` execute server-side and unknown reference types are ignored (forward compatibility), so all of those resolve to ``None``. r>tool_referencerNr)refs rrFrFns 6{&&6{ rcPt|tr |jSt|S)aRender an exception raised by a tool as tool-result content. A :class:`ToolError` carries its own structured content; anything else is rendered with ``repr`` (which, unlike ``str``, keeps the exception type). The caller owns the ``is_error`` flag and any logging. )r6r r2repr)excs rrrzs"#y!{{ 9rcrK|j|}tj|r |d{S|S7w)a Call ``tool`` with ``input``, awaiting the result if the tool is async. Bridges the sync (:class:`~anthropic.lib.tools.BetaFunctionTool`) and async (:class:`~anthropic.lib.tools.BetaAsyncFunctionTool`) runnable-tool shapes behind a single ``await``. N)r*inspect isawaitable)r0r)results rrrs6YYu F6"| Ms ,757)r/zIterable[NamedToolT]rzdict[str, NamedToolT])r8zIterable[BetaMessageParam]r9z Iterable[str]rset[str])r<rr:rSrNone)r<zJUnion[BetaRequestToolRemovalBlockParam, BetaRequestToolAdditionBlockParam]r:rSrrT)rK_ToolChangeReferencerz str | None)rN BaseExceptionrr )r0r'r)zdict[str, object]rr )#r$ __future__rrPtypingrrrrtyping_extensionsr_beta_functionsr r types.beta.beta_message_paramr #types.beta.beta_content_block_paramr0types.beta.beta_request_tool_removal_block_paramrrUr1types.beta.beta_request_tool_addition_block_paramr__all__rr'r,rrr7rCrFrrrrrr`s #66&B=Hc ^sHs\ 4 /& .  U bj      r